Update comunicacion_gmail.py

comunicacion_gmail.py

@@ -5,7 +5,6 @@ from email.mime.text import MIMEText
 import pickle
 import os.path
 import base64
-import email
 
 # Define los ámbitos que necesitas (ajusta según tus necesidades)
 SCOPES = ['https://mail.google.com/', 'https://www.googleapis.com/auth/gmail.readonly', 'https://www.googleapis.com/auth/gmail.send', 'https://www.googleapis.com/auth/gmail.modify']
@@ -24,73 +23,76 @@ def gmail_tool(accion, parametros={}):
             creds.refresh(Request())
         else:
             flow = InstalledAppFlow.from_client_secrets_file('credentials.json', SCOPES)
-            creds = flow.run_console()
-        with open('token.pickle', 'wb') as token:
-            pickle.dump(creds, token)
-
+            # Obtener URL de autorización
+            auth_url, _ = flow.authorization_url(prompt='consent')
+            return {"auth_url": auth_url, "requires_auth": True} # Indicar que se requiere autorización
+            
     service = build('gmail', 'v1', credentials=creds)
     result = {} # Inicializar result
-
-    if accion == "leer_correos":
-        # ... (implementación para leer correos) ...
-        results = service.users().messages().list(userId='me', maxResults=10).execute()
-        messages = results.get('messages', [])
-        for message in messages:
-            msg = service.users().messages().get(userId='me', id=message['id']).execute()
-
-
-            # Decodificar el mensaje (puede ser multipart)
-            payload = msg['payload']
-            parts = payload.get('parts', [])  # Verificar si hay partes
-
-            body = ""
-            if parts:
-              for part in parts:
-                  if part.get('mimeType') == 'text/plain':
-                      data = part['body'].get('data', '')
-                      body += base64.urlsafe_b64decode(data).decode()
-
-            else:  # Si no hay partes, el cuerpo está en payload['body']
-              data = payload['body'].get('data','')
-              body+= base64.urlsafe_b64decode(data).decode()
-
-            print(body)
-            result["message_id"] = send_message["id"] #  Almacenar el ID del mensaje en el resultado
-
-
-    elif accion == "enviar_correo":
-        # ... (implementación para enviar correo) ...
-        message = MIMEText('Este es el cuerpo del correo.')  # Importa MIMEText de email.mime.text
-        message['to'] = 'destinatario@example.com' #!CAMBIAR DESTINATARIO
-        message['from'] = '[email protected]'#!CAMBIAR TU CORREO
-        message['subject'] = 'Asunto del correo'
-
-        create_message = {'raw': base64.urlsafe_b64encode(message.as_bytes()).decode()}
-
-        send_message = service.users().messages().send(userId='me', body=create_message).execute()
-
-        print(F'Message Id: {send_message["id"]}')
-
-    elif accion == "verificar_almacenamiento":
-        try:
-            drive_service = build('drive', 'v3', credentials=creds)
-            about = drive_service.about().get(fields="storageQuota").execute()
-            storage_quota = about.get('storageQuota')
-                # print(storage_quota) # Para depurar
-            return storage_quota  # Devuelve la información del almacenamiento
-
-        except Exception as e:
-            print(f"Error al verificar el almacenamiento: {e}")
-            return {"error": "Error al verificar el almacenamiento."}
-
-
-
-    # ... (otras acciones)
-
-    return result
+    
+    try:
+        if accion == "leer_correos":
+            # ... (implementación para leer correos) ...
+            results = service.users().messages().list(userId='me', maxResults=parametros.get("maxResults", 10)).execute() # Usa parametros para maxResults
+            messages = results.get('messages', [])
+            result["messages"] = [] # Inicializa una lista para almacenar los mensajes
+            for message in messages:
+                msg = service.users().messages().get(userId='me', id=message['id']).execute()    
+                # Decodificar el mensaje (puede ser multipart)
+                payload = msg['payload']
+                parts = payload.get('parts', [])  # Verificar si hay partes    
+                body = ""
+                if parts:
+                    for part in parts:
+                        if part.get('mimeType') == 'text/plain':
+                            data = part['body'].get('data', '')
+                            body += base64.urlsafe_b64decode(data).decode()
+    
+                else:  # Si no hay partes, el cuerpo está en payload['body']
+                    data = payload['body'].get('data','')
+                    body+= base64.urlsafe_b64decode(data).decode()
+    
+                print(body)
+                result["messages"].append({"body": body, "id": message['id']}) #  Asocia el ID al mensaje
+    
+    
+        elif accion == "enviar_correo":
+            # ... (implementación para enviar correo) ...
+            destinatario = parametros.get("destinatario")
+            asunto = parametros.get("asunto")
+            cuerpo_correo = parametros.get("cuerpo")
+
+            message = MIMEText(cuerpo_correo)
+            message['to'] = destinatario
+            message['from'] = '[email protected]' #!CAMBIAR TU CORREO
+            message['subject'] = asunto
+    
+            create_message = {'raw': base64.urlsafe_b64encode(message.as_bytes()).decode()}
+    
+            send_message = service.users().messages().send(userId='me', body=create_message).execute()
+    
+            print(F'Message Id: {send_message["id"]}')
+            result["message_id"] = send_message["id"]  #  Almacenar el ID
+            
+        elif accion == "verificar_almacenamiento":
+            try:
+                drive_service = build('drive', 'v3', credentials=creds)
+                about = drive_service.about().get(fields="storageQuota").execute()
+                storage_quota = about.get('storageQuota')
+                    # print(storage_quota) # Para depurar
+                return storage_quota  # Devuelve la información del almacenamiento
+    
+            except Exception as e:
+                print(f"Error al verificar el almacenamiento: {e}")
+                return {"error": "Error al verificar el almacenamiento."}
+    
+    
+        return result #  Retorna el resultado al final del try...except
+        # ... (otras acciones)
         
-
-
-
+    except Exception as e:
+        print(f"Error en gmail_tool: {e}")
+        result["error"] = str(e)
+        return result   
 
 # ... (Integración con el prompt del sistema)