from router.common_model import BaseResponse
from provider.aufe.jwc.model import (
    AcademicInfo,
    TrainingPlanInfo,
    Course,
    ExamInfoResponse,
    TermScoreResponse,
)
from typing import List, Dict, Optional
from pydantic import BaseModel, Field


# 统一响应模型
class AcademicInfoResponse(BaseResponse[AcademicInfo]):
    """学业信息响应"""

    pass


class TrainingPlanInfoResponse(BaseResponse[TrainingPlanInfo]):
    """培养方案信息响应"""

    pass


class CourseListResponse(BaseResponse[List[Course]]):
    """评教课程列表响应"""

    pass


class ExamInfoAPIResponse(BaseResponse[ExamInfoResponse]):
    """考试信息响应"""

    pass


# ==================== 学期和成绩相关响应模型 ====================


class FetchTermScoreRequest(BaseModel):
    """获取学期成绩请求模型"""

    term_id: str = Field(..., description="学期ID，如：2024-2025-2-1")
    course_code: str = Field("", description="课程代码（可选，用于筛选）")
    course_name: str = Field("", description="课程名称（可选，用于筛选）")
    page_num: int = Field(1, description="页码，默认为1", ge=1)
    page_size: int = Field(50, description="每页大小，默认为50", ge=1, le=100)


class AllTermsResponse(BaseResponse[Dict[str, str]]):
    """所有学期信息响应"""

    pass


class TermScoreAPIResponse(BaseResponse[TermScoreResponse]):
    """学期成绩响应"""

    pass


# ==================== 课表相关响应模型 ====================


class TimeSlot(BaseModel):
    """时间段模型"""
    
    session: int = Field(..., description="节次")
    session_name: str = Field(..., description="节次名称")
    start_time: str = Field(..., description="开始时间，格式：HHMM")
    end_time: str = Field(..., description="结束时间，格式：HHMM")
    time_length: str = Field(..., description="时长（分钟）")
    djjc: int = Field(..., description="大节节次")


class CourseTimeLocation(BaseModel):
    """课程时间地点模型"""
    
    class_day: int = Field(..., description="上课星期几（1-7）")
    class_sessions: int = Field(..., description="上课节次")
    continuing_session: int = Field(..., description="持续节次数")
    class_week: str = Field(..., description="上课周次（24位二进制字符串）")
    week_description: str = Field(..., description="上课周次描述")
    campus_name: str = Field(..., description="校区名称")
    teaching_building_name: str = Field(..., description="教学楼名称")
    classroom_name: str = Field(..., description="教室名称")


class ScheduleCourse(BaseModel):
    """课表课程模型"""
    
    course_name: str = Field(..., description="课程名称")
    course_code: str = Field(..., description="课程代码")
    course_sequence: str = Field(..., description="课程序号")
    teacher_name: str = Field(..., description="授课教师")
    course_properties: str = Field(..., description="课程性质")
    exam_type: str = Field(..., description="考试类型")
    unit: float = Field(..., description="学分")
    time_locations: List[CourseTimeLocation] = Field(..., description="时间地点列表")
    is_no_schedule: bool = Field(False, description="是否无具体时间安排")


class ScheduleData(BaseModel):
    """课表数据模型"""
    
    total_units: float = Field(..., description="总学分")
    time_slots: List[TimeSlot] = Field(..., description="时间段列表")
    courses: List[ScheduleCourse] = Field(..., description="课程列表")
    semester_info: Dict[str, str] = Field(..., description="学期信息")
    

class ScheduleResponse(BaseResponse[ScheduleData]):
    """课表响应模型"""
    
    pass


class FetchScheduleRequest(BaseModel):
    """获取课表请求模型"""
    
    plan_code: str = Field(..., description="培养方案代码，如：2024-2025-2-1")