From 68080cc89af4f173732e0cb40f99c9d2cfcc4feb Mon Sep 17 00:00:00 2001
From: Gizmotronn <liam@skinetics.tech>
Date: Tue, 15 Oct 2024 23:19:34 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A5=F0=9F=AB=80=20=E2=86=9D=20[SSP-30?=
 =?UTF-8?q?=20SSP-29=20SSP-9]:=20Script=20for=20new=20folders=20anomalies?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 buckets/uploadForDir.py | 80 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 buckets/uploadForDir.py

diff --git a/buckets/uploadForDir.py b/buckets/uploadForDir.py
new file mode 100644
index 00000000..2a72c1f7
--- /dev/null
+++ b/buckets/uploadForDir.py
@@ -0,0 +1,80 @@
+import os
+from supabase import create_client, Client
+from pathlib import Path
+
+# Initialize Supabase client
+def init_supabase_client():
+    url = "http://127.0.0.1:54321"
+    key = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZS1kZW1vIiwicm9sZSI6ImFub24iLCJleHAiOjE5ODM4MTI5OTZ9.CRXP1A7WOeoJeXxjNni43kdQwgnWNReilDMblYTn_I0"
+    return create_client(url, key)
+
+def upload_file_to_supabase(supabase: Client, bucket_name: str, file_path: str, destination_path: str):
+    with open(file_path, "rb") as file:
+        try:
+            response = supabase.storage.from_(bucket_name).upload(destination_path, file)
+            print(f"Uploaded {file_path} -> {destination_path}")
+            return True
+        except Exception as e:
+            if "Duplicate" in str(e):
+                print(f"File already exists: {file_path}. Proceeding with database insertion.")
+                return True  
+            print(f"Failed to upload {file_path} -> {destination_path}: {e}")
+            return False
+
+def check_anomaly_exists(supabase: Client, anomaly_id):
+    try:
+        response = supabase.table('anomalies').select("*").eq("id", anomaly_id).execute()
+        return len(response.data) > 0
+    except Exception as e:
+        print(f"Error checking for anomaly {anomaly_id}: {e}")
+        return False
+
+def insert_into_anomalies(supabase: Client, anomaly_id, content, anomaly_set: str):
+    if not check_anomaly_exists(supabase, anomaly_id):
+        try:
+            data = {
+                "id": anomaly_id, 
+                "content": content, 
+                "anomalytype": "zoodexOthers",
+                "anomalySet": 'zoodex-nestQuestGo',
+            }
+            response = supabase.table('anomalies').insert(data).execute()
+            print(f"Inserted anomaly with id {anomaly_id} into 'anomalies' table.")
+        except Exception as e:
+            print(f"Failed to insert anomaly {anomaly_id}: {e}")
+    else:
+        print(f"Anomaly {anomaly_id} already exists in the database. Skipping insertion.")
+
+def upload_directory_to_supabase(supabase: Client, bucket_name: str, local_directory: str):
+    # Iterate over each subfolder in the local directory
+    for anomaly_folder in os.listdir(local_directory):
+        full_path = os.path.join(local_directory, anomaly_folder)
+        if os.path.isdir(full_path):
+            # Treat each subfolder as an anomaly
+            anomaly_id = anomaly_folder  # Use the folder name as the anomaly ID
+            anomaly_set = "telescope-minorPlanet"  # Set a consistent anomaly set
+            
+            # Insert the anomaly into the database
+            insert_into_anomalies(supabase, anomaly_id, anomaly_id, anomaly_set)
+            
+            # Upload all files from the subfolder
+            for root, _, files in os.walk(full_path):
+                for file_name in files:
+                    if file_name.startswith('.'):
+                        continue
+                    
+                    file_path = os.path.join(root, file_name)
+                    relative_path = os.path.relpath(file_path, local_directory)
+                    destination_path = f"{anomaly_id}/{Path(relative_path).as_posix()}"
+
+                    upload_file_to_supabase(supabase, bucket_name, file_path, destination_path)
+
+def main():
+    supabase = init_supabase_client()
+    bucket_name = "zoodex/zoodex-nestQuestGo"
+    local_directory = "zoodex/zoodex-nestQuestGo"
+    
+    upload_directory_to_supabase(supabase, bucket_name, local_directory)
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file