247 lines
10 KiB
Python
247 lines
10 KiB
Python
|
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,
|
|||
|
|
)
|