summaryrefslogtreecommitdiff
path: root/ATRI/rule.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/rule.py')
-rw-r--r--ATRI/rule.py20
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()