source: n4d-network/trunk/fuentes/n4d-nat-frontend.install/usr/share/n4d-nat-frontend/nat-manager.py @ 144

from gi.repository import Gtk
import xmlrpclib
import sys
import os
import os.path

import signal

signal.signal(signal.SIGINT, signal.SIG_DFL)

class NatManager:
        
        def __init__(self,ip="localhost"):
                status=self.read_key()
                #hack for testing
                #ip="server"
                #self.key="oua470v052tt1os1H8mpW8HhwUqqELGzEn40GUqdWKTpBsg67F"
                #
                self.client=xmlrpclib.ServerProxy("https://"+ip+":9779")
                
                if not status:
                        print("[!] You need root privileges to run this program [!]")
                        label = Gtk.Label("You need root privileges to run nat-manager")
                        dialog = Gtk.Dialog("Warning", None, Gtk.DialogFlags.MODAL | Gtk.DialogFlags.DESTROY_WITH_PARENT, (Gtk.STOCK_OK, Gtk.ResponseType.ACCEPT))
                        dialog.vbox.pack_start(label,True,True,10)
                        label.show()
                        dialog.set_border_width(6)
                        response = dialog.run()
                        dialog.destroy()
                        sys.exit(0)
                self.status={}
                self.get_status_list()
                self.build_gui()
        
        def read_key(self):
                
                try:
                        f=open("/etc/n4d/key")
                        self.key=f.readline().strip("\n")
                        f.close()
                        #hack
                        return True
                except:
                        return False

        #def check_perms
        
        def get_status_list(self):
                
                
                
                try:
                        
                        ret=self.client.get_nat("","NetworkManager")
                        
                        self.status["nat"]=ret["status"]
                        ret=self.client.get_routing("","NetworkManager")
                        self.status["routing"]=ret["status"]
                        self.status["nat_persistence"]=True
                        self.status["routing_persistence"]=True
                        
                        
                except Exception as e:
                        self.msg_label.set_text("N4D error: " + str(e))
                        #set msg error
                
                #print self.status
                
        #def get_status_list
        
        def build_gui(self):
                
                builder=Gtk.Builder()
                
                if os.path.exists("/srv/svn/pandora/n4d-network/trunk/n4d-nat-frontend.install/usr/share/n4d-nat-frontend/rsrc/nat-manager.glade"):
                        builder.add_from_file("/srv/svn/pandora/n4d-network/trunk/n4d-nat-frontend.install/usr/share/n4d-nat-frontend/rsrc/nat-manager.glade")
                else:
                        builder.add_from_file("/usr/share/n4d-nat-frontend/rsrc/nat-manager.glade")
                        
                self.window=builder.get_object("window")
                
                self.routing_enabled_rbutton=builder.get_object("routing_enabled_radiobutton")
                self.routing_disabled_rbutton=builder.get_object("routing_disabled_radiobutton")
                
                self.routing_enabled_rbutton.set_active(self.status["routing"])
                self.routing_disabled_rbutton.set_active(not self.status["routing"])
                
                self.nat_enabled_rbutton=builder.get_object("nat_enabled_radiobutton")
                self.nat_disabled_rbutton=builder.get_object("nat_disabled_radiobutton")
                
                self.nat_enabled_rbutton.set_active(self.status["nat"])
                self.nat_disabled_rbutton.set_active(not self.status["nat"])

                
                #self.nat_persistence_cbutton=builder.get_object("nat_persistence_checkbutton")
                #self.nat_persistence_cbutton.set_active(self.status["nat_persistence"])
                
                #self.routing_persistence_cbutton=builder.get_object("routing_persistence_checkbutton")
                #self.routing_persistence_cbutton.set_active(self.status["routing_persistence"])
                
                self.msg_label=builder.get_object("msg_label")
                
                self.apply_button=builder.get_object("apply_button")
                self.close_button=builder.get_object("close_button")
                
                self.apply_button.connect("clicked",self.apply_changes)
                self.close_button.connect("clicked",self.close_window)
                self.window.connect("destroy",self.close_window)
                
                self.window.show_all()
                Gtk.main()
                
                
                
        #def build_gui
        
        
        def apply_changes(self,widget):
                
                self.status["routing"]=self.routing_enabled_rbutton.get_active()
                self.status["nat"]=self.nat_enabled_rbutton.get_active()
                #self.status["routing_persiscence"]=self.routing_persistence_cbutton.get_active()
                #self.status["nat_persiscente"]=self.nat_persistence_cbutton.get_active()
                
                
                
                
                try:
                        self.client.set_nat(self.key,"NetworkManager",self.status["nat"],self.status["nat_persistence"])
                        self.client.set_routing(self.key,"NetworkManager",self.status["routing"],self.status["routing_persistence"])
                        self.msg_label.set_text("Changes saved successfuly")
                except Exception as e:
                        self.msg_label.set_text("Operation failed because: " + str(e))
                
                
        #def apply_changes
        
        def close_window(self,widget):
                
                Gtk.main_quit()
                sys.exit(0)
                
        #def close_window
        
        
#class NatManager

def usage():
        print("USAGE:")
        print("\tnat-manager [ -ip TEMPLATE_FILE -u USER -p PASSWORD ]")

if __name__=="__main__":
        
        nm=NatManager()