80 lines
3.3 KiB
Python
80 lines
3.3 KiB
Python
|
import os
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from stuff.singlefile import download_page
|
||
|
|
from stuff.const import MAX_FOLDER_NAME_SIZE
|
||
|
|
from stuff.download import download_file
|
||
|
|
from stuff.get_canvas import get_extra_assignment_files
|
||
|
|
from stuff.helpers import make_valid_filename, shorten_file_name
|
||
|
|
|
||
|
|
|
||
|
|
def download_module_item(module, item, modules_dir, cookies_path):
|
||
|
|
# If problems arise due to long pathnames, changing module.name to module.id might help, this can also be done with item.title
|
||
|
|
# A change would also have to be made in findCourseModules(course, course_view)
|
||
|
|
module_name = make_valid_filename(str(module.name))
|
||
|
|
module_name = shorten_file_name(module_name, len(module_name) - MAX_FOLDER_NAME_SIZE)
|
||
|
|
items_dir = os.path.join(modules_dir, module_name)
|
||
|
|
|
||
|
|
if item.url != "":
|
||
|
|
if not os.path.exists(items_dir):
|
||
|
|
os.makedirs(items_dir)
|
||
|
|
|
||
|
|
filename = make_valid_filename(str(item.title)) + ".html"
|
||
|
|
module_item_dir = os.path.join(items_dir, filename)
|
||
|
|
|
||
|
|
# Download the module page.
|
||
|
|
if not os.path.exists(module_item_dir):
|
||
|
|
download_page(item.url, cookies_path, items_dir, filename)
|
||
|
|
|
||
|
|
|
||
|
|
def download_assignment(cookies_path, cookie_jar, base_assign_dir, assignment):
|
||
|
|
assignment_title = make_valid_filename(str(assignment.title))
|
||
|
|
assignment_title = shorten_file_name(assignment_title, len(assignment_title) - MAX_FOLDER_NAME_SIZE)
|
||
|
|
assign_dir = os.path.join(base_assign_dir, assignment_title)
|
||
|
|
|
||
|
|
if assignment.html_url != "":
|
||
|
|
if not os.path.exists(assign_dir):
|
||
|
|
os.makedirs(assign_dir)
|
||
|
|
|
||
|
|
assignment_page_path = os.path.join(assign_dir, "assignment.html")
|
||
|
|
|
||
|
|
if not os.path.exists(assignment_page_path):
|
||
|
|
download_page(assignment.html_url, cookies_path, assign_dir, "assignment.html")
|
||
|
|
|
||
|
|
extra_files = get_extra_assignment_files(assignment.description, cookie_jar)
|
||
|
|
for name, url in extra_files:
|
||
|
|
download_file(url, Path(assign_dir, name), cookie_jar)
|
||
|
|
|
||
|
|
for submission in assignment.submissions:
|
||
|
|
download_submission(assignment, submission, assign_dir, cookies_path)
|
||
|
|
|
||
|
|
|
||
|
|
def download_submission(assignment, submission, assign_dir, cookies_path):
|
||
|
|
submission_dir = assign_dir
|
||
|
|
|
||
|
|
if len(assignment.submissions) != 1:
|
||
|
|
submission_dir = os.path.join(assign_dir, str(submission.user_id))
|
||
|
|
|
||
|
|
if submission.preview_url != "":
|
||
|
|
if not os.path.exists(submission_dir):
|
||
|
|
os.makedirs(submission_dir)
|
||
|
|
|
||
|
|
submission_page_dir = os.path.join(submission_dir, "submission.html")
|
||
|
|
|
||
|
|
if not os.path.exists(submission_page_dir):
|
||
|
|
download_page(submission.preview_url, cookies_path, submission_dir, "submission.html")
|
||
|
|
|
||
|
|
if (submission.attempt != 1 and assignment.updated_url != "" and assignment.html_url != ""
|
||
|
|
and assignment.html_url.rstrip("/") != assignment.updated_url.rstrip("/")):
|
||
|
|
submission_dir = os.path.join(assign_dir, "attempts")
|
||
|
|
|
||
|
|
if not os.path.exists(submission_dir):
|
||
|
|
os.makedirs(submission_dir)
|
||
|
|
|
||
|
|
for i in range(submission.attempt):
|
||
|
|
filename = "attempt_" + str(i + 1) + ".html"
|
||
|
|
submission_page_attempt_dir = os.path.join(submission_dir, filename)
|
||
|
|
|
||
|
|
if not os.path.exists(submission_page_attempt_dir):
|
||
|
|
download_page(assignment.updated_url + "/history?version=" + str(i + 1), cookies_path, submission_dir, filename)
|