Update function/dropbox.py

function/dropbox.py

@@ -1,22 +1,17 @@
 import dropbox.files
 import os
 import shutil
-import requests, base64
+import requests,base64
 from fastapi import HTTPException
-from dotenv import load_dotenv
-import os
-load_dotenv()
-DROPBOX_APP_KEY=os.getenv('DROPBOX_APP_KEY')
-DROPBOX_APP_SECRET=os.getenv('DROPBOX_APP_SECRET')
-DROPBOX_REFRESH_TOKEN=os.getenv('DROPBOX_REFRESH_TOKEN')
 
 def refresh_token_dropbox():
-    app_key = DROPBOX_APP_KEY
-    app_secret = DROPBOX_APP_SECRET
-    refresh_token = DROPBOX_REFRESH_TOKEN
+    app_key = "m79mo7gg7mh8pez"
+    app_secret = "by93yed7as4qlmo"
+    refresh_token = 'VG4SP0Ugo4EAAAAAAAAAAbHf7esmak_gieNGvLHxiiAXGfSeKPVXbX1lm28DMz_o'
     url = 'https://api.dropbox.com/oauth2/token'
     auth_string = f"{app_key}:{app_secret}"
     base64authorization = base64.b64encode(auth_string.encode()).decode('utf-8')
+
     headers = {
         'Authorization': f'Basic {base64authorization}',
         'Content-Type': 'application/x-www-form-urlencoded'
@@ -90,8 +85,9 @@ def download_folder(id):
     try:
         TOKEN = refresh_token_dropbox()
         dbx = dropbox.Dropbox(TOKEN)
-        local_path = f"/code/user_file/{id}"
+        local_path = f"/code/temp/{id}"
         os.makedirs(local_path, exist_ok=True)
+        clear_local_folder(local_path)
         result = dbx.files_list_folder(f"/{id}")
         for entry in result.entries:
             if isinstance(entry, dropbox.files.FileMetadata):
@@ -107,13 +103,14 @@ def download_file_id(file_name, id):
     try:
         TOKEN = refresh_token_dropbox()
         dbx = dropbox.Dropbox(TOKEN)
-        local_folder_path = f"/code/user_file/{id}"
+        local_folder_path = f"/code/temp/{id}"
         os.makedirs(local_folder_path, exist_ok=True)
         local_file_path = os.path.join(local_folder_path, file_name)
         with open(local_file_path, "wb") as f:
             metadata, response = dbx.files_download(f"/{id}/{file_name}")
             f.write(response.content)
         print(f"Downloaded file '{file_name}' to '{local_file_path}'")
+
     except dropbox.exceptions.ApiError as e:
         print(f"Error downloading file '{file_name}': {e}")
         raise HTTPException(status_code=500, detail="Internal Server Error")
@@ -128,13 +125,14 @@ def search_and_download_file(start_char, id):
             print(f"No file found starting with '{start_char}' in folder '{id}'")
             return
         file_name = files_starting_with_char[0]
-        local_folder_path = f"/code/user_file/{id}"
+        local_folder_path = f"/code/temp/{id}"
         os.makedirs(local_folder_path, exist_ok=True)
         local_file_path = os.path.join(local_folder_path, file_name)
         with open(local_file_path, "wb") as f:
             metadata, response = dbx.files_download(f"/{id}/{file_name}")
             f.write(response.content)
         print(f"Downloaded file '{file_name}' to '{local_file_path}'")
+
     except dropbox.exceptions.ApiError as e:
         print(f"Error searching or downloading file: {e}")
         raise HTTPException(status_code=500, detail="Internal Server Error")
@@ -149,6 +147,7 @@ def delete_all_files_in_folder(folder_id):
                 file_path = entry.path_display
                 dbx.files_delete_v2(file_path)
                 print(f"Deleted file '{file_path}'")
+        
         print(f"All files in folder '{folder_id}' have been deleted.")
     except dropbox.exceptions.ApiError as e:
         print(f"Error deleting files: {e}")