PamGram/plugins/genshin/quiz.py

import random

from telegram import Poll, Update
from telegram.constants import ChatAction
from telegram.error import BadRequest
from telegram.ext import filters, CallbackContext

from core.plugin import Plugin, handler
from core.services.quiz.services import QuizService
from core.services.users.services import UserService
from utils.log import logger

__all__ = ("QuizPlugin",)


class QuizPlugin(Plugin):
    """派蒙的十万个为什么"""

    def __init__(self, quiz_service: QuizService = None, user_service: UserService = None):
        self.user_service = user_service
        self.quiz_service = quiz_service
        self.time_out = 120

    @handler.message(filters=filters.Regex("来一道题"))
    @handler.command(command="quiz", block=False)
    async def command_start(self, update: Update, _: CallbackContext) -> None:
        message = update.effective_message
        user = update.effective_user
        chat = update.effective_chat
        await message.reply_chat_action(ChatAction.TYPING)
        question_id_list = await self.quiz_service.get_question_id_list()
        if filters.ChatType.GROUPS.filter(message):
            logger.info("用户 %s[%s] 在群 %s[%s] 发送挑战问题命令请求", user.full_name, user.id, chat.title, chat.id)
            if len(question_id_list) == 0:
                return None
        if len(question_id_list) == 0:
            return None
        question_id = random.choice(question_id_list)  # nosec
        question = await self.quiz_service.get_question(question_id)
        _options = []
        correct_option = None
        for answer in question.answers:
            _options.append(answer.text)
            if answer.is_correct:
                correct_option = answer.text
        if correct_option is None:
            question_id = question["question_id"]
            logger.warning("Quiz模块 correct_option 异常 question_id[%s]", question_id)
            return None
        random.shuffle(_options)
        index = _options.index(correct_option)
        try:
            poll_message = await message.reply_poll(
                question.text,
                _options,
                correct_option_id=index,
                is_anonymous=False,
                open_period=self.time_out,
                type=Poll.QUIZ,
            )
        except BadRequest as exc:
            if "Not enough rights" in exc.message:
                poll_message = await message.reply_text("出错了呜呜呜 ~ 权限不足，请请检查投票权限是否开启")
            else:
                raise exc
        if filters.ChatType.GROUPS.filter(message):
            self.add_delete_message_job(message, delay=300)
            self.add_delete_message_job(poll_message, delay=300)