103 lines
3.6 KiB
Python
103 lines
3.6 KiB
Python
|
from typing import Optional
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
import re
|
|||
|
|
from loguru import logger
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SemesterWeekInfo(BaseModel):
|
|||
|
|
"""学期周数信息"""
|
|||
|
|
|
|||
|
|
academic_year: str = Field("", description="学年，如 2025-2026")
|
|||
|
|
semester: str = Field("", description="学期，如 秋、春")
|
|||
|
|
week_number: int = Field(0, description="当前周数")
|
|||
|
|
is_end: bool = Field(False, description="是否为学期结束")
|
|||
|
|
weekday: str = Field("", description="星期几")
|
|||
|
|
raw_text: str = Field("", description="原始文本")
|
|||
|
|
|
|||
|
|
def calculate_statistics(self):
|
|||
|
|
"""计算统计信息（如果需要的话）"""
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def from_calendar_text(cls, calendar_text: str) -> "SemesterWeekInfo":
|
|||
|
|
"""从日历文本解析学期周数信息
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
calendar_text: 日历文本，例如 "2025-2026 秋 第1周 星期三"
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
SemesterWeekInfo: 学期周数信息对象
|
|||
|
|
"""
|
|||
|
|
# 清理文本
|
|||
|
|
clean_text = re.sub(r'\s+', ' ', calendar_text.strip())
|
|||
|
|
|
|||
|
|
# 初始化默认值
|
|||
|
|
academic_year = ""
|
|||
|
|
semester = ""
|
|||
|
|
week_number = 0
|
|||
|
|
is_end = False
|
|||
|
|
weekday = ""
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 解析学年：2025-2026
|
|||
|
|
year_match = re.search(r'(\d{4}-\d{4})', clean_text)
|
|||
|
|
if year_match:
|
|||
|
|
academic_year = year_match.group(1)
|
|||
|
|
|
|||
|
|
# 解析学期：秋、春
|
|||
|
|
semester_match = re.search(r'(春|秋|夏)', clean_text)
|
|||
|
|
if semester_match:
|
|||
|
|
semester = semester_match.group(1)
|
|||
|
|
|
|||
|
|
# 解析周数：第1周、第15周等
|
|||
|
|
week_match = re.search(r'第(\d+)周', clean_text)
|
|||
|
|
if week_match:
|
|||
|
|
week_number = int(week_match.group(1))
|
|||
|
|
|
|||
|
|
# 解析星期：星期一、星期二等
|
|||
|
|
weekday_match = re.search(r'星期([一二三四五六日天])', clean_text)
|
|||
|
|
if weekday_match:
|
|||
|
|
weekday = weekday_match.group(1)
|
|||
|
|
|
|||
|
|
# 判断是否为学期结束（通常第16周以后或包含"结束"等关键词）
|
|||
|
|
if week_number >= 16 or "结束" in clean_text or "考试" in clean_text:
|
|||
|
|
is_end = True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"解析学期周数信息时出错: {str(e)}")
|
|||
|
|
|
|||
|
|
return cls(
|
|||
|
|
academic_year=academic_year,
|
|||
|
|
semester=semester,
|
|||
|
|
week_number=week_number,
|
|||
|
|
is_end=is_end,
|
|||
|
|
weekday=weekday,
|
|||
|
|
raw_text=clean_text
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ErrorSemesterWeekInfo(SemesterWeekInfo):
|
|||
|
|
"""错误的学期周数信息"""
|
|||
|
|
|
|||
|
|
academic_year: str = Field("解析失败", description="学年")
|
|||
|
|
semester: str = Field("解析失败", description="学期")
|
|||
|
|
week_number: int = Field(-1, description="当前周数")
|
|||
|
|
is_end: bool = Field(False, description="是否为学期结束")
|
|||
|
|
weekday: str = Field("解析失败", description="星期几")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SemesterWeekResponse(BaseModel):
|
|||
|
|
"""学期周数信息响应模型"""
|
|||
|
|
|
|||
|
|
code: int = Field(0, description="响应码")
|
|||
|
|
message: str = Field("获取成功", description="响应消息")
|
|||
|
|
data: Optional[SemesterWeekInfo] = Field(None, description="学期周数数据")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ErrorSemesterWeekResponse(BaseModel):
|
|||
|
|
"""错误的学期周数信息响应"""
|
|||
|
|
|
|||
|
|
code: int = Field(-1, description="响应码")
|
|||
|
|
message: str = Field("请求失败，请稍后重试", description="响应消息")
|
|||
|
|
data: Optional[ErrorSemesterWeekInfo] = Field(default=None, description="错误数据")
|