source: open-sysclone/trunk/fuentes/open-sysclone-server.install/usr/share/n4d/python-plugins/OpenSysClone.py @ 2381

Last change on this file since 2381 was 2381, checked in by daduve, 3 years ago

New translations to n4d results in OpenSysClone? GUI

File size: 11.1 KB
Line 
1
2import sys
3import tempfile
4import threading
5import os
6import glob
7from jinja2 import Environment
8from jinja2.loaders import FileSystemLoader
9
10
11class OpenSysClone:
12
13        # Templates variables
14        TEMPLATES_PATH="/usr/share/n4d/templates/OpenSysClone/"
15        SAVE_IMAGE_TPL="SaveImage.tpl"
16        PXE_DESTINATION="/var/www/ipxeboot/pxemenu.d/70-OpenSysClone.php"
17        PATH_FILE="/net/OpenSysClone/"
18        USER_PATH_FILE="imagenes"
19        NAME_FILE="autoname"
20        HDD_DISK="sda1"
21        FINAL_ACTION="poweroff"
22        MIN_CLIENTS = "8"
23        MAX_WAIT = "100"
24        UDP_PATH="/net/OpenSysClone/prueba"
25        INTERFACE="eth0"
26        HOST_OPERATION="hostnamesaver.sh"
27        EXPORT_NFS_PATH="/etc/exports.d/"
28        EXPORT_NFS_NAME="opensysclone_nfs.exports"
29       
30
31        def __init__(self):
32                pass
33        #def init
34       
35        def startup(self,options):
36                if objects["VariablesManager"].get_variable('OPENSYSCLONE_SQUASHFS_PROTOCOL') is None:
37                        objects["VariablesManager"].init_variable('OPENSYSCLONE_SQUASHFS_PROTOCOL')
38
39        #def startup
40
41        def write_menu_pxe(self, aux_save_image_tpl=SAVE_IMAGE_TPL, aux_user_path_file = USER_PATH_FILE, aux_name_file = NAME_FILE , aux_hdd_disk = HDD_DISK ,aux_final_action=FINAL_ACTION, aux_host_operation=HOST_OPERATION):
42                       
43                try:   
44                        # save_path, name_file, hdd_disk, final_action
45                        environment_variables = {}
46                        aux_path_file=self.PATH_FILE+aux_user_path_file
47                       
48                        # Get the values from free server
49                        if  objects.has_key("VariablesManager"):
50                                environment_variables=objects["VariablesManager"].get_variable_list(["SRV_IP","OPENSYSCLONE_SQUASHFS_PROTOCOL"])
51                               
52                                #Prepare PATH to TFTP or HTTP protocol
53                                protocolo=objects["VariablesManager"].get_variable("OPENSYSCLONE_SQUASHFS_PROTOCOL")
54                               
55                                if protocolo=="http":
56                                        path_squashfs="ipxeboot/opensysclone-system"
57                                else:
58                                        path_squashfs="opensysclone-system"
59                       
60                       
61                        environment_variables["PATH_SQUASHFS"]=path_squashfs
62                        environment_variables["PATH_FILE"]=aux_path_file
63                        environment_variables["NAME_FILE"]=aux_name_file
64                        environment_variables["HDD_DISK"]=aux_hdd_disk
65                        environment_variables["FINAL_ACTION"]=aux_final_action
66                        environment_variables["HOST_OPERATION"]=aux_host_operation
67                       
68                       
69                        path_to_work=tempfile.mkdtemp()
70                        filename=path_to_work+"OpenSysClone.pxe"
71                       
72                        # Create temporal environment for jinja
73                        env = Environment(loader=FileSystemLoader(OpenSysClone.TEMPLATES_PATH))
74                        tmpl = env.get_template(aux_save_image_tpl)
75                       
76                        # Render the template with diferent values             
77                        textrendered=tmpl.render(environment_variables)
78                       
79                        #Create a temporal for nsswitch
80                        tmp,filename=tempfile.mkstemp()
81                        f = open(filename,'w')
82                        f.writelines(textrendered)
83                        f.close()
84                       
85                        # Using the ultimate chmod
86                        self.uchmod(filename,0644)
87                       
88                        # Copy unitaria
89                        shutil.copy(filename,OpenSysClone.PXE_DESTINATION)
90                        COMMENT_END = "PXE menu is prepared with protocol %s for SQUASHFS, your requirements have writed in this file %s" %(protocolo,OpenSysClone.PXE_DESTINATION)
91                        return [True,str(COMMENT_END),[protocolo,OpenSysClone.PXE_DESTINATION]]
92               
93                except Exception as e:
94
95                        return [False,str(e)]   
96       
97       
98        #def write_menu_pxe
99       
100       
101       
102        def del_menu_pxe(self, AUX_PXE_DESTINATION=PXE_DESTINATION):
103               
104                try:   
105                        # Remove obsolete file for PXE menu
106                        if os.path.isfile(AUX_PXE_DESTINATION):
107                                os.remove (AUX_PXE_DESTINATION)
108                                COMMENT_END = "Remove obsolete file %s for PXE menu" %(AUX_PXE_DESTINATION)
109                       
110                                return [True, str(COMMENT_END),[AUX_PXE_DESTINATION]]
111                        else:
112                                COMMENT_END = "PXE file was delete before"
113                                return [True, str(COMMENT_END)]
114                       
115                except Exception as e:
116
117                                return [False,str(e)]
118        #def del_menu_pxe
119       
120       
121       
122        def nfs_export_start (self, AUX_USER_PATH_FILE=USER_PATH_FILE):
123               
124                try:
125                        AUX_PATH_FILE=self.PATH_FILE+AUX_USER_PATH_FILE
126                        # Get the values from free server
127                        if  objects.has_key("VariablesManager"):
128                                INTERNAL_NETWORK_VARIABLES=objects["VariablesManager"].get_variable_list(["INTERNAL_NETWORK","INTERNAL_MASK"])
129                               
130                        #Export file to write or read OpenSysClone in the server to network range
131                        #With new file arguments
132                        if os.path.dirname(AUX_PATH_FILE):
133                                result = os.system('mkdir -p %s'%self.EXPORT_NFS_PATH)
134                                try:
135                                        fich = open(self.EXPORT_NFS_PATH+"/"+self.EXPORT_NFS_NAME,"w")
136                                        fich.write(str(AUX_PATH_FILE)+"\t"+str(INTERNAL_NETWORK_VARIABLES["INTERNAL_NETWORK"])+"/"+str(INTERNAL_NETWORK_VARIABLES["INTERNAL_MASK"])+"(rw,insecure,no_root_squash,async,no_subtree_check)\n")
137                                        fich.close()
138                                except Exception as e:
139                                        return [False,'Export failed: '+str(e)]
140                                #cmd='exportfs -o rw,insecure,no_root_squash,async,no_subtree_check  %s/%s:%s'%(INTERNAL_NETWORK_VARIABLES["INTERNAL_NETWORK"],INTERNAL_NETWORK_VARIABLES["INTERNAL_MASK"],AUX_PATH_FILE)
141                                #cmd='service nfs-kernel-server restart'
142                                cmd='exportfs -ra'
143                                #print cmd
144                                os.system(cmd)
145                                COMMENT_END = "NFS is sharing %s to my internal network %s/%s" %(AUX_PATH_FILE,INTERNAL_NETWORK_VARIABLES["INTERNAL_NETWORK"],INTERNAL_NETWORK_VARIABLES["INTERNAL_MASK"])
146                                return [True, str(COMMENT_END)]
147                        else:
148                                return [False,str(e)]
149                               
150                except Exception as e:
151
152                        return [False,str(e)]
153        #def nfs_export
154       
155       
156       
157        def nfs_export_stop (self,AUX_USER_PATH_FILE=USER_PATH_FILE):
158       
159                        #Restart nfs service to restart default parameters and stop extra parameters
160               
161                #try:   
162                #       os.system('service nfs-kernel-server restart')
163                #       COMMENT_END = "The NFS system is restarted with defaults parameters and suprimed OpeSysClone shared parameters"
164                #       return [True, str(COMMENT_END)]
165                       
166                #except Exception as e:
167
168                #       return [False,str(e)]   
169                try:
170                        AUX_PATH_FILE=self.PATH_FILE+AUX_USER_PATH_FILE
171                       
172                        # Get the values from free server
173                        if  objects.has_key("VariablesManager"):
174                                INTERNAL_NETWORK_VARIABLES=objects["VariablesManager"].get_variable_list(["INTERNAL_NETWORK","INTERNAL_MASK"])
175                               
176                       
177                        #Cancel the export file to write or read OpenSysClone in the server to network range
178                        #Deleting file with arguments and restarting the service
179                       
180                        if os.path.dirname(AUX_PATH_FILE):
181                                result = os.system("rm -f %s"%self.EXPORT_NFS_PATH+"/"+self.EXPORT_NFS_NAME)
182                                os.system('service nfs-kernel-server restart')
183                                #os.system('exportfs -u  %s/%s:%s'%(INTERNAL_NETWORK_VARIABLES["INTERNAL_NETWORK"],INTERNAL_NETWORK_VARIABLES["INTERNAL_MASK"],AUX_PATH_FILE))
184                                COMMENT_END = "NFS is NOT SHARING %s to my internal network %s/%s" %(AUX_PATH_FILE,INTERNAL_NETWORK_VARIABLES["INTERNAL_NETWORK"],INTERNAL_NETWORK_VARIABLES["INTERNAL_MASK"])
185                                return [True, str(COMMENT_END)]
186                        else:
187                                return [False,str(e)]
188                               
189                except Exception as e:
190
191                        return [False,str(e)]
192               
193               
194        #def nfs_export_stop
195       
196       
197        def uchmod(self,file,mode):
198               
199                try:
200                #Method to change file attributes
201               
202                        prevmask = os.umask(0)
203                        os.chmod(file,mode)
204                        os.umask(prevmask)
205                       
206                except Exception as e:
207
208                        return [False,str(e)]   
209        #def uchmod
210       
211       
212        def aux_send_multicast_file (self, AUX_USER_PATH_FILE=USER_PATH_FILE, AUX_NAME_FILE=NAME_FILE, AUX_MIN_CLIENTS=MIN_CLIENTS, AUX_MAX_WAIT=MAX_WAIT,AUX_INTERFACE=INTERFACE,AUX_UDP_PATH=UDP_PATH):
213       
214                        try:
215                                #Method to restart nfs service to restart default parameters and stop extra parameters
216                                AUX_PATH_FILE=self.PATH_FILE+AUX_USER_PATH_FILE
217                                os.system('killall udp-sender')
218                                #Inicializo el puerto de UDP que debe ser 2232 para la primera particion y +2 para cada una de las siguientes
219                                AUX_UDP_PORT=2232
220                                INSTRUCTION = ''
221                               
222                                #Obtengo las particiones a clonar y las meto en una lista
223                                tmp=set()
224                               
225                                for item in glob.glob(AUX_UDP_PATH+"/*.gz.*"):
226                                        #Para obtener las particiones he de eliminar primero la primera parte del path puesto que puede llevar puntos y luego no haria bien el split
227                                        item=item.replace(AUX_UDP_PATH+"/","")
228                                        #Corto por el primer punto y me quedo con la primera parte
229                                        item=item.split(".")[0]
230                                        #Elimino las cadenas iguales
231                                        tmp.add(item)
232                               
233                                AUX_PARTITIONS_LIST=sorted(tmp)
234                               
235                                print AUX_PARTITIONS_LIST
236                               
237                                #Recorro la lista de particiones y creo la instruccion
238                               
239                                for PARTITION in AUX_PARTITIONS_LIST :
240                                        INSTRUCTION += 'cat %s/%s*.gz.* | udp-sender --full-duplex --retries-until-drop 100 --min-clients %s --max-wait %s --interface %s --nokbd --mcast-all-addr 224.0.0.1 --portbase %s --ttl 1;'%((AUX_UDP_PATH),(PARTITION),(AUX_MIN_CLIENTS),(AUX_MAX_WAIT),(AUX_INTERFACE),(AUX_UDP_PORT))
241                                        AUX_UDP_PORT=AUX_UDP_PORT+2
242                               
243                                os.system ('%s'%INSTRUCTION)
244                                COMMENT_END = "Instruction in server for MULTICAST is: %s" %(INSTRUCTION)
245                       
246                                #print COMMENT_END
247                       
248                                return [True, str(COMMENT_END)]
249                       
250                        except Exception as e:
251
252                                return [False,str(e)]
253                       
254                        #os.system('cat %(AUX_UDP_PATH)s/*.gz.* | udp-sender --full-duplex --min-clients %(AUX_MIN_CLIENTS)s --max-wait %(AUX_MAX_WAIT)s --interface %(AUX_INTERFACE)s --nokbd --mcast-all-addr 224.0.0.1 --portbase %(AUX_UDP_PORT)s --ttl 1' % locals ())
255                       
256        #def aux_send_multicast_file
257       
258       
259        def send_multicast_file (self, AUX_USER_PATH_FILE=USER_PATH_FILE, AUX_NAME_FILE=NAME_FILE, AUX_MIN_CLIENTS=MIN_CLIENTS, AUX_MAX_WAIT=MAX_WAIT):
260               
261                        #Method to execute multiprocess
262                       
263                        try:
264                                AUX_PATH_FILE=self.PATH_FILE+AUX_USER_PATH_FILE
265                                # Get the values from free server
266                                if  objects.has_key("VariablesManager"):
267                                        INTERFACE=objects["VariablesManager"].get_variable("INTERNAL_INTERFACE")
268                               
269                       
270                                #Get the UDP path to export in multicast mode
271                                UDP_PATH=AUX_PATH_FILE+"/"+AUX_NAME_FILE
272                               
273                                if os.path.exists(UDP_PATH):
274                               
275                                        COMMENT_END = "UDP is in multicast mode for my internal network interface %s" %(INTERFACE)
276                                       
277                                       
278                                        #Instructions to execute multiprocess
279                                        t = threading.Thread(target=self.aux_send_multicast_file,args=(AUX_PATH_FILE,AUX_NAME_FILE,AUX_MIN_CLIENTS,AUX_MAX_WAIT,INTERFACE,UDP_PATH))
280                                        t.daemon=True
281                                        t.start()
282                                        return [True, str(COMMENT_END)]
283                                else:
284                                        COMMENT_END = "Your image don't exists in this directory"
285                                        return [False,str(COMMENT_END)]
286                                       
287                        except Exception as e:
288
289                                return [False,str(e)]
290                       
291        #def send_multicast_file
292       
293       
294        def list_img (self, AUX_USER_PATH_FILE=USER_PATH_FILE):
295               
296                #Method to list our ISOS in Server
297               
298                try:
299                        AUX_PATH_FILE=self.PATH_FILE+AUX_USER_PATH_FILE
300                        LIST_DIR=os.listdir(AUX_PATH_FILE)
301               
302                        return [True,LIST_DIR]
303
304                except Exception as e:
305
306                                return [False,str(e)]
307                               
308        def check_path (self, AUX_USER_PATH_FILE=USER_PATH_FILE):
309               
310                #Method to make our dir to save images
311               
312                try:
313                        AUX_PATH_FILE=self.PATH_FILE+AUX_USER_PATH_FILE
314                       
315                        if os.path.exists(AUX_PATH_FILE):
316                                COMMENT_END = "Path file exists, do nothing"
317                                return [True, str(COMMENT_END)]
318                        else:
319                                os.makedirs(AUX_PATH_FILE,0777)
320                                os.chmod(AUX_PATH_FILE,0777)
321                                COMMENT_END = "Path file is created now %s" %(AUX_PATH_FILE)
322                                return [True, str(COMMENT_END)]                 
323
324                except Exception as e:
325
326                                return [False,str(e)]
327                               
328                               
329        def del_image (self, DIR_FILE_DELETE):
330               
331                #Method to delete our save images
332               
333                try:
334                        AUX1_PATH_FILE=self.PATH_FILE+self.USER_PATH_FILE
335                        AUX_PATH_FILE=AUX1_PATH_FILE+"/"+DIR_FILE_DELETE
336                       
337                        if os.path.exists(AUX_PATH_FILE):
338                                os.system('rm -R  %s'%(AUX_PATH_FILE))
339                                COMMENT_END = "Path file is deleted %s" %(AUX_PATH_FILE)
340                                return [True, str(COMMENT_END)]
341                        else:
342                                COMMENT_END = "Path file isn't exists %s" %(AUX_PATH_FILE)
343                                return [True, str(COMMENT_END)]                 
344
345                except Exception as e:
346
347                                return [False,str(e)]
348                               
349        #def list_img
350       
351       
352       
353
354
355         
Note: See TracBrowser for help on using the repository browser.