Fetches holidays from Google calendar

google-calendar-fetch.py

from __future__ import print_function

import datetime
import json

import httplib2
from apiclient import discovery
from oauth2client import client
from oauth2client import tools
from oauth2client.file import Storage

try:
    import argparse

    flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args()
except ImportError:
    flags = None

# If modifying these scopes, delete your previously saved credentials
# at ~/.credentials/calendar-python-quickstart.json
SCOPES = 'https://www.googleapis.com/auth/calendar.readonly'
CLIENT_SECRET_FILE = 'client_secret.json'
APPLICATION_NAME = 'Google Calendar API Python Quickstart'

START_YEAR = 2014
END_YEAR = 2022


def get_credentials():
    """Gets valid user credentials from storage.

    If nothing has been stored, or if the stored credentials are invalid,
    the OAuth2 flow is completed to obtain the new credentials.

    Returns:
        Credentials, the obtained credential.
    """
    # home_dir = os.path.expanduser('.')
    # credential_dir = os.path.join(home_dir, '.credentials')
    # if not os.path.exists(credential_dir):
    #     os.makedirs(credential_dir)
    credential_path = 'calendar-credentials.json'

    store = Storage(credential_path)
    credentials = store.get()
    if not credentials or credentials.invalid:
        flow = client.flow_from_clientsecrets(CLIENT_SECRET_FILE, SCOPES)
        flow.user_agent = APPLICATION_NAME
        if flags:
            credentials = tools.run_flow(flow, store, flags)
        else:  # Needed only for compatibility with Python 2.6
            credentials = tools.run(flow, store)
        print('Storing credentials to ' + credential_path)
    return credentials


def main():
    """Shows basic usage of the Google Calendar API.

    Creates a Google Calendar API service object and outputs a list of the next
    10 events on the user's calendar.
    """
    credentials = get_credentials()
    http = credentials.authorize(httplib2.Http())
    service = discovery.build('calendar', 'v3', http=http)

    startDate = datetime.datetime(START_YEAR, 1, 1).isoformat() + 'Z'
    endDate = datetime.datetime(END_YEAR, 12, 31).isoformat() + 'Z'

    calendarIds = (
        ('ge', 'en.ge#[email protected]'),
        ('ru', 'en.russian#[email protected]'),
        ('de', 'en.german#[email protected]'),
        ('us', 'en.usa#[email protected]'),
        ('ua', 'en.ukrainian#[email protected]'),
    )

    all_events = []

    for countryKey, calendarId in calendarIds:
        eventsResult = service.events().list(
            # calendarId='#[email protected]', timeMin=now, maxResults=10, singleEvents=True,
            calendarId=calendarId, timeMin=startDate, timeMax=endDate, singleEvents=True,
            orderBy='startTime').execute()
        events = eventsResult.get('items', [])

        if not events:
            print('No upcoming events found.')

        for event in events:
            event['countryKey'] = countryKey
            start = event['start'].get('dateTime', event['start'].get('date'))
            print(countryKey, start, event['summary'])

        all_events.extend(events)

    with open('../web/holidays.json'.format(START_YEAR, END_YEAR), 'w') as outfile:
        json.dump(all_events, outfile)


if __name__ == '__main__':
    main()