feat: 上传 ziantt 题库

Author: Karben233

bili-hardcore/client/ziantt.py

@@ -0,0 +1,32 @@
+from tools.request_b import session
+from tools.logger import logger
+
+def save_question(json):
+    '''
+    保存题目信息
+    {
+        "qid": qid, // int类型
+        "question": question, // 题面
+        "ans_1": ans_1, // 无顺序限制，提交ans_text而非ans_hash
+        "ans_2": ans_2,
+        "ans_3": ans_3,
+        "ans_4": ans_4,
+        "answer": correct_answer, // 若已知正确答案提交正确的ans_text，未知请提交null
+        "source": source, // int类型，与author字段二选一，以b站接口返回的实际内容为准
+        "author": author, // str类型，与source字段二选一，以b站接口返回的实际内容为准
+        "category": category // str类型，请传入单一类型id，如"2"，如用户选择的是多个类型，不确定id请传入null
+    }
+    '''
+    res = session.post('https://senior.ziantt.top/submit', 
+    json = json,  
+    headers={
+        "User-Agent": "bili-hardcore Script Report", 
+    });
+    # resp = res.json()
+    # logger.info(resp)
+    # if resp["status"] == "success":
+    #     logger.info("题库服务器提交成功")
+    # elif resp["status"] == "exist":
+    #     logger.info("题目已存在")
+    # else:
+    #     logger.info("题库服务器提交失败")

bili-hardcore/scripts/start_senior.py

@@ -6,16 +6,26 @@
 
 from config.config import model_choice
 from scripts.check_config import clear_config
+from client.ziantt import save_question
+from time import sleep
 
 class QuizSession:
     def __init__(self):
         self.question_id = None
         self.answers = None
+        self.question_json = None
         self.question_num = 0
         self.question = None
+        self.current_score = 0
+        self.category = None
+        self.confirm = False
 
     def start(self):
         """开始答题会话"""
+        print("我们正在构建一个硬核会员题库，您是否愿意上传答题信息帮助我们构建题库？")
+        print("此操作仅会上传题目和题目对应的答案信息，并不会上传您的其他信息")
+        confirm = input('是否授权上传本次答题内容？[1]是 [2]否: ')
+        self.confirm = confirm == '1'
         try:
             while self.question_num < 100:
                 if not self.get_question():
@@ -48,6 +58,13 @@ def start(self):
                 if not self.submit_answer(result):
                     logger.error("提交答案失败")
                     return
+                score = question_result().get('score');
+                if self.current_score < score:
+                    logger.info("回答正确, 当前得分:{}".format(score))
+                    self.current_score = score
+                    self.upload_question(answer)
+                else:
+                    logger.info("回答错误, 当前得分:{}".format(score))
             self.print_result()
         except KeyboardInterrupt:
             logger.info("答题会话已终止")
@@ -69,6 +86,7 @@ def get_question(self):
                 return self.handle_verification()
 
             data = question.get('data', {})
+            self.question_json = data
             self.question = data.get('question')
             self.answers = data.get('answers', [])
             self.question_id = data.get('id')
@@ -96,6 +114,7 @@ def handle_verification(self):
                 logger.info(f"ID: {cat.get('id')} - {cat.get('name')}")
             logger.info("tips: 输入多个分类ID请用 *英文逗号* 隔开,例如:1,2,3(最多三个分类)")
             ids = input('请输入分类ID: ')
+            self.category = ids
             logger.info("获取验证码...")
             captcha_res = captcha_get()
             logger.info("请打开链接查看验证码内容:{}".format(captcha_res.get('url')))
@@ -109,7 +128,6 @@ def handle_verification(self):
             else:
                 logger.error("验证失败")
                 return False
-
         except Exception as e:
             logger.error(f"验证过程发生错误: {str(e)}")
             return False
@@ -180,6 +198,26 @@ def print_result(self):
         except Exception as e:
             logger.error(f"获取答题结果失败: {str(e)}")
 
+    def upload_question(self, answer):
+        # 保存题目
+        if self.confirm:
+            try:
+                question_submit_content = {
+                    'qid': self.question_json.get('id'),
+                    'question': self.question_json.get('question'),
+                    'ans_1': self.answers[0].get('ans_text'),
+                    'ans_2': self.answers[1].get('ans_text'),
+                    'ans_3': self.answers[2].get('ans_text'),
+                    'ans_4': self.answers[3].get('ans_text'),
+                    'answer': self.answers[answer-1].get('ans_text'),
+                    'source': self.question_json.get('source'),
+                    'author': self.question_json.get('author'),
+                    'category': self.category and len(self.category) == 1 and self.category[0] or None,
+                }
+                save_question(question_submit_content)
+            except Exception as e:
+                logger.error(f"题库上传失败，此报错不影响正常答题: {str(e)}")
+
 # 创建答题会话实例
 quiz_session = QuizSession()