diff options
author | Kyomotoi <[email protected]> | 2021-07-08 22:09:00 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2021-07-08 22:09:00 +0800 |
commit | be2747e4d4b820ca0f1f988d3b77a628da26fe7b (patch) | |
tree | e1a59dd79ecd973a7d704568dcdc018f1f1b651a /ATRI/rule.py | |
parent | a4e1b9d1581d756ef79ad063d1c0bd6b2fd13c1d (diff) | |
download | ATRI-be2747e4d4b820ca0f1f988d3b77a628da26fe7b.tar.gz ATRI-be2747e4d4b820ca0f1f988d3b77a628da26fe7b.tar.bz2 ATRI-be2747e4d4b820ca0f1f988d3b77a628da26fe7b.zip |
🔖♻️🐛🔧🔥📝 更新版本:YHN-001-A03
🔖 更新版本至:YHN-001-A03
✨ 新增插件:
- 涩图
- 闲聊(文爱
♻️ 重构:
- Service
- 所有插件
🐛 修复部分小bug
🔧 暂时移除部分设置
🔥 删除:
- 插件:nsfw、wife。日后加回
- 插件 essential 中部分内容
📝 更新README
Diffstat (limited to 'ATRI/rule.py')
-rw-r--r-- | ATRI/rule.py | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/ATRI/rule.py b/ATRI/rule.py index 907a65d..3d75f84 100644 --- a/ATRI/rule.py +++ b/ATRI/rule.py @@ -1,17 +1,24 @@ +from nonebot.adapters.cqhttp.event import PrivateMessageEvent from nonebot.rule import Rule -from nonebot.adapters.cqhttp import GroupMessageEvent, PokeNotifyEvent +from nonebot.adapters.cqhttp import MessageEvent, GroupMessageEvent -from .service import Service as sv +from .service import ServiceTools def is_in_service(service: str) -> Rule: async def _is_in_service(bot, event, state) -> bool: - user = str(event.user_id) - if isinstance(event, GroupMessageEvent): - return sv.auth_service(service, user, str(event.group_id)) + if isinstance(event, PrivateMessageEvent): + user_id = event.get_user_id() + result = ServiceTools().auth_service(service, user_id) + return result + elif isinstance(event, GroupMessageEvent): + user_id = event.get_event_name() + group_id = str(event.group_id) + result = ServiceTools().auth_service(service, user_id, group_id) + return result else: - return sv.auth_service(service, user, None) - + return True + return Rule(_is_in_service) @@ -20,10 +27,3 @@ def to_bot() -> Rule: return event.is_tome() return Rule(_to_bot) - - -def poke(bot, event: PokeNotifyEvent, state): - if event.is_tome(): - return True - else: - return False |