Support to upload file to Google Shared Drive (#21319)

apache · Feb 6, 2022 · 6b88d43 · 6b88d43
1 parent 1a77bc6
commit 6b88d43
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/airflow/providers/google/suite/hooks/drive.py b/airflow/providers/google/suite/hooks/drive.py
@@ -200,7 +200,7 @@ def upload_file(self, local_location: str, remote_location: str) -> str:
         media = MediaFileUpload(local_location)
         file = (
             service.files()
-            .create(body=file_metadata, media_body=media, fields="id")
+            .create(body=file_metadata, media_body=media, fields="id", supportsAllDrives=True)
             .execute(num_retries=self.num_retries)
         )
         self.log.info("File %s uploaded to gdrive://%s.", local_location, remote_location)

diff --git a/tests/providers/google/suite/hooks/test_drive.py b/tests/providers/google/suite/hooks/test_drive.py
@@ -245,6 +245,7 @@ def test_upload_file_to_root_directory(
                     body={"name": "remote_path", "parents": ["root"]},
                     fields="id",
                     media_body=mock_media_file_upload.return_value,
+                    supportsAllDrives=True,
                 )
             ]
         )
@@ -274,6 +275,7 @@ def test_upload_file_to_subdirectory(
                     body={"name": "remote_path", "parents": ["PARENT_ID"]},
                     fields="id",
                     media_body=mock_media_file_upload.return_value,
+                    supportsAllDrives=True,
                 )
             ]
         )