AutoDocsDB/script.py at main · psaunderualberta/AutoDocsDB · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
from nerodia.browser import Browser
import pandas as pd
import numpy as np
import glob
import os


def main():
    """
    Set these accordingly for both yourself and the assignment.
    The csv should be acquired from the class' eClass page -> Grades -> Export
    """
    file_location = os.path.dirname(os.path.realpath(__file__))
    mark_csv_path = glob.glob(os.path.join(file_location, "csv", "*.csv"))[0]
    primary_key = "ID"  # 7 digit student ID

    df = pd.read_csv(mark_csv_path)
    df.columns = df.columns.str.strip()
    df["name"] = df["First name"] + " " + df["Last name"]
    df.sort_values(primary_key, inplace=True)
    br = Browser(browser="firefox")
    br.goto("https://docsdb.cs.ualberta.ca/")

    # Login to docsdb
    # Set these environment variables, or hardcode them here.
    # IF YOU HARDCODE THEM, DO NOT COMMIT THIS FILE TO GITHUB
    br.text_field(name="oracle.login").value = os.environ["DOCSDB_USERNAME"]
    br.text_field(name="oracle.password").value = os.environ["DOCSDB_PASSWORD"]
    br.button(value="Get Menu").click()

    # Go to section marks
    br.radio(value="entersec").click()
    br.button(value="Run Form").click()

    # Get specific assignment
    course_num = "379"  # Change this to the correct course number

    # These attributes might take some manual fiddling to get right
    assignment_type = "Assign"  # Change this to the correct assignment type
    assignment_no = "1"  # Change this to the correct assignment number

    # Input the course number and assignment type
    br.text_field(name="coursenum").value = course_num
    br.text_field(name="type").value = assignment_type
    br.text_field(name="num").value = assignment_no
    br.button(value="Get List").click()

    # The column in the CSV file containing the marks
    assignment_name = "Assignment #1 submission/feedback"

    # Input marks
    valid_df = df[~df[assignment_name].isna()]
    for _, (student_id, name, mark) in valid_df[[primary_key, "name", assignment_name]].iterrows():
        print(f"Inputting mark for student '{name}': {mark}")
        br.text_field(css=f"[value='{int(student_id)}'] + input").value = mark

    # Warn about missing marks
    missing_df = df[df[assignment_name].isna()]
    ccid = "CCID"
    if len(missing_df) > 0:
        print("The following students are missing marks:")
        print(missing_df[ccid].values)
        print("These might be excused absences, or they might be missing marks.")
        print()

    # Display warning
    print("Please double check the marks before submission!")
    print("Press enter to continue.")
    input()


if __name__ == "__main__":
    main()