mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-22 06:17:54 +00:00
71 lines
2.9 KiB
Python
71 lines
2.9 KiB
Python
import random
|
|
|
|
from telegram import Update, Poll
|
|
from telegram.constants import ChatAction
|
|
from telegram.error import BadRequest
|
|
from telegram.ext import CallbackContext, CommandHandler, filters
|
|
|
|
from core.admin import BotAdminService
|
|
from core.baseplugin import BasePlugin
|
|
from core.plugin import Plugin, handler
|
|
from core.quiz import QuizService
|
|
from utils.decorators.error import error_callable
|
|
from utils.decorators.restricts import restricts
|
|
from utils.log import logger
|
|
|
|
|
|
class QuizPlugin(Plugin, BasePlugin):
|
|
"""派蒙的十万个为什么"""
|
|
|
|
def __init__(self, quiz_service: QuizService = None, bot_admin_service: BotAdminService = None):
|
|
self.bot_admin_service = bot_admin_service
|
|
self.quiz_service = quiz_service
|
|
self.time_out = 120
|
|
|
|
@handler(CommandHandler, command="quiz", block=False)
|
|
@restricts(restricts_time_of_groups=20)
|
|
@error_callable
|
|
async def command_start(self, update: Update, context: CallbackContext) -> None:
|
|
user = update.effective_user
|
|
message = update.effective_message
|
|
chat = message.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(f"用户 {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(f"Quiz模块 correct_option 异常 question_id[{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(context, message.chat_id, message.message_id, 300)
|
|
self._add_delete_message_job(context, poll_message.chat_id, poll_message.message_id, 300)
|