scholar.py

from __future__ import print_function

import os.path


from google.auth.transport.requests import Request

from google.oauth2.credentials import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow

from getCourseWork import getCourseWork
from updateCalendar import updateCalendar


SCOPES = ["https://www.googleapis.com/auth/classroom.courses.readonly", "https://www.googleapis.com/auth/classroom.coursework.me",
          "https://www.googleapis.com/auth/classroom.student-submissions.me.readonly", "https://www.googleapis.com/auth/classroom.courses"]


def getCreds():
    creds = None

    if os.path.exists('token.json'):
        creds = Credentials.from_authorized_user_file('token.json', SCOPES)
    # If there are no (valid) credentials available, let the user log in.
    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(
                'credentials.json', SCOPES)
            creds = flow.run_local_server(port=0)
        # Save the credentials for the next run
        with open('token.json', 'w') as token:
            token.write(creds.to_json())
    return creds


def main():
    creds = getCreds()
    print(('-' * 30))
    print('Starting\033[1;36m Scholar\033[0;0m')
    print(('-' * 30) + '\n')
    assignments = getCourseWork(creds)
    print('\n\n' + ('-' * 30) + (' ' * 40))
    print('Completed\033[1;32m Get Course Work\033[0;0m')
    updateCalendar(assignments)
    print(('-' * 30))
    print('Completed\033[1;32m Update Calendar\033[0;0m')
    print(('-' * 30))


if __name__ == "__main__":
    main()