Changeset 6432 for lliurex-gdrive/trunk/WIP_files/lliurexGdriveIndicator

Timestamp:

Dec 5, 2017, 6:00:35 PM (3 years ago)

Author:

jrpelegrina

Message:

WIP in notify changes

File:

• lliurex-gdrive/trunk/WIP_files/lliurexGdriveIndicator (modified) (10 diffs)

lliurex-gdrive/trunk/WIP_files/lliurexGdriveIndicator

@@ -8 +8 @@
 
 import os
+from os import listdir
+from os.path import isfile, isfile,isdir,join
 import threading
 import subprocess
@@ -16 +18 @@
 import notify2
 import time
+import datetime
+import copy
 
 
@@ -45 +49 @@
 SP8=RSRC+"rsrc/sp8.png"
 
+formato = '%d-%m-%y %H:%M:%S'
 TIME_CHECK_STATUS=900000
 
 class LliurexGdriveIndicator:
 
-        WATCH_DIR=os.path.expanduser("~/.config/lliurex-google-drive-profiles/configProfiles")
+        WATCH_DIR=os.path.expanduser("~/.config/lliurex-google-drive-profiles/")
         
         def __init__(self,icon_name):
@@ -66 +71 @@
                 self.profiles_info=self.LliurexGoogleDriveManager.profiles_config.copy()
                 self.current_status={}
+                self.blacklist=['.Trash','.Trash-1000']
+                self.init_files=[]
+                self.update_files=[]
+                self.local_changes={}
 
                         
@@ -146 +155 @@
                                 self.main.update_info()
                                 self.main.initial_connection=True
-                                for profile in self.main.update_profiles:
-                                        if not profile in self.main.profiles_info:
-                                                mountpoint=self.main.update_profiles[profile]["mountpoint"]
-                                                self.main.add_menu_item(profile,mountpoint)
-
-                                for item in self.main.menu.get_children():
-                                        if item.profile !="":
-                                                if not item.profile in self.main.update_profiles:
-                                                        self.main.menu.remove(item)     
-                                
-                                self.main.profiles_info=self.main.update_profiles.copy()
-                                self.main.menu.show_all()                                       
+
+                                if os.path.expanduser(event.pathname)==os.path.join(LliurexGdriveIndicator.WATCH_DIR,"configProfiles"):
+                                        
+                                        for profile in self.main.update_profiles:
+                                                mountpoint_new=self.main.update_profiles[profile]["mountpoint"]
+                                                gdriveFolder_new=self.main.update_profiles[profile]["gdrive_folder"]
+                                                if not profile in self.main.profiles_info:
+                                                        self.main.add_menu_item(profile,mountpoint_new)
+                                                        wdd=wm.add_watch(mountpoint_new,mask,rec=True)
+                                                else:
+                                                        mountpoint_old=self.main.profiles_info[profile]["mountpoint"]
+                                                        gdriveFolder_old=self.main.profiles_info[profile]["gdrive_folder"]
+                                                        if (mountpoint_new != mountpoint_old) or (gdriveFolder_new != gdriveFolder_old): 
+                                                                self.main.remove_folder_info(profile)
+                                                                ide=wm.get_wd(mountpoint_old)
+                                                                wdd=wm.rm_watch(ide,rec=True)
+                                                                self.main.init_folders_info(profile,mountpoint_new)
+                                                                wdd=wm.add_watch(mountpoint_new,mask,rec=True)
+
+                                        profile_rm=""
+                                        for item in self.main.menu.get_children():
+                                                if item.profile !="":
+                                                        if not item.profile in self.main.update_profiles:
+                                                                self.main.menu.remove(item)
+                                                                profile_rm=item.profile
+
+                                        if profile_rm !="":
+                                                mountpoint=self.main.profiles_info[profile_rm]["mountpoint"]
+                                                self.main.remove_folder_info(profile_rm)
+                                                ide=wm.get_wd(mountpoint)
+                                                wdd=wm.rm_watch(ide,rec=True)                           
                                                                 
-        
+
+                                        self.main.profiles_info=self.main.update_profiles.copy()
+                                        self.main.menu.show_all()
+
+                                else:
+                                        if 'mountToken' in os.path.basename(event.pathname):
+                                                        tmp_profile=os.path.basename(event.pathname).split("__")[0]
+                                                        print tmp_profile
+                                                        for item in self.main.profiles_info:
+                                                                if item==tmp_profile:
+                                                                        tmp_mountpoint=self.main.profiles_info[item]["mountpoint"]
+                                                                        if os.path.exists(event.pathname):
+                                                                                self.main.remove_folder_info(item)
+                                                                                ide=wm.get_wd(tmp_mountpoint)
+                                                                                wdd=wm.rm_watch(ide,rec=True)
+                                                                                self.main.init_folders_info(item,tmp_mountpoint)
+                                                                                wdd=wm.add_watch(tmp_mountpoint,mask,rec=True)
+                                                                        else:
+                                                                                self.main.remove_folder_info(item)
+                                                                                ide=wm.get_wd(tmp_mountpoint)
+                                                                                wdd=wm.rm_watch(ide,rec=True)   
+
+                                        else:                           
+                                                time.sleep(2)
+                                                t=join(event.path,event.name)
+                                                print t
+                                                if isfile(t):
+                                                        try:
+                                                                
+                                                                estado=os.stat(t)
+                                                        except Exception as e:
+                                                                print str(e)
+                                                                t=t.split('trashinfo')
+                                                                t=t[0]+'trashinfo'
+                                                                estado=os.stat(t)
+                                                                
+                                                                                
+                                                        acceso=datetime.datetime.fromtimestamp(estado.st_mtime)
+                                                        self.main.local_changes[t]=acceso
+                                                                                
+                
                 notifier=Notifier(wm,Process_handler(self))
                 wdd=wm.add_watch(LliurexGdriveIndicator.WATCH_DIR,mask,rec=True)
-                        
+                
+                for profile in self.profiles_info:
+                        path=os.path.expanduser(self.profiles_info[profile]["mountpoint"])
+                        wdd=wm.add_watch(path,mask,rec=True)
+
+                
                 while True:
                         try:
-                                        
+                                
                                 notifier.process_events()
                                 if notifier.check_events():
-                                        notifier.read_events()
-                                
+                                                notifier.read_events()
+
                         except Exception as e:
+                                print str(e)
                                 notifier.stop()
-                                        
+                
                 return False
         
@@ -249 +323 @@
                 item.status=False
                 self.menu.insert(item,4)
+                self.init_folders_info(profile,mountpoint)
 
         #def add_menu_item      
@@ -298 +373 @@
         #def populate_menu
         
-        
+        def init_folders_info(self,profile,mountpoint):
+
+                self.folders_info={}
+                self.folders_info['profile']=profile
+                self.files=set()
+                self.controlDirectory(mountpoint)
+                self.folders_info['files']=self.files
+                self.init_files.append(self.folders_info)
+                        
+        #def init_folders_info                          
+
+
+        def controlDirectory(self,path):
+                
+                
+                for item in listdir(path):
+                        if item not in self.blacklist:
+                                
+                                t=join(path,item)
+
+                                if isfile(t):
+                                        
+                                        compartido=False
+                                        estado=os.stat(t)
+                                        acceso=datetime.datetime.fromtimestamp(estado.st_mtime)
+                                        last_update=acceso.strftime(formato)
+                                        tmp=t+":__"+str(acceso)
+                                        
+                                        self.files.add(tmp)
+                                else:
+                                        if isdir(t):
+                                                self.controlDirectory(t)        
+
+        #def controlDirectory                                   
+                                                
+        def remove_folder_info(self,profile):
+
+                i=0
+                for item in self.init_files:
+                        print item
+                        if profile ==item["profile"]:
+                                print "borrado OK"
+                                self.init_files.pop(i)
+                        i=i+1   
+
+        #def remove_folder_info         
+        
+
         def item_clicked(self,widget,profile):
         
@@ -487 +609 @@
                         for profile in self.profiles_info:
                                 if self.connection:
+                                        cmd="google-drive-ocamlfuse -cc -label %s"%profile
+                                        os.system(cmd)
                                         mountpoint=self.profiles_info[profile]["mountpoint"]
                                         status_info=self.LliurexGoogleDriveManager.check_mountpoint_status(mountpoint)
+                                        self.update_folders_info(profile,mountpoint)
+
                                 else:
                                         status_info={}
@@ -507 +633 @@
                 
                 #self.result_connect_threads.pop(id)            
+                self.notify_changes()
                 return False    
 
@@ -516 +643 @@
 
         #def check_connect      
+
+        
+        def update_folders_info(self,profile,mountpoint):
+
+                self.folders_info={}
+                self.folders_info['profile']=profile
+                self.files=set()
+                self.controlDirectory(mountpoint)
+                self.folders_info['files']=self.files
+                self.update_files.append(self.folders_info)     
+
+        #def update_folders_info        
+
+        def notify_changes(self):
+
+
+                if self.init_files!=self.update_files:
+                        toInfo=self.remove_local_changes()
+                        
+                        for item in toInfo:
+                                numFiles_up=0
+                                numFiles_rm=0
+                                numFiles_add=0
+                                p=item['profile']
+                                if 'files_changes' in item:
+                                        numFiles_up=len(item['files_changes'])
+                                if 'files_delete' in item:      
+                                        numFiles_rm=len(item['files_delete'])
+                                if 'files_add' in item: 
+                                        numFiles_add=len(item['files_add'])
+                                if numFiles_up>0 or numFiles_rm>0 or numFiles_add>0:
+                                        message1=_(': changes in last 15 minutes:\n')
+                                        message_up=""
+                                        message_rm=""
+                                        message_add=""          
+                                        if numFiles_up>0:
+                                                message_up=_("-Files updated: ")
+                                                if numFiles_up==1:
+                                                        for element in item['files_changes']:
+                                                                file=os.path.basename(element.split(":__")[0])
+                                                                file_extension=os.path.splitex(file)
+                                                                if file_extension[1]==".desktop":
+                                                                        file=file_extension[0]
+
+                                                                message_up=message_up + file +'\n'
+                                
+                                                else:   
+                                                        message_up=message_up+str(numFiles_up) + '\n'
+
+                                        if numFiles_add>0:
+                                                message_add=_("-Files added: ")
+                                                if numFiles_add==1:
+                                                                for element in item['files_add']:
+                                                                        file=os.path.basename(element.split(":__")[0])
+                                                                        file_extension=os.path.splitex(file)
+                                                                        if file_extension[1]==".desktop":
+                                                                                file=file_extension[0]
+                                                                        message_add=message_add + file +'\n'
+                                                        else:
+                                                                message_add=message_add+str(numFiles_add) + '\n'
+
+                                        if numFiles_rm>0:       
+                                                message_rm=_("-Files deleted: ")
+                                                if numFiles_rm==1:
+                                                        for element in item['files_delete']:
+                                                                file=os.path.basename(element.split(":__")[0])  
+                                                                file_extension=os.path.splitex(file)
+                                                                if file_extension[1]==".desktop":
+                                                                        file=file_extension[0]
+                                                                message_rm= message_rm + file +'\n'
+                                                else:           
+                                                        message_rm= message_rm+str(numFiles_rm)
+                                                        print message_rm
+                                        
+                                        self.show_message(p + message1 + message_add + message_up + message_rm)
+
+                self.init_files=copy.copy(self.update_files)
+                self.local_changes={}
+
+        def remove_local_changes(self):
+
+        changes=self.detect_changes()
+        changes_toNotify=[]
+
+        if len(changes)>0:
+                for item in changes:
+                        tmp_changes=[]
+                        tmp_deletes=[]
+                        tmp_add=[]
+                        to_info={}
+                        to_info['profile']=item['profile']
+
+                        if 'files_changes' in item:
+                                tmp_changes=item['files_changes']
+                                toInfo_up=tmp_changes.copy()
+                        if 'files_deletes' in item:
+                                tmp_deletes=item['files_deletes']       
+                                toInfo_rm=tmp_deletes.copy()
+                        if 'files_add' in item:
+                                tmp_add=item['files_add']       
+                                toInfo_add=tmp_add.copy()
+
+                        if len(self.local_changes)>0:
+                                if len(tmp_changes)>0:
+                                        for item in tmp_changes:
+                                                #file=item.keys()[0]
+                                                file=item.split(":__")[0]
+                                                data=item.split(":__")[1]
+                                                for i in self.local_changes:
+                                                        if file==i:
+                                                                data=datetime.datetime.strptime(data,'%Y-%m-%d %H:%M:%S')
+                                                                if data<=self.local_changes[i]:
+                                                                        toInfo_up.remove(item)
+                                        to_info['files_changes']=toInfo_up                              
+                                if len(tmp_deletes)>0:
+                                        for item in tmp_deletes:
+                                                file=os.path.basename(item.split(":__")[0])
+                                                file=os.path.splitext(file)[0]
+                                                data=item.split(":__")[1]
+                                                for i in self.local_changes:
+                                                        print i
+                                                        if file in i:
+                                                                data=datetime.datetime.strptime(data,'%Y-%m-%d %H:%M:%S')
+                                                                if data<=self.local_changes[i]:
+                                                                        toInfo_rm.remove(item)
+                                        to_info['files_delete']=toInfo_rm                                       
+
+                                if len(tmp_add)>0:
+                                        for item in tmp_add:
+                                                file=item.split(":__")[0]
+                                                data=item.split(":__")[1]
+                                                for i in self.local_changes:
+                                                        if file==i:
+                                                                data=datetime.datetime.strptime(data,'%Y-%m-%d %H:%M:%S')
+                                                                if data<=self.local_changes[i]:
+                                                                        toInfo_add.remove(item) 
+                                        to_info['files_add']=toInfo_add
+                        else:
+                                to_info['files_changes']=tmp_changes
+                                to_info['files_delete']=tmp_deletes
+                                to_info['files_add']=tmp_add            
+
+                        changes_toNotify.append(to_info)
+
+        return changes_toNotify 
+
+        #remove_local_changes
+
+        def detect_changes(self):
+
+                changes=[]
+                                
+                
+                for item in self.update_files:
+                        info_changes={}
+                        info_changes['profile']=item['profile']
+                        final_files=item['files']
+
+                        for item in self.init_files:
+                                profile_init=item['profile']
+                                original_files=item['files']
+                                if len(original_files)>0:
+                                        if len(final_files)>0:
+                                                if info_changes['profile']==profile_init:
+                                                        diff_changes=final_files.difference(original_files)
+                                                        diff_deletes=original_files.difference(final_files)
+                                                        if len(diff_changes)>0:
+                                                                changes.append(info_changes)
+                                                                if len(diff_deletes)>0: 
+                                                                                tmp_del=diff_deletes.copy()
+                                                                                tmp_add=diff_changes.copy()
+                                                                                for item in diff_deletes:
+                                                                                        file_d=item.split(":__")[0]
+                                                                                        for element in diff_changes:
+                                                                                                file_c=element.split(":__")[0]
+                                                                                                if file_d == file_c:
+                                                                                                        tmp_del.remove(item)
+                                                                                                        tmp_add.remove(element)
+                                                                                if len(tmp_del)>0:
+                                                                                        info_changes['files_deletes']=tmp_del
+
+                                                                                if len(tmp_add)>0:
+                                                                                        info_changes['files_add']=tmp_add
+                                                                                        tmp_up=diff_changes.difference(tmp_add)
+                                                                                        info_changes['files_changes']=tmp_up
+                                                                                else:
+                                                                                        info_changes['files_changes']=diff_changes      
+                                                                                                
+                                                                else:
+                                                                        info_changes['files_add']=diff_changes                  
+                                                        else:
+                                                                if len(original_files)>len(final_files) and len(diff_deletes)>0:
+                                                                        info_changes['files_deletes']=diff_deletes
+                                                                        changes.append(info_changes)
+                                                        
+                return changes                                  
+
+        #def detect_changes