source: zero-server-wizard/trunk/fuentes/install-files/usr/share/zero-server-wizard/slave.py @ 825

Last change on this file since 825 was 825, checked in by hectorgh, 4 years ago

missing imports

File size: 16.9 KB
Line 
1import os
2import os.path
3import xmlrpclib
4import datetime
5import subprocess
6import sys
7import multiprocessing
8
9from cStringIO import StringIO
10
11
12import lliurex.net
13
14try:
15        import gi
16        gi.require_version('Gtk', '3.0')
17        from gi.repository import Gtk,Gdk,GLib
18except:
19        pass
20
21DNS1="172.27.111.5"
22DNS2="172.27.111.6"
23
24class Slave:
25       
26
27        def __init__(self,core,server_conf=None):
28               
29                self.id="SLAVE"
30                self.scripts_path="/usr/share/zero-server-wizard/types/slave/actions/"
31                self.template={}
32                if server_conf!=None:
33                        try:
34                                for item in server_conf["variables"]:
35                                        self.template[item]=None
36                        except:
37                                pass
38                               
39               
40                self.core=core
41               
42               
43               
44        #def __init__
45       
46        def start_gui(self,standalone=False):
47               
48                self.standalone=standalone
49               
50                builder=self.core.builder
51
52               
53               
54                self.apply_button=builder.get_object("apply_button2")
55                self.internal_speed_label=builder.get_object("internal_iface_speed_label2")
56                self.external_speed_label=builder.get_object("external_iface_speed_label2")
57                self.password_entry=builder.get_object("password_entry4")
58                self.password_entry2=builder.get_object("password_entry5")
59                self.remote_user_entry=builder.get_object("remote_user")
60                #self.password_entry.connect("changed",self.password_changed)
61                self.srv_name_entry=builder.get_object("srv_name_entry2")
62                self.srv_domain_entry=builder.get_object("srv_domain_entry2")
63                self.internal_combobox=builder.get_object("internal_combobox2")
64                self.internal_combobox.set_name("Button")
65                self.external_combobox=builder.get_object("external_combobox2")
66               
67                self.internal_ip_entry=builder.get_object("srvip_entry2")
68                self.internal_mask_entry=builder.get_object("internal_mask_entry2")
69                self.external_ip_entry=builder.get_object("external_ip_entry2")
70                self.external_mask_entry=builder.get_object("external_mask_entry2")
71                self.external_gateway_entry=builder.get_object("external_gateway_entry2")
72                self.dns1_entry=builder.get_object("dns1_entry2")
73                self.dns2_entry=builder.get_object("dns2_entry2")
74                self.pass_label=builder.get_object("pass_label2")
75                self.dhcp_radiobutton=builder.get_object("dhcp_radiobutton2")
76                self.dhcp_radiobutton.connect("toggled",self.radio_button_changed)
77                self.manual_radiobutton=builder.get_object("manual_radiobutton2")
78                self.manual_expander=builder.get_object("manual_expander2")
79                self.manual_expander.set_sensitive(False)
80                self.apply_button.connect("clicked",self.apply_clicked)
81                #self.apply_button.set_sensitive(False)
82                self.apply_button.set_name("Button")
83                self.replication_cbutton=builder.get_object("data_cbutton1")
84                self.gluster_msg=builder.get_object("error_label2")
85                if self.core.platform != "x86_64":
86                        self.replication_cbutton.set_sensitive(False)
87                else:
88                        self.gluster_msg.hide()
89               
90                self.croom_spinbutton=builder.get_object("croom_spinbutton")
91                self.croom_spinbutton.connect("changed",self.spin_changed)
92                self.set_default_gui_values()
93               
94                if standalone:
95                       
96                        Gtk.main()
97
98               
99        #def start_gui
100       
101        def spin_changed(self,widget):
102               
103                value=self.croom_spinbutton.get_chars(0,-1)
104               
105                if len(value)>0:
106               
107               
108                        self.croom_spinbutton.set_value(int(value))
109               
110                        ip="10.2.%s.254"%str(int(self.croom_spinbutton.get_value()))
111                        self.internal_ip_entry.set_text(ip)
112               
113               
114        #def spin_changed
115       
116       
117        def radio_button_changed(self,widget):
118
119                self.manual_expander.set_sensitive(not self.dhcp_radiobutton.get_active())
120                self.manual_expander.set_expanded(not self.dhcp_radiobutton.get_active())
121               
122        #def radio_button_changed
123       
124        def password_changed(self,widget):
125               
126                if self.password_entry.get_text()!=self.password_entry1.get_text():
127                        self.pass_label.set_markup("<span fgcolor='red'>" + "Passwords do not match" +"</span>")
128                        self.apply_button.set_sensitive(False)
129                else:
130                        self.pass_label.set_markup("")
131                        self.apply_button.set_sensitive(True)
132                       
133        #def password_changed
134       
135        def set_default_gui_values(self):
136               
137
138                self.iiface_model=Gtk.ListStore(str)
139                self.eiface_model=Gtk.ListStore(str)
140               
141                self.remote_user_entry.set_text("netadmin")
142               
143                self.internal_combobox.set_model(self.iiface_model)
144                self.external_combobox.set_model(self.eiface_model)
145                rendi=Gtk.CellRendererText()
146                self.internal_combobox.pack_start(rendi,True)
147                self.internal_combobox.add_attribute(rendi,"text",0)
148                self.internal_combobox.connect("changed",self.get_link_speed,0)
149                rende=Gtk.CellRendererText()
150                self.external_combobox.pack_start(rende,True)
151                self.external_combobox.add_attribute(rende,"text",0)
152                self.external_combobox.connect("changed",self.get_link_speed,1)
153                self.interfaces=lliurex.net.get_devices_info()         
154               
155               
156                for item in self.interfaces:
157                        if "eth" in item["name"]:
158                                self.iiface_model.append([item["name"]])
159                                self.eiface_model.append([item["name"]])
160                               
161                       
162                self.internal_combobox.set_active(0)
163                if len(self.iiface_model)>1:
164                        self.external_combobox.set_active(1)
165                else:
166                        self.external_combobox.set_active(0)
167                       
168               
169                self.srv_domain_entry.set_text(self.template.setdefault("srv_domain_name","lliurex"))
170                self.srv_domain_entry.set_sensitive(False)
171
172                self.internal_ip_entry.set_text(self.template.setdefault("srv_ip","10.2.1.254"))
173                self.internal_ip_entry.set_sensitive(False)
174                self.internal_mask_entry.set_text(self.template.setdefault("internal_mask","255.255.255.0"))
175                self.internal_mask_entry.set_sensitive(False)
176                self.dns1_entry.set_text(DNS1)
177                self.dns2_entry.set_text(DNS2)
178               
179                try:
180                       
181                        f=open("/etc/hostname")
182                        line=f.readline().strip("\n")
183                        f.close()
184                        self.srv_name_entry.set_text(self.template.setdefault("srv_name",line))
185                except:
186                        pass
187                       
188               
189               
190        #def set_default_gui_values
191       
192        def apply_clicked(self,widget):
193               
194
195                self.get_gui_values()
196               
197               
198        #def apply_clicked
199       
200        def get_gui_values(self):
201               
202                iter=self.internal_combobox.get_active_iter()
203                if iter!=None:
204                        self.template["internal_iface"]=self.iiface_model.get(iter,0)[0]
205                else:
206                        self.template["internal_iface"]=None
207                iter=self.external_combobox.get_active_iter()
208                if iter!=None:
209                        self.template["external_iface"]=self.eiface_model.get(iter,0)[0]
210                else:
211                        self.template["external_iface"]=None
212               
213                self.template["adminpassword"]=self.password_entry.get_text()
214                self.template["remote_user"]=self.remote_user_entry.get_text()
215                self.template["remote_password"]=self.password_entry.get_text()
216                self.template["remote_password2"]=self.password_entry2.get_text()
217                self.template["number_classroom"]=int(self.croom_spinbutton.get_value())
218                self.template["enable_data_replication"]=str(self.replication_cbutton.get_active()).lower()
219                self.template["srv_name"]=self.srv_name_entry.get_text()
220                self.template["srv_name"]=''.join(e for e in self.template["srv_name"] if e.isalnum())
221                self.template["srv_domain_name"]=self.srv_domain_entry.get_text()
222                self.template["srv_domain_name"]=''.join(e for e in self.template["srv_domain_name"] if e.isalnum())
223                self.template["srv_ip"]=self.internal_ip_entry.get_text()
224                self.template["internal_mask"]=self.internal_mask_entry.get_text()
225                if self.dhcp_radiobutton.get_active():
226                        self.template["external_mode"]="dhcp"
227
228                else:
229                        self.template["external_mode"]="manual"
230
231                self.template["external_ip"]=self.external_ip_entry.get_text()
232                self.template["external_mask"]=self.external_mask_entry.get_text()
233                self.template["external_gateway"]=self.external_gateway_entry.get_text()
234                self.template["dns1"]=self.dns1_entry.get_text()
235                self.template["dns2"]=self.dns2_entry.get_text()
236               
237               
238               
239                self.check_values()
240       
241       
242        def check_values(self):
243               
244
245                        error_msg=""
246                        '''
247                        if self.template["internal_iface"]==self.template["external_iface"]:
248                                error_msg+="* External and internal interfaces must be different.\n"
249                        '''
250                       
251                        if self.template["remote_password"]!=self.template["remote_password2"]:
252                                error_msg+="* Remote passwords do not match"
253                       
254                        if not lliurex.net.is_valid_ip(self.template["srv_ip"]):
255                                error_msg+="* Internal IP must be a valid IP.\n"
256
257                        if not lliurex.net.is_valid_ip(self.template["internal_mask"]):
258                                error_msg+="* Internal mask must be a valid IP.\n"
259                               
260                        if self.template["srv_name"]=="":
261                                error_msg+="* Server name cannot be an empty string.\n"
262                        if self.template["srv_domain_name"]=="":
263                                error_msg+="* Server domain name cannot be an empty string.\n"
264                       
265                       
266                        if self.template["external_mode"]=="manual":
267
268                                if self.template["external_ip"]=='' or not lliurex.net.is_valid_ip(self.template["external_ip"]):
269                                        error_msg+="* External IP must be a valid IP.\n"
270                                if self.template["external_mask"]=="" or not lliurex.net.is_valid_ip(self.template["external_mask"]):
271                                        error_msg+="* External mask must be a valid IP.\n"
272                                if self.template["external_gateway"]=="" or not lliurex.net.is_valid_ip(self.template["external_gateway"]):
273                                        error_msg+="* External gateway must be a valid IP.\n"
274                               
275                               
276                        if self.template["dns1"]=="" or not lliurex.net.is_valid_ip(self.template["dns1"]):
277                                error_msg+="* DNS1 must be a valid IP.\n"
278                        if self.template["dns2"]=="" or not lliurex.net.is_valid_ip(self.template["dns2"]):
279                                error_msg+="* DNS2 must be a valid IP.\n"
280                               
281                        if error_msg!="":
282                                builder = Gtk.Builder()
283                                if os.path.exists("/srv/svn/pandora/zero-server-wizard/trunk/install-files/usr/share/zero-server-wizard/rsrc/warning.glade"):
284                                        builder.add_from_file("/srv/svn/pandora/zero-server-wizard/trunk/install-files/usr/share/zero-server-wizard/rsrc/warning.glade")
285                                else:
286                                        builder.add_from_file("/usr/share/zero-server-wizard/rsrc/warning.glade")
287                                       
288                                       
289                                window=builder.get_object("window1")
290                                msg_label=builder.get_object("msg_label")
291                                close_button=builder.get_object("close_button")
292                                def hide(widget):
293                                        window.destroy()
294                                close_button.connect("clicked",hide)
295                                msg_label.set_text(error_msg.strip("\n"))
296                                window.show()                           
297                       
298                        else:
299                               
300                                builder=Gtk.Builder()
301                                if os.path.exists("/srv/svn/pandora/zero-server-wizard/trunk/install-files/usr/share/zero-server-wizard/rsrc/progress_window.glade"):
302                                        builder.add_from_file("/srv/svn/pandora/zero-server-wizard/trunk/install-files/usr/share/zero-server-wizard/rsrc/progress_window.glade")
303                                else:
304                                        builder.add_from_file("/usr/share/zero-server-wizard/rsrc/progress_window.glade")                               
305                                       
306                                       
307                                self.error_msg=None
308                                window=builder.get_object("window1")
309                                review_textview=builder.get_object("textbuffer1")
310                                exec_textview=builder.get_object("textbuffer2")
311                                apply_button=builder.get_object("apply_button")
312                                close_button=builder.get_object("close_button")
313                                pbar=builder.get_object("progressbar1")
314                                terminal_box=builder.get_object("scrolledwindow2")
315                               
316                               
317                               
318                               
319                                widgets=[close_button,pbar]
320                                self.pulsating=False
321                               
322                               
323                                msg=""
324                               
325                                msg+="[ " + "REMOTE USER" + " ] = " + self.template["remote_user"]+ "\n"
326                                msg+="[ " + "CLASSROOM" + " ] = " + str(self.template["number_classroom"])+ "\n"
327                                msg+="[ " + "DATA REPLICATION" + " ] = " + self.template["enable_data_replication"]+ "\n"
328                                msg+="[ "+"SERVER NAME"+" ] = " + self.template["srv_name"] + "\n"
329                                msg+="[ "+"SERVER DOMAIN NAME"+" ] = " + self.template["srv_domain_name"] + "\n"
330                                msg+="[ "+"INTERNAL INTERFACE"+" ] = " + self.template["internal_iface"] + "\n"
331                                msg+="[ "+"INTERNAL IP"+" ] = " + self.template["srv_ip"] + "\n"
332                                msg+="[ "+"INTERNAL MASK"+" ] = " + self.template["internal_mask"] + "\n"
333                                msg+="[ "+"EXTERNAL INTERFACE"+" ] = " + self.template["external_iface"] + "\n"
334                                msg+="[ "+"EXTERNAL MODE"+" ] = " + self.template["external_mode"] + "\n"
335                                if self.template["external_mode"]!="dhcp":
336                                        msg+="[ "+"EXTERNAL IP"+" ] = " + self.template["external_ip"] + "\n"
337                                        msg+="[ "+"EXTERNAL MASK"+" ] = " + self.template["external_mask"] + "\n"
338                                        msg+="[ "+"EXTERNAL GATEWAY"+" ] = " + self.template["external_gateway"] + "\n"
339                                msg+="[ "+"DNS"+" ] = " + str(self.template["dns1"]) + ", " + str(self.template["dns2"])+ "\n"
340                                       
341
342                               
343                                review_textview.set_text(msg)
344                                       
345                               
346                               
347                                def pulsating():
348                                        pbar.pulse()
349                                       
350
351                                        try:
352                                                f=open("/tmp/.zsw-log","r")
353                                                tmp="".join(f.readlines())
354                                                if self.msg_thread!=tmp:
355                                                        self.msg_thread=tmp
356                                                        exec_textview.set_text(tmp)
357                                                       
358                                                f.close()
359                                        except:
360                                                pass
361                                       
362                                        if not os.path.exists("/tmp/.zsw-on"):
363                                                pbar.set_fraction(1)
364                                                close_button.set_sensitive(True)
365                                               
366                                                if os.path.exists("/tmp/.zsw-error"):
367                                               
368                                                        f=open("/tmp/.zsw-error","r")
369                                                        self.error_msg=f.readline()
370                                                        f.close()
371                                                        os.remove("/tmp/.zsw-error")
372                                                       
373                                                        builder = Gtk.Builder()
374                                                        if os.path.exists("/srv/svn/pandora/zero-server-wizard/trunk/install-files/usr/share/zero-server-wizard/rsrc/warning.glade"):
375                                                                builder.add_from_file("/srv/svn/pandora/zero-server-wizard/trunk/install-files/usr/share/zero-server-wizard/rsrc/warning.glade")
376                                                        else:
377                                                                builder.add_from_file("/usr/share/zero-server-wizard/rsrc/warning.glade")
378                                                               
379                                                               
380                                                        window2=builder.get_object("window1")
381                                                        msg_label=builder.get_object("msg_label")
382                                                        close_button2=builder.get_object("close_button")
383                                                        def hide(widget):
384                                                                window2.destroy()
385                                                        close_button2.connect("clicked",hide)
386                                                        msg_label.set_markup("<b>"+self.error_msg+"</b>")
387                                                        window2.set_title("Error")
388                                                        window2.show()
389                                                       
390                                                        return False
391                                                       
392                                                else:
393                                                        label = Gtk.Label("Initialization complete. A reboot is required.")
394                                                        dialog = Gtk.Dialog("Zero Server Wizard", None, Gtk.DialogFlags.MODAL | Gtk.DialogFlags.DESTROY_WITH_PARENT, (Gtk.STOCK_OK, Gtk.ResponseType.ACCEPT))
395                                                        img=Gtk.Image.new_from_stock(Gtk.STOCK_APPLY,Gtk.IconSize.DIALOG)
396                                                        hbox=Gtk.HBox()
397                                                        hbox.pack_start(img,True,True,5)
398                                                        hbox.pack_start(label,True,True,5)
399                                                        hbox.show_all()
400                                                        dialog.vbox.pack_start(hbox,True,True,10)
401                                                        label.show()
402                                                        dialog.set_border_width(6)
403                                                        response = dialog.run()
404
405                                                        sys.exit(0)
406                                               
407                                               
408                                               
409                                        return True
410                               
411                                def apply_clicked(widget,data):
412                                       
413                                        self.pulsating=True
414                                        f=open("/tmp/.zsw-on","w")
415                                        f.close()
416                                        self.msg_thread=""
417                                        self.prv_msg=None
418                                        widget.set_sensitive(False)
419                                        t=multiprocessing.Process(target=self.execute,args=(self.template,True,))
420                                        #t.daemon=True
421                                        t.start()
422                                        close_button.set_sensitive(False)
423                                        GLib.timeout_add(100,pulsating)
424                                       
425                                def destroy(uno,dos):
426                                        try:
427                                                if self.pulsating:
428                                                        return True
429                                                else:
430                                                        window.destroy()
431                                        except:
432                                                pass
433                                               
434
435                                       
436                                #def apply_clicked
437                               
438                               
439                                def close(widget):
440                                        window.destroy()
441
442
443
444                                apply_button.connect("clicked",apply_clicked,None)
445                                close_button.connect("clicked",close)
446                               
447                                window.connect("delete_event",destroy)
448                               
449                                window.show()
450                               
451                                pass
452                                #def init_done
453                               
454
455        #def check_values
456       
457        def get_link_speed(self,widget,id):
458               
459                tree_iter = widget.get_active_iter()
460                if tree_iter != None:
461                        model = widget.get_model()
462                        try:
463                                speed=lliurex.net.get_device_info(model[tree_iter][0])["Speed"][0]
464                        except:
465                                speed="Unknown speed"
466                        if id==0:
467                                self.internal_speed_label.set_text(speed)
468                        else:
469                                self.external_speed_label.set_text(speed)
470                               
471               
472        #def get_link_speed
473       
474        def external_mode_signal(self,widget):
475               
476                self.manual_external_box.set_sensitive(not self.dhcp_radiobutton.get_active())
477               
478        #def ems
479       
480       
481        def close_window(self,widget):
482                if self.standalone:
483                        Gtk.main_quit()
484               
485               
486        #def close_window
487       
488       
489        def execute(self,template,gui=False):
490               
491                print("[SLAVE] Executing slave configuration...")
492                       
493                #self.template=template
494                self.log("Executing Slave configuration...")
495
496                self.core.template=dict(self.core.template.items() + self.template.items())
497                self.template=self.core.template
498
499                if not self.template.has_key("remote_ip"):
500                        self.template["remote_ip"]="localhost"
501
502               
503                if os.path.exists("/tmp/.zsw-log"):
504                        os.remove("/tmp/.zsw-log")
505               
506                for f in sorted(os.listdir("/usr/share/zero-server-wizard/types/slave/actions/")):
507                        if os.path.isfile(self.scripts_path+f):
508                               
509                                try:
510                                        if f.endswith(".py"):
511                                                self.log("Excuting " + f + " ...")
512                                                print("[SLAVE] Executing " + f + " ...")
513                                                if not gui:
514                                                        execfile(self.scripts_path+f,locals())
515                                                else:
516                                                       
517                                                        #.set_text(ret_textview.get_buffer().get_text(ret_textview.get_buffer().get_start_iter(),ret_textview.get_buffer().get_end_iter(),True)+ "\npor aqui")
518                                                        #msg+=("[INDEPENDENT] Executing " + f + " ... ")
519                                                        f_=open("/tmp/.zsw-log","a")
520                                                        f_.write("[SLAVE] Executing " + f + " ... ")
521                                                        f_.close()                                                     
522                                                        execfile(self.scripts_path+f,locals())
523                                                        f_=open("/tmp/.zsw-log","a")
524                                                        f_.write(" OK\n")
525                                                        f_.close()
526                                                       
527
528
529                                                       
530                                except Exception as e:
531                                        print "[ERROR!] " + str(e)
532
533                                        try:
534                                                f=open("/tmp/.zsw-error","w")
535                                                f.write(e.message)
536                                                f.close()
537                                        except:
538                                                pass
539                                       
540                                        if os.path.exists("/tmp/.zsw-on"):
541                                                os.remove("/tmp/.zsw-on")
542                                        return False
543                                       
544                if gui:
545                        if os.path.exists("/tmp/.zsw-on"):
546                                os.remove("/tmp/.zsw-on")
547               
548                return True     
549               
550        #def execute
551       
552        def log(self,line):
553               
554                independent_log="/tmp/slave"
555                f=open(independent_log,"a")
556                f.write( "[ "+datetime.datetime.today().strftime("%d/%m/%y %H:%M:%S") + " ] " + line + "\n")
557                f.close()
558               
559               
560        #def log
561       
562       
563       
564       
565#class Independent
Note: See TracBrowser for help on using the repository browser.