zhenxun_bot/plugins/draw_card/rule.py
2021-06-04 18:01:33 +08:00

43 lines
948 B
Python

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)