mirror of
https://github.com/cqwu-ehall/cqwu-ehall.git
synced 2024-11-22 02:55:33 +00:00
37 lines
1.2 KiB
Python
37 lines
1.2 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]
|
||
|
|
||
|
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
|