import asyncio import re from bs4 import BeautifulSoup from fastapi import APIRouter, Depends from fastapi.responses import JSONResponse from pydantic import ValidationError from loveace.router.endpoint.jwc.model.base import JWCConfig from loveace.router.endpoint.jwc.model.schedule import ScheduleData from loveace.router.schemas.error import ProtectRouterErrorToCode from loveace.router.schemas.uniresponse import UniResponseModel from loveace.service.remote.aufe import AUFEConnection from loveace.service.remote.aufe.depends import get_aufe_conn jwc_schedules_router = APIRouter( prefix="/schedule", responses=ProtectRouterErrorToCode.gen_code_table(), ) ENDPOINTS = { "student_schedule_pre": "/student/courseSelect/calendarSemesterCurriculum/index", "student_schedule": "/student/courseSelect/thisSemesterCurriculum/{dynamic_path}/ajaxStudentSchedule/past/callback", "section_and_time": "/ajax/getSectionAndTime", } @jwc_schedules_router.get( "/{term_code}/table", summary="获取课表信息", response_model=UniResponseModel[ScheduleData], ) async def get_schedule_table( term_code: str, conn: AUFEConnection = Depends(get_aufe_conn) ) -> UniResponseModel[ScheduleData] | JSONResponse: """ 获取指定学期的课程表 ✅ 功能特性： - 获取指定学期的完整课程表 - 显示课程名称、教室、时间、教师等信息 - 支持按周查询 💡 使用场景： - 查看本周课程安排 - 了解完整学期课程表 - 课表分享和导出 Args: term_code: 学期代码（如：2023-2024-1） Returns: ScheduleData: 包含课程表数据和课程详情 """ try: conn.logger.info(f"开始获取学期 {term_code} 的课表信息") # 第一步：访问课表预备页面，获取动态路径 dynamic_page = JWCConfig().to_full_url(ENDPOINTS["student_schedule_pre"]) dynamic_page_response = await conn.client.get( dynamic_page, follow_redirects=True, timeout=conn.timeout ) if dynamic_page_response.status_code != 200: conn.logger.error( f"获取课表预备页面失败，状态码: {dynamic_page_response.status_code}" ) return ProtectRouterErrorToCode.remote_service_error.to_json_response( conn.logger.trace_id ) soup = BeautifulSoup(dynamic_page_response.text, "lxml") # 尝试从页面中提取动态路径 scripts = soup.find_all("script") dynamic_path = "B2RMNJkT95" # 默认值 for script in scripts: try: script_text = script.string # type: ignore if script_text and "ajaxStudentSchedule" in script_text: # 使用正则表达式提取路径 match = re.search( r"/([A-Za-z0-9]+)/ajaxStudentSchedule", script_text ) if match: dynamic_path = match.group(1) break except AttributeError: continue section_and_time_headers = { **conn.client.headers, "Referer": JWCConfig().to_full_url(ENDPOINTS["student_schedule"]), } select_and_time_url = JWCConfig().to_full_url(ENDPOINTS["section_and_time"]) select_and_time_data = { "planNumber": "", "ff": "f", "sf_request_type": "ajax", } section_and_time_response_coro = conn.client.post( select_and_time_url, data=select_and_time_data, headers=section_and_time_headers, follow_redirects=True, timeout=conn.timeout, ) student_schedule_url = JWCConfig().to_full_url( ENDPOINTS["student_schedule"].format(dynamic_path=dynamic_path) ) schedule_params = { "planCode": term_code, "sf_request_type": "ajax", } student_schedule_response_coro = conn.client.get( student_schedule_url, params=schedule_params, follow_redirects=True, timeout=conn.timeout, ) section_and_time_response, student_schedule_response = await asyncio.gather( section_and_time_response_coro, student_schedule_response_coro ) if section_and_time_response.status_code != 200: conn.logger.error( f"获取节次时间信息失败，状态码: {section_and_time_response.status_code}" ) return ProtectRouterErrorToCode.remote_service_error.to_json_response( conn.logger.trace_id, message="无法获取节次时间信息，请稍后再试" ) if student_schedule_response.status_code != 200: conn.logger.error( f"获取课表信息失败，状态码: {student_schedule_response.status_code}" ) return ProtectRouterErrorToCode.remote_service_error.to_json_response( conn.logger.trace_id, message="无法获取课表信息，请稍后再试" ) time_data = section_and_time_response.json() schedule_data = student_schedule_response.json() # 处理时间段信息 time_slots = [] section_time = time_data.get("sectionTime", []) for time_slot in section_time: time_slots.append( { "session": time_slot.get("id", {}).get("session", 0), "session_name": time_slot.get("sessionName", ""), "start_time": time_slot.get("startTime", ""), "end_time": time_slot.get("endTime", ""), "time_length": time_slot.get("timeLength", ""), "djjc": time_slot.get("djjc", 0), } ) # 处理课程信息 courses = [] xkxx_list = schedule_data.get("xkxx", []) for xkxx_item in xkxx_list: if isinstance(xkxx_item, dict): for course_key, course_data in xkxx_item.items(): if isinstance(course_data, dict): # 提取基本课程信息 course_name = course_data.get("courseName", "") course_code = course_data.get("id", {}).get("coureNumber", "") course_sequence = course_data.get("id", {}).get( "coureSequenceNumber", "" ) teacher_name = ( course_data.get("attendClassTeacher", "") .replace("* ", "") .strip() ) course_properties = course_data.get("coursePropertiesName", "") exam_type = course_data.get("examTypeName", "") unit = float(course_data.get("unit", 0)) # 处理时间地点列表 time_locations = [] time_place_list = course_data.get("timeAndPlaceList", []) # 检查是否有具体时间安排 is_no_schedule = len(time_place_list) == 0 for time_place in time_place_list: # 过滤掉无用的字段，只保留关键信息 time_location = { "class_day": time_place.get("classDay", 0), "class_sessions": time_place.get("classSessions", 0), "continuing_session": time_place.get( "continuingSession", 0 ), "class_week": time_place.get("classWeek", ""), "week_description": time_place.get( "weekDescription", "" ), "campus_name": time_place.get("campusName", ""), "teaching_building_name": time_place.get( "teachingBuildingName", "" ), "classroom_name": time_place.get("classroomName", ""), } time_locations.append(time_location) # 只保留有效的课程（有课程名称的） if course_name: course = { "course_name": course_name, "course_code": course_code, "course_sequence": course_sequence, "teacher_name": teacher_name, "course_properties": course_properties, "exam_type": exam_type, "unit": unit, "time_locations": time_locations, "is_no_schedule": is_no_schedule, } courses.append(course) # 构建最终数据 processed_data = { "total_units": float(schedule_data.get("allUnits", 0)), "time_slots": time_slots, "courses": courses, } conn.logger.info( f"成功处理课表数据：共{len(courses)}门课程，{len(time_slots)}个时间段" ) result = ScheduleData.model_validate(processed_data) return UniResponseModel[ScheduleData]( success=True, data=result, message="获取课表信息成功", error=None, ) except ValidationError as ve: conn.logger.error(f"数据验证错误: {ve}") return ProtectRouterErrorToCode().validation_error.to_json_response( conn.logger.trace_id, "数据验证错误" ) except Exception as e: conn.logger.exception(e) return ProtectRouterErrorToCode().server_error.to_json_response( conn.logger.trace_id, )