Update get_supabase.py

get_supabase.py

@@ -1,27 +1,46 @@
 import os 
-from supabase import create_client,Client
-from  dotenv  import load_dotenv
+from supabase import create_client, Client
+from dotenv import load_dotenv
+from storage3.utils import StorageException
 
-class Supabase : 
+class Supabase:
     def __init__(self):
-        pass
+        self.client = None
 
-    def init_supabase_client(self): 
+    def init_supabase_client(self):
         load_dotenv()
         try:
-            client :Client = create_client(os.getenv("SUPABASE_URL"),os.getenv("SUPABASE_KEY"))
-            
-            return client
-        except  Exception as e : 
+            self.client = create_client(os.getenv("SUPABASE_URL"), os.getenv("SUPABASE_KEY"))
+            return self.client
+        except Exception as e:
             print(f"Error initializing Supabase client: {e}")
             return None
 
-    def download_bucket_storage(self,bucket_name):
-        client = self.init_supabase_client()
-        files_list = client.storage.from_(bucket_name).list()
-        for file in files_list : 
-            client.storage.from_(bucket_name).download(file)
-                  
-L = Supabase()
-print(L.init_supabase_client())
-L.download_bucket_storage("CVs")
+    def download_bucket_storage(self, bucket_name, download_path='.'):
+        if not self.client:
+            self.init_supabase_client()
+        
+        if not self.client:
+            print("Failed to initialize Supabase client.")
+            return
+
+        try:
+            files_list = self.client.storage.from_(bucket_name).list()
+            for file in files_list:
+                file_name = file['name']
+                try:
+                    with open(os.path.join(download_path, file_name), 'wb+') as f:
+                        res = self.client.storage.from_(bucket_name).download(file_name)
+                        f.write(res)
+                    print(f"Successfully downloaded: {file_name}")
+                except StorageException as se:
+                    print(f"Error downloading {file_name}: {se}")
+                except Exception as e:
+                    print(f"Unexpected error downloading {file_name}: {e}")
+        except Exception as e:
+            print(f"Error listing files in bucket {bucket_name}: {e}")
+
+# Example usage
+if __name__ == "__main__":
+    supabase = Supabase()
+    supabase.download_bucket_storage("CVs")