suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-09-03 06:22:25 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

adapted to a16, #18

Browse Source
This commit is contained in:
felinae98 2021-10-21 16:34:20 +08:00
parent ad6bd9c4dd
commit 4d2cad5369
No known key found for this signature in database
GPG Key ID: 00C8B010587FF610
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/plugins/nonebot_hk_reporter/config_manager.py
View File

@ -14,6 +14,11 @@ from .send import send_msgs
from .utils import parse_text from .utils import parse_text
from .types import Target from .types import Target
def _gen_prompt_template(prompt: str):
if hasattr(Message, 'template'):
return Message.template(prompt)
return prompt
common_platform = [p.platform_name for p in \ common_platform = [p.platform_name for p in \
filter(lambda platform: platform.enabled and platform.is_common, filter(lambda platform: platform.enabled and platform.is_common,
platform_manager.values()) platform_manager.values())
@ -46,7 +51,7 @@ def do_add_sub(add_sub: Type[Matcher]):
else: else:
await add_sub.reject('平台输入错误') await add_sub.reject('平台输入错误')
@add_sub.got('platform', '{_prompt}', parse_platform) @add_sub.got('platform', _gen_prompt_template('{_prompt}'), parse_platform)
@add_sub.handle() @add_sub.handle()
async def init_id(bot: Bot, event: Event, state: T_State): async def init_id(bot: Bot, event: Event, state: T_State):
if platform_manager[state['platform']].has_target: if platform_manager[state['platform']].has_target:
@ -63,7 +68,7 @@ def do_add_sub(add_sub: Type[Matcher]):
state['id'] = target state['id'] = target
state['name'] = name state['name'] = name
@add_sub.got('id', '{_prompt}', parse_id) @add_sub.got('id', _gen_prompt_template('{_prompt}'), parse_id)
@add_sub.handle() @add_sub.handle()
async def init_cat(bot: Bot, event: Event, state: T_State): async def init_cat(bot: Bot, event: Event, state: T_State):
if not platform_manager[state['platform']].categories: if not platform_manager[state['platform']].categories:
@ -80,7 +85,7 @@ def do_add_sub(add_sub: Type[Matcher]):
res.append(platform_manager[state['platform']].reverse_category[cat]) res.append(platform_manager[state['platform']].reverse_category[cat])
state['cats'] = res state['cats'] = res
@add_sub.got('cats', '{_prompt}', parser_cats) @add_sub.got('cats', _gen_prompt_template('{_prompt}'), parser_cats)
@add_sub.handle() @add_sub.handle()
async def init_tag(bot: Bot, event: Event, state: T_State): async def init_tag(bot: Bot, event: Event, state: T_State):
if not platform_manager[state['platform']].enable_tag: if not platform_manager[state['platform']].enable_tag:
@ -94,7 +99,7 @@ def do_add_sub(add_sub: Type[Matcher]):
else: else:
state['tags'] = str(event.get_message()).strip().split() state['tags'] = str(event.get_message()).strip().split()
@add_sub.got('tags', '{_prompt}', parser_tags) @add_sub.got('tags', _gen_prompt_template('{_prompt}'), parser_tags)
@add_sub.handle() @add_sub.handle()
async def add_sub_process(bot: Bot, event: Event, state: T_State): async def add_sub_process(bot: Bot, event: Event, state: T_State):
config = Config() config = Config()