263 lines
10 KiB
Python
263 lines
10 KiB
Python
|
import re
|
|||
|
|
|
|||
|
|
import ujson
|
|||
|
|
from bs4 import BeautifulSoup
|
|||
|
|
from fastapi import APIRouter, Depends
|
|||
|
|
from fastapi.responses import JSONResponse
|
|||
|
|
from httpx import HTTPError
|
|||
|
|
from pydantic import ValidationError
|
|||
|
|
from ujson import JSONDecodeError
|
|||
|
|
|
|||
|
|
from loveace.router.endpoint.jwc.model.base import JWCConfig
|
|||
|
|
from loveace.router.endpoint.jwc.model.plan import (
|
|||
|
|
PlanCompletionCategory,
|
|||
|
|
PlanCompletionInfo,
|
|||
|
|
)
|
|||
|
|
from loveace.router.endpoint.jwc.utils.plan import populate_category_children
|
|||
|
|
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
|
|||
|
|
|
|||
|
|
ENDPOINT = {
|
|||
|
|
"plan": "/student/integratedQuery/planCompletion/index",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
jwc_plan_router = APIRouter(
|
|||
|
|
prefix="/plan",
|
|||
|
|
responses=ProtectRouterErrorToCode().gen_code_table(),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@jwc_plan_router.get(
|
|||
|
|
"/current",
|
|||
|
|
summary="获取当前培养方案完成信息",
|
|||
|
|
response_model=UniResponseModel[PlanCompletionInfo],
|
|||
|
|
)
|
|||
|
|
async def get_current_plan_completion(
|
|||
|
|
conn: AUFEConnection = Depends(get_aufe_conn),
|
|||
|
|
) -> UniResponseModel[PlanCompletionInfo] | JSONResponse:
|
|||
|
|
"""
|
|||
|
|
获取用户的培养方案完成情况
|
|||
|
|
|
|||
|
|
✅ 功能特性：
|
|||
|
|
- 获取培养方案的总体完成进度
|
|||
|
|
- 按类别显示各类课程的完成情况
|
|||
|
|
- 显示已完成、未完成、可选课程等
|
|||
|
|
|
|||
|
|
💡 使用场景：
|
|||
|
|
- 查看毕业要求的完成进度
|
|||
|
|
- 了解还需要修读哪些课程
|
|||
|
|
- 规划后续选课
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
PlanCompletionInfo: 包含方案完成情况和各类别详情
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
conn.logger.info("获取当前培养方案完成信息")
|
|||
|
|
response = await conn.client.get(
|
|||
|
|
JWCConfig().to_full_url(ENDPOINT["plan"]),
|
|||
|
|
follow_redirects=True,
|
|||
|
|
timeout=600,
|
|||
|
|
)
|
|||
|
|
if response.status_code != 200:
|
|||
|
|
conn.logger.error(f"获取培养方案信息失败，状态码: {response.status_code}")
|
|||
|
|
return ProtectRouterErrorToCode().remote_service_error.to_json_response(
|
|||
|
|
conn.logger.trace_id
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
html_content = response.text
|
|||
|
|
|
|||
|
|
# 使用BeautifulSoup解析HTML
|
|||
|
|
soup = BeautifulSoup(html_content, "lxml")
|
|||
|
|
|
|||
|
|
# 提取培养方案名称
|
|||
|
|
plan_name = ""
|
|||
|
|
|
|||
|
|
# 查找包含"培养方案"的h4标签
|
|||
|
|
h4_elements = soup.find_all("h4")
|
|||
|
|
for h4 in h4_elements:
|
|||
|
|
text = h4.get_text(strip=True) if h4 else ""
|
|||
|
|
if "培养方案" in text:
|
|||
|
|
plan_name = text
|
|||
|
|
conn.logger.info(f"找到培养方案标题: {plan_name}")
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
# 解析专业和年级信息
|
|||
|
|
major = ""
|
|||
|
|
grade = ""
|
|||
|
|
if plan_name:
|
|||
|
|
grade_match = re.search(r"(\d{4})级", plan_name)
|
|||
|
|
if grade_match:
|
|||
|
|
grade = grade_match.group(1)
|
|||
|
|
|
|||
|
|
major_match = re.search(r"\d{4}级(.+?)本科", plan_name)
|
|||
|
|
if major_match:
|
|||
|
|
major = major_match.group(1)
|
|||
|
|
|
|||
|
|
# 查找zTree数据
|
|||
|
|
ztree_data = []
|
|||
|
|
|
|||
|
|
# 在script标签中查找zTree初始化数据
|
|||
|
|
scripts = soup.find_all("script")
|
|||
|
|
for script in scripts:
|
|||
|
|
try:
|
|||
|
|
script_text = script.get_text() if script else ""
|
|||
|
|
if "$.fn.zTree.init" in script_text and "flagId" in script_text:
|
|||
|
|
conn.logger.info("找到包含zTree初始化的script标签")
|
|||
|
|
|
|||
|
|
# 提取zTree数据
|
|||
|
|
# 尝试多种模式匹配
|
|||
|
|
patterns = [
|
|||
|
|
r'\$\.fn\.zTree\.init\(\$\("#treeDemo"\),\s*setting,\s*(\[.*?\])\s*\);',
|
|||
|
|
r"\.zTree\.init\([^,]+,\s*[^,]+,\s*(\[.*?\])\s*\);",
|
|||
|
|
r'init\(\$\("#treeDemo"\)[^,]*,\s*[^,]*,\s*(\[.*?\])',
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
json_part = None
|
|||
|
|
for pattern in patterns:
|
|||
|
|
match = re.search(pattern, script_text, re.DOTALL)
|
|||
|
|
if match:
|
|||
|
|
json_part = match.group(1)
|
|||
|
|
conn.logger.info(
|
|||
|
|
f"使用模式匹配成功提取zTree数据: {len(json_part)}字符"
|
|||
|
|
)
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
if json_part:
|
|||
|
|
# 清理和修复JSON格式
|
|||
|
|
# 移除JavaScript注释和多余的逗号
|
|||
|
|
json_part = re.sub(r"//.*?\n", "\n", json_part)
|
|||
|
|
json_part = re.sub(r"/\*.*?\*/", "", json_part, flags=re.DOTALL)
|
|||
|
|
json_part = re.sub(r",\s*}", "}", json_part)
|
|||
|
|
json_part = re.sub(r",\s*]", "]", json_part)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
ztree_data = ujson.loads(json_part)
|
|||
|
|
conn.logger.info(f"JSON解析成功，共{len(ztree_data)}个节点")
|
|||
|
|
break
|
|||
|
|
except JSONDecodeError as e:
|
|||
|
|
conn.logger.warning(f"JSON解析失败: {str(e)}")
|
|||
|
|
# 如果JSON解析失败，不使用手动解析，直接跳过
|
|||
|
|
continue
|
|||
|
|
else:
|
|||
|
|
conn.logger.warning("未能通过模式匹配提取zTree数据")
|
|||
|
|
continue
|
|||
|
|
except Exception:
|
|||
|
|
continue
|
|||
|
|
if not ztree_data:
|
|||
|
|
conn.logger.warning("未找到有效的zTree数据")
|
|||
|
|
|
|||
|
|
# 输出调试信息
|
|||
|
|
conn.logger.info(f"HTML内容长度: {len(html_content)}")
|
|||
|
|
conn.logger.info(f"找到的script标签数量: {len(soup.find_all('script'))}")
|
|||
|
|
|
|||
|
|
# 检查是否包含关键词
|
|||
|
|
contains_ztree = "zTree" in html_content
|
|||
|
|
contains_flagid = "flagId" in html_content
|
|||
|
|
contains_plan = "培养方案" in html_content
|
|||
|
|
conn.logger.info(
|
|||
|
|
f"HTML包含关键词: zTree={contains_ztree}, flagId={contains_flagid}, 培养方案={contains_plan}"
|
|||
|
|
)
|
|||
|
|
conn.logger.warning("未找到有效的zTree数据")
|
|||
|
|
|
|||
|
|
if contains_plan:
|
|||
|
|
conn.logger.warning(
|
|||
|
|
"检测到培养方案内容，但zTree数据解析失败，可能页面结构已变化"
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
conn.logger.warning(
|
|||
|
|
"未检测到培养方案相关内容，可能需要重新登录或检查访问权限"
|
|||
|
|
)
|
|||
|
|
return ProtectRouterErrorToCode().remote_service_error.to_json_response(
|
|||
|
|
conn.logger.trace_id,
|
|||
|
|
message="未找到有效的培养方案数据，请检查登录状态或稍后再试",
|
|||
|
|
)
|
|||
|
|
# 解析zTree数据构建分类和课程信息
|
|||
|
|
try:
|
|||
|
|
# 按层级组织数据
|
|||
|
|
nodes_by_id = {node["id"]: node for node in ztree_data}
|
|||
|
|
root_categories = []
|
|||
|
|
|
|||
|
|
# 统计根分类和所有节点信息，用于调试
|
|||
|
|
all_parent_ids = set()
|
|||
|
|
root_nodes = []
|
|||
|
|
|
|||
|
|
for node in ztree_data:
|
|||
|
|
parent_id = node.get("pId", "")
|
|||
|
|
all_parent_ids.add(parent_id)
|
|||
|
|
|
|||
|
|
# 根分类的判断条件：pId为"-1"（这是zTree中真正的根节点标识）
|
|||
|
|
# 从HTML示例可以看出，真正的根分类的pId是"-1"
|
|||
|
|
is_root_category = parent_id == "-1"
|
|||
|
|
|
|||
|
|
if is_root_category:
|
|||
|
|
root_nodes.append(node)
|
|||
|
|
|
|||
|
|
conn.logger.info(
|
|||
|
|
f"zTree数据分析: 总节点数={len(ztree_data)}, 根节点数={len(root_nodes)}, 不同父ID数={len(all_parent_ids)}"
|
|||
|
|
)
|
|||
|
|
conn.logger.debug(f"所有父ID: {sorted(all_parent_ids)}")
|
|||
|
|
|
|||
|
|
# 构建分类树
|
|||
|
|
for node in root_nodes:
|
|||
|
|
category = PlanCompletionCategory.from_ztree_node(node)
|
|||
|
|
# 填充分类的子分类和课程（支持多层嵌套）
|
|||
|
|
try:
|
|||
|
|
populate_category_children(category, node["id"], nodes_by_id, conn)
|
|||
|
|
except Exception as e:
|
|||
|
|
conn.logger.error(f"填充分类子项异常: {str(e)}")
|
|||
|
|
conn.logger.error(
|
|||
|
|
f"异常节点信息: category_id={node['id']}, 错误详情: {str(e)}"
|
|||
|
|
)
|
|||
|
|
root_categories.append(category)
|
|||
|
|
conn.logger.info(
|
|||
|
|
f"创建根分类: {category.category_name} (ID: {node['id']})"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 创建完成情况信息
|
|||
|
|
completion_info = PlanCompletionInfo(
|
|||
|
|
plan_name=plan_name,
|
|||
|
|
major=major,
|
|||
|
|
grade=grade,
|
|||
|
|
categories=root_categories,
|
|||
|
|
total_categories=0,
|
|||
|
|
total_courses=0,
|
|||
|
|
passed_courses=0,
|
|||
|
|
failed_courses=0,
|
|||
|
|
unread_courses=0,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 计算统计信息
|
|||
|
|
completion_info.calculate_statistics()
|
|||
|
|
conn.logger.info(
|
|||
|
|
f"培养方案完成信息统计: 分类数={completion_info.total_categories}, 课程数={completion_info.total_courses}, 已过课程={completion_info.passed_courses}, 未过课程={completion_info.failed_courses}, 未修读课程={completion_info.unread_courses}"
|
|||
|
|
)
|
|||
|
|
return UniResponseModel[PlanCompletionInfo](
|
|||
|
|
success=True,
|
|||
|
|
data=completion_info,
|
|||
|
|
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
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
except HTTPError as he:
|
|||
|
|
conn.logger.error(f"HTTP请求错误: {he}")
|
|||
|
|
return ProtectRouterErrorToCode().remote_service_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
|
|||
|
|
)
|