wp-cal-integration/adapters/google.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import datetime
import logging

from google.auth.transport.requests import Request
#from google.oauth2.credentials import Credentials
from google.oauth2.service_account import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError

logger = logging.getLogger(f"wp_cal.{__name__}")

# If modifying these scopes, delete the file token.json.
SCOPES = ['https://www.googleapis.com/auth/calendar.readonly']


class Google():

    def __init__(self, calendar_id, *, credentials, token_file):
        self.calendar_id = calendar_id
        self.credentials_info = credentials
        self.credentials = None

    def login(self):
        self.credentials = Credentials.from_service_account_info(self.credentials_info, scopes=SCOPES)
        if not self.credentials.valid:
            self.credentials.refresh(Request())
        logger.debug('credentials.valid = %s', repr(self.credentials.valid))


    def get_events(self, start: datetime.datetime | None=None, until: datetime.timedelta | None=None, limit=None):
        until = until if until else datetime.timedelta(days=365)
        now = start if start else datetime.datetime.utcnow().astimezone()
        now_365 = now + until

        try:
            service = build('calendar', 'v3', credentials=self.credentials)
            events = service.events().list(
                calendarId=self.calendar_id,
                timeMin=now.isoformat(),
                timeMax=now_365.isoformat(),
                maxResults=limit,
            ).execute()
        except HttpError:
            logger.exception("a http error occured")
            raise
        events = events.get('items', [])
        return events