nonebot-bison/src/plugins/auto_agree.py
2021-02-09 12:59:17 +08:00

19 lines
839 B
Python

from nonebot import on_request, logger
from nonebot.adapters.cqhttp import Bot, Event
from nonebot.permission import SUPERUSER
from nonebot.typing import T_State
from nonebot.adapters.cqhttp.permission import PRIVATE_FRIEND
from nonebot.adapters.cqhttp.event import FriendRequestEvent, GroupRequestEvent, RequestEvent
friend_req = on_request(priority=5)
@friend_req.handle()
async def add_superuser(bot: Bot, event: RequestEvent, state: T_State):
if str(event.user_id) in bot.config.superusers and event.request_type == 'private':
await event.approve(bot)
logger.info('add user {}'.format(event.user_id))
elif event.sub_type == 'invite' and str(event.user_id) in bot.config.superusers and event.request_type == 'group':
await event.approve(bot)
logger.info('add group {}'.format(event.group_id))