source: epi/trunk/fuentes/python3-epi/epimanager.py @ 6611

Last change on this file since 6611 was 6611, checked in by jrpelegrina, 3 years ago

WIP: Added support to download function in the script

File size: 16.1 KB
Line 
1#!/usr/bin/env python3
2
3import os
4import subprocess
5import sys
6import json
7import platform
8#import socket
9import tempfile
10import time
11import datetime
12import urllib.request
13
14import lliurexstore.storeManager as StoreManager
15
16
17class EpiManager:
18       
19        def __init__(self):
20
21
22                self.storeManager=StoreManager.StoreManager()
23                self.reposPath="/etc/apt/sources.list.d/"
24                self.sources_list="epi.list"
25                self.epi_sources=os.path.join(self.reposPath,self.sources_list)
26                self.epi_keyring="/tmp/epi_keyring"
27                                               
28                self.order=0
29                self.epiFiles={}
30                self.arquitecture=False
31                self.update=False
32                self.zomando_name={}
33                self.epi_base={"repository":[],
34                                        "force32" : False,
35                                        "required_x" : False,
36                                        "script": {},
37                                        "depends":"",
38                                        "zomando":"",
39                                        "required_root":False,
40                                        "download_byscript":False
41                                        }
42
43                #self.read_conf(epi_file)
44       
45               
46        #def __init__   
47
48       
49        def check_connection(self):
50       
51                try:
52                        '''
53                        s=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
54                        host=socket.gethostbyname('lliurex.net')
55                        result=s.connect_ex((host, 80))
56                        s.close()
57                        if result:
58                                return False
59                       
60                        return True
61                        '''
62                        #req=urllib.request("http://lliurex.net")
63                        res=urllib.request.urlopen("http://lliurex.net")
64                        return True
65                       
66                except:
67                        return False   
68
69        #def check_connection   
70
71        def read_conf(self,epi_file):
72
73
74                #file=os.path.join(CONF_FOLDER,conf_file)
75                #self.zomando_name[self.order]=os.path.splitext(conf_file)[0]
76               
77                if os.path.exists(epi_file):
78                        f=open(epi_file)
79                        epi_load=json.load(f)
80                        epi_conf=self.epi_base.copy()
81                        f.close()
82                        epi_conf.update(epi_load)
83                        self.epiFiles[self.order]=epi_conf
84                        self.zomando_name[self.order]=epi_conf["zomando"]
85                        try:
86                                if epi_conf["depends"]!="":
87                                        self.order=self.order+1
88                                        self.read_conf(epi_conf["depends"])
89                        except :
90                                pass           
91
92        #def read_conf         
93
94
95        def get_pkg_info(self):
96
97                pkg_list=[]
98                self.pkg_info={}
99                tmp_list=self.epiFiles.copy()
100
101                for item in tmp_list:
102                        pkg_list=[]
103                        #if tmp_list[item]["type"]!="file":
104                        pkg_list=tmp_list[item]["pkg_list"]
105                       
106                        '''
107                        else:
108                               
109                                element={}
110                                element["name"]=tmp_list[item]["name"]
111                                element["key_store"]=tmp_list[item]["key_store"]
112                                pkg_list.append(element)
113                        '''
114                        info=self.get_store_info(pkg_list,item)
115
116                        cont=0
117
118                        for element in pkg_list:
119                                name=element["name"]
120                                if info[name]["status"]=="installed":
121                                        cont=cont+1
122
123                        if cont==len(pkg_list):
124                                if item>0:
125                                        self.epiFiles.pop(item)
126                                        self.zomando_name.pop(item)
127                                else:
128                                        self.epiFiles[item]["status"]="installed"
129                                        self.pkg_info.update(info)
130                        else:
131                                self.epiFiles[item]["status"]="availabled"
132                                self.pkg_info.update(info)     
133
134        #def get_pkg_info                               
135                                                       
136
137        def get_store_info(self,pkg_list,order):                       
138
139                        pkg_info={}
140                       
141                        for item in pkg_list:
142                                app=item["name"]
143                                name=""
144                                summary=""
145                                status=""
146                                description=""
147                                icon=""
148                                debian_name=""
149                                component=""
150                                pkg=item["key_store"]
151                                action="info"
152                                self.storeManager.execute_action(action,pkg)
153                                while self.storeManager.is_action_running(action):
154                                        time.sleep(0.2)
155
156                                ret=self.storeManager.get_status(action)
157
158                                if ret["status"]==0:
159                                        data=self.storeManager.get_result(action)
160
161                                        if len(data)>0:
162                                                if (data["info"][0]["state"]) !="":
163                                                        status=data["info"][0]["state"]
164                                                else:
165                                                        status=self.check_pkg_status(app,order)
166                                                       
167
168                                                description=data["info"][0]["description"]
169                                                icon=data["info"][0]["icon"]
170                                                name=data["info"][0]["name"]
171                                                summary=data["info"][0]["summary"]
172                                                debian_name=data["info"][0]["package"]
173                                                component=data["info"][0]["component"]
174
175                                        else:
176                                                status=self.check_pkg_status(app,order)         
177                                else:
178                                        status=self.check_pkg_status(app,order)         
179                               
180                                pkg_info[app]={}
181                                pkg_info[app]["debian_name"]=debian_name
182                                pkg_info[app]["component"]=component
183                                pkg_info[app]["status"]=status
184                                pkg_info[app]["description"]=description
185                                pkg_info[app]["icon"]=icon
186                                pkg_info[app]["name"]=name
187                                pkg_info[app]["summary"]=summary
188
189                        return pkg_info
190
191        #def get_store_info                     
192
193        def check_pkg_status(self,pkg,order=None):
194       
195
196                cmd='dpkg -l '+ pkg + '| grep "^i[i]"'
197                p=subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
198                poutput,perror=p.communicate()
199
200                if len(poutput)>0:
201                        return "installed"
202                else:
203                        if order !=None:
204                                if len(self.epiFiles[order]["script"])>0:
205                                        try:
206                                                if self.epiFiles[order]["script"]["getStatus"]:
207
208                                                        if os.path.exists(self.epiFiles[order]["script"]["name"]):
209                                                                cmd=self.epiFiles[order]["script"]["name"] +' getStatus';
210                                                                p=subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE)
211                                                                poutput=p.communicate()
212                                                                if len(poutput)>0:
213                                                                        if poutput[0].decode("utf-8").split("\n")[0]=='0':
214                                                                                return "installed"
215                                               
216                                        except Exception as e:
217                                                #print (str(e))
218                                                pass   
219                               
220                return "available"     
221
222        #def check_pkg_status                                   
223               
224        def check_root(self):
225
226                self.root=False
227
228                try:
229                        f=open("/etc/epi.token","w")
230                        f.close()
231                        os.remove("/etc/epi.token")
232                        self.download_path="/var/cache/epi-downloads"
233                        self.root=True
234                except:
235                        if not os.path.exists(os.path.expanduser("~/.cache/epi-downloads/")):
236                                os.mkdir(os.path.expanduser("~/.cache/epi-downloads/"))
237                        self.download_path=os.path.expanduser("~/.cache/epi-downloads/")
238
239        #def check_root         
240
241
242        def required_root (self):
243
244                cont=0
245
246                if not self.root:
247                        for item in self.epiFiles:
248                                if self.epiFiles[item]["type"]!="file":
249                                        cont=cont+1
250
251                                else:
252                                        if self.epiFiles[item]["required_root"]:
253                                                cont=cont+1
254                                                       
255                        if cont>0:
256                                return True
257                        else:
258                                return False   
259
260                else:
261                        return False           
262
263        #def required_root             
264
265
266       
267        def check_update_repos(self):
268               
269                #Only update repos if needed
270                current_date=datetime.date.today().strftime('%y%m%d')
271                filename='/var/cache/apt/pkgcache.bin'
272                lastmod=os.path.getmtime(filename)
273                lastupdate=datetime.datetime.fromtimestamp(lastmod).strftime('%y%m%d')
274                cmd=""
275
276                if current_date !=lastupdate or self.update:
277                        cmd="LANG=C LANGUAGE=en apt-get update; "
278                else:
279                        for item in self.epi_conf["pkg_list"]:
280                                app=item["name"]
281                                command="LANG=C LANGUAGE=en apt-cache policy %s"%app
282                                p=subprocess.Popen(command,shell=True, stdout=subprocess.PIPE)
283                                output=p.communicate()
284                                if str(output[0]) != '':
285                                        if str(output[0]).split("\\n")[2].split(":")[1]=="":
286                                                cmd="LANG=C LANGUAGE=en apt-get update; "
287                                                return cmd
288
289                                else:
290                                        cmd="LANG=C LANGUAGE=en apt-get update; "
291                                        return cmd             
292
293                return cmd
294               
295               
296
297        def check_arquitecture(self):
298
299                self.force32=self.epi_conf['force32']
300                cmd=""
301               
302                if self.force32:
303                        if platform.architecture()[0]=='64bit':
304                                cmd='dpkg --add-architecture i386; '
305                                                               
306                self.arquitecture=True
307                return cmd             
308
309        #def check_arquitecture
310       
311        def add_repository_keys(self,order):
312
313                self.epi_conf=self.epiFiles[order]
314
315                cmd=""
316                self.type=self.epi_conf["type"]
317
318                if self.type=="apt":
319
320                        repos_list=self.epi_conf["repository"]
321
322                        if len(repos_list)>0:
323                                f = open(self.epi_sources,'w')
324                                for item in repos_list:
325                                        if item["url"]!="":
326                                                f.write(item["url"]+'\n')
327                                        try:
328                                                key_cmd=item["key_cmd"]
329                                                if key_cmd !="":
330                                                        cmd=cmd+key_cmd+';'     
331                                        except Exception as e:
332                                                if len(self.epi_conf["script"])>0:
333                                                        try:
334                                                                if self.epi_conf["script"]["addRepoKeys"]:
335                                                                        if os.path.exists(self.epi_conf["script"]["name"]):
336                                                                                command=self.epi_conf["script"]["name"] + ' addRepoKeys;'
337                                                                                cmd=cmd+command
338                                                        except Exception as e:
339                                                                print (str(e))
340                                                                pass
341
342                                f.close()
343                                self.update=True
344
345                       
346                return cmd             
347
348        #def add_repository_keys       
349
350
351        def get_app_version(self,item=None):
352
353                self.force32=self.epi_conf["force32"]
354
355                if self.force32:
356                       
357                        #if self.type=="deb":
358                        version=item["version"]["32b"]
359                        '''
360                        else:
361                                version=self.epi_conf["version"]["32b"]
362                        '''     
363                else:
364                        try:
365                                version=item["version"]["all"]
366                        except Exception as e:
367                                if platform.architecture()[0]=='64bit': 
368                        #if self.type=="deb":
369                        #if item["version"]["all"]=="":
370                                        version=item["version"]["64b"]
371                                else:
372                                        version=item["version"]["32b"]
373                        '''
374                        else:
375                                if self.epi_conf["version"]["all"]=="":
376                                        version=self.epi_conf["version"]["64b"]
377                                else:
378                                        version=self.epi_conf["version"]["all"]                 
379                        '''
380                return version
381
382        #def get_app_version   
383                                               
384        def download_app(self):
385
386                self.manage_download=True
387                self.download_folder={}
388                cmd=""
389
390                self.type=self.epi_conf["type"]
391                if self.type !='apt':
392
393                        self.token_result_download=tempfile.mkstemp("_result_download")
394
395                       
396                        if self.type=="file":
397                                if self.epi_conf["download_byscript"]:
398                                        self.manage_download=False
399                                        if len(self.epi_conf["script"])>0:
400                                                script=self.epi_conf["script"]["name"]
401                                                if os.path.exists(script):
402                                                        cmd=script +' download; echo $? >' + self.token_result_download[1] +';'
403
404                        if self.manage_download:
405
406                                for item in self.epi_conf["pkg_list"]:
407                                        version=self.get_app_version(item)
408                                        if self.type=="deb":
409                                                name=item["name"]+".deb"
410                                                tmp_file=os.path.join(self.download_path,name)
411                                        else:
412                                                try:
413                                                        tmp_file=os.path.join(self.download_path,item["alias_download"])
414                                                except Exception as e: 
415                                                        #name=item["name"]
416                                                        tmp_file=os.path.join(self.download_path,version)
417                               
418                                        url=item["url_download"]
419                                       
420                                        if os.path.exists(tmp_file):
421                                                cmd=cmd+'rm -f '+ tmp_file +';'
422                                        self.download_folder["name"]=tmp_file
423                                        cmd=cmd+'wget ' +url+version + ' --progress=bar:force -O ' + tmp_file +'; '
424                                        #cmd=cmd+'curl -o ' + tmp_file +' -C - '+ url + version
425
426                                cmd=cmd + ' echo $? >' + self.token_result_download[1] +';'     
427                        '''
428                        if self.type=="deb":
429
430                                for item in self.epi_conf["pkg_list"]:
431                                        name=item["name"]+".deb"
432                                        version=self.get_app_version(item)
433                                        url=item["url_download"]
434                                        tmp_file=os.path.join(self.download_path,name)
435                                        if os.path.exists(tmp_file):
436                                                cmd=cmd+'rm -f '+ tmp_file +';'
437                                        self.download_folder["name"]=tmp_file
438                                        #cmd=cmd+'wget ' +url+version + ' --progress=bar:force -O ' + tmp_file
439                                        cmd=cmd+'curl -o ' + tmp_file +' -C - '+ url + version
440
441                                cmd=cmd + '; echo $? >' + self.token_result_download[1] +';'     
442
443                        else:
444                                if self.epi_conf["url_download"]!="":
445                                        name=self.epi_conf["name"]
446                                        version=self.get_app_version()
447                                        url=self.epi_conf["url_download"]
448                                        tmp_file=os.path.join(self.download_path,version)
449                                        if os.path.exists(tmp_file):
450                                                cmd=cmd+'rm -f '+ tmp_file +';'
451                                        self.download_folder["name"]=tmp_file   
452                                        #cmd=cmd+'wget ' +url+version + ' --progress=bar:force -O ' + tmp_file + '; echo $? >' + self.token_result_download[1] +';'
453                                        cmd=cmd+'curl -o ' + tmp_file +' -C - '+ url + version + '; echo $? >' + self.token_result_download[1] +';'
454
455                        '''
456
457                return cmd                     
458                                       
459        #def download_app               
460
461
462        def check_download(self):
463
464               
465                result=True
466
467                if self.type !='apt':
468
469                       
470                        if os.path.exists(self.token_result_download[1]):
471                                file=open(self.token_result_download[1])
472                                content=file.readline()
473                                if '0' not in content:
474                                        result=False
475                                file.close()
476                                os.remove(self.token_result_download[1])
477
478                                if result:     
479                                        if self.manage_download:
480                                                pkgs_todownload=len(self.download_folder)
481                                                cont=0
482
483                                                for item in self.download_folder:
484                                                        if os.path.exists(self.download_folder[item]):
485                                                                cont=cont+1
486
487                                                if cont != pkgs_todownload:
488                                                        result=False
489
490                return result
491
492        #def check_download             
493
494        def preinstall_app(self):
495       
496
497                cmd=""
498
499                if len(self.epi_conf["script"])>0:
500                        self.token_result_preinstall=tempfile.mkstemp("_result_preinstall")
501                        script=self.epi_conf["script"]["name"]
502                        if os.path.exists(script):
503                                cmd=script +' preInstall; echo $? >' + self.token_result_preinstall[1] +';'
504
505                return cmd             
506
507
508        def check_preinstall(self):
509               
510                result=True
511
512                try:
513                        if os.path.exists(self.token_result_preinstall[1]):
514                                file=open(self.token_result_preinstall[1])
515                                content=file.readline()
516                                if '0' not in content:
517                                        result=False
518                                file.close()
519                                os.remove(self.token_result_preinstall[1])
520
521                except:                 
522                        pass
523
524                return result
525
526
527        #def preinstall_app     
528
529        def install_app(self):
530       
531                add_i386=""
532               
533                if not self.arquitecture:
534                        add_i386=self.check_arquitecture()
535
536
537                cmd=""
538               
539                if self.type=="apt":
540
541                        update_repos=self.check_update_repos()
542                        cmd=update_repos+add_i386 + "LANG=C LANGUAGE=en DEBIAN_FRONTEND=noninteractive apt-get install --reinstall --allow-downgrades --allow-remove-essential --allow-change-held-packages --yes "
543                        for item in self.epi_conf["pkg_list"]:
544                                app=item["name"]
545                                cmd=cmd + app +" "
546
547       
548                       
549                elif self.type=="deb":
550                       
551                        cmd=add_i386 + "dpkg -i "
552                        for item in self.epi_conf["pkg_list"]:
553                                name=item["name"]+".deb"
554                                pkg=self.download_folder["name"]
555                                cmd=cmd+pkg +" "
556
557                       
558                else:
559                        self.token_result_install=tempfile.mkstemp("_result")
560                        script=self.epi_conf["script"]["name"]
561                        if os.path.exists(script):
562                                cmd=script + ' install; echo $? >' + self.token_result_install[1]
563
564                cmd=cmd+";"
565                return cmd     
566
567        #def install_app       
568
569
570        def check_install_remove(self,action):
571
572                dpkg_status={}
573                cont=0
574                token=""
575               
576                if action=="install":
577                               
578                        if self.type !="file":
579                                pkgs=self.epi_conf["pkg_list"]
580                       
581                                for item in pkgs:
582                                        status=self.check_pkg_status(item["name"])
583                               
584                                        if status=="installed":
585                                                cont=cont+1
586                               
587                                        dpkg_status[item["name"]]=status
588
589                                if cont==len(pkgs):
590                                        result=True
591               
592                                else:
593                                        result=False
594
595                        else:
596                                token=self.token_result_install[1]     
597                                if os.path.exists(token):
598                                        file=open(token)
599                                        content=file.readline()
600                                        if '0' not in content:
601                                                result=False
602                                        else:
603                                                result=True     
604                                                                               
605                                        file.close()
606                                        os.remove(token)
607
608                else:
609               
610                        if self.epiFiles[0]["type"] !="file":
611                                        pkgs=self.epiFiles[0]["pkg_list"]                       
612                                        for item in pkgs:
613                                                status=self.check_pkg_status(item["name"])
614                                                if status!="installed":
615                                                        cont=cont+1
616                                                dpkg_status[item["name"]]=status
617
618               
619                        token=self.token_result_remove[1]
620                        if os.path.exists(token):
621                                        file=open(token)
622                                        content=file.readline()
623                                        if '0' not in content:
624                                                result=False
625                                        else:
626                                                result=True     
627                                                       
628                                        file.close()
629                                        os.remove(token)
630
631                return dpkg_status,result                       
632
633       
634               
635        #def check_install_remove       
636
637        def postinstall_app(self):
638       
639
640                cmd=""
641               
642                if len(self.epi_conf["script"])>0:
643                        self.token_result_postinstall=tempfile.mkstemp("_result_postinstall")
644                        script=self.epi_conf["script"]["name"]
645                        if os.path.exists(script):
646                                cmd=script + ' postInstall; echo $? >' + self.token_result_postinstall[1] +';'
647
648                return cmd     
649
650        #def postinstall_app   
651       
652        def check_postinstall(self):
653               
654                result=True
655
656                try:
657                        if os.path.exists(self.token_result_postinstall[1]):
658                                file=open(self.token_result_postinstall[1])
659                                content=file.readline()
660                                if '0' not in content:
661                                        result=False
662                                file.close()
663                                os.remove(self.token_result_postinstall[1])
664                except:
665                        pass                   
666
667                return result
668
669        #def check_postinstall 
670
671        def remove_repo_keys(self):
672       
673                if os.path.exists(self.epi_sources):
674                        os.remove(self.epi_sources)     
675
676                if os.path.exists(self.epi_keyring):
677                        os.remove(self.epi_keyring)     
678
679        #def remove_repo_keys   
680
681        def unistall_app(self,order):
682
683                cmd=""
684
685                if self.epiFiles[order]["script"]["remove"]:
686                        self.token_result_remove=tempfile.mkstemp("_result_remove")
687                        script=self.epiFiles[order]["script"]["name"]
688                        if os.path.exists(script):
689                                cmd=script + ' remove; echo $? >' + self.token_result_remove[1] + ';'
690
691                return cmd
692
693        def zerocenter_feedback(self,order,action,result=None):
694
695                zomando_name=self.zomando_name[order]
696
697                if zomando_name!="":
698                        if action=="init":
699                                cmd="zero-center add-pulsating-color " +zomando_name
700                        elif action=="install":
701                                if result:
702                                        cmd="zero-center remove-pulsating-color "+zomando_name + " ;zero-center set-configured " +zomando_name
703                                       
704                                else:
705                                        cmd="zero-center remove-pulsating-color "+zomando_name + " ;zero-center set-failed " +zomando_name
706                        elif action=="uninstall":
707                                if result:
708                                        cmd="zero-center remove-pulsating-color "+zomando_name + " ;zero-center set-non-configured " +zomando_name
709                                else:
710                                        cmd="zero-center remove-pulsating-color "+zomando_name + " ;zero-center set-failed " +zomando_name
711
712                        os.system(cmd)         
713
714        #def zero-center_feedback       
715
716
717#class ApplicationInstallerManager
718
719
720if __name__=="__main__":
721       
722        epi=EpiManager()
Note: See TracBrowser for help on using the repository browser.