modify vedio_upload_to_gofile.py

vedio_upload_to_gofile.py

@@ -1,44 +1,60 @@
-import requests
-import shutil
 import os
+import json
+import requests
 
 import ImportantVariables
 
 
-def upload_folder_to_gofile(folder_path, zip_name="folder.zip"):
-    # Compress the folder into a ZIP file
-    print("Compress the folder into a ZIP file")
-    zip_path = f"{zip_name}"
-    shutil.make_archive(zip_path.replace(".zip", ""), 'zip', folder_path)
-
-    # Upload the ZIP file to Gofile
-    print("Upload the ZIP file to Gofile")
+def upload_file_to_gofile(file_path):
     url = "https://store1.gofile.io/uploadFile"
-    with open(zip_path, "rb") as f:
+    with open(file_path, "rb") as f:
         response = requests.post(url, files={"file": f})
 
-    # Handle response
     if response.status_code == 200:
         result = response.json()
         if result["status"] == "ok":
-            public_url = result["data"]["downloadPage"]
-            print(f"✅ Folder uploaded: {public_url}")
-            # Cleanup: Remove the ZIP file after upload
-            os.remove(zip_path)
-            return public_url
+            return result["data"]["downloadPage"]
         else:
             print("❌ Upload failed:", result)
     else:
         print(f"❌ Error: HTTP {response.status_code}")
+    return None
+
+
+def upload_folder_to_gofile(folder_path, json_name="files.json"):
+    file_urls = {}
+
+    # Upload each file in the folder
+    for root, _, files in os.walk(folder_path):
+        for file in files:
+            file_path = os.path.join(root, file)
+            print(f"Uploading {file}...")
+            file_url = upload_file_to_gofile(file_path)
+            if file_url:
+                file_urls[file] = file_url
+
+    # Save file URLs in a JSON file
+    json_path = os.path.join(folder_path, json_name)
+    with open(json_path, "w") as json_file:
+        json.dump(file_urls, json_file, indent=4)
+
+    # Upload the JSON file
+    print("Uploading JSON file...")
+    json_file_url = upload_file_to_gofile(json_path)
 
+    return json_file_url
 
 
+# Example usage
+# folder_url = upload_folder_to_gofile("your_folder_path_here")
+# print("JSON file URL:", folder_url)
 
 
 def main():
     # Example usage
     folder_path = ImportantVariables.PATHS["output_folder_path"]
     public_url=upload_folder_to_gofile(folder_path)
+    print("JSON file URL:", public_url)
     return public_url