ChoChoX
059e1de984
通过 browser-act 浏览器会话直接调用教务系统 API， 支持命令行和图形界面操作选课/退课，自适应不同选课轮次。 Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
209 lines
8.1 KiB
Python
209 lines
8.1 KiB
Python
"""
|
||
莞工教务系统 - 核心API模块
|
||
============================
|
||
被 CLI (dgut.py) 和 GUI (dgut-gui/dgut_gui.py) 共用。
|
||
自适应探测选课类型标签页，不依赖硬编码的课程类型。
|
||
"""
|
||
import json
|
||
import re
|
||
import subprocess
|
||
import time
|
||
from typing import Any, Optional
|
||
|
||
ORIGIN = "https://jwx.dgut.edu.cn"
|
||
BROWSER_ACT = "browser-act"
|
||
SESSION_NAME = "jwx-s1"
|
||
|
||
# 已知的标签页ID → iframe页面 → API 映射
|
||
# 新增选课类型时只需在此添加
|
||
TAB_REGISTRY: dict[str, dict[str, Optional[str]]] = {
|
||
"选课学分情况": {"id": "xkzx", "iframe": "/xsxk/xsxk_tzsm", "api": None},
|
||
"必修选课": {"id": "bxxksfkf", "iframe": "/xsxkkc/getBxxkxx", "api": "/xsxkkc/xsxkXxxk"},
|
||
"选修选课": {"id": "xxxksfkf", "iframe": "/xsxkkc/getXxxk", "api": "/xsxkkc/xsxkXxxk"},
|
||
"本学期计划选课": {"id": "bxqxksfkf", "iframe": "/xsxkkc/getBxqjxk", "api": "/xsxkkc/xsxkXxxk"},
|
||
"公共选修": {"id": "ggksfkf", "iframe": "/xsxkkc/getGgxxk", "api": "/xsxkkc/xsxkXxxk"},
|
||
"跨年级选课": {"id": "knjxksfkf", "iframe": "/xsxkkc/getKnjxk", "api": "/xsxkkc/xsxkXxxk"},
|
||
"跨专业选课": {"id": "kzyxksfkf", "iframe": "/xsxkkc/getKzyxk", "api": "/xsxkkc/xsxkXxxk"},
|
||
"辅修选课": {"id": "fxzyxksfkf", "iframe": "/xsxkkc/getFxzy", "api": "/xsxkkc/xsxkXxxk"},
|
||
"重修选课": {"id": "cxxkkf", "iframe": "/xsxkkc/comeInCxxk", "api": None},
|
||
"体育选课": {"id": "tyxksfkf", "iframe": "/xsxkkc/newcomeInTyxk", "api": None},
|
||
"选课组选课": {"id": "xkzxksfkf", "iframe": "/xsxkkc/getXkzxk", "api": None},
|
||
}
|
||
|
||
|
||
# ─── 底层通信 ────────────────────────────────────────────────────────────
|
||
|
||
def _browser_eval(js_code: str) -> str:
|
||
escaped = js_code.replace("'", "'\\''")
|
||
cmd = f"{BROWSER_ACT} --session {SESSION_NAME} eval '{escaped}'"
|
||
try:
|
||
result = subprocess.run(cmd, shell=True, capture_output=True, text=True, timeout=30)
|
||
if result.returncode != 0:
|
||
raise RuntimeError(result.stderr.strip() or result.stdout.strip())
|
||
return result.stdout.strip()
|
||
except FileNotFoundError:
|
||
raise RuntimeError("browser-act not found")
|
||
|
||
|
||
def fetch_api(method: str, path: str, body: Optional[str] = None,
|
||
referer: Optional[str] = None) -> Any:
|
||
"""在浏览器上下文中发 fetch 请求（自动携带 cookie）。"""
|
||
url = f"{ORIGIN}{path}"
|
||
headers: dict[str, str] = {"X-Requested-With": "XMLHttpRequest"}
|
||
if referer:
|
||
headers["Referer"] = referer
|
||
opts = {"method": method, "credentials": "include", "headers": headers}
|
||
if body:
|
||
opts["body"] = body
|
||
opts["headers"]["Content-Type"] = "application/x-www-form-urlencoded; charset=UTF-8"
|
||
|
||
opts_json = json.dumps(opts)
|
||
js = (
|
||
f"(async()=>{{"
|
||
f"try{{const r=await fetch('{url}',{opts_json});const t=await r.text();"
|
||
f"try{{return JSON.stringify({{ok:!0,data:JSON.parse(t)}})}}"
|
||
f"catch(e){{return JSON.stringify({{ok:!0,raw:t}})}}}}"
|
||
f"catch(e){{return JSON.stringify({{ok:!1,error:e.message}})}}}})()"
|
||
)
|
||
output = _browser_eval(js)
|
||
result = json.loads(output)
|
||
if not result.get("ok"):
|
||
raise RuntimeError(result.get("error", "unknown error"))
|
||
return result.get("data") or result.get("raw", {})
|
||
|
||
|
||
def fetch_html(path: str) -> str:
|
||
return fetch_api("GET", path)
|
||
|
||
|
||
def _browser_click_enter_button() -> bool:
|
||
"""在浏览器中点击"进入选课"按钮。"""
|
||
js = """
|
||
(function(){
|
||
var f = document.querySelector('iframe[id=NEW_XSD_PYGL_XKGL_NXSXKZX]');
|
||
if(!f) return 'no iframe';
|
||
var d = f.contentDocument;
|
||
var btns = d.querySelectorAll('button');
|
||
for(var b of btns) {
|
||
if(b.textContent.includes('进入选课')) { b.click(); return 'ok'; }
|
||
}
|
||
return 'btn not found';
|
||
})()
|
||
"""
|
||
result = _browser_eval(js)
|
||
time.sleep(3)
|
||
return "ok" in result
|
||
|
||
|
||
def ensure_in_course_selection() -> bool:
|
||
"""
|
||
确保浏览器已在选课界面内。若在列表页则自动点击"进入选课"。
|
||
"""
|
||
try:
|
||
r = subprocess.run(
|
||
f"{BROWSER_ACT} --session {SESSION_NAME} state",
|
||
shell=True, capture_output=True, text=True, timeout=15,
|
||
)
|
||
out = r.stdout
|
||
if "selectBottom" in out and "进入选课" not in out:
|
||
return True
|
||
if "进入选课" in out:
|
||
return _browser_click_enter_button()
|
||
return False
|
||
except Exception:
|
||
return False
|
||
|
||
|
||
# ─── 业务 API ────────────────────────────────────────────────────────────
|
||
|
||
def get_cycles() -> list[dict]:
|
||
"""获取可选课的选课轮次列表。"""
|
||
return fetch_api("GET", "/xsxk/xklc_list_data?xkmc=").get("data", [])
|
||
|
||
|
||
def discover_tabs(jx0502zbid: str) -> list[dict]:
|
||
"""
|
||
动态探测选课界面开放了哪些标签页。
|
||
通过分析 selectBottom 页面的 HTML 和 JavaScript 实现。
|
||
返回: [{"id": str, "label": str, "iframe": str, "api": str|None}, ...]
|
||
"""
|
||
html = fetch_html(f"/xsxk/selectBottom?jx0502zbid={jx0502zbid}&sfylxkstr=")
|
||
|
||
# 方式1: 从 <li onclick="jpbxxk('xxxksfkf')">选修选课</li> 或 'xxxksfkf' 提取
|
||
tabs: list[dict] = []
|
||
seen_ids: set[str] = set()
|
||
for m in re.finditer(
|
||
r"""<li[^>]*?onclick="jpbxxk\(['']{1}(.+?)['']{1}\)"[^>]*?>([^<]+)</li>""",
|
||
html,
|
||
):
|
||
tid, label = m.group(1), m.group(2).strip()
|
||
if tid in seen_ids:
|
||
continue
|
||
seen_ids.add(tid)
|
||
known = TAB_REGISTRY.get(label, {})
|
||
tabs.append({
|
||
"id": tid,
|
||
"label": label,
|
||
"iframe": known.get("iframe"),
|
||
"api": known.get("api"),
|
||
})
|
||
|
||
# 方式2: 从 jpbxxk 函数解析 iframe 映射
|
||
iframe_map: dict[str, str] = {}
|
||
for m in re.finditer(
|
||
r"case\s+'([^']+)':\s*[^#]*?#selectTable\)\.attr\('src',\s*\"([^\"]+)\"",
|
||
html,
|
||
):
|
||
iframe_map[m.group(1)] = m.group(2)
|
||
for tab in tabs:
|
||
if not tab["iframe"]:
|
||
tab["iframe"] = iframe_map.get(tab["id"])
|
||
|
||
# 方式3: 对新标签尝试自动推断 API
|
||
for tab in tabs:
|
||
if tab["api"] is None and tab.get("iframe"):
|
||
# 如果 iframe 路径包含 getXxx 模式，推测对应 xsxkXxx API
|
||
iframe = tab["iframe"]
|
||
m = re.match(r"/xsxkkc/get(\w+)", iframe)
|
||
if m:
|
||
suffix = m.group(1)
|
||
tab["api"] = f"/xsxkkc/xsxk{suffix}"
|
||
|
||
return tabs
|
||
|
||
|
||
def get_courses(api_path: str, referer: str) -> list[dict]:
|
||
"""获取某选课类型的课程列表。"""
|
||
post_data = (
|
||
"sEcho=1&iColumns=14&sColumns=&iDisplayStart=0&iDisplayLength=50"
|
||
"&mDataProp_0=kch&mDataProp_1=kcmc&mDataProp_2=skfsmc"
|
||
"&mDataProp_3=fzmc&mDataProp_4=ktmc&mDataProp_5=xf"
|
||
"&mDataProp_6=skls&mDataProp_7=sksj&mDataProp_8=skdd"
|
||
"&mDataProp_9=xqmc&mDataProp_10=syrs&mDataProp_11=skfsmc"
|
||
"&mDataProp_12=ctsm&mDataProp_13=czOper"
|
||
)
|
||
query = "?1=1&kcxx=&skls=&skfs=" if "?" not in api_path else ""
|
||
data = fetch_api("POST", f"{api_path}{query}", body=post_data, referer=referer)
|
||
return data.get("aaData", [])
|
||
|
||
|
||
def select_course(kcid: str, jx0404id: str, referer: str) -> dict:
|
||
"""选课。"""
|
||
params = f"kcid={kcid}&cfbs=null&jx0404id={jx0404id}&xkzy=&trjf=&sfsyjc=&sfkvtj="
|
||
return fetch_api("GET", f"/xsxkkc/xxxkOper?{params}", referer=referer)
|
||
|
||
|
||
def drop_course(jx0404id: str, referer: str) -> dict:
|
||
"""退课。"""
|
||
return fetch_api("GET", f"/xsxkjg/xstkOper?jx0404id={jx0404id}", referer=referer)
|
||
|
||
|
||
def ensure_session() -> bool:
|
||
"""检查 browser-act session 是否存在。"""
|
||
try:
|
||
r = subprocess.run(f"{BROWSER_ACT} session list", shell=True,
|
||
capture_output=True, text=True, timeout=10)
|
||
return SESSION_NAME in r.stdout
|
||
except FileNotFoundError:
|
||
return False
|