rush_b.py

import os
import json
import signal
import time
import platform


def check_answer(a1: str, a2: str):
    if a1 == a2:
        return True
    if a1.upper() == a2.upper():
        return True
    b1 = a1.upper().replace("1", "A").replace("2", "B").replace("3", "C").replace("4", "D")\
        .replace("对", "A").replace("错", "B").replace("T", "A").replace("F", "B")
    b2 = a2.upper().replace("1", "A").replace("2", "B").replace("3", "C").replace("4", "D")\
        .replace("对", "A").replace("错", "B").replace("T", "A").replace("F", "B")
    if b1 == b2:
        return True
    return False


do_what = "毛泽东思想和中国特色社会主义理论体系概论题库.json"
problems = []

# 跳过个数
bypass_count = 0
# 只做错题？
only_do_error = False
# 错误阈值
error_threshold = 1
# 是否清屏
do_clear = False
# do_clear = True

# custom
clear_command = "cls" if platform.system() == 'Windows' else "clear"
input_hint = "\033[34m>>\033[0m "
blank = "\n-----------------------------------------------"

auto_save = True
finish_cnt = 0


def save_exit(signum, frame):
    with open("process.json", "w", encoding='UTF-8') as f:
        f.write(json.dumps(problems, ensure_ascii=False))
    exit(0)


signal.signal(signal.SIGINT, save_exit)
signal.signal(signal.SIGTERM, save_exit)
if os.path.exists("process.json"):
    with open("process.json", "r", encoding='UTF-8') as f:
        problems = json.loads(f.read())
else:
    with open(do_what, "r", encoding='UTF-8') as f:
        problems = json.loads(f.read())

for pid, problem in enumerate(problems):
    if pid == len(problems) - 1:
        with open("process.json", "w", encoding='UTF-8') as f:
            f.write(json.dumps(problems, ensure_ascii=False))
    if pid < bypass_count:
        continue
    if only_do_error:
        try:
            if problem['error_times'] < error_threshold:
                continue
        except KeyError:
            continue
    try:
        if check_answer(problem['answer'], problem['my_answer']):
            continue
    except KeyError:
        pass
    type=problem['type']
    if type=='单选题':
        print("\033[32m[%s]\033[0m 第%d题：\n" %
              (problem['type'].replace("题", ""), pid + 1))
    elif type=='多选题':
        print("\033[31m[%s]\033[0m 第%d题：\n" %
              (problem['type'].replace("题", ""), pid + 1))
    elif type=='判断题':
        print("\033[36m[%s]\033[0m 第%d题：\n" %
              (problem['type'].replace("题", ""), pid + 1))
    else:
        raise ValueError()
    print(problem['problem'])
    if problem['type'] != "判断题":
        print(problem['A'])
        print(problem['B'])
        print(problem['C'])
        print(problem['D'])
    my_answer = input(input_hint)
    if my_answer.upper() == "SAVE":
        with open("process.json", "w", encoding='UTF-8') as f:
            f.write(json.dumps(problems, ensure_ascii=False))
        my_answer = input(input_hint)
    if my_answer.upper() == "QUIT":
        with open("process.json", "w", encoding='UTF-8') as f:
            f.write(json.dumps(problems, ensure_ascii=False))
        exit(0)

    my_answer = "".join((lambda x: (x.sort(), x)[1])(list(my_answer)))
    problems[pid]['my_answer'] = my_answer

    if check_answer(problem['answer'], my_answer):
        print("\033[32m答对了！\033[0m")
    else:
        print("\033[31m答错了！\033[0m")
        print("正确答案：\033[32m" + problem['answer'] + "\033[0m")
        print("我的答案：\033[31m" + my_answer + "\033[0m", end='')
        try:
            problems[pid]['error_times'] += 1
        except KeyError:
            problems[pid]['error_times'] = 1
        input()

    # 记录做过的次数
    try:
        problems[pid]['do_times'] += 1
    except KeyError:
        problems[pid]['do_times'] = 1

    if do_clear:
        # time.sleep(1)
        os.system(clear_command)
    else:
        print(blank)

    finish_cnt += 1

    if (auto_save and finish_cnt % 1 == 0) or pid == len(problems) - 1:
        with open("process.json", "w", encoding='UTF-8') as f:
            f.write(json.dumps(problems, ensure_ascii=False))