mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
45 lines
2.0 KiB
Python
45 lines
2.0 KiB
Python
from nonebot import on_request
|
|
from nonebot.adapters.cqhttp import Bot, FriendRequestEvent, GroupRequestEvent
|
|
from models.friend_user import FriendUser
|
|
from datetime import datetime
|
|
from configs.config import AUTO_ADD_FRIEND
|
|
from nonebot.adapters.cqhttp.exception import ActionFailed
|
|
|
|
__plugin_name__ = '处理请求'
|
|
|
|
friend_req = on_request(priority=5)
|
|
|
|
|
|
@friend_req.handle()
|
|
async def _(bot: Bot, event: FriendRequestEvent, state: dict):
|
|
if AUTO_ADD_FRIEND:
|
|
nickname = ''
|
|
for user in await bot.get_friend_list():
|
|
if user['user_id'] == event.user_id:
|
|
nickname = user['nickname']
|
|
await FriendUser.add_friend_info(user['user_id'], user['nickname'])
|
|
break
|
|
await bot.send_private_msg(user_id=int(list(bot.config.superusers)[0]), message=f"{nickname}({event.user_id})"
|
|
f" 添加小真寻好友（已自动同意）")
|
|
await bot.set_friend_add_request(flag=event.flag, approve=True)
|
|
|
|
|
|
group_req = on_request(priority=5, block=True)
|
|
|
|
|
|
@group_req.handle()
|
|
async def _(bot: Bot, event: GroupRequestEvent, state: dict):
|
|
if event.sub_type == 'invite':
|
|
nickname = await FriendUser.get_user_name(event.user_id)
|
|
if str(event.user_id) in bot.config.superusers:
|
|
try:
|
|
await bot.set_group_add_request(flag=event.flag, sub_type='invite', approve=True)
|
|
except ActionFailed:
|
|
pass
|
|
else:
|
|
await bot.send_private_msg(user_id=int(list(bot.config.superusers)[0]),
|
|
message=f"报告..\n{nickname}({event.user_id})"
|
|
f" 希望邀请我加入 {event.group_id}\n邀请日期：{str(datetime.now()).split('.')[0]}")
|
|
await bot.send_private_msg(user_id=event.user_id,
|
|
message="想要邀请我偷偷入群嘛~~已经提醒管理员了\n等待管理员处理吧~")
|