mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
112 lines
3.5 KiB
Python
112 lines
3.5 KiB
Python
from nonebot.adapters import Bot
|
|
from tortoise import fields
|
|
|
|
from zhenxun.services.db_context import Model
|
|
from zhenxun.utils.enum import RequestHandleType, RequestType
|
|
from zhenxun.utils.exception import NotFoundError
|
|
|
|
|
|
class FgRequest(Model):
|
|
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
|
"""自增id"""
|
|
request_type = fields.CharEnumField(RequestType, default=None, description="请求类型")
|
|
"""请求类型"""
|
|
platform = fields.CharField(255, description="平台")
|
|
"""平台"""
|
|
bot_id = fields.CharField(255, description="Bot Id")
|
|
"""botId"""
|
|
flag = fields.CharField(max_length=255, default="", description="flag")
|
|
"""flag"""
|
|
user_id = fields.CharField(max_length=255, description="请求用户id")
|
|
"""请求用户id"""
|
|
group_id = fields.CharField(max_length=255, null=True, description="邀请入群id")
|
|
"""邀请入群id"""
|
|
nickname = fields.CharField(max_length=255, description="请求人名称")
|
|
"""对象名称"""
|
|
comment = fields.CharField(max_length=255, null=True, description="验证信息")
|
|
"""验证信息"""
|
|
handle_type = fields.CharEnumField(RequestHandleType, null=True, description="处理类型")
|
|
"""处理类型"""
|
|
|
|
class Meta:
|
|
table = "fg_request"
|
|
table_description = "好友群组请求"
|
|
|
|
@classmethod
|
|
async def approve(cls, bot: Bot, id: int, request_type: RequestType):
|
|
"""同意请求
|
|
|
|
参数:
|
|
bot: Bot
|
|
id: 请求id
|
|
request_type: 请求类型
|
|
|
|
异常:
|
|
NotFoundError: 未发现请求
|
|
"""
|
|
await cls._handle_request(bot, id, request_type, RequestHandleType.APPROVE)
|
|
|
|
@classmethod
|
|
async def refused(cls, bot: Bot, id: int, request_type: RequestType):
|
|
"""拒绝请求
|
|
|
|
参数:
|
|
bot: Bot
|
|
id: 请求id
|
|
request_type: 请求类型
|
|
|
|
异常:
|
|
NotFoundError: 未发现请求
|
|
"""
|
|
await cls._handle_request(bot, id, request_type, RequestHandleType.REFUSED)
|
|
|
|
@classmethod
|
|
async def ignore(cls, bot: Bot, id: int, request_type: RequestType):
|
|
"""忽略请求
|
|
|
|
参数:
|
|
bot: Bot
|
|
id: 请求id
|
|
request_type: 请求类型
|
|
|
|
异常:
|
|
NotFoundError: 未发现请求
|
|
"""
|
|
await cls._handle_request(bot, id, request_type, RequestHandleType.IGNORE)
|
|
|
|
@classmethod
|
|
async def _handle_request(
|
|
cls,
|
|
bot: Bot,
|
|
id: int,
|
|
request_type: RequestType,
|
|
handle_type: RequestHandleType,
|
|
):
|
|
"""处理请求
|
|
|
|
参数:
|
|
bot: Bot
|
|
id: 请求id
|
|
request_type: 请求类型
|
|
handle_type: 处理类型
|
|
|
|
异常:
|
|
NotFoundError: 未发现请求
|
|
"""
|
|
req = await cls.get_or_none(id=id, request_type=request_type)
|
|
if not req:
|
|
raise NotFoundError
|
|
req.handle_type = RequestHandleType
|
|
await req.save(update_fields=["handle_type"])
|
|
if handle_type != RequestHandleType.IGNORE:
|
|
if request_type == RequestType.FRIEND:
|
|
await bot.set_friend_add_request(
|
|
flag=req.flag, approve=handle_type == RequestHandleType.APPROVE
|
|
)
|
|
else:
|
|
await bot.set_group_add_request(
|
|
flag=req.flag,
|
|
sub_type="invite",
|
|
approve=handle_type == RequestHandleType.APPROVE,
|
|
)
|