Update easy_sync.py

easy_sync.py

@@ -1,14 +1,15 @@
 import subprocess, time, threading
 from typing import List, Union
+import os, shutil, fnmatch
 
 class Channel:
-    def __init__(self,source,destination,sync_deletions=False,sync_time=60,exclude: Union[str, List, None] = None):#What args to take
+    def __init__(self,source,destination,sync_deletions=False,every=60,exclude: Union[str, List, None] = None):
         self.source = source
         self.destination = destination
-        self.stop = threading.Event()
+        self.event = threading.Event()
         self.syncing_thread = threading.Thread(target=self._sync,args=())
         self.sync_deletions = sync_deletions
-        self.sync_time = sync_time
+        self.every = every
         if not exclude:
             exclude = []
         if isinstance(exclude,str):
@@ -29,9 +30,9 @@ class Channel:
         command.extend([f'{self.source}/',f'{self.destination}/'])
         if self.sync_deletions:    
             command.append('--delete')
-        while not self.stop.is_set():
+        while not self.event.is_set():
             subprocess.run(command)
-            time.sleep(self.sync_time)
+            time.sleep(self.every)
 
     def copy(self):#Sync once
         command = self.command
@@ -43,22 +44,79 @@ class Channel:
         subprocess.run(command)
         return True
     
-    def open(self):#Handle threads
+    def start(self):#Handle threads
         if self.syncing_thread.is_alive():#Check if it's running
-            self.stop.set()
+            self.event.set()
             self.syncing_thread.join()
-        if self.stop.is_set():
-            self.stop.clear()
+        if self.event.is_set():
+            self.event.clear()
         if self.syncing_thread._started.is_set():#If it has been started before
             self.syncing_thread = threading.Thread(target=self._sync,args=())#Create a FRESH thread
         self.syncing_thread.start()#Start the thread
         return self.alive()
 
-    def close(self):#Stop the thread and close the process
+    def stop(self):#Stop the thread and close the process
         if self.alive():
-            self.stop.set()
+            self.event.set()
             self.syncing_thread.join()
             while self.alive():
                 if not self.alive():
                     break
-        return not self.alive()
+        return not self.alive()
+    
+class GarbageMan:
+    def __init__(self) -> None:
+        self.thread = threading.Thread(target=self.take_out,args=())
+        self.event = threading.Event()
+
+    def destroy(self, trash):
+        if not isinstance(trash,dict):
+            if os.path.isdir(os.path.join(self.path,trash)):
+                shutil.rmtree(os.path.join(self.path,trash))
+            elif os.path.isfile(os.path.join(self.path,trash)):
+                os.remove(os.path.join(self.path,trash))
+        else:
+            trash.Delete()
+
+    def take_out(self) -> None:
+        while not self.event.is_set():
+            for object in self.garbage:
+                trash = object["title"] if isinstance(object,dict) else object
+                if fnmatch.fnmatch(trash,self.pattern):
+                    self.destroy(object)
+            time.sleep(self.every)
+
+    def stop(self) -> None:
+        if not self.event.is_set():
+            self.event.set()
+            self.thread.join()
+        self.event.clear()
+        if self.thread._started.is_set():
+            self.thread = threading.Thread(target=self.take_out,args=())
+
+    def start(self,path: Union[str,List],every:int=30,pattern: str='') -> None:
+        if isinstance(path,list):
+            self.path = None
+            self.garbage = path
+        elif isinstance(path,str):
+            self.path = path
+            self.garbage = os.listdir(path)
+        else:
+            return "Error"
+        self.every = every
+        self.pattern = pattern
+        if self.thread.is_alive():
+            self.stop()
+        self.thread.start()
+
+    def _fake(self, trash):
+        if not isinstance(trash,dict):
+            if os.path.isdir(os.path.join(self.path,trash)):
+                with open("log.txt","a") as f:
+                    f.write(f"Fake deleted dir: {trash}")
+            elif os.path.isfile(os.path.join(self.path,trash)):
+                with open("log.txt","a") as f:
+                    f.write(f"Fake deleted file: {trash}")
+        else:
+            with open("log.txt","a") as f:
+                    f.write(f"Fake permanently deleted: {trash['title']}")