add_events.py

## Adds your timetable from `data.txt` to Google Calendar.
from __future__ import print_function
import httplib2
import os

from apiclient import discovery
import oauth2client
from oauth2client import client
from oauth2client import tools
from oauth2client import file
import json
import datetime

DEBUG = False

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

SCOPES = 'https://www.googleapis.com/auth/calendar'
CLIENT_SECRET_FILE = 'client_secret.json'
APPLICATION_NAME = 'gyft'

def next_weekday(d, weekday):
    days_ahead = weekday - d.weekday()
    if days_ahead <= 0: # Target day already happened this week
        days_ahead += 7
    return d + datetime.timedelta(days_ahead)

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 = os.path.join(credential_dir,
                                   'calendar-python-quickstart.json')

    store = file.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

### days to number
days = {}
days["Monday"] = 0
days["Tuesday"] = 1
days["Wednesday"] = 2
days["Thursday"] = 3
days["Friday"] = 4
days["Saturday"] = 5
###

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.
    """
    now = datetime.datetime.now()
    credentials = get_credentials()
    http = credentials.authorize(httplib2.Http())
    service = discovery.build('calendar', 'v3', http=http)
    # Get your timetable
    with open('data.txt') as data_file:    
        data = json.load(data_file)
    # Get subjects code and their respective name
    with open('subjects.json') as data_file:    
        subjects = json.load(data_file)
    # Get latlong of classrooms
    with open('full_location.json') as data_file:    
        latlong = json.load(data_file)    
    for day in data:
        startDate = next_weekday(now, days[day])
        for time in data[day]:
            # parsing time from time_table dict
            # currently we only parse the starting time
            # the end time might be having error of 5 minutes
            # ex. : A class ending at 17:55 might be shown ending at 18:00
            startHour = ""
            startMinute = ""
            startMeridiem = ""
            counter = 0
            for i in range(len(time)):
                if (time[i] == ":"):
                    counter += 1
                    continue
                if (time[i] == "-"):
                    break
                if counter == 0:
                    startHour += time[i]
                elif counter == 1:
                    startMinute += time[i]
                else:
                    startMeridiem += time[i]
            replaceHour = 0
            if (startMeridiem == "PM"):
                replaceHour += 12
            if (startHour != "12"):
                replaceHour += int(startHour)
            event = {}
            # Currently there are no labs in `subjects.json`
            # if (data[day][time][0] in subjects.keys()):
            if (data[day][time][0] in subjects.keys()):
                event['summary'] = subjects[data[day][time][0]].title()
            else: 
                event['summary'] = data[day][time][0]
            if (data[day][time][1] in latlong.keys()):
            	event['location'] = latlong[data[day][time][1]].title()
            else:                  
            	event['location'] = data[day][time][1]
            event['start'] = {}
            start_time = startDate.replace(hour = replaceHour, minute = int(startMinute))
            event['start']['dateTime'] = start_time.__str__().replace(" ", "T")
            event['start']['timeZone'] = "Asia/Kolkata"
            event['end'] = {}
            event['end']['dateTime'] = (start_time + datetime.timedelta(hours = int(data[day][time][2]))).__str__().replace(" ", "T")
            event['end']['timeZone'] = "Asia/Kolkata"
            event['recurrence'] = ['RRULE:FREQ=WEEKLY;UNTIL=20170419T000000Z']
            recurring_event = service.events().insert(calendarId='primary', body=event).execute()
            if (DEBUG):
                print (event)
                break
        if (DEBUG):
            break
    print ("\n\nEvents added to calendar\n")

if __name__ == '__main__':
    main()