You've already forked HeurAMS-legacy
243 lines
7.9 KiB
Python
243 lines
7.9 KiB
Python
# mcq.py
|
|
from .base import BasePuzzle
|
|
import random
|
|
from typing import List, Dict, Optional, Union
|
|
from heurams.services.logger import get_logger
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class MCQPuzzle(BasePuzzle):
|
|
"""选择题谜题生成器
|
|
|
|
该类用于生成和管理选择题谜题, 支持多个题目同时生成,
|
|
每个题目包含问题, 正确答案和干扰项选项.
|
|
|
|
Attributes:
|
|
prefix (str): 题目前缀文本
|
|
mapping (Dict[str, str]): 问题和正确答案的映射字典
|
|
jammer (List[str]): 干扰项列表
|
|
max_riddles_num (int): 最大题目数量限制
|
|
wording (Union[str, List[str]]): 题目文本内容
|
|
answer (Union[str, List[str]]): 正确答案列表
|
|
options (List[List[str]]): 每个题目的选项列表
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
mapping: Dict[str, str],
|
|
jammer: List[str],
|
|
max_riddles_num: int = 2,
|
|
prefix: str = "",
|
|
) -> None:
|
|
"""初始化选择题谜题生成器
|
|
|
|
Args:
|
|
mapping: 问题和正确答案的映射字典, 键为问题, 值为正确答案
|
|
jammer: 干扰项列表, 用于生成错误选项
|
|
max_riddles_num: 每次生成的最大题目数量, 范围限制在1-5之间
|
|
prefix: 题目前缀文本, 会显示在每个题目之前
|
|
"""
|
|
logger.debug(
|
|
"MCQPuzzle.__init__: mapping size=%d, jammer size=%d, max_riddles_num=%d",
|
|
len(mapping),
|
|
len(jammer),
|
|
max_riddles_num,
|
|
)
|
|
self.prefix = prefix
|
|
self.mapping = mapping
|
|
self.max_riddles_num = max(1, min(max_riddles_num, 5))
|
|
|
|
# 初始化干扰项, 确保至少有4个选项
|
|
self._init_jammer(jammer)
|
|
|
|
# 初始化题目状态
|
|
self._reset_puzzle_state()
|
|
|
|
def _init_jammer(self, jammer: List[str]) -> None:
|
|
"""初始化干扰项列表
|
|
|
|
合并传入的干扰项和所有正确答案, 确保去重后至少有4个干扰项.
|
|
|
|
Args:
|
|
jammer: 传入的干扰项列表
|
|
"""
|
|
# 合并正确答案和传入的干扰项, 并去重
|
|
logger.debug(f"答案映射: {self.mapping}, {type(self.mapping)}")
|
|
logger.debug(f"干扰项: {jammer}, {type(jammer)}")
|
|
unique_jammers = set(jammer + list(self.mapping.values()))
|
|
self.jammer = list(unique_jammers)
|
|
|
|
# 确保至少有4个干扰项
|
|
while len(self.jammer) < 4:
|
|
self.jammer.append(" " * (4 - len(self.jammer)))
|
|
|
|
unique_jammers = set(jammer + list(self.mapping.values()))
|
|
|
|
def _reset_puzzle_state(self) -> None:
|
|
"""重置谜题状态为初始值
|
|
|
|
将题目文本, 答案和选项重置为默认状态.
|
|
"""
|
|
self.wording: Union[str, List[str]] = "选择题 - 尚未刷新谜题"
|
|
self.answer: Union[str, List[str]] = ["选择题 - 尚未刷新谜题"]
|
|
self.options: List[List[str]] = []
|
|
|
|
def refresh(self) -> None:
|
|
"""刷新谜题, 生成指定数量的选择题
|
|
|
|
从mapping中随机选择指定数量的问题, 为每个问题生成包含正确答案
|
|
和干扰项的选项列表, 并更新谜题状态.
|
|
|
|
Raises:
|
|
ValueError: 当mapping为空时不会抛出异常, 但会设置空谜题状态
|
|
"""
|
|
logger.debug("MCQPuzzle.refresh 开始, mapping size=%d", len(self.mapping))
|
|
if not self.mapping:
|
|
self._set_empty_puzzle()
|
|
return
|
|
|
|
num_questions = min(self.max_riddles_num, len(self.mapping))
|
|
selected_questions = random.sample(list(self.mapping.items()), num_questions)
|
|
|
|
puzzles: List[str] = []
|
|
answers: List[str] = []
|
|
all_options: List[List[str]] = []
|
|
|
|
for question, correct_answer in selected_questions:
|
|
options = self._generate_options(correct_answer)
|
|
puzzles.append(question)
|
|
answers.append(correct_answer)
|
|
all_options.append(options)
|
|
|
|
self.wording = self._format_questions(puzzles)
|
|
self.answer = answers
|
|
self.options = all_options
|
|
|
|
def _set_empty_puzzle(self) -> None:
|
|
"""设置为空谜题状态
|
|
|
|
当没有可用的题目时, 设置相应的提示信息.
|
|
"""
|
|
self.wording = "无可用题目"
|
|
self.answer = ["无答案"]
|
|
self.options = []
|
|
|
|
def _generate_options(self, correct_answer: str) -> List[str]:
|
|
"""为单个问题生成选项列表(包含正确答案和干扰项)
|
|
|
|
Args:
|
|
correct_answer: 当前问题的正确答案
|
|
|
|
Returns:
|
|
包含4个选项的列表, 其中一个是正确答案, 三个是干扰项
|
|
|
|
Note:
|
|
如果可用干扰项不足3个, 会使用重复的干扰项填充
|
|
"""
|
|
options = [correct_answer]
|
|
|
|
# 获取可用的干扰项(排除正确答案)
|
|
available_jammers = [
|
|
jammer for jammer in self.jammer if jammer != correct_answer
|
|
]
|
|
|
|
# 选择3个干扰项
|
|
if len(available_jammers) >= 3:
|
|
selected_jammers = random.sample(available_jammers, 3)
|
|
else:
|
|
selected_jammers = random.choices(available_jammers, k=3)
|
|
|
|
options.extend(selected_jammers)
|
|
random.shuffle(options)
|
|
|
|
return options
|
|
|
|
def _format_questions(self, puzzles: List[str]) -> List[str]:
|
|
"""格式化问题列表为可读的文本
|
|
|
|
Args:
|
|
puzzles: 原始问题文本列表
|
|
|
|
Returns:
|
|
格式化后的问题文本列表, 包含编号和前缀
|
|
|
|
Example:
|
|
输入: ["问题1", "问题2"]
|
|
输出: ["前缀:\\n 1. 问题1", "前缀:\\n 2. 问题2"]
|
|
"""
|
|
if not puzzles:
|
|
return []
|
|
|
|
formatted_questions = []
|
|
for i, puzzle in enumerate(puzzles, 1):
|
|
question_text = (
|
|
f"{self.prefix}:\n {i}. {puzzle}" if self.prefix else f"{i}. {puzzle}"
|
|
)
|
|
formatted_questions.append(question_text)
|
|
|
|
return formatted_questions
|
|
|
|
def __str__(self) -> str:
|
|
"""返回谜题的字符串表示
|
|
|
|
Returns:
|
|
包含所有问题和正确答案的格式化字符串
|
|
|
|
Example:
|
|
选择题 - 尚未刷新谜题
|
|
正确答案: 选择题 - 尚未刷新谜题
|
|
"""
|
|
if isinstance(self.wording, list):
|
|
question_text = "\n".join(self.wording)
|
|
else:
|
|
question_text = self.wording
|
|
|
|
if isinstance(self.answer, list):
|
|
answer_text = ", ".join(self.answer)
|
|
else:
|
|
answer_text = str(self.answer)
|
|
|
|
return f"{question_text}\n正确答案: {answer_text}"
|
|
|
|
def get_question_count(self) -> int:
|
|
"""获取当前生成的题目数量
|
|
|
|
Returns:
|
|
当前题目的数量, 如果尚未刷新则返回 0
|
|
"""
|
|
if isinstance(self.wording, list):
|
|
return len(self.wording)
|
|
elif self.wording == "选择题 - 尚未刷新谜题" or self.wording == "无可用题目":
|
|
return 0
|
|
else:
|
|
return 1
|
|
|
|
def get_correct_answer_for_question(self, question_index: int) -> Optional[str]:
|
|
"""获取指定题目的正确答案
|
|
|
|
Args:
|
|
question_index: 题目索引(从0开始)
|
|
|
|
Returns:
|
|
指定题目的正确答案, 如果索引无效则返回None
|
|
"""
|
|
if not isinstance(self.answer, list):
|
|
return None
|
|
if 0 <= question_index < len(self.answer):
|
|
return self.answer[question_index]
|
|
return None
|
|
|
|
def get_options_for_question(self, question_index: int) -> Optional[List[str]]:
|
|
"""获取指定题目的选项列表
|
|
|
|
Args:
|
|
question_index: 题目索引(从0开始)
|
|
|
|
Returns:
|
|
指定题目的选项列表, 如果索引无效则返回None
|
|
"""
|
|
if 0 <= question_index < len(self.options):
|
|
return self.options[question_index]
|
|
return None
|