diff --git a/config.py b/config.py index aa281d2..b68f292 100644 --- a/config.py +++ b/config.py @@ -1,10 +1,11 @@ # 课程 id， 前往选课界面填在这里 -course_ids = ["COMP6101P.01", "COMP6107P.01"] +course_ids = [] # 以下cookie和student_id 按 F12 设置 +turn_id = "" student_id = "" cookie = "" # 请求频率，丧心病狂可以填0.1， 正常3-5就可以 -hz = 3 \ No newline at end of file +hz = 5 \ No newline at end of file diff --git a/main.py b/main.py index 6e2f834..ec3608f 100644 --- a/main.py +++ b/main.py @@ -5,8 +5,11 @@ import requests import threading -import time, json +import time +import json +import random from config import * + url = "https://jw.ustc.edu.cn/ws/for-std/course-select/add-request" drop_url = 'https://jw.ustc.edu.cn/ws/for-std/course-select/add-drop-response' all_class_url = "https://jw.ustc.edu.cn/ws/for-std/course-select/addable-lessons" @@ -21,19 +24,18 @@ accept-encoding: gzip, deflate, br accept-language: zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7 cache-control: no-cache -content-length: 98 content-type: application/x-www-form-urlencoded; charset=UTF-8 cookie: {} origin: https://jw.ustc.edu.cn pragma: no-cache -referer: https://jw.ustc.edu.cn/for-std/course-select/{}/turn/381/select +referer: https://jw.ustc.edu.cn/for-std/course-select/{}/turn/{}/select sec-ch-ua: "Google Chrome";v="87", " Not;A Brand";v="99", "Chromium";v="87" sec-ch-ua-mobile: ?0 sec-fetch-dest: empty sec-fetch-mode: cors sec-fetch-site: same-origin user-agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.141 Safari/537.36 -x-requested-with: XMLHttpRequest""".format(cookie, student_id) +x-requested-with: XMLHttpRequest""".format(cookie, student_id, turn_id) dp_row_header = """authority: jw.ustc.edu.cn method: POST path: /ws/for-std/course-select/add-drop-response @@ -42,19 +44,18 @@ accept-encoding: gzip, deflate, br accept-language: zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7 cache-control: no-cache -content-length: 63 content-type: application/x-www-form-urlencoded; charset=UTF-8 cookie: {} origin: https://jw.ustc.edu.cn pragma: no-cache -referer: https://jw.ustc.edu.cn/for-std/course-select/{}/turn/381/select +referer: https://jw.ustc.edu.cn/for-std/course-select/{}/turn/{}/select sec-ch-ua: "Google Chrome";v="87", " Not;A Brand";v="99", "Chromium";v="87" sec-ch-ua-mobile: ?0 sec-fetch-dest: empty sec-fetch-mode: cors sec-fetch-site: same-origin user-agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.141 Safari/537.36 -x-requested-with: XMLHttpRequest""".format(cookie, student_id) +x-requested-with: XMLHttpRequest""".format(cookie, student_id, turn_id) def deal_row(s): return {i.split(": ")[0].strip(): i.split(": ")[1] for i in s.split("\n")} @@ -63,44 +64,70 @@ def deal_row(s): dp_headers = deal_row(dp_row_header) def get_allclass(): - c = requests.post(all_class_url, headers=headers, data={"turnId":"381", "studentId":student_id}) - all_data = json.loads(c.text) - - for each in all_data: - class_dict[each["code"]] = each["id"], each["course"].get("nameZh", "") - -get_allclass() + try: + c = requests.post(all_class_url, headers=headers, data={"turnId": turn_id, "studentId": student_id}) + c.raise_for_status() + all_data = json.loads(c.text) + for each in all_data: + class_dict[each["code"]] = each["id"], each["course"].get("nameZh", "") + # print("可用课程：", class_dict) # 调试：打印可用课程 + except (requests.RequestException, json.JSONDecodeError) as e: + print(f"获取课程列表失败：{e}") + class_dict.clear() def ppp(course_id, index): + attempt = 0 + if course_id not in class_dict: + print(f"错误：课程代码 {course_id} 未在可选课程中找到。") + return False while True: - form_d = """studentAssoc: {} - lessonAssoc: {} - courseSelectTurnAssoc: 381 - scheduleGroupAssoc: - virtualCost: 0""".format(student_id, class_dict[course_id][0]) - dp_form_row = """studentId: {} - requestId: {}""" - form_data = deal_row(form_d) - c = requests.post(url, headers=headers, data=form_data) - dp_form_data = deal_row(dp_form_row.format(student_id, c.text)) - d = requests.post(drop_url, headers=dp_headers, data=dp_form_data) - data = json.loads(d.text) - if data["success"]: - print("恭喜你， {}选课成功！！！！".format(class_dict[course_id][1])) - return True - else: - print("{}选课失败 TAT : {}".format(class_dict[course_id][1], data["errorMessage"]["text"])) - time.sleep(hz) - + try: + form_d = """studentAssoc: {} + lessonAssoc: {} + courseSelectTurnAssoc: {} + scheduleGroupAssoc: + virtualCost: 0""".format(student_id, class_dict[course_id][0], turn_id) + form_data = deal_row(form_d) + c = requests.post(url, headers=headers, data=form_data) + c.raise_for_status() + # print(f"课程 {course_id} 的 add-request 响应：{c.text}, HTTP 状态码：{c.status_code}") + dp_form_row = """studentId: {} + requestId: {}""" + dp_form_data = deal_row(dp_form_row.format(student_id, c.text)) + d = requests.post(drop_url, headers=dp_headers, data=dp_form_data) + d.raise_for_status() + # print(f"课程 {course_id} 的 drop_url 响应：{d.text}, HTTP 状态码：{d.status_code}") + try: + data = json.loads(d.text) + if data is None: + print(f"课程 {course_id} 返回 null 响应，可能是服务器限流或错误") + attempt += 1 + time.sleep(random.uniform(2, 3)) # 随机延迟 2-3 秒 + continue + if data["success"]: + print("恭喜你， {}选课成功！！！！".format(class_dict[course_id][1])) + return True + else: + print("{}选课失败 TAT : {}".format(class_dict[course_id][1], data["errorMessage"]["text"])) + attempt += 1 + time.sleep(random.uniform(2, 3)) + except json.JSONDecodeError as e: + print(f"课程 {course_id} JSON 解析错误：{e}，响应内容：{d.text}") + attempt += 1 + time.sleep(random.uniform(2, 3)) + except requests.RequestException as e: + print(f"课程 {course_id} 网络错误：{e}") + attempt += 1 + time.sleep(random.uniform(2, 3)) if __name__ == "__main__": threads = [] + get_allclass() + # print("要抢的课程：", course_ids) # 调试：打印目标课程 for i in range(len(course_ids)): - task = threading.Thread(target=ppp, args=(course_ids[i], i, )) + task = threading.Thread(target=ppp, args=(course_ids[i], i)) threads.append(task) task.start() for task in threads: task.join() - print("全部抢完") - - + print("全部抢完") \ No newline at end of file