Python 后端提交

Python 后端（FastAPI + FastMCP + ...）的初始版本号设定为 0.1.0，这是 uv 在 pypriject.toml
里给我自动设置的，我觉得有道理。

router/api_router.py

@@ -0,0 +1,100 @@
+from pathlib import Path
+from typing import Annotated
+
+from fastapi import APIRouter, Depends, HTTPException, status
+from fastapi.responses import FileResponse
+from sqlmodel import Session, select
+
+from njupt_api.baselib import logger
+from njupt_api.zhengfang import (
+    ZhengFang,
+    course_dict_serializer,
+    course_list_serializer,
+)
+from router.enhance.lib import ReturnDto, ScheduleQueryDto, apply_enhance, get_session
+from router.enhance.model import Course
+
+TEMP_DIR = Path.cwd() / "temp"
+
+
+api_router = APIRouter(prefix="/api", tags=["API"])
+
+
+@api_router.post("/schedule/class")
+async def post_schedule_class(
+    student: ScheduleQueryDto,
+    session: Annotated[Session, Depends(get_session)],
+) -> ReturnDto:
+    if student.username is None and student.password is None:
+        logger.debug("未提供学号和密码参数，尝试从数据库中返回一次性存储的班级课表。")
+        course_dtos = session.exec(select(Course)).all()
+        course_list: list[dict] = [course_dict_serializer(course) for course in course_dtos]
+        logger.success(f"{student.week=} 从数据库中返回一次性存储的班级课表。")
+        return await apply_enhance(course_list, student.week, student.img)
+    if student.username and student.password:
+        async with ZhengFang() as zf:
+            if await zf.login(student.username, student.password):
+                course_list = course_list_serializer(await zf.get_class_schedule())
+                logger.success(
+                    f"{student.username} | {student.week=} 获取指定学生的班级课表成功。",
+                )
+                return await apply_enhance(course_list, student.week, student.img)
+            logger.error(
+                f"{student.username} | 获取课程表失败，请检查账号密码是否正确后再试。",
+            )
+            return ReturnDto(
+                success=False,
+                message="获取课程表失败，请检查账号密码是否正确后再试。",
+            )
+    else:
+        logger.error(
+            f"参数错误，请同时携带或同时不携带学号和密码参数: {student.username=} | {student.password=}",
+        )
+        return ReturnDto(
+            success=False,
+            message="参数错误，请同时携带或同时不携带学号和密码参数。",
+        )
+
+
+@api_router.post("/schedule/student")
+async def post_schedule_student(student: ScheduleQueryDto) -> ReturnDto:
+    if student.username is None or student.password is None:
+        logger.error("查询学生课表需要同时提供学号和密码参数。")
+        return ReturnDto(
+            success=False,
+            message="查询学生课表需要同时提供学号和密码参数。",
+        )
+
+    async with ZhengFang() as zf:
+        if await zf.login(student.username, student.password):
+            course_list = course_list_serializer(await zf.get_student_schedule())
+            logger.success(f"{student.username} | 获取学生个人课表成功。")
+            return await apply_enhance(course_list, student.week, student.img)
+        logger.error(
+            f"{student.username} | 获取课程表失败，请检查账号密码是否正确后再试。",
+        )
+        return ReturnDto(
+            success=False,
+            message="获取课程表失败，请检查账号密码是否正确后再试。",
+        )
+
+
+@api_router.get("/schedule/img/{name}")
+async def get_schedule_img(name: str) -> FileResponse:
+    """
+    从 temp 工作目录中读取指定图片并返回。如果图片不存在则报 404。
+
+    Returns:
+        FileResponse: 图片。
+
+    Raises:
+        HTTPException: 404 - 查找的图片不存在。
+    """
+    image_file = TEMP_DIR / name
+    logger.debug(f"尝试获取 {image_file!s}")
+    if image_file.exists():
+        return FileResponse(path=str(image_file), media_type="image/png")
+    raise HTTPException(
+        status_code=status.HTTP_404_NOT_FOUND,
+        detail="Name wrong or too late.",
+    )