zhenxun_bot/plugins/draw_card/rule.py

from nonebot.rule import Rule
from nonebot.adapters.cqhttp import Bot, MessageEvent
from nonebot.typing import T_State
from .config import GENSHIN_FLAG, PRTS_FLAG, PRETTY_FLAG, GUARDIAN_FLAG, PCR_FLAG, AZUR_FLAG, FGO_FLAG, ONMYOJI_FLAG


def is_switch(game_name: str) -> Rule:

    async def _is_switch(bot: Bot, event: MessageEvent, state: T_State) -> bool:
        if game_name == 'prts':
            return PRTS_FLAG
        if game_name == 'genshin':
            return GENSHIN_FLAG
        if game_name == 'pretty':
            return PRETTY_FLAG
        if game_name == 'guardian':
            return GUARDIAN_FLAG
        if game_name == 'pcr':
            return PCR_FLAG
        if game_name == 'azur':
            return AZUR_FLAG
        if game_name == 'fgo':
            return FGO_FLAG
        if game_name == 'onmyoji':
            return ONMYOJI_FLAG
        else:
            return False

    return Rule(_is_switch)