mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-05 11:26:43 +08:00
commit
05bd064a27
1
.gitignore
vendored
1
.gitignore
vendored
@ -274,6 +274,7 @@ dist
|
|||||||
|
|
||||||
# vuepress build output
|
# vuepress build output
|
||||||
.vuepress/dist
|
.vuepress/dist
|
||||||
|
docs/.vuepress/.temp/
|
||||||
|
|
||||||
# Serverless directories
|
# Serverless directories
|
||||||
.serverless/
|
.serverless/
|
||||||
|
@ -64,6 +64,8 @@ yarn && yarn build
|
|||||||
本项目使用了 Python 3.9 的语法,请将 Python 版本升级到 3.9 及以上,推荐使用 docker 部署
|
本项目使用了 Python 3.9 的语法,请将 Python 版本升级到 3.9 及以上,推荐使用 docker 部署
|
||||||
2. bot 不理我
|
2. bot 不理我
|
||||||
请确认自己是群主或者管理员,并且检查`COMMAND_START`环境变量是否设为`[""]`
|
请确认自己是群主或者管理员,并且检查`COMMAND_START`环境变量是否设为`[""]`
|
||||||
|
或者按照`COMMAND_START`中的设置添加命令前缀,例:
|
||||||
|
`COMMAND_START=["/"]`则应发送`/添加订阅`
|
||||||
3. 微博漏订阅了
|
3. 微博漏订阅了
|
||||||
微博更新了新的风控措施,某些含有某些关键词的微博会获取不到。
|
微博更新了新的风控措施,某些含有某些关键词的微博会获取不到。
|
||||||
|
|
||||||
|
@ -2,7 +2,7 @@ module.exports = {
|
|||||||
title: 'Nonebot Bison',
|
title: 'Nonebot Bison',
|
||||||
description: 'Docs for Nonebot Bison',
|
description: 'Docs for Nonebot Bison',
|
||||||
themeConfig: {
|
themeConfig: {
|
||||||
nav: [
|
navbar: [
|
||||||
{ text: '主页', link: '/' },
|
{ text: '主页', link: '/' },
|
||||||
{ text: '部署与使用', link: '/usage/' },
|
{ text: '部署与使用', link: '/usage/' },
|
||||||
{ text: '开发', link: '/dev/' },
|
{ text: '开发', link: '/dev/' },
|
||||||
|
BIN
docs/.vuepress/public/images/forward-msg-simple1.png
Normal file
BIN
docs/.vuepress/public/images/forward-msg-simple1.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 36 KiB |
BIN
docs/.vuepress/public/images/forward-msg-simple2.png
Normal file
BIN
docs/.vuepress/public/images/forward-msg-simple2.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 12 KiB |
@ -2,8 +2,9 @@
|
|||||||
home: true
|
home: true
|
||||||
heroText: Nonebot Bison
|
heroText: Nonebot Bison
|
||||||
tagline: 本bot励志做全泰拉骑车最快的信使
|
tagline: 本bot励志做全泰拉骑车最快的信使
|
||||||
actionText: 快速部署
|
actions:
|
||||||
actionLink: /usage/
|
- text: 快速部署
|
||||||
|
link: /usage/
|
||||||
features:
|
features:
|
||||||
- title: 拓展性强
|
- title: 拓展性强
|
||||||
details: 没有自己想要的网站?只要简单的爬虫知识就可以给它适配一个新的网站
|
details: 没有自己想要的网站?只要简单的爬虫知识就可以给它适配一个新的网站
|
||||||
|
@ -115,13 +115,32 @@ sidebar: auto
|
|||||||
开启,默认关
|
开启,默认关
|
||||||
- `BISON_USE_QUEUE`: 是否用队列的方式发送消息,降低发送频率,默认开
|
- `BISON_USE_QUEUE`: 是否用队列的方式发送消息,降低发送频率,默认开
|
||||||
- `BISON_RESEND_TIMES`: 最大重发次数,默认 0
|
- `BISON_RESEND_TIMES`: 最大重发次数,默认 0
|
||||||
|
- `BISON_USE_PIC_MERGE`: 是否启用多图片时合并转发(仅限群)
|
||||||
|
|
||||||
|
- `0`: 不启用(默认)
|
||||||
|
- `1`: 首条消息单独发送,剩余图片合并转发
|
||||||
|
- `2`: 所有消息全部合并转发
|
||||||
|
|
||||||
|
::: details 配置项示例
|
||||||
|
|
||||||
|
- 当`BISON_USE_PIC_MERGE=1`时:
|
||||||
|

|
||||||
|
- 当`BISON_USE_PIC_MERGE=2`时:
|
||||||
|

|
||||||
|
|
||||||
|
:::
|
||||||
|
::: warning
|
||||||
|
启用此功能时,可能会因为待推送图片过大/过多而导致文字消息与合并转发图片消息推送间隔过大(选择模式`1`时),请谨慎考虑开启。或者选择模式`2`,使图文消息一同合并转发(可能会使消息推送延迟过长)
|
||||||
|
:::
|
||||||
|
|
||||||
## 使用
|
## 使用
|
||||||
|
|
||||||
::: warning
|
::: warning
|
||||||
本节假设`COMMAND_START`设置中包含`''`,如果出现 bot 不响应的问题,请先
|
本节假设`COMMAND_START`设置中包含`''`
|
||||||
排查这个设置
|
|
||||||
:::
|
- 如果出现 bot 不响应的问题,请先排查这个设置
|
||||||
|
- 尝试在命令前添加设置的命令前缀,如`COMMAND_START=['/']`,则尝试使用`/添加订阅`
|
||||||
|
:::
|
||||||
|
|
||||||
### 命令
|
### 命令
|
||||||
|
|
||||||
|
@ -110,9 +110,7 @@ def do_add_sub(add_sub: Type[Matcher]):
|
|||||||
)
|
)
|
||||||
async def init_id(state: T_State):
|
async def init_id(state: T_State):
|
||||||
if platform_manager[state["platform"]].has_target:
|
if platform_manager[state["platform"]].has_target:
|
||||||
state[
|
state["_prompt"] = "请输入订阅用户的id:\n查询id获取方法请回复:“查询”"
|
||||||
"_prompt"
|
|
||||||
] = "请输入订阅用户的id,详情查阅https://nonebot-bison.vercel.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84uid"
|
|
||||||
else:
|
else:
|
||||||
state["id"] = "default"
|
state["id"] = "default"
|
||||||
state["name"] = await platform_manager[state["platform"]].get_target_name(
|
state["name"] = await platform_manager[state["platform"]].get_target_name(
|
||||||
@ -124,6 +122,8 @@ def do_add_sub(add_sub: Type[Matcher]):
|
|||||||
return
|
return
|
||||||
target = str(event.get_message()).strip()
|
target = str(event.get_message()).strip()
|
||||||
try:
|
try:
|
||||||
|
if target == "查询":
|
||||||
|
raise LookupError
|
||||||
if target == "取消":
|
if target == "取消":
|
||||||
raise KeyboardInterrupt
|
raise KeyboardInterrupt
|
||||||
name = await check_sub_target(state["platform"], target)
|
name = await check_sub_target(state["platform"], target)
|
||||||
@ -131,10 +131,23 @@ def do_add_sub(add_sub: Type[Matcher]):
|
|||||||
raise ValueError
|
raise ValueError
|
||||||
state["id"] = target
|
state["id"] = target
|
||||||
state["name"] = name
|
state["name"] = name
|
||||||
|
except (LookupError):
|
||||||
|
url = "https://nonebot-bison.vercel.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84-uid"
|
||||||
|
title = "Bison所支持的平台UID"
|
||||||
|
content = "查询相关平台的uid格式或获取方式"
|
||||||
|
image = "https://s3.bmp.ovh/imgs/2022/03/ab3cc45d83bd3dd3.jpg"
|
||||||
|
getId_share = f"[CQ:share,url={url},title={title},content={content},image={image}]" # 缩短字符串格式长度,以及方便后续修改为消息段格式
|
||||||
|
await add_sub.reject(Message(getId_share))
|
||||||
except (KeyboardInterrupt):
|
except (KeyboardInterrupt):
|
||||||
await add_sub.finish("已中止订阅")
|
await add_sub.finish("已中止订阅")
|
||||||
except (ValueError):
|
except (ValueError):
|
||||||
await add_sub.reject("id输入错误")
|
await add_sub.reject("id输入错误")
|
||||||
|
else:
|
||||||
|
await add_sub.send(
|
||||||
|
"即将订阅的用户为:{} {} {}\n如有错误请输入“取消”重新订阅".format(
|
||||||
|
state["platform"], state["name"], state["id"]
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
@add_sub.got("id", _gen_prompt_template("{_prompt}"), [Depends(parse_id)])
|
@add_sub.got("id", _gen_prompt_template("{_prompt}"), [Depends(parse_id)])
|
||||||
async def init_cat(state: T_State):
|
async def init_cat(state: T_State):
|
||||||
@ -304,11 +317,16 @@ del_sub_matcher = on_command(
|
|||||||
del_sub_matcher.handle()(set_target_user_info)
|
del_sub_matcher.handle()(set_target_user_info)
|
||||||
do_del_sub(del_sub_matcher)
|
do_del_sub(del_sub_matcher)
|
||||||
|
|
||||||
group_manage_matcher = on_command("群管理")
|
group_manage_matcher = on_command("群管理", rule=to_me(), permission=SUPERUSER, priority=4)
|
||||||
|
|
||||||
|
|
||||||
@group_manage_matcher.handle()
|
@group_manage_matcher.handle()
|
||||||
async def send_group_list(bot: Bot, state: T_State):
|
async def send_group_list(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||||
|
await group_manage_matcher.finish(Message("该功能只支持私聊使用,请私聊Bot"))
|
||||||
|
|
||||||
|
|
||||||
|
@group_manage_matcher.handle()
|
||||||
|
async def send_group_list(bot: Bot, event: PrivateMessageEvent, state: T_State):
|
||||||
groups = await bot.call_api("get_group_list")
|
groups = await bot.call_api("get_group_list")
|
||||||
res_text = "请选择需要管理的群:\n"
|
res_text = "请选择需要管理的群:\n"
|
||||||
group_number_idx = {}
|
group_number_idx = {}
|
||||||
|
@ -4,7 +4,7 @@ from httpx import Response
|
|||||||
from nonebug.app import App
|
from nonebug.app import App
|
||||||
|
|
||||||
from .platforms.utils import get_json
|
from .platforms.utils import get_json
|
||||||
from .utils import fake_admin_user, fake_group_message_event
|
from .utils import BotReply, fake_admin_user, fake_group_message_event
|
||||||
|
|
||||||
|
|
||||||
# 选择platform阶段中止
|
# 选择platform阶段中止
|
||||||
@ -43,18 +43,7 @@ async def test_abort_add_on_platform(app: App):
|
|||||||
ctx.should_pass_rule()
|
ctx.should_pass_rule()
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_1,
|
event_1,
|
||||||
Message(
|
Message(BotReply.add_reply_on_platform(platform_manager, common_platform)),
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
|
||||||
+ "".join(
|
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_abort = fake_group_message_event(
|
event_abort = fake_group_message_event(
|
||||||
@ -63,7 +52,7 @@ async def test_abort_add_on_platform(app: App):
|
|||||||
ctx.receive_event(bot, event_abort)
|
ctx.receive_event(bot, event_abort)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_abort,
|
event_abort,
|
||||||
"已中止订阅",
|
BotReply.add_reply_abort,
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
ctx.should_finished()
|
ctx.should_finished()
|
||||||
@ -105,18 +94,7 @@ async def test_abort_add_on_id(app: App):
|
|||||||
ctx.should_pass_rule()
|
ctx.should_pass_rule()
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_1,
|
event_1,
|
||||||
Message(
|
Message(BotReply.add_reply_on_platform(platform_manager, common_platform)),
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
|
||||||
+ "".join(
|
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_2 = fake_group_message_event(
|
event_2 = fake_group_message_event(
|
||||||
@ -125,9 +103,7 @@ async def test_abort_add_on_id(app: App):
|
|||||||
ctx.receive_event(bot, event_2)
|
ctx.receive_event(bot, event_2)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_2,
|
event_2,
|
||||||
Message(
|
Message(BotReply.add_reply_on_id),
|
||||||
"请输入订阅用户的id,详情查阅https://nonebot-bison.vercel.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84uid"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_abort = fake_group_message_event(
|
event_abort = fake_group_message_event(
|
||||||
@ -136,7 +112,7 @@ async def test_abort_add_on_id(app: App):
|
|||||||
ctx.receive_event(bot, event_abort)
|
ctx.receive_event(bot, event_abort)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_abort,
|
event_abort,
|
||||||
"已中止订阅",
|
BotReply.add_reply_abort,
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
ctx.should_finished()
|
ctx.should_finished()
|
||||||
@ -179,16 +155,9 @@ async def test_abort_add_on_cats(app: App):
|
|||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_1,
|
event_1,
|
||||||
Message(
|
Message(
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
BotReply.add_reply_on_platform(
|
||||||
+ "".join(
|
platform_manager=platform_manager, common_platform=common_platform
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
)
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
),
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
@ -198,9 +167,7 @@ async def test_abort_add_on_cats(app: App):
|
|||||||
ctx.receive_event(bot, event_2)
|
ctx.receive_event(bot, event_2)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_2,
|
event_2,
|
||||||
Message(
|
Message(BotReply.add_reply_on_id),
|
||||||
"请输入订阅用户的id,详情查阅https://nonebot-bison.vercel.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84uid"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_3 = fake_group_message_event(
|
event_3 = fake_group_message_event(
|
||||||
@ -209,20 +176,23 @@ async def test_abort_add_on_cats(app: App):
|
|||||||
ctx.receive_event(bot, event_3)
|
ctx.receive_event(bot, event_3)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_3,
|
event_3,
|
||||||
Message(
|
BotReply.add_reply_on_target_confirm(
|
||||||
"请输入要订阅的类别,以空格分隔,支持的类别有:{}".format(
|
"weibo", "明日方舟Arknights", "6279793937"
|
||||||
" ".join(list(platform_manager["weibo"].categories.values()))
|
|
||||||
)
|
|
||||||
),
|
),
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
|
ctx.should_call_send(
|
||||||
|
event_3,
|
||||||
|
Message(BotReply.add_reply_on_cats(platform_manager, "weibo")),
|
||||||
|
True,
|
||||||
|
)
|
||||||
event_abort = fake_group_message_event(
|
event_abort = fake_group_message_event(
|
||||||
message=Message("取消"), sender=Sender(card="", nickname="test", role="admin")
|
message=Message("取消"), sender=Sender(card="", nickname="test", role="admin")
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_abort)
|
ctx.receive_event(bot, event_abort)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_abort,
|
event_abort,
|
||||||
"已中止订阅",
|
BotReply.add_reply_abort,
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
ctx.should_finished()
|
ctx.should_finished()
|
||||||
@ -265,16 +235,9 @@ async def test_abort_add_on_tag(app: App):
|
|||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_1,
|
event_1,
|
||||||
Message(
|
Message(
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
BotReply.add_reply_on_platform(
|
||||||
+ "".join(
|
platform_manager=platform_manager, common_platform=common_platform
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
)
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
),
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
@ -284,9 +247,7 @@ async def test_abort_add_on_tag(app: App):
|
|||||||
ctx.receive_event(bot, event_2)
|
ctx.receive_event(bot, event_2)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_2,
|
event_2,
|
||||||
Message(
|
Message(BotReply.add_reply_on_id),
|
||||||
"请输入订阅用户的id,详情查阅https://nonebot-bison.vercel.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84uid"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_3 = fake_group_message_event(
|
event_3 = fake_group_message_event(
|
||||||
@ -295,25 +256,28 @@ async def test_abort_add_on_tag(app: App):
|
|||||||
ctx.receive_event(bot, event_3)
|
ctx.receive_event(bot, event_3)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_3,
|
event_3,
|
||||||
Message(
|
BotReply.add_reply_on_target_confirm(
|
||||||
"请输入要订阅的类别,以空格分隔,支持的类别有:{}".format(
|
"weibo", "明日方舟Arknights", "6279793937"
|
||||||
" ".join(list(platform_manager["weibo"].categories.values()))
|
|
||||||
)
|
|
||||||
),
|
),
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
|
ctx.should_call_send(
|
||||||
|
event_3,
|
||||||
|
Message(BotReply.add_reply_on_cats(platform_manager, "weibo")),
|
||||||
|
True,
|
||||||
|
)
|
||||||
event_4 = fake_group_message_event(
|
event_4 = fake_group_message_event(
|
||||||
message=Message("图文 文字"), sender=fake_admin_user
|
message=Message("图文 文字"), sender=fake_admin_user
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_4)
|
ctx.receive_event(bot, event_4)
|
||||||
ctx.should_call_send(event_4, Message('请输入要订阅的tag,订阅所有tag输入"全部标签"'), True)
|
ctx.should_call_send(event_4, Message(BotReply.add_reply_on_tags), True)
|
||||||
event_abort = fake_group_message_event(
|
event_abort = fake_group_message_event(
|
||||||
message=Message("取消"), sender=Sender(card="", nickname="test", role="admin")
|
message=Message("取消"), sender=Sender(card="", nickname="test", role="admin")
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_abort)
|
ctx.receive_event(bot, event_abort)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_abort,
|
event_abort,
|
||||||
"已中止订阅",
|
BotReply.add_reply_abort,
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
ctx.should_finished()
|
ctx.should_finished()
|
||||||
|
@ -4,7 +4,7 @@ from httpx import Response
|
|||||||
from nonebug.app import App
|
from nonebug.app import App
|
||||||
|
|
||||||
from .platforms.utils import get_json
|
from .platforms.utils import get_json
|
||||||
from .utils import fake_admin_user, fake_group_message_event
|
from .utils import BotReply, fake_admin_user, fake_group_message_event
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
@ -49,18 +49,7 @@ async def test_configurable_at_me_false(app: App):
|
|||||||
ctx.receive_event(bot, event)
|
ctx.receive_event(bot, event)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event,
|
event,
|
||||||
Message(
|
Message(BotReply.add_reply_on_platform(platform_manager, common_platform)),
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
|
||||||
+ "".join(
|
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
ctx.should_pass_rule()
|
ctx.should_pass_rule()
|
||||||
@ -104,16 +93,9 @@ async def test_add_with_target(app: App):
|
|||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_1,
|
event_1,
|
||||||
Message(
|
Message(
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
BotReply.add_reply_on_platform(
|
||||||
+ "".join(
|
platform_manager=platform_manager, common_platform=common_platform
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
)
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
),
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
@ -124,15 +106,7 @@ async def test_add_with_target(app: App):
|
|||||||
ctx.should_rejected()
|
ctx.should_rejected()
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_2,
|
event_2,
|
||||||
(
|
BotReply.add_reply_on_platform_input_allplatform(platform_manager),
|
||||||
"全部平台\n"
|
|
||||||
+ "\n".join(
|
|
||||||
[
|
|
||||||
"{}:{}".format(platform_name, platform.name)
|
|
||||||
for platform_name, platform in platform_manager.items()
|
|
||||||
]
|
|
||||||
)
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_3 = fake_group_message_event(
|
event_3 = fake_group_message_event(
|
||||||
@ -141,16 +115,14 @@ async def test_add_with_target(app: App):
|
|||||||
ctx.receive_event(bot, event_3)
|
ctx.receive_event(bot, event_3)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_3,
|
event_3,
|
||||||
Message(
|
Message(BotReply.add_reply_on_id),
|
||||||
"请输入订阅用户的id,详情查阅https://nonebot-bison.vercel.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84uid"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_4_err = fake_group_message_event(
|
event_4_err = fake_group_message_event(
|
||||||
message=Message("000"), sender=fake_admin_user
|
message=Message("000"), sender=fake_admin_user
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_4_err)
|
ctx.receive_event(bot, event_4_err)
|
||||||
ctx.should_call_send(event_4_err, "id输入错误", True)
|
ctx.should_call_send(event_4_err, BotReply.add_reply_on_id_input_error, True)
|
||||||
ctx.should_rejected()
|
ctx.should_rejected()
|
||||||
event_4_ok = fake_group_message_event(
|
event_4_ok = fake_group_message_event(
|
||||||
message=Message("6279793937"), sender=fake_admin_user
|
message=Message("6279793937"), sender=fake_admin_user
|
||||||
@ -158,29 +130,36 @@ async def test_add_with_target(app: App):
|
|||||||
ctx.receive_event(bot, event_4_ok)
|
ctx.receive_event(bot, event_4_ok)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_4_ok,
|
event_4_ok,
|
||||||
Message(
|
BotReply.add_reply_on_target_confirm(
|
||||||
"请输入要订阅的类别,以空格分隔,支持的类别有:{}".format(
|
"weibo", "明日方舟Arknights", "6279793937"
|
||||||
" ".join(list(platform_manager["weibo"].categories.values()))
|
|
||||||
)
|
|
||||||
),
|
),
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
|
ctx.should_call_send(
|
||||||
|
event_4_ok,
|
||||||
|
Message(BotReply.add_reply_on_cats(platform_manager, "weibo")),
|
||||||
|
True,
|
||||||
|
)
|
||||||
event_5_err = fake_group_message_event(
|
event_5_err = fake_group_message_event(
|
||||||
message=Message("图文 文字 err"), sender=fake_admin_user
|
message=Message("图文 文字 err"), sender=fake_admin_user
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_5_err)
|
ctx.receive_event(bot, event_5_err)
|
||||||
ctx.should_call_send(event_5_err, "不支持 err", True)
|
ctx.should_call_send(
|
||||||
|
event_5_err, BotReply.add_reply_on_cats_input_error("err"), True
|
||||||
|
)
|
||||||
ctx.should_rejected()
|
ctx.should_rejected()
|
||||||
event_5_ok = fake_group_message_event(
|
event_5_ok = fake_group_message_event(
|
||||||
message=Message("图文 文字"), sender=fake_admin_user
|
message=Message("图文 文字"), sender=fake_admin_user
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_5_ok)
|
ctx.receive_event(bot, event_5_ok)
|
||||||
ctx.should_call_send(event_5_ok, Message('请输入要订阅的tag,订阅所有tag输入"全部标签"'), True)
|
ctx.should_call_send(event_5_ok, Message(BotReply.add_reply_on_tags), True)
|
||||||
event_6 = fake_group_message_event(
|
event_6 = fake_group_message_event(
|
||||||
message=Message("全部标签"), sender=fake_admin_user
|
message=Message("全部标签"), sender=fake_admin_user
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_6)
|
ctx.receive_event(bot, event_6)
|
||||||
ctx.should_call_send(event_6, ("添加 明日方舟Arknights 成功"), True)
|
ctx.should_call_send(
|
||||||
|
event_6, BotReply.add_reply_subscribe_success("明日方舟Arknights"), True
|
||||||
|
)
|
||||||
ctx.should_finished()
|
ctx.should_finished()
|
||||||
subs = config.list_subscribe(10000, "group")
|
subs = config.list_subscribe(10000, "group")
|
||||||
assert len(subs) == 1
|
assert len(subs) == 1
|
||||||
@ -220,18 +199,7 @@ async def test_add_with_target_no_cat(app: App):
|
|||||||
ctx.should_pass_rule()
|
ctx.should_pass_rule()
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_1,
|
event_1,
|
||||||
Message(
|
Message(BotReply.add_reply_on_platform(platform_manager, common_platform)),
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
|
||||||
+ "".join(
|
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_3 = fake_group_message_event(
|
event_3 = fake_group_message_event(
|
||||||
@ -240,16 +208,21 @@ async def test_add_with_target_no_cat(app: App):
|
|||||||
ctx.receive_event(bot, event_3)
|
ctx.receive_event(bot, event_3)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_3,
|
event_3,
|
||||||
Message(
|
Message(BotReply.add_reply_on_id),
|
||||||
"请输入订阅用户的id,详情查阅https://nonebot-bison.vercel.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84uid"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_4_ok = fake_group_message_event(
|
event_4_ok = fake_group_message_event(
|
||||||
message=Message("32540734"), sender=fake_admin_user
|
message=Message("32540734"), sender=fake_admin_user
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_4_ok)
|
ctx.receive_event(bot, event_4_ok)
|
||||||
ctx.should_call_send(event_4_ok, ("添加 塞壬唱片-MSR 成功"), True)
|
ctx.should_call_send(
|
||||||
|
event_4_ok,
|
||||||
|
BotReply.add_reply_on_target_confirm("ncm-artist", "塞壬唱片-MSR", "32540734"),
|
||||||
|
True,
|
||||||
|
)
|
||||||
|
ctx.should_call_send(
|
||||||
|
event_4_ok, BotReply.add_reply_subscribe_success("塞壬唱片-MSR"), True
|
||||||
|
)
|
||||||
ctx.should_finished()
|
ctx.should_finished()
|
||||||
subs = config.list_subscribe(10000, "group")
|
subs = config.list_subscribe(10000, "group")
|
||||||
assert len(subs) == 1
|
assert len(subs) == 1
|
||||||
@ -284,18 +257,7 @@ async def test_add_no_target(app: App):
|
|||||||
ctx.should_pass_rule()
|
ctx.should_pass_rule()
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_1,
|
event_1,
|
||||||
Message(
|
Message(BotReply.add_reply_on_platform(platform_manager, common_platform)),
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
|
||||||
+ "".join(
|
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_3 = fake_group_message_event(
|
event_3 = fake_group_message_event(
|
||||||
@ -304,18 +266,16 @@ async def test_add_no_target(app: App):
|
|||||||
ctx.receive_event(bot, event_3)
|
ctx.receive_event(bot, event_3)
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_3,
|
event_3,
|
||||||
Message(
|
Message(BotReply.add_reply_on_cats(platform_manager, "arknights")),
|
||||||
"请输入要订阅的类别,以空格分隔,支持的类别有:{}".format(
|
|
||||||
" ".join(list(platform_manager["arknights"].categories.values()))
|
|
||||||
)
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_4 = fake_group_message_event(
|
event_4 = fake_group_message_event(
|
||||||
message=Message("游戏公告"), sender=fake_admin_user
|
message=Message("游戏公告"), sender=fake_admin_user
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_4)
|
ctx.receive_event(bot, event_4)
|
||||||
ctx.should_call_send(event_4, ("添加 明日方舟游戏信息 成功"), True)
|
ctx.should_call_send(
|
||||||
|
event_4, BotReply.add_reply_subscribe_success("明日方舟游戏信息"), True
|
||||||
|
)
|
||||||
ctx.should_finished()
|
ctx.should_finished()
|
||||||
subs = config.list_subscribe(10000, "group")
|
subs = config.list_subscribe(10000, "group")
|
||||||
assert len(subs) == 1
|
assert len(subs) == 1
|
||||||
@ -348,18 +308,7 @@ async def test_platform_name_err(app: App):
|
|||||||
ctx.should_pass_rule()
|
ctx.should_pass_rule()
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_1,
|
event_1,
|
||||||
Message(
|
Message(BotReply.add_reply_on_platform(platform_manager, common_platform)),
|
||||||
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
|
||||||
+ "".join(
|
|
||||||
[
|
|
||||||
"{}:{}\n".format(
|
|
||||||
platform_name, platform_manager[platform_name].name
|
|
||||||
)
|
|
||||||
for platform_name in common_platform
|
|
||||||
]
|
|
||||||
)
|
|
||||||
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
|
||||||
),
|
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_2 = fake_group_message_event(
|
event_2 = fake_group_message_event(
|
||||||
@ -370,100 +319,87 @@ async def test_platform_name_err(app: App):
|
|||||||
ctx.should_rejected()
|
ctx.should_rejected()
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event_2,
|
event_2,
|
||||||
"平台输入错误",
|
BotReply.add_reply_on_platform_input_error,
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_query_sub(app: App):
|
@respx.mock
|
||||||
from nonebot.adapters.onebot.v11.message import Message
|
async def test_add_with_get_id(app: App):
|
||||||
|
from nonebot.adapters.onebot.v11.event import Sender
|
||||||
|
from nonebot.adapters.onebot.v11.message import Message, MessageSegment
|
||||||
from nonebot_bison.config import Config
|
from nonebot_bison.config import Config
|
||||||
from nonebot_bison.config_manager import query_sub_matcher
|
from nonebot_bison.config_manager import add_sub_matcher, common_platform
|
||||||
from nonebot_bison.platform import platform_manager
|
from nonebot_bison.platform import platform_manager
|
||||||
|
|
||||||
config = Config()
|
config = Config()
|
||||||
config.user_target.truncate()
|
config.user_target.truncate()
|
||||||
config.add_subscribe(
|
|
||||||
10000,
|
ak_list_router = respx.get(
|
||||||
"group",
|
"https://m.weibo.cn/api/container/getIndex?containerid=1005056279793937"
|
||||||
"6279793937",
|
|
||||||
"明日方舟Arknights",
|
|
||||||
"weibo",
|
|
||||||
[platform_manager["weibo"].reverse_category["图文"]],
|
|
||||||
["明日方舟"],
|
|
||||||
)
|
)
|
||||||
async with app.test_matcher(query_sub_matcher) as ctx:
|
ak_list_router.mock(
|
||||||
|
return_value=Response(200, json=get_json("weibo_ak_profile.json"))
|
||||||
|
)
|
||||||
|
ak_list_bad_router = respx.get(
|
||||||
|
"https://m.weibo.cn/api/container/getIndex?containerid=100505000"
|
||||||
|
)
|
||||||
|
ak_list_bad_router.mock(
|
||||||
|
return_value=Response(200, json=get_json("weibo_err_profile.json"))
|
||||||
|
)
|
||||||
|
|
||||||
|
async with app.test_matcher(add_sub_matcher) as ctx:
|
||||||
bot = ctx.create_bot()
|
bot = ctx.create_bot()
|
||||||
event = fake_group_message_event(message=Message("查询订阅"), to_me=True)
|
event_1 = fake_group_message_event(
|
||||||
ctx.receive_event(bot, event)
|
message=Message("添加订阅"),
|
||||||
ctx.should_pass_rule()
|
sender=Sender(card="", nickname="test", role="admin"),
|
||||||
ctx.should_pass_permission()
|
to_me=True,
|
||||||
ctx.should_call_send(
|
|
||||||
event, Message("订阅的帐号为:\nweibo 明日方舟Arknights 6279793937 [图文] 明日方舟\n"), True
|
|
||||||
)
|
)
|
||||||
|
ctx.receive_event(bot, event_1)
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
|
||||||
async def test_del_sub(app: App):
|
|
||||||
from nonebot.adapters.onebot.v11.bot import Bot
|
|
||||||
from nonebot.adapters.onebot.v11.message import Message
|
|
||||||
from nonebot_bison.config import Config
|
|
||||||
from nonebot_bison.config_manager import del_sub_matcher
|
|
||||||
from nonebot_bison.platform import platform_manager
|
|
||||||
|
|
||||||
config = Config()
|
|
||||||
config.user_target.truncate()
|
|
||||||
config.add_subscribe(
|
|
||||||
10000,
|
|
||||||
"group",
|
|
||||||
"6279793937",
|
|
||||||
"明日方舟Arknights",
|
|
||||||
"weibo",
|
|
||||||
[platform_manager["weibo"].reverse_category["图文"]],
|
|
||||||
["明日方舟"],
|
|
||||||
)
|
|
||||||
async with app.test_matcher(del_sub_matcher) as ctx:
|
|
||||||
bot = ctx.create_bot(base=Bot)
|
|
||||||
assert isinstance(bot, Bot)
|
|
||||||
event = fake_group_message_event(
|
|
||||||
message=Message("删除订阅"), to_me=True, sender=fake_admin_user
|
|
||||||
)
|
|
||||||
ctx.receive_event(bot, event)
|
|
||||||
ctx.should_pass_rule()
|
ctx.should_pass_rule()
|
||||||
ctx.should_pass_permission()
|
|
||||||
ctx.should_call_send(
|
ctx.should_call_send(
|
||||||
event,
|
event_1,
|
||||||
Message(
|
Message(
|
||||||
"订阅的帐号为:\n1 weibo 明日方舟Arknights 6279793937\n [图文] 明日方舟\n请输入要删除的订阅的序号"
|
BotReply.add_reply_on_platform(
|
||||||
|
platform_manager=platform_manager, common_platform=common_platform
|
||||||
|
)
|
||||||
),
|
),
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
event_1_err = fake_group_message_event(
|
event_3 = fake_group_message_event(
|
||||||
message=Message("2"), sender=fake_admin_user
|
message=Message("weibo"), sender=fake_admin_user
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_1_err)
|
ctx.receive_event(bot, event_3)
|
||||||
ctx.should_call_send(event_1_err, "删除错误", True)
|
ctx.should_call_send(
|
||||||
|
event_3,
|
||||||
|
Message(BotReply.add_reply_on_id),
|
||||||
|
True,
|
||||||
|
)
|
||||||
|
event_4_query = fake_group_message_event(
|
||||||
|
message=Message("查询"), sender=fake_admin_user
|
||||||
|
)
|
||||||
|
ctx.receive_event(bot, event_4_query)
|
||||||
ctx.should_rejected()
|
ctx.should_rejected()
|
||||||
event_1_ok = fake_group_message_event(
|
ctx.should_call_send(
|
||||||
message=Message("1"), sender=fake_admin_user
|
event_4_query,
|
||||||
|
[MessageSegment(*BotReply.add_reply_on_id_input_search())],
|
||||||
|
True,
|
||||||
|
)
|
||||||
|
"""
|
||||||
|
line 362:
|
||||||
|
鬼知道为什么要在这里这样写,
|
||||||
|
没有[]的话assert不了(should_call_send使用[MessageSegment(...)]的格式进行比较)
|
||||||
|
不在这里MessageSegment()的话也assert不了(指不能让add_reply_on_id_input_search直接返回一个MessageSegment对象)
|
||||||
|
amen
|
||||||
|
"""
|
||||||
|
event_abort = fake_group_message_event(
|
||||||
|
message=Message("取消"), sender=Sender(card="", nickname="test", role="admin")
|
||||||
|
)
|
||||||
|
ctx.receive_event(bot, event_abort)
|
||||||
|
ctx.should_call_send(
|
||||||
|
event_abort,
|
||||||
|
BotReply.add_reply_abort,
|
||||||
|
True,
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_1_ok)
|
|
||||||
ctx.should_call_send(event_1_ok, "删除成功", True)
|
|
||||||
ctx.should_finished()
|
ctx.should_finished()
|
||||||
subs = config.list_subscribe(10000, "group")
|
|
||||||
assert len(subs) == 0
|
|
||||||
|
|
||||||
|
|
||||||
async def test_test(app: App):
|
|
||||||
from nonebot.adapters.onebot.v11.bot import Bot
|
|
||||||
from nonebot.adapters.onebot.v11.message import Message
|
|
||||||
from nonebot_bison.config_manager import test_matcher
|
|
||||||
|
|
||||||
async with app.test_matcher(test_matcher) as ctx:
|
|
||||||
bot = ctx.create_bot(base=Bot)
|
|
||||||
event = fake_group_message_event(message=Message("testtt"))
|
|
||||||
ctx.receive_event(bot, event)
|
|
||||||
ctx.should_pass_permission()
|
|
||||||
ctx.should_pass_rule()
|
|
||||||
ctx.should_call_send(event, "666", True)
|
|
@ -1,9 +1,11 @@
|
|||||||
|
import pytest
|
||||||
from nonebug import App
|
from nonebug import App
|
||||||
|
|
||||||
from .utils import fake_admin_user, fake_private_message_event, fake_superuser
|
from .utils import fake_group_message_event, fake_private_message_event, fake_superuser
|
||||||
|
|
||||||
|
|
||||||
async def test_query(app: App):
|
@pytest.mark.asyncio
|
||||||
|
async def test_query_with_superuser_private(app: App):
|
||||||
from nonebot.adapters.onebot.v11.bot import Bot
|
from nonebot.adapters.onebot.v11.bot import Bot
|
||||||
from nonebot.adapters.onebot.v11.message import Message
|
from nonebot.adapters.onebot.v11.message import Message
|
||||||
from nonebot_bison.config_manager import group_manage_matcher
|
from nonebot_bison.config_manager import group_manage_matcher
|
||||||
@ -11,7 +13,10 @@ async def test_query(app: App):
|
|||||||
async with app.test_matcher(group_manage_matcher) as ctx:
|
async with app.test_matcher(group_manage_matcher) as ctx:
|
||||||
bot = ctx.create_bot(base=Bot)
|
bot = ctx.create_bot(base=Bot)
|
||||||
event = fake_private_message_event(
|
event = fake_private_message_event(
|
||||||
message=Message("群管理"), sender=fake_superuser
|
message=Message("群管理"),
|
||||||
|
sender=fake_superuser,
|
||||||
|
to_me=True,
|
||||||
|
user_id=fake_superuser.user_id,
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event)
|
ctx.receive_event(bot, event)
|
||||||
ctx.should_pass_rule()
|
ctx.should_pass_rule()
|
||||||
@ -23,23 +28,57 @@ async def test_query(app: App):
|
|||||||
event, Message("请选择需要管理的群:\n1. 101 - test group\n请输入左侧序号"), True
|
event, Message("请选择需要管理的群:\n1. 101 - test group\n请输入左侧序号"), True
|
||||||
)
|
)
|
||||||
event_1_err = fake_private_message_event(
|
event_1_err = fake_private_message_event(
|
||||||
message=Message("0"), sender=fake_superuser
|
message=Message("0"),
|
||||||
|
sender=fake_superuser,
|
||||||
|
to_me=True,
|
||||||
|
user_id=fake_superuser.user_id,
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_1_err)
|
ctx.receive_event(bot, event_1_err)
|
||||||
ctx.should_rejected()
|
|
||||||
ctx.should_call_send(event_1_err, "请输入正确序号", True)
|
ctx.should_call_send(event_1_err, "请输入正确序号", True)
|
||||||
|
ctx.should_rejected()
|
||||||
event_1_ok = fake_private_message_event(
|
event_1_ok = fake_private_message_event(
|
||||||
message=Message("1"), sender=fake_superuser
|
message=Message("1"),
|
||||||
|
sender=fake_superuser,
|
||||||
|
to_me=True,
|
||||||
|
user_id=fake_superuser.user_id,
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_1_ok)
|
ctx.receive_event(bot, event_1_ok)
|
||||||
ctx.should_call_send(event_1_ok, "请输入需要使用的命令:添加订阅,查询订阅,删除订阅", True)
|
ctx.should_call_send(event_1_ok, "请输入需要使用的命令:添加订阅,查询订阅,删除订阅", True)
|
||||||
event_2_err = fake_private_message_event(
|
event_2_err = fake_private_message_event(
|
||||||
message=Message("222"), sender=fake_superuser
|
message=Message("222"),
|
||||||
|
sender=fake_superuser,
|
||||||
|
to_me=True,
|
||||||
|
user_id=fake_superuser.user_id,
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_2_err)
|
ctx.receive_event(bot, event_2_err)
|
||||||
ctx.should_rejected()
|
|
||||||
ctx.should_call_send(event_2_err, "请输入正确的命令", True)
|
ctx.should_call_send(event_2_err, "请输入正确的命令", True)
|
||||||
|
ctx.should_rejected()
|
||||||
event_2_ok = fake_private_message_event(
|
event_2_ok = fake_private_message_event(
|
||||||
message=Message("查询订阅"), sender=fake_superuser
|
message=Message("查询订阅"),
|
||||||
|
sender=fake_superuser,
|
||||||
|
to_me=True,
|
||||||
|
user_id=fake_superuser.user_id,
|
||||||
)
|
)
|
||||||
ctx.receive_event(bot, event_2_ok)
|
ctx.receive_event(bot, event_2_ok)
|
||||||
|
ctx.should_pass_rule()
|
||||||
|
ctx.should_pass_permission()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_query_with_superuser_group_tome(app: App):
|
||||||
|
from nonebot.adapters.onebot.v11.bot import Bot
|
||||||
|
from nonebot.adapters.onebot.v11.message import Message
|
||||||
|
from nonebot_bison.config_manager import group_manage_matcher
|
||||||
|
|
||||||
|
async with app.test_matcher(group_manage_matcher) as ctx:
|
||||||
|
bot = ctx.create_bot(base=Bot)
|
||||||
|
event = fake_group_message_event(
|
||||||
|
message=Message("群管理"),
|
||||||
|
sender=fake_superuser,
|
||||||
|
to_me=True,
|
||||||
|
user_id=fake_superuser.user_id,
|
||||||
|
)
|
||||||
|
ctx.receive_event(bot, event)
|
||||||
|
ctx.should_pass_rule()
|
||||||
|
ctx.should_pass_permission()
|
||||||
|
ctx.should_call_send(event, Message("该功能只支持私聊使用,请私聊Bot"), True)
|
||||||
|
101
tests/test_config_manager_query_del.py
Normal file
101
tests/test_config_manager_query_del.py
Normal file
@ -0,0 +1,101 @@
|
|||||||
|
import pytest
|
||||||
|
import respx
|
||||||
|
from httpx import Response
|
||||||
|
from nonebug.app import App
|
||||||
|
|
||||||
|
from .platforms.utils import get_json
|
||||||
|
from .utils import fake_admin_user, fake_group_message_event
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_query_sub(app: App):
|
||||||
|
from nonebot.adapters.onebot.v11.message import Message
|
||||||
|
from nonebot_bison.config import Config
|
||||||
|
from nonebot_bison.config_manager import query_sub_matcher
|
||||||
|
from nonebot_bison.platform import platform_manager
|
||||||
|
|
||||||
|
config = Config()
|
||||||
|
config.user_target.truncate()
|
||||||
|
config.add_subscribe(
|
||||||
|
10000,
|
||||||
|
"group",
|
||||||
|
"6279793937",
|
||||||
|
"明日方舟Arknights",
|
||||||
|
"weibo",
|
||||||
|
[platform_manager["weibo"].reverse_category["图文"]],
|
||||||
|
["明日方舟"],
|
||||||
|
)
|
||||||
|
async with app.test_matcher(query_sub_matcher) as ctx:
|
||||||
|
bot = ctx.create_bot()
|
||||||
|
event = fake_group_message_event(message=Message("查询订阅"), to_me=True)
|
||||||
|
ctx.receive_event(bot, event)
|
||||||
|
ctx.should_pass_rule()
|
||||||
|
ctx.should_pass_permission()
|
||||||
|
ctx.should_call_send(
|
||||||
|
event, Message("订阅的帐号为:\nweibo 明日方舟Arknights 6279793937 [图文] 明日方舟\n"), True
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_del_sub(app: App):
|
||||||
|
from nonebot.adapters.onebot.v11.bot import Bot
|
||||||
|
from nonebot.adapters.onebot.v11.message import Message
|
||||||
|
from nonebot_bison.config import Config
|
||||||
|
from nonebot_bison.config_manager import del_sub_matcher
|
||||||
|
from nonebot_bison.platform import platform_manager
|
||||||
|
|
||||||
|
config = Config()
|
||||||
|
config.user_target.truncate()
|
||||||
|
config.add_subscribe(
|
||||||
|
10000,
|
||||||
|
"group",
|
||||||
|
"6279793937",
|
||||||
|
"明日方舟Arknights",
|
||||||
|
"weibo",
|
||||||
|
[platform_manager["weibo"].reverse_category["图文"]],
|
||||||
|
["明日方舟"],
|
||||||
|
)
|
||||||
|
async with app.test_matcher(del_sub_matcher) as ctx:
|
||||||
|
bot = ctx.create_bot(base=Bot)
|
||||||
|
assert isinstance(bot, Bot)
|
||||||
|
event = fake_group_message_event(
|
||||||
|
message=Message("删除订阅"), to_me=True, sender=fake_admin_user
|
||||||
|
)
|
||||||
|
ctx.receive_event(bot, event)
|
||||||
|
ctx.should_pass_rule()
|
||||||
|
ctx.should_pass_permission()
|
||||||
|
ctx.should_call_send(
|
||||||
|
event,
|
||||||
|
Message(
|
||||||
|
"订阅的帐号为:\n1 weibo 明日方舟Arknights 6279793937\n [图文] 明日方舟\n请输入要删除的订阅的序号"
|
||||||
|
),
|
||||||
|
True,
|
||||||
|
)
|
||||||
|
event_1_err = fake_group_message_event(
|
||||||
|
message=Message("2"), sender=fake_admin_user
|
||||||
|
)
|
||||||
|
ctx.receive_event(bot, event_1_err)
|
||||||
|
ctx.should_call_send(event_1_err, "删除错误", True)
|
||||||
|
ctx.should_rejected()
|
||||||
|
event_1_ok = fake_group_message_event(
|
||||||
|
message=Message("1"), sender=fake_admin_user
|
||||||
|
)
|
||||||
|
ctx.receive_event(bot, event_1_ok)
|
||||||
|
ctx.should_call_send(event_1_ok, "删除成功", True)
|
||||||
|
ctx.should_finished()
|
||||||
|
subs = config.list_subscribe(10000, "group")
|
||||||
|
assert len(subs) == 0
|
||||||
|
|
||||||
|
|
||||||
|
async def test_test(app: App):
|
||||||
|
from nonebot.adapters.onebot.v11.bot import Bot
|
||||||
|
from nonebot.adapters.onebot.v11.message import Message
|
||||||
|
from nonebot_bison.config_manager import test_matcher
|
||||||
|
|
||||||
|
async with app.test_matcher(test_matcher) as ctx:
|
||||||
|
bot = ctx.create_bot(base=Bot)
|
||||||
|
event = fake_group_message_event(message=Message("testtt"))
|
||||||
|
ctx.receive_event(bot, event)
|
||||||
|
ctx.should_pass_permission()
|
||||||
|
ctx.should_pass_rule()
|
||||||
|
ctx.should_call_send(event, "666", True)
|
@ -69,3 +69,69 @@ from nonebot.adapters.onebot.v11.event import Sender
|
|||||||
|
|
||||||
fake_admin_user = Sender(nickname="test", role="admin")
|
fake_admin_user = Sender(nickname="test", role="admin")
|
||||||
fake_superuser = Sender(user_id=10001, nickname="superuser")
|
fake_superuser = Sender(user_id=10001, nickname="superuser")
|
||||||
|
|
||||||
|
|
||||||
|
class BotReply:
|
||||||
|
@staticmethod
|
||||||
|
def add_reply_on_platform(platform_manager, common_platform):
|
||||||
|
return (
|
||||||
|
"请输入想要订阅的平台,目前支持,请输入冒号左边的名称:\n"
|
||||||
|
+ "".join(
|
||||||
|
[
|
||||||
|
"{}:{}\n".format(
|
||||||
|
platform_name, platform_manager[platform_name].name
|
||||||
|
)
|
||||||
|
for platform_name in common_platform
|
||||||
|
]
|
||||||
|
)
|
||||||
|
+ "要查看全部平台请输入:“全部”\n中止订阅过程请输入:“取消”"
|
||||||
|
)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def add_reply_on_platform_input_allplatform(platform_manager):
|
||||||
|
return "全部平台\n" + "\n".join(
|
||||||
|
[
|
||||||
|
"{}:{}".format(platform_name, platform.name)
|
||||||
|
for platform_name, platform in platform_manager.items()
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def add_reply_on_id_input_search():
|
||||||
|
search_url = "https://nonebot-bison.vercel.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84-uid"
|
||||||
|
search_title = "Bison所支持的平台UID"
|
||||||
|
search_content = "查询相关平台的uid格式或获取方式"
|
||||||
|
search_image = "https://s3.bmp.ovh/imgs/2022/03/ab3cc45d83bd3dd3.jpg"
|
||||||
|
type = "share"
|
||||||
|
data = {
|
||||||
|
"url": search_url,
|
||||||
|
"title": search_title,
|
||||||
|
"content": search_content,
|
||||||
|
"image": search_image,
|
||||||
|
}
|
||||||
|
msg = [type, data]
|
||||||
|
return msg
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def add_reply_on_target_confirm(platform, name, id):
|
||||||
|
return f"即将订阅的用户为:{platform} {name} {id}\n如有错误请输入“取消”重新订阅"
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def add_reply_on_cats(platform_manager, platform: str):
|
||||||
|
return "请输入要订阅的类别,以空格分隔,支持的类别有:{}".format(
|
||||||
|
" ".join(list(platform_manager[platform].categories.values()))
|
||||||
|
)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def add_reply_on_cats_input_error(cat: str):
|
||||||
|
return "不支持 {}".format(cat)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def add_reply_subscribe_success(name):
|
||||||
|
return "添加 {} 成功".format(name)
|
||||||
|
|
||||||
|
add_reply_on_id_input_error = "id输入错误"
|
||||||
|
add_reply_on_platform_input_error = "平台输入错误"
|
||||||
|
add_reply_on_id = "请输入订阅用户的id:\n查询id获取方法请回复:“查询”"
|
||||||
|
add_reply_on_tags = '请输入要订阅的tag,订阅所有tag输入"全部标签"'
|
||||||
|
add_reply_abort = "已中止订阅"
|
||||||
|
Loading…
x
Reference in New Issue
Block a user