Update google_drive_upload.py

google_drive_upload.py

@@ -1,7 +1,6 @@
 import os
-from google.oauth2.credentials import Credentials
-from google_auth_oauthlib.flow import InstalledAppFlow
-from google.auth.transport.requests import Request
+import json
+from google.oauth2.service_account import Credentials
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaFileUpload
 
@@ -9,25 +8,27 @@ from googleapiclient.http import MediaFileUpload
 SCOPES = ['https://www.googleapis.com/auth/drive.file']
 
 def authenticate_google_drive():
-    creds = None
-    if os.path.exists('token.json'):
-        creds = Credentials.from_authorized_user_file('token.json', SCOPES)
-    if not creds or not creds.valid:
-        if creds and creds.expired and creds.refresh_token:
-            creds.refresh(Request())
-        else:
-            flow = InstalledAppFlow.from_client_secrets_file('credentials.json', SCOPES)
-            creds = flow.run_local_server(port=0)
-        with open('token.json', 'w') as token:
-            token.write(creds.to_json())
+    """Autentica usando una cuenta de servicio."""
+    # Cargar las credenciales desde el secreto
+    service_account_info = json.loads(os.getenv('GOOGLE_SERVICE_ACCOUNT', '{}'))
+    if not service_account_info:
+        raise ValueError("No se encontró la información de la cuenta de servicio.")
+
+    # Crear credenciales
+    creds = Credentials.from_service_account_info(service_account_info, scopes=SCOPES)
     return creds
 
 def upload_to_google_drive(file_path):
-    creds = authenticate_google_drive()
-    service = build('drive', 'v3', credentials=creds)
+    """Sube un archivo a Google Drive."""
+    try:
+        creds = authenticate_google_drive()
+        service = build('drive', 'v3', credentials=creds)
 
-    file_metadata = {'name': os.path.basename(file_path)}
-    media = MediaFileUpload(file_path, resumable=True)
-    file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
-    print(f"Archivo subido con ID: {file.get('id')}")
-    return file.get('id')
+        file_metadata = {'name': os.path.basename(file_path)}
+        media = MediaFileUpload(file_path, resumable=True)
+        file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
+        print(f"Archivo subido con ID: {file.get('id')}")
+        return file.get('id')
+    except Exception as e:
+        print(f"Error subiendo a Google Drive: {e}")
+        return None