summaryrefslogtreecommitdiff
path: root/ATRI/rule.py
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2021-07-08 22:09:00 +0800
committerKyomotoi <[email protected]>2021-07-08 22:09:00 +0800
commitbe2747e4d4b820ca0f1f988d3b77a628da26fe7b (patch)
treee1a59dd79ecd973a7d704568dcdc018f1f1b651a /ATRI/rule.py
parenta4e1b9d1581d756ef79ad063d1c0bd6b2fd13c1d (diff)
downloadATRI-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.py28
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