Lucas Oskorep
59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
from datetime import datetime
|
|
from time import sleep
|
|
|
|
from google.oauth2 import service_account
|
|
from googleapiclient.discovery import build
|
|
from googleapiclient.errors import HttpError
|
|
|
|
from event import Event
|
|
|
|
|
|
class CalGrab(object):
|
|
|
|
def __init__(self, auth_file, calendars=[], callbacks=[]):
|
|
self.calendars = calendars
|
|
# TODO: Add multiple calendar support
|
|
self.callbacks = callbacks
|
|
self.creds = service_account.Credentials.from_service_account_file(
|
|
auth_file, scopes=['https://www.googleapis.com/auth/calendar.readonly'])
|
|
try:
|
|
self.service = build('calendar', 'v3', credentials=self.creds)
|
|
except HttpError as error:
|
|
raise error
|
|
|
|
def update_at_interval(self, frequency, time_to_update=-1):
|
|
"""
|
|
|
|
:param frequency: Amount of time to wait between updates in seconds
|
|
:param time_to_update: normally set to -1 aka no limit. Time in minutes to continually ping the api for.
|
|
:return: None
|
|
"""
|
|
try:
|
|
start = None
|
|
while True:
|
|
now = datetime.utcnow() # 'Z' indicates UTC time
|
|
if start == None:
|
|
start = now
|
|
events_result = self.service.events().list(
|
|
calendarId='loskorep@productiveedge.com',
|
|
timeMin=now.isoformat() + 'Z',
|
|
maxResults=10,
|
|
singleEvents=True,
|
|
orderBy='startTime'
|
|
).execute()
|
|
events = events_result.get('items', [])
|
|
if not events:
|
|
print('No upcoming events found.')
|
|
return
|
|
|
|
events = [i for i in [Event.get_from_gcal_api_json(json) for json in events] if i is not None]
|
|
print(events)
|
|
|
|
for callback in self.callbacks:
|
|
callback(events)
|
|
if time_to_update > 0 and (now - start).total_seconds() > time_to_update:
|
|
return
|
|
sleep(frequency)
|
|
except HttpError as error:
|
|
print('An error occurred: %s' % error)
|