v2

added feature of downloading assingments files, download all files now

automatingv2.py

@@ -0,0 +1,166 @@
+from __future__ import print_function
+import pickle
+import os.path
+from googleapiclient.discovery import build
+from google_auth_oauthlib.flow import InstalledAppFlow
+from google.auth.transport.requests import Request
+from googleapiclient.http import MediaIoBaseDownload
+import io
+import os
+import os.path
+
+
+def main():
+
+    service = get_classroom_service()
+    courses = service.courses().list(pageSize=11).execute()
+
+    for course in courses['courses']:
+
+        course_name = course['name']
+        course_id = course['id']
+
+        try:
+            os.mkdir('./' + course_name)
+        except OSError:
+            print("Creation of {} directory failed".format(course_name))
+        else:
+            print("Created {} directory ".format(course_name))
+
+        anoncs = service.courses().announcements().list(
+            courseId=course_id).execute()
+        work = service.courses().courseWork().list(
+            courseId=course_id).execute()
+
+        download_annonc_files(anoncs, course_name)
+        download_works_files(work, course_name)
+
+
+def get_classroom_service():
+
+    SCOPES = [
+        'https://www.googleapis.com/auth/classroom.courses.readonly',
+        'https://www.googleapis.com/auth/classroom.announcements.readonly',
+        'https://www.googleapis.com/auth/classroom.student-submissions.me.readonly'
+    ]
+    """Shows basic usage of the Classroom API.
+    Prints the names of the first 10 courses the user has access to.
+    """
+    creds = None
+    # The file token.pickle stores the user's access and refresh tokens, and is
+    # created automatically when the authorization flow completes for the first
+    # time.
+    if os.path.exists('token.pickle'):
+        with open('token.pickle', 'rb') as token:
+            creds = pickle.load(token)
+    # 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.pickle', 'wb') as token:
+            pickle.dump(creds, token)
+
+    service = build('classroom', 'v1', credentials=creds)
+    return service
+
+
+def download_file(file_id, file_name, course_name):
+
+    SCOPES = ['https://www.googleapis.com/auth/drive']
+    """Shows basic usage of the Drive v3 API.
+    Prints the names and ids of the first 10 files the user has access to.
+    """
+    creds = None
+    # The file token.pickle stores the user's access and refresh tokens, and is
+    # created automatically when the authorization flow completes for the first
+    # time.
+    if os.path.exists('token.pickledrive'):
+        with open('token.pickledrive', 'rb') as token:
+            creds = pickle.load(token)
+    # 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(
+                'credential-drive.json', SCOPES)
+            creds = flow.run_local_server(port=0)
+        # Save the credentials for the next run
+        with open('token.pickledrive', 'wb') as token:
+            pickle.dump(creds, token)
+
+    service = build('drive', 'v3', credentials=creds)
+
+    request = service.files().get_media(fileId=file_id)
+
+    fh = io.BytesIO()
+    downloader = MediaIoBaseDownload(fh, request)
+    done = False
+    while done is False:
+        status, done = downloader.next_chunk()
+        print("Download %d%%." % int(status.progress() * 100))
+
+    fh.seek(0)
+
+    with open(os.path.join('./', course_name, file_name), 'wb') as f:
+        f.write(fh.read())
+        f.close()
+
+
+def download_annonc_files(announcements, course_name):
+    annonc_list = announcements.keys()
+    if (len(annonc_list) != 0):
+        for announcement in announcements['announcements']:
+            try:  #if this announcements contain a file then do this
+                for val in announcement['materials']:
+                    file_id = val['driveFile']['driveFile']['id']
+                    file_name = val['driveFile']['driveFile']['title']
+                    extension = (
+                        os.path.splitext(file_name)
+                    )[1]  #the extension exists in second elemnts of returned tuple
+                    print(extension)
+
+                    if (valid(extension[1:])):
+                        print(file_name)
+                        download_file(file_id, file_name, course_name)
+                        print("HAHAHA")
+            except KeyError as e:
+                print("this announcement doesn't have any file to download")
+
+
+def download_works_files(works, course_name):
+    works_list = works.keys()
+    if (len(works_list) != 0):
+        for work in works['courseWork']:
+            try:  #if this announcements contain a file then do this
+                for val in work['materials']:
+                    file_id = val['driveFile']['driveFile']['id']
+                    file_name = val['driveFile']['driveFile']['title']
+
+                    extension = (
+                        os.path.splitext(file_name)
+                    )[1]  #the extension exists in second elemnts of returned tuple
+                    print(extension)
+                    if (valid(extension[1:])):
+                        print(file_name)
+                        download_file(file_id, file_name, course_name)
+                        print("HAHAHA")
+            except KeyError as e:
+                print("this work doesn't have any file to download")
+
+
+def valid(ch):
+    return ch in [
+        'pdf', 'docx', 'pptx', 'png', 'jpg', 'html', 'css', 'js', 'java',
+        'class', 'txt', 'r', 'm'
+    ]
+
+
+#his
+if __name__ == '__main__':
+    main()