diff options
author | Kyomotoi <[email protected]> | 2021-04-11 15:03:35 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2021-04-11 15:03:35 +0800 |
commit | cbf4900c5c3b1de65e55ce480d4b453091528e84 (patch) | |
tree | 6bde5ad2d8f6f4010cb317b2b72be36f17b503d4 /ATRI/rule.py | |
parent | 212424da74fecbd77c9824e422ea8b028b330c9e (diff) | |
download | ATRI-cbf4900c5c3b1de65e55ce480d4b453091528e84.tar.gz ATRI-cbf4900c5c3b1de65e55ce480d4b453091528e84.tar.bz2 ATRI-cbf4900c5c3b1de65e55ce480d4b453091528e84.zip |
🚀✨⚡️🎨🐛 正式推送
Diffstat (limited to 'ATRI/rule.py')
-rw-r--r-- | ATRI/rule.py | 20 |
1 files changed, 3 insertions, 17 deletions
diff --git a/ATRI/rule.py b/ATRI/rule.py index 109d644..00a4c21 100644 --- a/ATRI/rule.py +++ b/ATRI/rule.py @@ -1,34 +1,20 @@ from nonebot.rule import Rule from nonebot.adapters.cqhttp import GroupMessageEvent, PokeNotifyEvent -from .config import config from .service import Service as sv 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, event.group_id) + return sv.auth_service(service, user, str(event.group_id)) else: - return sv.auth_service(service, None) + return sv.auth_service(service, user, None) return Rule(_is_in_service) -def is_block() -> Rule: - async def _is_in_banlist(bot, event, state) -> bool: - return sv.BlockSystem.auth_user(int(event.get_user_id())) - - return Rule(_is_in_banlist) - - -def is_in_dormant() -> Rule: - async def _is_in_dormant(bot, event, state) -> bool: - return sv.Dormant.is_dormant() - - return Rule(_is_in_dormant) - - def to_bot() -> Rule: async def _to_bot(bot, event, state) -> bool: return event.is_tome() |