ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-16 19:11:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🔧 change command and add builtin plugin

Browse Source
This commit is contained in:
yanyongyu
2020-08-29 21:59:36 +08:00
parent def5caedbc
commit f5b655ef71
6 changed files with 79 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
nonebot/rule.py
View File

@ -74,13 +74,21 @@ class TrieRule:
suffix = cls.suffix.longest_prefix(
message_r.data["text"].rstrip()[::-1])
state["_prefix"] = {prefix.key: prefix.value} if prefix else {}
state["_suffix"] = {suffix.key: suffix.value} if suffix else {}
state["_prefix"] = {
"raw_command": prefix.key,
"command": prefix.value
} if prefix else {}
state["_suffix"] = {
"raw_command": suffix.key,
"command": suffix.value
} if suffix else {}
return ({
prefix.key: prefix.value
"raw_command": prefix.key,
"command": prefix.value
} if prefix else {}, {
suffix.key: suffix.value
"raw_command": suffix.key,
"command": suffix.value
} if suffix else {})
@ -122,7 +130,7 @@ def command(command: Tuple[str, ...]) -> Rule:
TrieRule.add_prefix(f"{start}{sep.join(command)}", command)
async def _command(bot: Bot, event: Event, state: dict) -> bool:
return command in state["_prefix"].values()
return command == state["_prefix"]["command"]
return Rule(_command)