GoogleCalendar.py

from __future__ import print_function
import datetime, os.path, pickle
from googleapiclient.discovery import build
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request
from Constants import TOKEN_FILE, CREDS_FILE, API_URL

class Calendar:
    def __init__(self):
        self.Service = None
        SCOPES = [API_URL]
        CREDS = None # Google API Function, generating the current Auth without opening the browser, so cool :).
        if os.path.exists(TOKEN_FILE):
            with open(TOKEN_FILE, 'rb') as token:
                CREDS = pickle.load(token)
        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(CREDS_FILE, SCOPES)
                CREDS = flow.run_local_server(port=0)
            with open(TOKEN_FILE, 'wb') as token:
                pickle.dump(CREDS, token)

        service = build('calendar', 'v3', credentials=CREDS)

        self.Service = service

    def addEvent(self, subjectID, eventName="", eventLocation="", eventDescription="", start="", end="", timeZone="Europe/Madrid", colorID='undefined', calendarID='primary'):
        """
        :param subjectID: This has to be unique, contain onlye Base32HEX encoding characters and length between 5 and 1024.
        :param eventName: Event name, that name is going to appear at Google Calendar GUI.
        :param eventLocation: Event location, Google Maps can use it.
        :param eventDescription: Just text about the event.
        :param start: YYYY-MM-DDTHours:Minutes:Seconds
        :param end: YYYY-MM-DDTHours:Minutes:Seconds
        :param timeZone: TimeZone of the event inside the calendar.
        :param calendarID: ID to detect the subcalendar where you want to add it.
        :return: The event object.
        """

        newEvent = \
        {
            'summary': eventName, 'location': eventLocation, 'description': eventDescription, 'colorId': colorID,
            'start':
            {
                'dateTime': start,
                'timeZone': timeZone,
            },
            'end':
            {
                'dateTime': end,
                'timeZone': timeZone,
            },
            "reminders": # By default, Google Calendar will send a reminder to the user.
            {
                "useDefault": False,
            },
            "extendedProperties":
            {
                "private":
                    {
                        "subjectID": subjectID,
                        "fromUPF": True, # Value to identify UPF events.
                    }
            }
        }

        newEvent = self.Service.events().insert(calendarId=calendarID, body=newEvent).execute()
        return newEvent

    def deleteEvent(self, eventID, calendarID='primary'):
        self.Service.events().delete(calendarId=calendarID, eventId=eventID).execute()

    def getEvent(self, eventID, calendarID='primary'):
        self.Service.events().get(calendarId=calendarID, eventId=eventID)

    def getEvents(self, fromDate=None, toDate=None, calendarID='primary'):
        eventsList = []
        pageToken = None
        while True:
            events = self.Service.events().list(calendarId=calendarID, pageToken=pageToken).execute()
            for event in events['items']:
                try:
                    eventsList.append((event['description'], event['id'], event['extendedProperties']['private']['subjectID']))
                except KeyError:
                    None
            pageToken = events.get('nextPageToken')
            if not pageToken: break
        return eventsList # List of tuples (description, eventID, subjectID)