feat(work): 新增刷作业模式，具体使用请查看 main 函数

main.py

@@ -26,6 +26,9 @@ from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
 import warnings
 
+from deepseek import DeepSeek
+from doubao import DOUBAO
+
 # 修复 ANTIALIAS 错误 - 添加猴子补丁
 import PIL.Image
 
@@ -79,7 +82,20 @@ def replace_html_entities(text, replace_mode="keep_char"):
 
 
 class CKWK:
-    def __init__(self, un: str, pw: str, host: str) -> None:
+    def __init__(self, un: str, pw: str, host: str, ai_type="doubao") -> None:
+        """
+        un: 用户名
+        pw: 密码
+        host: 网站域名
+        ai_type: ai类别取值为 'doubao' 或者 'deepseek'
+        """
+        if ai_type == "doubao":
+            self.ai = DOUBAO()
+        elif ai_type == "deepseek":
+            self.ai = DeepSeek()
+        else:
+            self.ai = DOUBAO()
+
         self.session = requests.Session()
         # 定义重试策略
         retry_strategy = Retry(
@@ -93,18 +109,20 @@ class CKWK:
         adapter = HTTPAdapter(max_retries=retry_strategy)
         self.session.mount("http://", adapter)
         self.session.mount("https://", adapter)
-        # self.session.verify = False
-        # self.session.proxies.update(
-        #     {
-        #         "http": "http://127.0.0.1:9000",
-        #         "https": "http://127.0.0.1:9000",
-        #     }
-        # )
+        self.session.verify = False
+        self.session.proxies.update(
+            {
+                "http": "http://127.0.0.1:9000",
+                "https": "http://127.0.0.1:9000",
+            }
+        )
         self.username = un
         self.host = host
         self.password = pw
         self.user = dict()
         self.courses = []
+        self.works = []
+        self.qas = []
         self.ocr = ddddocr.DdddOcr()
         self.session.headers.update(
             {
@@ -215,6 +233,82 @@ class CKWK:
         finally:
             return video, state
 
+    def parse_q_a(self, html: str) -> None:
+        tree = etree.HTML(html)
+        qas = tree.xpath('//div[@class="topic-item"]/form')
+        for qa in qas:
+            items = {
+                "id": qa.xpath(
+                    './div[@class="courseexamcon-submit"]/input[@name="answerId"]/@value'
+                )[0],
+                "index": qa.xpath(
+                    './div[@class="courseexamcon-main"]/div/div[@class="num"]/span/text()'
+                )[0],
+                "type": qa.xpath(
+                    './div[@class="courseexamcon-main"]/div/div[@class="type"]/text()'
+                )[0],
+                "title": qa.xpath(
+                    './div[@class="courseexamcon-main"]/div[@class="name"]/text()'
+                )[0].strip(),
+                "chooies": [
+                    {
+                        "num": a.xpath('.//span[@class="num"]/text()')[0],
+                        "txt": a.xpath('.//span[@class="txt"]/text()')[0],
+                    }
+                    for a in qa.xpath(
+                        './div[@class="courseexamcon-main"]/div[@class="list"]//li'
+                    )
+                ],
+            }
+            self.qas.append(items)
+
+    def get_question_answer(self, courseID: str, nodeID: str, workID: str):
+        headers = self.session.headers
+        headers["x-requested-with"] = "XMLHttpRequest"
+        resp = self.session.get(
+            "https://{}/user/work/start".format(self.host),
+            params={"courseId": courseID, "nodeId": nodeID, "workId": workID},
+            headers=headers,
+        )
+        if resp.status_code == 200 and resp.json().get("status", False):
+            qa_path = resp.json().get("url", "")
+            self.log(f'开始作业: {resp.json().get("msg", "")} url -> {qa_path}')
+            resp2 = self.session.get("https://{}{}".format(self.host, qa_path))
+            if resp.status_code == 200:
+                self.parse_q_a(resp2.text)
+        else:
+            self.log(f'开始作业: {resp.json().get("msg", "")}')
+            return False
+        return True
+
+    def work_submit(self, answerID: str, workID: str, answer: str, final: bool):
+        headers = self.session.headers
+        headers["x-requested-with"] = "XMLHttpRequest"
+        data = {
+            "answerId": answerID,
+            "workId": workID,
+            "finish": "0" if final else "1",
+        }
+        if "," in answer:
+            answers = answer.split(",")
+            data["answer[]"] = answers
+            # for a in answers:
+            #     data.setdefault("answer[]", []).append(a)
+        else:
+            if answer == "":
+                data["answer"] = "C"
+            data["answer"] = answer
+
+        resp = self.session.post(
+            f"https://{self.host}/user/work/submit",
+            data=data,
+            headers=headers,
+        )
+        if resp.status_code == 200 and resp.json().get("status", False):
+            self.log(
+                f'提交答案: {resp.json().get("msg", "")} {answerID} => {answer} {final}'
+            )
+
     def get_course(self, _id: str) -> tuple[int, int]:
         resp = self.session.get("https://{}/user/node?nodeId={}".format(self.host, _id))
         if resp.status_code == 200 and "错误提示" not in resp.text:
@@ -306,7 +400,23 @@ class CKWK:
         if resp.status_code == 200 and resp.json().get("status", False):
             self.log(f'{resp.json().get("msg", "")} page -> {page}')
             self.courses.extend(resp.json().get("list", []))
-            if page <= resp.json().get("pageInfo", {}).get("pageCount", 0):
+            if page < resp.json().get("pageInfo", {}).get("pageCount", 0):
+                self.get_study_record(course_id, page + 1)
+
+    def get_work_record(self, course_id: str, user_id: str, page=1):
+        resp = self.session.get(
+            f"https://{self.host}/user/study_record/work.json",
+            params={
+                "courseId": course_id,
+                # "userId": user_id,
+                "page": page,
+                "_": str(int(time.time() * 1000)),
+            },
+        )
+        if resp.status_code == 200 and resp.json().get("status", False):
+            self.log(f'{resp.json().get("msg", "")} page -> {page}')
+            self.works.extend(resp.json().get("list", []))
+            if page < resp.json().get("pageInfo", {}).get("pageCount", 0):
                 self.get_study_record(course_id, page + 1)
 
     def log(self, *args, **kwargs) -> None:
@@ -414,6 +524,7 @@ class CKWK:
             self.online()
 
     def run(self) -> None:
+        print("当前为刷课模式: ")
         while self.login():
             self.get_user()
             thread = Thread(target=self.start_loop, daemon=True)
@@ -443,7 +554,7 @@ class CKWK:
                             f'{l.get("name", "")} --> {l.get("id", "")} --> 已经学习过了'
                         )
                         continue
-                    
+
                     if isReply:
                         self.reply(c.get("id", ""), l.get("id", ""))
 
@@ -470,6 +581,48 @@ class CKWK:
             self.log("所有课程已经结束")
             break
 
+    def work(self):
+        print("当前为刷作业模式: ")
+        while self.login():
+            self.get_user()
+            thread = Thread(target=self.start_loop, daemon=True)
+            thread.start()
+            cs = self.user.get("courses", [])
+            print(
+                f'姓名: {self.user.get("name", "")}\n\t互评: {self.user.get("huping", 0)}\n\t乐学园: {self.user.get("lexueyuan", 0)}\n\t讨论主题: {self.user.get("taolunzhuti", 0)}\n\t学习时长: {self.user.get("study_time", "")}'
+            )
+            if len(cs) > 1:
+                for i, _c in enumerate(cs):
+                    print(f'{i+1}: {_c.get("name", "")} {_c.get("progress", "")}')
+                cs_order = input("请输入刷课顺序(用英文逗号分隔,不需要耍的不用加): ")
+                cs_list = [cs[int(_c) - 1] for _c in cs_order.split(",")]
+            cs_list = cs
+            for c in cs_list:
+                self.get_work_record(c.get("id", ""), "")
+                # print(self.works)
+                for work in self.works:
+                    self.qas = []
+                    if "已阅" in work.get("state"):
+                        self.log(
+                            f'{work.get("title", "")} 已阅 {work.get("finalScore", "")}'
+                        )
+                        continue
+                    if not self.get_question_answer(
+                        c.get("id", ""), work.get("nodeId", ""), work.get("id", "")
+                    ):
+                        continue
+                    for q in self.qas:
+                        # print(q)
+                        chooise = self.ai.chat(q)
+                        self.work_submit(
+                            q.get("id", ""),
+                            work.get("id", ""),
+                            chooise,
+                            int(q.get("index", 1)) < len(self.qas),
+                        )
+                    # break
+            break
+
 
 if __name__ == "__main__":
     host_server = [
@@ -486,12 +639,17 @@ if __name__ == "__main__":
         u = input("请输入账号: ")
     if p == "":
         p = input("请输入密码: ")
-    r = input("是否需要评论(不需要[0],需要[1]): ")
-    if int(r) == 1:
-        isReply = True
-    else:
-        isReply = False
-    print(bool(isReply))
-    if u is None and p is None and c in [0, 1, 2]:
-        os._exit(0)
-    CKWK(f"{u}", f"{p}", host_server[c][0]).run()
+
+    # 刷课
+    # r = input("是否需要评论(不需要[0],需要[1]): ")
+    # if int(r) == 1:
+    #     isReply = True
+    # else:
+    #     isReply = False
+    # print(bool(isReply))
+    # if u is None and p is None and c in [0, 1, 2]:
+    #     os._exit(0)
+    # CKWK(f"{u}", f"{p}", host_server[c][0]).run()
+
+    # 刷作业
+    CKWK(f"{u}", f"{p}", host_server[c][0]).work()