8ae1e64987
Co-authored-by: Charles Packer <packercharles@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Shubham Naik <shub@memgpt.ai>
98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
# Enabling API control on Google Calendar requires a few steps:
|
|
# https://developers.google.com/calendar/api/quickstart/python
|
|
# including:
|
|
# pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib
|
|
|
|
import os
|
|
import os.path
|
|
import traceback
|
|
from typing import Optional
|
|
|
|
from google.auth.transport.requests import Request
|
|
from google.oauth2.credentials import Credentials
|
|
from google_auth_oauthlib.flow import InstalledAppFlow
|
|
from googleapiclient.discovery import build
|
|
from googleapiclient.errors import HttpError
|
|
|
|
# If modifying these scopes, delete the file token.json.
|
|
# SCOPES = ["https://www.googleapis.com/auth/calendar.readonly"]
|
|
SCOPES = ["https://www.googleapis.com/auth/calendar"]
|
|
TOKEN_PATH = os.path.expanduser("~/.letta/gcal_token.json")
|
|
CREDENTIALS_PATH = os.path.expanduser("~/.letta/google_api_credentials.json")
|
|
|
|
|
|
def schedule_event(
|
|
self,
|
|
title: str,
|
|
start: str,
|
|
end: str,
|
|
# attendees: Optional[List[str]] = None,
|
|
# attendees: Optional[list[str]] = None,
|
|
description: Optional[str] = None,
|
|
# timezone: Optional[str] = "America/Los_Angeles",
|
|
) -> str:
|
|
"""
|
|
Schedule an event on the user's Google Calendar. Start and end time must be in ISO 8601 format, e.g. February 1st 2024 at noon PT would be "2024-02-01T12:00:00-07:00".
|
|
|
|
Args:
|
|
title (str): Event name
|
|
start (str): Start time in ISO 8601 format (date, time, and timezone offset)
|
|
end (str): End time in ISO 8601 format (date, time, and timezone offset)
|
|
description (Optional[str]): Expanded description of the event
|
|
|
|
Returns:
|
|
str: The status of the event scheduling request.
|
|
"""
|
|
|
|
creds = None
|
|
# The file token.json stores the user's access and refresh tokens, and is
|
|
# created automatically when the authorization flow completes for the first
|
|
# time.
|
|
if os.path.exists(TOKEN_PATH):
|
|
creds = Credentials.from_authorized_user_file(TOKEN_PATH, SCOPES)
|
|
# If there are no (valid) credentials available, let the user log in.
|
|
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_PATH, SCOPES)
|
|
creds = flow.run_local_server(port=0)
|
|
# Save the credentials for the next run
|
|
with open(TOKEN_PATH, "w") as token:
|
|
token.write(creds.to_json())
|
|
|
|
#### Create an event
|
|
# Refer to the Python quickstart on how to setup the environment:
|
|
# https://developers.google.com/calendar/quickstart/python
|
|
# Change the scope to 'https://www.googleapis.com/auth/calendar' and delete any
|
|
# stored credentials.
|
|
try:
|
|
service = build("calendar", "v3", credentials=creds)
|
|
|
|
event = {
|
|
"summary": title,
|
|
# "location": "800 Howard St., San Francisco, CA 94103",
|
|
"start": {
|
|
"dateTime": start,
|
|
"timeZone": "America/Los_Angeles",
|
|
},
|
|
"end": {
|
|
"dateTime": end,
|
|
"timeZone": "America/Los_Angeles",
|
|
},
|
|
}
|
|
|
|
# if attendees is not None:
|
|
# event["attendees"] = attendees
|
|
|
|
if description is not None:
|
|
event["description"] = description
|
|
|
|
event = service.events().insert(calendarId="primary", body=event).execute()
|
|
return "Event created: %s" % (event.get("htmlLink"))
|
|
|
|
except HttpError as error:
|
|
traceback.print_exc()
|
|
|
|
return f"An error occurred while trying to create an event: {str(error)}"
|