from ysu_sdk.cas import CASClient, CASCredential

# First login (interactive — callbacks read user input)
cas = CASClient()
cred = cas.login(
    "<student id>",
    "<password>",
    mfa_handler=lambda c: input(f"Enter {c.method} code ({c.mobile_hint}): "),
)
cred.save()

# Reuse credential to issue tickets for any service
cas = CASClient(credential=CASCredential.load())
sess = cas.authorize("https://jwxt.ysu.edu.cn/jwapp/sys/emaphome/portal/index.do")
resp = sess.get("https://jwxt.ysu.edu.cn/jwapp/sys/emaphome/portal/index.do")

from ysu_sdk.cas import CASClient, CASCredential
from ysu_sdk.jwxt import JWXTClient

# 1. Authenticate via CAS
cas = CASClient(credential=CASCredential.load())

# 2. Create JWXT client
#    - Business methods are decorated with lazy re-auth: the client trusts
#      existing jwxt cookies by default and falls back to CAS authorize
#      only when the server returns 401/403 or redirects to the CAS login
#      page. This avoids hammering cer.ysu.edu.cn on every call.
#    - If you already have a JWXTSession snapshot (e.g. from a previous
#      request), pass it via jwxt_session to skip even the first authorize.
jwxt = JWXTClient(cas)

# 3. Query information
grades = jwxt.query_grades()                   # grades
gpa = jwxt.query_gpa_stats()                   # GPA stats
schedule = jwxt.query_schedule()               # lecture schedule (current term)
exp_schedule = jwxt.query_schedule_experimental()  # experiment schedule
unscheduled = jwxt.query_unscheduled_courses() # unscheduled courses
exams = jwxt.query_exams()                     # exam arrangements
info = jwxt.query_student_info()               # student profile
plan = jwxt.query_training_plan()              # training plan
completion = jwxt.query_academic_completion()  # academic completion
warnings = jwxt.query_academic_warnings()      # academic warnings

# Query by term
grades = jwxt.query_grades(term="2024-2025-1")
schedule = jwxt.query_schedule(term="2024-2025-2")
exams = jwxt.query_exams(term="2024-2025-2")

from ysu_sdk.jwxt import JWXTSession

# After any business call, capture the rotated session
snapshot: JWXTSession = jwxt.session_snapshot()
json_str = snapshot.to_json()          # → transport / store

# On the next run (or in a stateless server), feed it back
session = JWXTSession.from_json(json_str)
jwxt = JWXTClient(cas, jwxt_session=session)

g = jwxt.query_grades(term="2025-2026-2")[0]

# 1) max / min / average
class_stat = jwxt.query_grade_statistics(class_id=g.class_id)
course_stat = jwxt.query_grade_statistics(course_code=g.course_code)

# 2) bucketed distribution (excellent / good / average / pass / fail)
class_dist = jwxt.query_grade_distribution(class_id=g.class_id)
course_dist = jwxt.query_grade_distribution(course_code=g.course_code)

# 3) personal ranking (defaults to the logged-in student)
class_rank = jwxt.query_grade_ranking(class_id=g.class_id)
course_rank = jwxt.query_grade_ranking(course_code=g.course_code)
# class_rank.rank / class_rank.total / class_rank.score

from ysu_sdk.jwxt import EvaluationAnswer, JWXTBusinessError

# 1. Query evaluation types and pending counts for the current term
types = jwxt.query_evaluation_types()
# [EvaluationType(name="Student evaluation", code="01", count=3), ...]

# 2. Pull pending tasks for a given type
tasks = jwxt.query_pending_evaluations(eval_type="01")
# [EvaluationTask(wid=..., wjid=..., course_name=..., teacher_name=..., group_no=...), ...]

# 3. Pull questionnaire details
detail = jwxt.get_evaluation_detail(
    group_no=tasks[0].group_no,
    eval_type=tasks[0].eval_type,
    sequence=tasks[0].sequence,
)

# 4. Build answers (single/multi choice → option_ids, fill-in → text)
answers = [
    EvaluationAnswer(tmid=q.tmid, question_type=q.question_type, option_ids=[q.options[0].wid])
    for q in detail.questions if q.question_type in ("01", "07")
]

# 5. Pre-check (strongly recommended) — server validates completeness and score
try:
    jwxt.calculate_evaluation_score(
        group_no=tasks[0].group_no,
        wjid=detail.wjid,
        eval_type=tasks[0].eval_type,
        answers=answers,
    )
except JWXTBusinessError as e:
    print(f"Invalid answers: {e.msg}")
    raise

# 6. Submit (irreversible)
jwxt.submit_evaluation(
    group_no=tasks[0].group_no,
    wjid=detail.wjid,
    eval_type=tasks[0].eval_type,
    answers=answers,
)

from ysu_sdk.jwxt import (
    JWXTError,
    NotLoggedInError,
    JWXTProtocolError,
    JWXTBusinessError,
)

try:
    jwxt.submit_evaluation(...)
except NotLoggedInError:
    # CAS cookie expired or kicked offline — re-login required
    ...
except JWXTBusinessError as e:
    # Server returned a non-zero business code (e.g. "evaluation window not open",
    # "incomplete answers"). Inspect e.code / e.msg / e.url programmatically.
    if "evaluation window not open" in (e.msg or ""):
        ...
except JWXTProtocolError:
    # Response is not JSON, envelope is malformed, or network/HTTP error
    ...
except JWXTError:
    # Catch-all base for everything under the jwxt package
    ...