mirror of
https://github.com/cqwu-ehall/cqwu-ehall.git
synced 2024-11-26 04:02:51 +00:00
82 lines
3.3 KiB
Python
82 lines
3.3 KiB
Python
from typing import Union, List
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
import cqwu
|
|
from cqwu.enums import ExamRound
|
|
from cqwu.errors import NoExamData
|
|
from cqwu.types import AiExam
|
|
|
|
|
|
class GetExamCalendar:
|
|
async def get_exam_calendar(
|
|
self: "cqwu.Client",
|
|
exam_round: Union[str, ExamRound] = ExamRound.Supplementation,
|
|
xue_nian: int = None,
|
|
xue_qi: int = None,
|
|
use_model: bool = False,
|
|
) -> Union[str, List[AiExam]]:
|
|
""" 获取考试安排表 """
|
|
xue_nian = xue_nian or self.xue_nian
|
|
xue_qi = xue_qi or self.xue_qi
|
|
exam_round = ExamRound(exam_round)
|
|
jw_html = await self.login_jwmis()
|
|
jw_host = self.get_web_vpn_host(jw_html.url, https=True)
|
|
jw_url = f"{jw_host}/cqwljw/student/ksap.ksapb_date.jsp"
|
|
headers = {
|
|
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
|
|
'Accept-Language': 'zh-CN,zh;q=0.9,zh-Hans;q=0.8,und;q=0.7,en;q=0.6,zh-Hant;q=0.5,ja;q=0.4',
|
|
'Cache-Control': 'no-cache',
|
|
'Connection': 'keep-alive',
|
|
'Content-Type': 'application/x-www-form-urlencoded',
|
|
'DNT': '1',
|
|
'Pragma': 'no-cache',
|
|
'Referer': f'{jw_host}/cqwljw/student/ksap.ksapb.html?menucode=S20403',
|
|
'Sec-Fetch-Dest': 'iframe',
|
|
'Sec-Fetch-Mode': 'navigate',
|
|
'Sec-Fetch-Site': 'same-origin',
|
|
'Sec-Fetch-User': '?1',
|
|
'Upgrade-Insecure-Requests': '1',
|
|
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
|
|
'sec-ch-ua': '"Chromium";v="112", "Not:A-Brand";v="99"',
|
|
'sec-ch-ua-mobile': '?0',
|
|
'sec-ch-ua-platform': '"Windows"',
|
|
}
|
|
data = {
|
|
'xn': str(xue_nian),
|
|
'xq': str(xue_qi),
|
|
'title': '',
|
|
'xnxq': f'{xue_nian}{xue_qi}',
|
|
'kslc': exam_round.value,
|
|
}
|
|
jw_html = await self.request.post(jw_url, data=data, headers=headers, timeout=60, follow_redirects=True)
|
|
if "没有检索到记录!" in jw_html.text:
|
|
raise NoExamData("没有检索到记录!")
|
|
jw_html = jw_html.text.replace("""<script type="text/javascript" src="//clientvpn.cqwu.edu.cn/webvpn/bundle.debug.js" charset="utf-8"></script>""", "")
|
|
jw_html = jw_html.replace("""<script language='javascript' type='text/javascript' src='../js/Print.js'></script>""", "")
|
|
jw_html = jw_html.replace("charset=GBK", 'charset=UTF-8')
|
|
if not use_model:
|
|
return jw_html
|
|
return parse_html(jw_html, exam_round)
|
|
|
|
|
|
def parse_html(html: str, exam_round: ExamRound) -> List[AiExam]:
|
|
data: List[AiExam] = []
|
|
soup = BeautifulSoup(html, "html.parser")
|
|
trs = soup.find_all("tr")[1:]
|
|
for tr in trs:
|
|
tds = tr.find_all("td")
|
|
if len(tds) != 5:
|
|
continue
|
|
data.append(
|
|
AiExam(
|
|
name=tds[0].text,
|
|
credit=float(tds[1].text),
|
|
time=tds[2].text,
|
|
position=tds[3].text,
|
|
seat=tds[4].text,
|
|
exam_round=exam_round,
|
|
)
|
|
)
|
|
return data
|