From a955af5513cf9fe78649ae09dbf4bc41bacd56ff Mon Sep 17 00:00:00 2001
From: BIRK Renaud <renaud.birk@imt-atlantique.net>
Date: Sun, 9 Mar 2025 11:54:36 +0100
Subject: [PATCH] feat: Google Drive controller to handle wrong file names
---
src/controllers/gdrive_controller.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/src/controllers/gdrive_controller.py b/src/controllers/gdrive_controller.py
index 2a111ca..fa1d17f 100644
--- a/src/controllers/gdrive_controller.py
+++ b/src/controllers/gdrive_controller.py
@@ -9,7 +9,15 @@ class GDriveController:
SERVICE_ACCOUNT_FILE = "service_account.json"
SCOPES = ["https://www.googleapis.com/auth/drive.file"]
- creds = service_account.Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)
+ creds = None
+ try:
+ creds = service_account.Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)
+ except Exception:
+ try:
+ creds = service_account.Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE + ".json", scopes=SCOPES)
+ except Exception:
+ creds = service_account.Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE[:-5], scopes=SCOPES)
+
self.drive_service = build("drive", "v3", credentials=creds)
def post_gdrive(self, local_file_path: str, folder_id: str) -> Response:
--
GitLab