diff --git a/genshinuid_enka/__init__.py b/genshinuid_enka/__init__.py index 8cc154e6..71677beb 100644 --- a/genshinuid_enka/__init__.py +++ b/genshinuid_enka/__init__.py @@ -5,16 +5,30 @@ from pathlib import Path from .draw_char_card import * from .draw_char_card import draw_char_img from ..all_import import * # noqa: F401,F403 +from ..utils.enka_api.get_enka_data import switch_api from ..utils.enka_api.enka_to_data import enka_to_data from ..utils.db_operation.db_operation import get_all_uid from ..utils.message.error_reply import * # noqa: F401,F403 -from ..utils.mhy_api.convert_mysid_to_uid import convert_mysid from ..utils.alias.alias_to_char_name import alias_to_char_name AUTO_REFRESH = False PLAYER_PATH = Path(__file__).parents[1] / 'player' +@sv.on_fullmatch('切换api') +async def send_change_api_info(bot: HoshinoBot, ev: CQEvent): + if ev.sender: + qid = int(ev.sender['user_id']) + else: + return + + if qid not in bot.config.SUPERUSERS: + return + + im = await switch_api() + await bot.send(ev, im) + + @sv.on_rex( r'^(\[CQ:at,qq=[0-9]+\])?( )?' r'(uid|查询|mys)([0-9]+)?' diff --git a/genshinuid_signin/__init__.py b/genshinuid_signin/__init__.py index 4136e603..9db6c1f1 100644 --- a/genshinuid_signin/__init__.py +++ b/genshinuid_signin/__init__.py @@ -24,6 +24,13 @@ async def get_sign_func(bot: HoshinoBot, ev: CQEvent): @sv.on_fullmatch('全部重签') async def recheck(bot: HoshinoBot, ev: CQEvent): + if ev.sender: + qid = int(ev.sender['user_id']) + else: + return + if qid not in bot.config.SUPERUSERS: + return + logger.info('开始执行[全部重签]') await bot.send(ev, '已开始执行') await send_daily_sign()