mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-04 02:26:11 +08:00
add outer manager #6
This commit is contained in:
felinae98
parent
74b754f283
commit
286772a6db
@ -1,3 +1,4 @@
|
||||
from typing import Type
|
||||
from nonebot import logger, on_command
|
||||
from nonebot.adapters.cqhttp import Bot, Event, GroupMessageEvent
|
||||
from nonebot.adapters.cqhttp.message import Message
|
||||
@ -5,6 +6,7 @@ from nonebot.adapters.cqhttp.permission import GROUP_ADMIN, GROUP_MEMBER, GROUP_
|
||||
from nonebot.permission import Permission, SUPERUSER
|
||||
from nonebot.rule import to_me
|
||||
from nonebot.typing import T_State
|
||||
from nonebot.matcher import Matcher
|
||||
|
||||
from .config import Config, NoSuchSubscribeException
|
||||
from .platform import platform_manager, check_sub_target
|
||||
@ -23,152 +25,158 @@ async def send_help(bot: Bot, event: Event, state: T_State):
|
||||
message = '使用方法：\n@bot 添加订阅（仅管理员）\n@bot 查询订阅\n@bot 删除订阅（仅管理员）'
|
||||
await help_match.finish(Message(await parse_text(message)))
|
||||
|
||||
add_sub = on_command("添加订阅", rule=to_me(), permission=GROUP_ADMIN | GROUP_OWNER | SUPERUSER, priority=5)
|
||||
@add_sub.handle()
|
||||
async def add_sub_handle_platform(bot: Bot, event: Event, state: T_State):
|
||||
if 'platform' in state:
|
||||
return
|
||||
await bot.send(event=event, message='请输入想要订阅的平台，目前支持：\n' +
|
||||
''.join(['{}：{}\n'.format(platform_name, platform_manager[platform_name].name) \
|
||||
for platform_name in common_platform]) +
|
||||
'要查看全部平台请输入：“全部”'
|
||||
)
|
||||
await add_sub.pause()
|
||||
|
||||
@add_sub.handle()
|
||||
async def add_sub_parse_platform(bot: Bot, event: Event, state: T_State):
|
||||
if 'platform' in state:
|
||||
return
|
||||
platform = str(event.get_message()).strip()
|
||||
if platform == '全部':
|
||||
message = '全部平台\n' + \
|
||||
'\n'.join(['{}：{}'.format(platform_name, platform.name) \
|
||||
for platform_name, platform in platform_manager.items()])
|
||||
await add_sub.reject(message)
|
||||
elif platform in platform_manager:
|
||||
state['platform'] = platform
|
||||
else:
|
||||
await add_sub.reject('平台输入错误')
|
||||
# @add_sub.got('platform', '请输入想要订阅的平台，目前支持：{}'.format(', '.join(platform_manager.keys())))
|
||||
# @add_sub.got('id', '请输入订阅用户的id，详情查阅https://github.com/felinae98/nonebot-hk-reporter')
|
||||
@add_sub.handle()
|
||||
async def add_sub_handle_id(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].has_target or 'id' in state:
|
||||
return
|
||||
await bot.send(event=event, message='请输入订阅用户的id，详情查阅https://github.com/felinae98/nonebot-hk-reporter')
|
||||
await add_sub.pause()
|
||||
def do_add_sub(add_sub: Type[Matcher]):
|
||||
@add_sub.handle()
|
||||
async def init_promote(bot: Bot, event: Event, state: T_State):
|
||||
state['_prompt'] = '请输入想要订阅的平台，目前支持：\n' + \
|
||||
''.join(['{}：{}\n'.format(platform_name, platform_manager[platform_name].name) \
|
||||
for platform_name in common_platform]) + \
|
||||
'要查看全部平台请输入：“全部”'
|
||||
|
||||
@add_sub.handle()
|
||||
async def add_sub_parse_id(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].has_target or 'id' in state:
|
||||
return
|
||||
target = str(event.get_message()).strip()
|
||||
name = await check_sub_target(state['platform'], target)
|
||||
if not name:
|
||||
await add_sub.reject('id输入错误')
|
||||
state['id'] = target
|
||||
state['name'] = name
|
||||
async def parse_platform(bot: Bot, event: Event, state: T_State) -> None:
|
||||
platform = str(event.get_message()).strip()
|
||||
if platform == '全部':
|
||||
message = '全部平台\n' + \
|
||||
'\n'.join(['{}：{}'.format(platform_name, platform.name) \
|
||||
for platform_name, platform in platform_manager.items()])
|
||||
await add_sub.reject(message)
|
||||
elif platform in platform_manager:
|
||||
state['platform'] = platform
|
||||
else:
|
||||
await add_sub.reject('平台输入错误')
|
||||
|
||||
@add_sub.handle()
|
||||
async def add_sub_handle_cat(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].categories:
|
||||
return
|
||||
if 'cats' in state:
|
||||
return
|
||||
await bot.send(event=event, message='请输入要订阅的类别，以空格分隔，支持的类别有：{}'.format(
|
||||
','.join(list(platform_manager[state['platform']].categories.values()))
|
||||
))
|
||||
await add_sub.pause()
|
||||
@add_sub.got('platform', '{_prompt}', parse_platform)
|
||||
@add_sub.handle()
|
||||
async def init_id(bot: Bot, event: Event, state: T_State):
|
||||
state['_prompt'] = '请输入订阅用户的id，详情查阅https://github.com/felinae98/nonebot-hk-reporter'
|
||||
|
||||
@add_sub.handle()
|
||||
async def add_sub_parse_cat(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].categories:
|
||||
return
|
||||
if 'cats' in state:
|
||||
return
|
||||
res = []
|
||||
for cat in str(event.get_message()).strip().split():
|
||||
if cat not in platform_manager[state['platform']].reverse_category:
|
||||
await add_sub.reject('不支持 {}'.format(cat))
|
||||
res.append(platform_manager[state['platform']].reverse_category[cat])
|
||||
state['cats'] = res
|
||||
async def parse_id(bot: Bot, event: Event, state: T_State):
|
||||
target = str(event.get_message()).strip()
|
||||
name = await check_sub_target(state['platform'], target)
|
||||
if not name:
|
||||
await add_sub.reject('id输入错误')
|
||||
state['id'] = target
|
||||
state['name'] = name
|
||||
|
||||
@add_sub.handle()
|
||||
async def add_sub_handle_tag(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].enable_tag:
|
||||
return
|
||||
if 'tags' in state:
|
||||
return
|
||||
await bot.send(event=event, message='请输入要订阅的tag，订阅所有tag输入"全部标签"')
|
||||
await add_sub.pause()
|
||||
@add_sub.got('id', '{_prompt}', parse_id)
|
||||
@add_sub.handle()
|
||||
async def init_cat(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].categories:
|
||||
state['cats'] = []
|
||||
return
|
||||
state['_prompt'] = '请输入要订阅的类别，以空格分隔，支持的类别有：{}'.format(
|
||||
','.join(list(platform_manager[state['platform']].categories.values())))
|
||||
|
||||
@add_sub.handle()
|
||||
async def add_sub_parse_tag(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].enable_tag:
|
||||
return
|
||||
if 'tags' in state:
|
||||
return
|
||||
if str(event.get_message()).strip() == '全部标签':
|
||||
state['tags'] = []
|
||||
else:
|
||||
state['tags'] = str(event.get_message()).strip().split()
|
||||
async def parser_cats(bot: Bot, event: Event, state: T_State):
|
||||
res = []
|
||||
for cat in str(event.get_message()).strip().split():
|
||||
if cat not in platform_manager[state['platform']].reverse_category:
|
||||
await add_sub.reject('不支持 {}'.format(cat))
|
||||
res.append(platform_manager[state['platform']].reverse_category[cat])
|
||||
state['cats'] = res
|
||||
|
||||
@add_sub.handle()
|
||||
async def add_sub_process(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||
if not platform_manager[state['platform']].has_target:
|
||||
state['name'] = await platform_manager[state['platform']].get_account_name(Target(''))
|
||||
state['id'] = 'default'
|
||||
config = Config()
|
||||
config.add_subscribe(event.group_id, user_type='group',
|
||||
target=state['id'],
|
||||
target_name=state['name'], target_type=state['platform'],
|
||||
cats=state.get('cats', []), tags=state.get('tags', []))
|
||||
await add_sub.finish('添加 {} 成功'.format(state['name']))
|
||||
@add_sub.got('cats', '{_prompt}', parser_cats)
|
||||
@add_sub.handle()
|
||||
async def init_tag(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].enable_tag:
|
||||
state['tags'] = []
|
||||
return
|
||||
state['_prompt'] = '请输入要订阅的tag，订阅所有tag输入"全部标签"'
|
||||
|
||||
query_sub = on_command("查询订阅", rule=to_me(), priority=5)
|
||||
@query_sub.handle()
|
||||
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||
config: Config = Config()
|
||||
sub_list = config.list_subscribe(event.group_id, "group")
|
||||
res = '订阅的帐号为：\n'
|
||||
for sub in sub_list:
|
||||
res += '{} {} {}'.format(sub['target_type'], sub['target_name'], sub['target'])
|
||||
platform = platform_manager[sub['target_type']]
|
||||
if platform.categories:
|
||||
res += ' [{}]'.format(', '.join(map(lambda x: platform.categories[x], sub['cats'])))
|
||||
if platform.enable_tag:
|
||||
res += ' {}'.format(', '.join(sub['tags']))
|
||||
res += '\n'
|
||||
# send_msgs(bot, event.group_id, 'group', [await parse_text(res)])
|
||||
await query_sub.finish(Message(await parse_text(res)))
|
||||
async def parser_tags(bot: Bot, event: Event, state: T_State):
|
||||
if str(event.get_message()).strip() == '全部标签':
|
||||
state['tags'] = []
|
||||
else:
|
||||
state['tags'] = str(event.get_message()).strip().split()
|
||||
|
||||
del_sub = on_command("删除订阅", rule=to_me(), permission=GROUP_ADMIN | GROUP_OWNER | SUPERUSER, priority=5)
|
||||
@del_sub.handle()
|
||||
async def send_list(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||
config: Config = Config()
|
||||
sub_list = config.list_subscribe(event.group_id, "group")
|
||||
res = '订阅的帐号为：\n'
|
||||
state['sub_table'] = {}
|
||||
for index, sub in enumerate(sub_list, 1):
|
||||
state['sub_table'][index] = {'target_type': sub['target_type'], 'target': sub['target']}
|
||||
res += '{} {} {} {}\n'.format(index, sub['target_type'], sub['target_name'], sub['target'])
|
||||
platform = platform_manager[sub['target_type']]
|
||||
if platform.categories:
|
||||
res += ' [{}]'.format(', '.join(map(lambda x: platform.categories[x], sub['cats'])))
|
||||
if platform.enable_tag:
|
||||
res += ' {}'.format(', '.join(sub['tags']))
|
||||
res += '\n'
|
||||
res += '请输入要删除的订阅的序号'
|
||||
await bot.send(event=event, message=Message(await parse_text(res)))
|
||||
|
||||
@del_sub.receive()
|
||||
async def do_del(bot, event: GroupMessageEvent, state: T_State):
|
||||
try:
|
||||
index = int(str(event.get_message()).strip())
|
||||
@add_sub.got('tags', '{_prompt}', parser_tags)
|
||||
@add_sub.handle()
|
||||
async def add_sub_process(bot: Bot, event: Event, state: T_State):
|
||||
if not platform_manager[state['platform']].has_target:
|
||||
state['name'] = await platform_manager[state['platform']].get_account_name(Target(''))
|
||||
state['id'] = 'default'
|
||||
config = Config()
|
||||
config.del_subscribe(event.group_id, 'group', **state['sub_table'][index])
|
||||
except Exception as e:
|
||||
await del_sub.reject('删除错误')
|
||||
logger.warning(e)
|
||||
else:
|
||||
await del_sub.finish('删除成功')
|
||||
config.add_subscribe(state.get('_user_id') or event.group_id, user_type='group',
|
||||
target=state['id'],
|
||||
target_name=state['name'], target_type=state['platform'],
|
||||
cats=state.get('cats', []), tags=state.get('tags', []))
|
||||
await add_sub.finish('添加 {} 成功'.format(state['name']))
|
||||
|
||||
def do_query_sub(query_sub: Type[Matcher]):
|
||||
@query_sub.handle()
|
||||
async def _(bot: Bot, event: Event, state: T_State):
|
||||
config: Config = Config()
|
||||
sub_list = config.list_subscribe(state.get('_user_id') or event.group_id, "group")
|
||||
res = '订阅的帐号为：\n'
|
||||
for sub in sub_list:
|
||||
res += '{} {} {}'.format(sub['target_type'], sub['target_name'], sub['target'])
|
||||
platform = platform_manager[sub['target_type']]
|
||||
if platform.categories:
|
||||
res += ' [{}]'.format(', '.join(map(lambda x: platform.categories[x], sub['cats'])))
|
||||
if platform.enable_tag:
|
||||
res += ' {}'.format(', '.join(sub['tags']))
|
||||
res += '\n'
|
||||
# send_msgs(bot, event.group_id, 'group', [await parse_text(res)])
|
||||
await query_sub.finish(Message(await parse_text(res)))
|
||||
|
||||
def do_del_sub(del_sub: Type[Matcher]):
|
||||
@del_sub.handle()
|
||||
async def send_list(bot: Bot, event: Event, state: T_State):
|
||||
config: Config = Config()
|
||||
sub_list = config.list_subscribe(state.get('_user_id') or event.group_id, "group")
|
||||
res = '订阅的帐号为：\n'
|
||||
state['sub_table'] = {}
|
||||
for index, sub in enumerate(sub_list, 1):
|
||||
state['sub_table'][index] = {'target_type': sub['target_type'], 'target': sub['target']}
|
||||
res += '{} {} {} {}\n'.format(index, sub['target_type'], sub['target_name'], sub['target'])
|
||||
platform = platform_manager[sub['target_type']]
|
||||
if platform.categories:
|
||||
res += ' [{}]'.format(', '.join(map(lambda x: platform.categories[x], sub['cats'])))
|
||||
if platform.enable_tag:
|
||||
res += ' {}'.format(', '.join(sub['tags']))
|
||||
res += '\n'
|
||||
res += '请输入要删除的订阅的序号'
|
||||
await bot.send(event=event, message=Message(await parse_text(res)))
|
||||
|
||||
@del_sub.receive()
|
||||
async def do_del(bot, event: Event, state: T_State):
|
||||
try:
|
||||
index = int(str(event.get_message()).strip())
|
||||
config = Config()
|
||||
config.del_subscribe(state.get('_user_id') or event.group_id, 'group', **state['sub_table'][index])
|
||||
except Exception as e:
|
||||
await del_sub.reject('删除错误')
|
||||
logger.warning(e)
|
||||
else:
|
||||
await del_sub.finish('删除成功')
|
||||
|
||||
async def parse_group_number(bot: Bot, event: Event, state: T_State):
|
||||
state[state["_current_key"]] = int(str(event.get_message()))
|
||||
|
||||
|
||||
add_sub_matcher = on_command("添加订阅", rule=to_me(), permission=GROUP_ADMIN | GROUP_OWNER | SUPERUSER, priority=5)
|
||||
do_add_sub(add_sub_matcher)
|
||||
manage_add_sub_mather = on_command('管理-添加订阅', permission=SUPERUSER, priority=5)
|
||||
@manage_add_sub_mather.got('_user_id', "群号", parse_group_number)
|
||||
async def handle(bot: Bot, event: Event, state: T_State):
|
||||
pass
|
||||
do_add_sub(manage_add_sub_mather)
|
||||
|
||||
|
||||
query_sub_macher = on_command("查询订阅", rule=to_me(), priority=5)
|
||||
do_query_sub(query_sub_macher)
|
||||
manage_query_sub_mather = on_command('管理-查询订阅', permission=SUPERUSER, priority=5)
|
||||
@manage_query_sub_mather.got('_user_id', "群号", parse_group_number)
|
||||
async def handle(bot: Bot, event: Event, state: T_State):
|
||||
pass
|
||||
do_query_sub(manage_query_sub_mather)
|
||||
|
||||
|
||||
del_sub_macher = on_command("删除订阅", rule=to_me(), permission=GROUP_ADMIN | GROUP_OWNER | SUPERUSER, priority=5)
|
||||
do_del_sub(del_sub_macher)
|
||||
manage_del_sub_mather = on_command('管理-删除订阅', permission=SUPERUSER, priority=5)
|
||||
@manage_del_sub_mather.got('_user_id', "群号", parse_group_number)
|
||||
async def handle(bot: Bot, event: Event, state: T_State):
|
||||
pass
|
||||
do_del_sub(manage_del_sub_mather)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user