Add download from folder

Add download from folder

Author: JE-Chen

file_automation/remote/google_drive/download/download_file.py

@@ -14,10 +14,30 @@ def download_file(file_id: str, file_name: str):
         done = False
         while done is False:
             status, done = downloader.next_chunk()
-            print(F'Download {int(status.progress() * 100)}.')
+            print(f"Download {file_name} {int(status.progress() * 100)}%.")
     except HttpError as error:
         print(f"An error occurred: {error}")
         return None
     with open(file_name, "wb") as output_file:
         output_file.write(file.getbuffer())
     return file
+
+
+def download_file_from_folder(folder_name: str):
+    try:
+        files = dict()
+        response = driver_instance.service.files().list(
+            q=f"mimeType = 'application/vnd.google-apps.folder' and name = '{folder_name}'"
+        ).execute()
+        folder = response.get("files", [])[0]
+        folder_id = folder.get("id")
+        response = driver_instance.service.files().list(
+            q=f"'{folder_id}' in parents"
+        ).execute()
+        for file in response.get("files", []):
+            download_file(file.get("id"), file.get("name"))
+            files.update({file.get("name"): file.get("id")})
+        return files
+    except HttpError as error:
+        print(f"An error occurred: {error}")
+        return None

file_automation/remote/google_drive/upload/upload_to_driver.py

@@ -8,7 +8,7 @@
 
 def upload_to_drive(file_name: str, file_path: str):
     try:
-        if Path(file_path).is_file():
+        if Path(file_path).exists():
             file_metadata = {
                 "name": file_name,
                 "mimeType": "*/*"
@@ -28,3 +28,28 @@ def upload_to_drive(file_name: str, file_path: str):
     except HttpError as error:
         print(f"An error occurred: {error}")
         return None
+
+
+def upload_to_dir(folder_id: str, file_name: str, file_path: str):
+    try:
+        if Path(file_path).exists():
+            file_metadata = {
+                "name": file_name,
+                "mimeType": "*/*",
+                "parents": [f"{folder_id}"]
+            }
+            media = MediaFileUpload(
+                file_path,
+                mimetype="*/*",
+                resumable=True
+            )
+            file_id = driver_instance.service.files().create(
+                body=file_metadata,
+                media_body=media,
+                fields="id"
+            ).execute()
+            return file_id
+        return False
+    except HttpError as error:
+        print(f"An error occurred: {error}")
+        return None

tests/unit_test/remote/google_drive/quick_test.py

@@ -1,5 +1,5 @@
+from file_automation.remote.google_drive.download.download_file import download_file_from_folder
 from file_automation.remote.google_drive.search.search_drive import search_all_file
-from file_automation.remote.google_drive.share.share_file import share_file_to_anyone
 
 print(search_all_file())
-print(share_file_to_anyone("1PlUh-iBEQF4n-0Ew0FfabVuoDZbuFpxc"))
+print(download_file_from_folder("new_folder"))