mirror of
https://github.com/cqwu-ehall/cqwu-ehall.git
synced 2024-11-22 11:01:20 +00:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import datetime
|
|
from typing import Tuple
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from cqwu.enums import ExamRound
|
|
|
|
|
|
class AiExam(BaseModel):
|
|
name: str
|
|
""" 课程名称 """
|
|
credit: float
|
|
""" 学分 """
|
|
time: str
|
|
""" 考试时间 """
|
|
position: str
|
|
""" 考试地点 """
|
|
seat: str
|
|
""" 座位号 """
|
|
exam_round: ExamRound
|
|
""" 考试轮次 """
|
|
|
|
@property
|
|
def name_no_id(self) -> str:
|
|
""" 获取课程名称(去除课程编号) """
|
|
return self.name.split("]")[-1]
|
|
|
|
@property
|
|
def days_left(self) -> int:
|
|
""" 获取距离考试的天数 """
|
|
return (self.get_time()[0] - datetime.datetime.now()).days
|
|
|
|
def get_time(self) -> Tuple[datetime.datetime, datetime.datetime]:
|
|
""" 获取格式化后的考试时间 """
|
|
# 2023-06-25(18周 星期日)09:00-11:00
|
|
day = datetime.datetime.strptime(self.time.split("(")[0], "%Y-%m-%d")
|
|
start_time = datetime.datetime.strptime(self.time.split(")")[1].split("-")[0], "%H:%M")
|
|
start_time = datetime.datetime(day.year, day.month, day.day, start_time.hour, start_time.minute)
|
|
end_time = datetime.datetime.strptime(self.time.split(")")[1].split("-")[1], "%H:%M")
|
|
end_time = datetime.datetime(day.year, day.month, day.day, end_time.hour, end_time.minute)
|
|
return start_time, end_time
|