Add drive download utility functions and test

services/google_drive_service.py

@@ -0,0 +1,38 @@
+from io import BytesIO
+import json
+import os
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaIoBaseDownload
+from googleapiclient.errors import HttpError
+from google.oauth2.service_account import Credentials
+
+SERVICE_ACCOUNT_FILE = os.environ['GOOGLE_SERVICE_ACCOUNT_FILE']
+
+from logger import create_log
+
+logger = create_log(__name__)
+
+service_account_info=json.loads(SERVICE_ACCOUNT_FILE)
+scopes = ['https://www.googleapis.com/auth/drive']
+credentials = Credentials.from_service_account_info(service_account_info, scopes=scopes)
+
+drive_service = build('drive', 'v3', credentials=credentials)
+
+def get_drive_file(file_id: str) -> BytesIO:
+    request = drive_service.files().get_media(fileId=file_id)
+    file = BytesIO()
+
+    try:
+        downloader = MediaIoBaseDownload(file, request)
+
+        done = False
+        while not done:
+            status, done = downloader.next_chunk()
+
+    except HttpError as error:
+        logger.warning(f"HTTP error occurred when downloading Drive file {file_id}: {error}")
+        return None
+    except Exception as err:
+        logger.exception(f"Error occurred when downloading Drive file {file_id}: {error}")
+
+    return file

tests/integration/test_google_integration.py

@@ -0,0 +1,14 @@
+import os
+from load_env import load_env_file
+
+load_env_file('local-compose', file_string='config/local-compose.yaml')
+
+from services.google_drive_service import get_drive_file
+
+TEST_ID = '1GBskazIv6j2BjOolNYqJKKIJC8iYhTcs'
+
+def test_get_drive_file():
+    file = get_drive_file(TEST_ID)
+
+    assert file != None
+    assert file.seek(0, 2) > 1