files

Author: itsmeakhil

packages/backend-server/app/api/v1/web/drive/files/services.py

@@ -2,7 +2,19 @@
 from app.managers.collection_names import FILES, FOLDERS
 from app.utils.utils import response_helper, create_uuid, get_file_extension, get_folders_from_path, create_timestamp
 from app.utils.i8ns import translate
-from app.utils.s3_utils import generate_upload_url
+from app.utils.s3_utils import generate_upload_url, generate_download_url
+
+
+
+def get_files_list(user, parent_id=None):
+    db = user.get("db")
+    query = {"created_by": user.get("user_id")}
+    if parent_id:
+        query["parent_id"] = parent_id
+    files = db_manager.find(db, FILES, query)
+    for file in files:
+        file["file_url"]= generate_download_url(file.get("key"), 360000)
+    return files 
 
 async def get_presigned_url(user, payload):
     db = user.get("db")

packages/backend-server/app/api/v1/web/drive/folders/services.py

@@ -3,6 +3,7 @@
 from app.managers.collection_names import FILES, FOLDERS
 from app.utils.utils import response_helper, create_uuid, create_timestamp
 from app.utils.i8ns import translate
+from app.api.v1.web.drive.files.services import get_files_list
 
 async def create_folder(user, payload):
     db = user.get("db")
@@ -74,21 +75,28 @@ async def move_folders(user, payload):
     return response_helper(200, translate("drive.folders.moved"), folders_moved=folders_moved)
 
 async def get_folders_list(user, parent_id):
-    db = user.get("db")
     query = {"created_by": user.get("user_id")}
     if parent_id:
         query["parent_id"] = parent_id
 
-    folders = db_manager.find(db, FOLDERS, query)
+    folders = get_folders(user, parent_id)
 
     for folder in folders:
-        parent_query={"parent_id": folder.get("doc_id"),"created_by": user.get("user_id")}
-        files = db_manager.find(db, FILES,parent_query )
-        folder["files"] = files
-        folder["sub_folders"] = db_manager.find(db, FOLDERS, parent_query)
+        folder["files"] = get_files_list(user, folder.get("doc_id"))
+        folder["sub_folders"] = get_folders(user, folder.get("doc_id"))
 
     data= {
         "folders": folders,
-        "files": db_manager.find(db, FILES, query)
+        "files": get_files_list(user, parent_id)
     }
-    return response_helper(200, translate("drive.folders.list"), data=data)
+    return response_helper(200, translate("drive.folders.list"), data=data)
+
+
+
+def get_folders(user, parent_id=None):
+    db = user.get("db")
+    query = {"created_by": user.get("user_id")}
+    if parent_id:
+        query["parent_id"] = parent_id
+    folders = db_manager.find(db, FOLDERS, query)
+    return folders