demo.py

import os

from pysjtu import Session, Client, CourseRange, JCSSRecognizer, create_client

# from datetime import datetime, timezone
# import arrow
# from ics import Calendar, Event

lesson_time = (((8, 0), (8, 45)),
               ((8, 55), (9, 40)),
               ((10, 0), (10, 45)),
               ((10, 55), (11, 40)),
               ((12, 0), (12, 45)),
               ((12, 55), (13, 40)),
               ((14, 0), (14, 45)),
               ((14, 55), (15, 40)),
               ((16, 0), (16, 45)),
               ((16, 55), (17, 40)),
               ((18, 0), (18, 45)),
               ((18, 55), (19, 40)),
               ((19, 55), (20, 40)),
               ((20, 55), (21, 40)))

try:
    sess_file = open("session", mode="r+b")
except FileNotFoundError:
    sess_file = None    # type: ignore

client = create_client(os.environ["SJTU_USER"], os.environ["SJTU_PASS"])

with Session(session_file=sess_file, ocr=JCSSRecognizer()) if sess_file else Session(username=os.environ["SJTU_USER"],
                                                               password=os.environ["SJTU_PASS"], ocr=JCSSRecognizer()) as sess:
    client = Client(session=sess)

    sectors = client.course_selection_sectors
    classes = sectors[0].classes

    print(client.student_id)
    schedule = client.schedule(2019, 1)
    print(schedule)
    print(schedule.filter(name="高等数学II"))
    print(list(map(lambda x: x.name, schedule.filter(week=2, day=range(1, 3), time=[range(10, 12), 1]))))

    '''
    c = Calendar()
    local_tz = datetime.now(timezone.utc).astimezone().tzinfo
    for lesson in schedule:
        e = Event()
        e.name = lesson.name
        e.begin = arrow.get(datetime(2020, 1, 5 + lesson.day, *lesson_time[lesson.time[0] - 1][0], tzinfo=local_tz))
        e.end = arrow.get(datetime(2020, 1, 5 + lesson.day, *lesson_time[lesson.time[-1] - 1][1], tzinfo=local_tz))
        e.location = lesson.location
        c.events.add(e)

    with open("test.ics", mode="w") as f:
        f.write(str(c))
    '''

    score = client.score(2019, 0)
    print(score)
    print(score[0].detail)

    exam = client.exam(2019, 0)
    print(exam)

    query = client.query_courses(2019, 0, name="高等数学")
    print(query[3:18])
    for item in query:
        print(item, end=" ")
    print()

    print(client.term_start_date)
    query_params = client.default_gpa_query_params

    print(client.gpa(query_params, 10))

    query_params.course_range = CourseRange.ALL
    print(client.gpa(query_params, 10))

    if not sess_file:
        sess.dump(open("session", mode="wb"))