Update google_function.py

google_function.py

@@ -21,7 +21,8 @@ SCOPES = ["https://www.googleapis.com/auth/gmail.compose",
           "https://www.googleapis.com/auth/calendar.readonly", 
           "https://www.googleapis.com/auth/documents.readonly",
           "https://www.googleapis.com/auth/forms.body",
-          "https://www.googleapis.com/auth/spreadsheets.readonly"]
+          "https://www.googleapis.com/auth/spreadsheets.readonly", 
+          "https://www.googleapis.com/auth/drive"]
 
 #---------------------------------------------------------- LETTURA EMAIL ---------------------------------------------------------
 def converti_email_txt(body):
@@ -33,10 +34,13 @@ def converti_email_txt(body):
         body = body
     return body
 
-def leggi_gmail(max_results=10):
+def leggi_gmail(max_results=10, data_inizio = None, data_fine = None):
     links = []
     service = build("gmail", "v1", credentials=st.session_state.creds)
-    results = service.users().messages().list(userId="me", labelIds=["INBOX"], q="is:unread",  maxResults=max_results).execute()
+    start_date_str = data_inizio.strftime("%Y/%m/%d")
+    end_date_str = data_fine.strftime("%Y/%m/%d")
+    query = f"is:unread after:{start_date_str} before:{end_date_str}"    
+    results = service.users().messages().list(userId="me", labelIds=["INBOX"], q=query,  maxResults=max_results).execute()
     messages = results.get("messages", [])
     testo_email = ''
     if not messages:
@@ -68,15 +72,18 @@ def leggi_gmail(max_results=10):
                                 body = converti_email_txt(body)
                                 break
                 else:
-                    body = base64.urlsafe_b64decode(msg["payload"]["body"]["data"]).decode("utf-8")
-                    body = converti_email_txt(body)
+                    if "data" in msg["payload"]["body"]:
+                        body = base64.urlsafe_b64decode(msg["payload"]["body"]["data"]).decode("utf-8")
+                        body = converti_email_txt(body)
             testo_email += 'Data Ricezione: ' + received_date[5:] + '\nMittente: ' + from_name + '\nOggetto: ' + ', '.join(subject) + '\nTesto Email: ' + body + '\n---------------------------------------------------\n'
             links.append(('Mittente: ' + from_name, 'Data: ' + received_date[5:] + '\n\nOggetto: ' + ', '.join(subject)))
+    if testo_email == '':  
+        testo_email = 'Non sono presenti email nell intervallo di date specificate! Riprova con altre date'
     return testo_email, links
             
 #---------------------------------------------------------- SCRITTURA BOZZA EMAIL ---------------------------------------------------------            
 def scrivi_bozza_gmail(testo):
-    draft_url = ''
+    draft_url = 'https://mail.google.com/mail/u/0/#drafts'
     try:
         service = build("gmail", "v1", credentials=st.session_state.creds)
         message = EmailMessage()
@@ -93,18 +100,87 @@ def scrivi_bozza_gmail(testo):
         print(draft_details)
     except HttpError as error:
         print(f"An error occurred: {error}")
-    return draft_url
+    return draft_url    
+
+#---------------------------------------------------------- CREA DOCUMENTO GOOGLE  ---------------------------------------------------------            
+def crea_documento_google(testo):
+    drive_service = build('drive', 'v3', credentials=st.session_state.creds)
+    docs_service = build('docs', 'v1', credentials=st.session_state.creds)
+    file_metadata = {'name': 'Documento di Bonsi AI', 'mimeType': 'application/vnd.google-apps.document'}
+    new_document = drive_service.files().create(body=file_metadata).execute()
+    document_id = new_document['id']
+   
+    title_text = "Documento di Bonsi AI"
+    title_index = 1
+    bold_title_request = {
+        "insertText": {
+            "location": {
+                "index": title_index,
+            },
+            "text": title_text
+        }
+    }
+    bold_request = {
+        "updateTextStyle": {
+            "range": {
+                "startIndex": title_index,
+                "endIndex": title_index + len(title_text)
+            },
+            "textStyle": {
+                "bold": True,
+                "fontSize": {
+                    "magnitude": 20,
+                    "unit": "PT"
+                }
+            },
+            "fields": "bold,fontSize"
+        }
+    }
+
+    normal_text = "\n\n" + testo
+    normal_text_index = title_index + len(title_text)
+    normal_text_request = {
+        "insertText": {
+            "location": {
+                "index": normal_text_index,
+            },
+            "text": normal_text
+        }
+    }
+    normal_request = {
+            "updateTextStyle": {
+                "range": {
+                    "startIndex": normal_text_index, 
+                    "endIndex": normal_text_index + len(normal_text)
+                },
+                "textStyle": {
+                    "bold": False,
+                    "fontSize": {
+                        "magnitude": 12,
+                        "unit": "PT"
+                    }
+                },
+                "fields": "bold,fontSize"
+            }
+        }
+
+    docs_service.documents().batchUpdate(documentId=document_id, body={
+        'requests': [bold_title_request, bold_request, normal_text_request, normal_request]
+    }).execute()
+    return 'https://docs.google.com/document/d/' + document_id
 
 #---------------------------------------------------------- LEGGI GOOGLE CALENDAR  ---------------------------------------------------------            
-def leggi_calendario_google(max_results=10):
+def leggi_calendario_google(max_results=10, data_inizio = None, data_fine = None):
     try:
         service = build("calendar", "v3", credentials=st.session_state.creds)
         calendar_list_result = service.calendarList().list().execute()
         calendars = calendar_list_result.get('items', [])
         descrizione_eventi = ''
         links = []
+        timeMin = datetime.datetime.combine(data_inizio, datetime.time()).isoformat() + 'Z'
+        timeMax = datetime.datetime.combine(data_fine, datetime.time(23, 59, 59)).isoformat() + 'Z'
         for calendar in calendars: 
-            events_result = (service.events().list(calendarId=calendar['id'], timeMin=datetime.datetime.now().isoformat() + 'Z', maxResults=max_results, singleEvents=True, orderBy="startTime", ).execute())
+            events_result = (service.events().list(calendarId=calendar['id'], timeMin=timeMin, timeMax = timeMax, maxResults=max_results, singleEvents=True, orderBy="startTime", ).execute())
             events = events_result.get("items", [])
             for event in events:
                 start = event["start"].get("dateTime", event["start"].get("date"))
@@ -129,6 +205,8 @@ def leggi_calendario_google(max_results=10):
                 links.append((f'Calendario: {calendario}', descrizione_link))
     except HttpError as error:
         print(f"An error occurred: {error}")
+    if descrizione_eventi == '':
+        descrizione_eventi = 'Non sono presenti eventi nel calendario per le date specificate. Riprova con altre date!'
     return descrizione_eventi, links
 
 #---------------------------------------------------------- CONNESSIONE ACCOUNT GOOGLE ---------------------------------------------------------