mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-14 00:20:06 +08:00
init🎉: 首次提交
This commit is contained in:
HibiKier
@@ -0,0 +1,111 @@
|
||||
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,
|
||||
)
|
||||
@@ -0,0 +1,70 @@
|
||||
from typing import Union
|
||||
|
||||
from tortoise import fields
|
||||
|
||||
from zhenxun.configs.config import Config
|
||||
from zhenxun.services.db_context import Model
|
||||
|
||||
|
||||
class FriendUser(Model):
|
||||
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
||||
"""自增id"""
|
||||
user_id = fields.CharField(255, unique=True, description="用户id")
|
||||
"""用户id"""
|
||||
user_name = fields.CharField(max_length=255, default="", description="用户名称")
|
||||
"""用户名称"""
|
||||
nickname = fields.CharField(max_length=255, null=True, description="用户自定义昵称")
|
||||
"""私聊下自定义昵称"""
|
||||
|
||||
class Meta:
|
||||
table = "friend_users"
|
||||
table_description = "好友信息数据表"
|
||||
|
||||
@classmethod
|
||||
async def get_user_name(cls, user_id: Union[int, str]) -> str:
|
||||
"""
|
||||
说明:
|
||||
获取好友用户名称
|
||||
参数:
|
||||
:param user_id: 用户id
|
||||
"""
|
||||
if user := await cls.get_or_none(user_id=str(user_id)):
|
||||
return user.user_name
|
||||
return ""
|
||||
|
||||
@classmethod
|
||||
async def get_user_nickname(cls, user_id: Union[int, str]) -> str:
|
||||
"""
|
||||
说明:
|
||||
获取用户昵称
|
||||
参数:
|
||||
:param user_id: 用户id
|
||||
"""
|
||||
if user := await cls.get_or_none(user_id=str(user_id)):
|
||||
if user.nickname:
|
||||
_tmp = ""
|
||||
if black_word := Config.get_config("nickname", "BLACK_WORD"):
|
||||
for x in user.nickname:
|
||||
_tmp += "*" if x in black_word else x
|
||||
return _tmp
|
||||
return ""
|
||||
|
||||
@classmethod
|
||||
async def set_user_nickname(cls, user_id: Union[int, str], nickname: str):
|
||||
"""
|
||||
说明:
|
||||
设置用户昵称
|
||||
参数:
|
||||
:param user_id: 用户id
|
||||
:param nickname: 昵称
|
||||
"""
|
||||
await cls.update_or_create(
|
||||
user_id=str(user_id), defaults={"nickname": nickname}
|
||||
)
|
||||
|
||||
@classmethod
|
||||
async def _run_script(cls):
|
||||
await cls.raw(
|
||||
"ALTER TABLE friend_users ALTER COLUMN user_id TYPE character varying(255);"
|
||||
)
|
||||
# 将user_id字段类型改为character varying(255))
|
||||
@@ -0,0 +1,31 @@
|
||||
from typing import List, Optional
|
||||
|
||||
from tortoise import fields
|
||||
|
||||
from zhenxun.services.db_context import Model
|
||||
from zhenxun.services.log import logger
|
||||
|
||||
|
||||
class GroupInfo(Model):
|
||||
group_id = fields.CharField(255, pk=True)
|
||||
"""群聊id"""
|
||||
group_name = fields.TextField(default="")
|
||||
"""群聊名称"""
|
||||
max_member_count = fields.IntField(default=0)
|
||||
"""最大人数"""
|
||||
member_count = fields.IntField(default=0)
|
||||
"""当前人数"""
|
||||
group_flag = fields.IntField(default=0)
|
||||
"""群认证标记"""
|
||||
|
||||
class Meta:
|
||||
table = "group_info"
|
||||
table_description = "群聊信息表"
|
||||
|
||||
@classmethod
|
||||
def _run_script(cls):
|
||||
return [
|
||||
"ALTER TABLE group_info ADD group_flag Integer NOT NULL DEFAULT 0;", # group_info表添加一个group_flag
|
||||
"ALTER TABLE group_info ALTER COLUMN group_id TYPE character varying(255);"
|
||||
# 将group_id字段类型改为character varying(255)
|
||||
]
|
||||
@@ -0,0 +1,30 @@
|
||||
from typing import List, Optional
|
||||
|
||||
from tortoise import fields
|
||||
|
||||
from zhenxun.services.db_context import Model
|
||||
|
||||
|
||||
class GroupInfo(Model):
|
||||
group_id = fields.CharField(255, pk=True, description="群组id")
|
||||
"""群聊id"""
|
||||
group_name = fields.TextField(default="", description="群组名称")
|
||||
"""群聊名称"""
|
||||
max_member_count = fields.IntField(default=0, description="最大人数")
|
||||
"""最大人数"""
|
||||
member_count = fields.IntField(default=0, description="当前人数")
|
||||
"""当前人数"""
|
||||
group_flag = fields.IntField(default=0, description="群认证标记")
|
||||
"""群认证标记"""
|
||||
|
||||
class Meta:
|
||||
table = "group_info"
|
||||
table_description = "群聊信息表"
|
||||
|
||||
@classmethod
|
||||
def _run_script(cls):
|
||||
return [
|
||||
"ALTER TABLE group_info ADD group_flag Integer NOT NULL DEFAULT 0;", # group_info表添加一个group_flag
|
||||
"ALTER TABLE group_info ALTER COLUMN group_id TYPE character varying(255);"
|
||||
# 将group_id字段类型改为character varying(255)
|
||||
]
|
||||
@@ -0,0 +1,127 @@
|
||||
from datetime import datetime
|
||||
from typing import List, Optional, Set, Union
|
||||
|
||||
from tortoise import fields
|
||||
|
||||
from zhenxun.configs.config import Config
|
||||
from zhenxun.services.db_context import Model
|
||||
from zhenxun.services.log import logger
|
||||
|
||||
|
||||
class GroupInfoUser(Model):
|
||||
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
||||
"""自增id"""
|
||||
user_id = fields.CharField(255)
|
||||
"""用户id"""
|
||||
user_name = fields.CharField(255, default="")
|
||||
"""用户昵称"""
|
||||
group_id = fields.CharField(255)
|
||||
"""群聊id"""
|
||||
user_join_time = fields.DatetimeField(null=True)
|
||||
"""用户入群时间"""
|
||||
nickname = fields.CharField(255, null=True)
|
||||
"""群聊昵称"""
|
||||
uid = fields.BigIntField(null=True)
|
||||
"""用户uid"""
|
||||
|
||||
class Meta:
|
||||
table = "group_info_users"
|
||||
table_description = "群员信息数据表"
|
||||
unique_together = ("user_id", "group_id")
|
||||
|
||||
@classmethod
|
||||
async def get_group_member_id_list(cls, group_id: Union[int, str]) -> Set[int]:
|
||||
"""
|
||||
说明:
|
||||
获取该群所有用户id
|
||||
参数:
|
||||
:param group_id: 群号
|
||||
"""
|
||||
return set(
|
||||
await cls.filter(group_id=str(group_id)).values_list("user_id", flat=True)
|
||||
) # type: ignore
|
||||
|
||||
@classmethod
|
||||
async def set_user_nickname(
|
||||
cls, user_id: Union[int, str], group_id: Union[int, str], nickname: str
|
||||
):
|
||||
"""
|
||||
说明:
|
||||
设置群员在该群内的昵称
|
||||
参数:
|
||||
:param user_id: 用户id
|
||||
:param group_id: 群号
|
||||
:param nickname: 昵称
|
||||
"""
|
||||
await cls.update_or_create(
|
||||
user_id=str(user_id),
|
||||
group_id=str(group_id),
|
||||
defaults={"nickname": nickname},
|
||||
)
|
||||
|
||||
@classmethod
|
||||
async def get_user_all_group(cls, user_id: Union[int, str]) -> List[int]:
|
||||
"""
|
||||
说明:
|
||||
获取该用户所在的所有群聊
|
||||
参数:
|
||||
:param user_id: 用户id
|
||||
"""
|
||||
return list(
|
||||
await cls.filter(user_id=str(user_id)).values_list("group_id", flat=True)
|
||||
) # type: ignore
|
||||
|
||||
@classmethod
|
||||
async def get_user_nickname(
|
||||
cls, user_id: Union[int, str], group_id: Union[int, str]
|
||||
) -> str:
|
||||
"""
|
||||
说明:
|
||||
获取用户在该群的昵称
|
||||
参数:
|
||||
:param user_id: 用户id
|
||||
:param group_id: 群号
|
||||
"""
|
||||
if user := await cls.get_or_none(user_id=str(user_id), group_id=str(group_id)):
|
||||
if user.nickname:
|
||||
nickname = ""
|
||||
if black_word := Config.get_config("nickname", "BLACK_WORD"):
|
||||
for x in user.nickname:
|
||||
nickname += "*" if x in black_word else x
|
||||
return nickname
|
||||
return user.nickname
|
||||
return ""
|
||||
|
||||
@classmethod
|
||||
async def get_group_member_uid(
|
||||
cls, user_id: Union[int, str], group_id: Union[int, str]
|
||||
) -> Optional[int]:
|
||||
logger.debug(
|
||||
f"GroupInfoUser 尝试获取 用户[<u><e>{user_id}</e></u>] 群聊[<u><e>{group_id}</e></u>] UID"
|
||||
)
|
||||
user, _ = await cls.get_or_create(user_id=str(user_id), group_id=str(group_id))
|
||||
_max_uid_user, _ = await cls.get_or_create(user_id="114514", group_id="114514")
|
||||
_max_uid = _max_uid_user.uid
|
||||
if not user.uid:
|
||||
all_user = await cls.filter(user_id=str(user_id)).all()
|
||||
for x in all_user:
|
||||
if x.uid:
|
||||
return x.uid
|
||||
user.uid = _max_uid + 1
|
||||
_max_uid_user.uid = _max_uid + 1
|
||||
await cls.bulk_update([user, _max_uid_user], ["uid"])
|
||||
logger.debug(
|
||||
f"GroupInfoUser 获取 用户[<u><e>{user_id}</e></u>] 群聊[<u><e>{group_id}</e></u>] UID: {user.uid}"
|
||||
)
|
||||
return user.uid
|
||||
|
||||
@classmethod
|
||||
async def _run_script(cls):
|
||||
return [
|
||||
"alter table group_info_users alter user_join_time drop not null;", # 允许 user_join_time 为空
|
||||
"ALTER TABLE group_info_users ALTER COLUMN user_join_time TYPE timestamp with time zone USING user_join_time::timestamp with time zone;",
|
||||
"ALTER TABLE group_info_users RENAME COLUMN user_qq TO user_id;", # 将user_id改为user_id
|
||||
"ALTER TABLE group_info_users ALTER COLUMN user_id TYPE character varying(255);",
|
||||
# 将user_id字段类型改为character varying(255)
|
||||
"ALTER TABLE group_info_users ALTER COLUMN group_id TYPE character varying(255);",
|
||||
]
|
||||
@@ -0,0 +1,134 @@
|
||||
from datetime import datetime
|
||||
from typing import Union
|
||||
|
||||
from tortoise import fields
|
||||
|
||||
from zhenxun.services.db_context import Model
|
||||
|
||||
|
||||
class LevelUser(Model):
|
||||
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
||||
"""自增id"""
|
||||
user_id = fields.CharField(255)
|
||||
"""用户id"""
|
||||
group_id = fields.CharField(255)
|
||||
"""群聊id"""
|
||||
user_level = fields.BigIntField()
|
||||
"""用户权限等级"""
|
||||
group_flag = fields.IntField(default=0)
|
||||
"""特殊标记，是否随群管理员变更而设置权限"""
|
||||
|
||||
class Meta:
|
||||
table = "level_users"
|
||||
table_description = "用户权限数据库"
|
||||
unique_together = ("user_id", "group_id")
|
||||
|
||||
@classmethod
|
||||
async def get_user_level(cls, user_id: int | str, group_id: int | str) -> int:
|
||||
"""
|
||||
获取用户在群内的等级
|
||||
|
||||
参数:
|
||||
user_id: 用户id
|
||||
group_id: 群组id
|
||||
|
||||
返回:
|
||||
int: 权限等级
|
||||
"""
|
||||
if user := await cls.get_or_none(user_id=str(user_id), group_id=str(group_id)):
|
||||
return user.user_level
|
||||
return 0
|
||||
|
||||
@classmethod
|
||||
async def set_level(
|
||||
cls,
|
||||
user_id: int | str,
|
||||
group_id: int | str,
|
||||
level: int,
|
||||
group_flag: int = 0,
|
||||
):
|
||||
"""
|
||||
设置用户在群内的权限
|
||||
|
||||
参数:
|
||||
user_id: 用户id
|
||||
group_id: 群组id
|
||||
level: 权限等级
|
||||
group_flag: 是否被自动更新刷新权限 0:是, 1:否.
|
||||
"""
|
||||
await cls.update_or_create(
|
||||
user_id=str(user_id),
|
||||
group_id=str(group_id),
|
||||
defaults={
|
||||
"user_level": level,
|
||||
"group_flag": group_flag,
|
||||
},
|
||||
)
|
||||
|
||||
@classmethod
|
||||
async def delete_level(cls, user_id: int | str, group_id: int | str) -> bool:
|
||||
"""
|
||||
删除用户权限
|
||||
|
||||
参数:
|
||||
user_id: 用户id
|
||||
group_id: 群组id
|
||||
|
||||
返回:
|
||||
bool: 是否含有用户权限
|
||||
"""
|
||||
if user := await cls.get_or_none(user_id=str(user_id), group_id=str(group_id)):
|
||||
await user.delete()
|
||||
return True
|
||||
return False
|
||||
|
||||
@classmethod
|
||||
async def check_level(
|
||||
cls, user_id: int | str, group_id: int | str, level: int
|
||||
) -> bool:
|
||||
"""
|
||||
检查用户权限等级是否大于 level
|
||||
|
||||
参数:
|
||||
user_id: 用户id
|
||||
group_id: 群组id
|
||||
level: 权限等级
|
||||
|
||||
返回:
|
||||
bool: 是否大于level
|
||||
"""
|
||||
if group_id:
|
||||
if user := await cls.get_or_none(
|
||||
user_id=str(user_id), group_id=str(group_id)
|
||||
):
|
||||
return user.user_level >= level
|
||||
else:
|
||||
user_list = await cls.filter(user_id=str(user_id)).all()
|
||||
user = max(user_list, key=lambda x: x.user_level)
|
||||
return user.user_level >= level
|
||||
return False
|
||||
|
||||
@classmethod
|
||||
async def is_group_flag(cls, user_id: int | str, group_id: int | str) -> bool:
|
||||
"""
|
||||
检测是否会被自动更新刷新权限
|
||||
|
||||
参数:
|
||||
user_id: 用户id
|
||||
group_id: 群组id
|
||||
|
||||
返回:
|
||||
bool: 是否会被自动更新权限刷新
|
||||
"""
|
||||
if user := await cls.get_or_none(user_id=str(user_id), group_id=str(group_id)):
|
||||
return user.group_flag == 1
|
||||
return False
|
||||
|
||||
@classmethod
|
||||
async def _run_script(cls):
|
||||
return [
|
||||
"ALTER TABLE level_users RENAME COLUMN user_qq TO user_id;", # 将user_id改为user_id
|
||||
"ALTER TABLE level_users ALTER COLUMN user_id TYPE character varying(255);",
|
||||
# 将user_id字段类型改为character varying(255)
|
||||
"ALTER TABLE level_users ALTER COLUMN group_id TYPE character varying(255);",
|
||||
]
|
||||
@@ -0,0 +1,51 @@
|
||||
from tortoise import fields
|
||||
|
||||
from zhenxun.services.db_context import Model
|
||||
from zhenxun.utils.enum import BlockType, PluginType
|
||||
|
||||
from .plugin_limit import PluginLimit
|
||||
|
||||
|
||||
class PluginInfo(Model):
|
||||
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
||||
"""自增id"""
|
||||
module = fields.CharField(255, description="模块名")
|
||||
"""模块名"""
|
||||
module_path = fields.CharField(255, description="模块路径", unique=True)
|
||||
"""模块路径"""
|
||||
name = fields.CharField(255, description="插件名称")
|
||||
"""插件名称"""
|
||||
status = fields.BooleanField(default=True, description="全局开关状态")
|
||||
"""全局开关状态"""
|
||||
block_type = fields.CharEnumField(
|
||||
BlockType, default=None, null=True, description="禁用类型"
|
||||
)
|
||||
"""禁用类型"""
|
||||
load_status = fields.BooleanField(default=True, description="加载状态")
|
||||
"""加载状态"""
|
||||
author = fields.CharField(255, null=True, description="作者")
|
||||
"""作者"""
|
||||
version = fields.CharField(max_length=255, null=True, description="版本")
|
||||
"""版本"""
|
||||
level = fields.IntField(default=5, description="所需群权限")
|
||||
"""所需群权限"""
|
||||
default_status = fields.BooleanField(default=True, description="进群默认开关状态")
|
||||
"""进群默认开关状态"""
|
||||
limit_superuser = fields.BooleanField(default=False, description="是否限制超级用户")
|
||||
"""是否限制超级用户"""
|
||||
menu_type = fields.CharField(max_length=255, default="功能", description="菜单类型")
|
||||
"""菜单类型"""
|
||||
plugin_type = fields.CharEnumField(PluginType, null=True, description="插件类型")
|
||||
"""插件类型"""
|
||||
cost_gold = fields.IntField(default=0, description="调用插件所需金币")
|
||||
"""调用插件所需金币"""
|
||||
plugin_limit = fields.ReverseRelation["PluginLimit"]
|
||||
"""插件限制"""
|
||||
admin_level = fields.IntField(default=0, null=True, description="调用所需权限等级")
|
||||
"""调用所需权限等级"""
|
||||
is_delete = fields.BooleanField(default=False, description="是否删除")
|
||||
"""是否删除"""
|
||||
|
||||
class Meta:
|
||||
table = "plugin_info"
|
||||
table_description = "插件基本信息"
|
||||
@@ -0,0 +1,44 @@
|
||||
from tortoise import fields
|
||||
|
||||
from zhenxun.services.db_context import Model
|
||||
from zhenxun.utils.enum import (
|
||||
BlockType,
|
||||
LimitCheckType,
|
||||
LimitWatchType,
|
||||
PluginLimitType,
|
||||
)
|
||||
|
||||
|
||||
class PluginLimit(Model):
|
||||
id = fields.IntField(pk=True, generated=True, auto_increment=True)
|
||||
"""自增id"""
|
||||
module = fields.CharField(255, description="模块名")
|
||||
"""模块名"""
|
||||
module_path = fields.CharField(255, description="模块路径")
|
||||
"""模块路径"""
|
||||
plugin = fields.ForeignKeyField(
|
||||
"models.PluginInfo",
|
||||
related_name="plugin_limit",
|
||||
on_delete=fields.CASCADE,
|
||||
description="所属插件",
|
||||
)
|
||||
limit_type = fields.CharEnumField(PluginLimitType, description="限制类型")
|
||||
"""限制类型"""
|
||||
watch_type = fields.CharEnumField(LimitWatchType, description="监听类型")
|
||||
"""限制类型"""
|
||||
status = fields.BooleanField(default=True, description="限制的开关状态")
|
||||
"""限制的开关状态"""
|
||||
check_type = fields.CharEnumField(
|
||||
LimitCheckType, default=BlockType.ALL, description="检查类型"
|
||||
)
|
||||
"""检查类型"""
|
||||
result = fields.CharField(max_length=255, null=True, description="返回信息")
|
||||
"""返回信息"""
|
||||
cd = fields.IntField(null=True, description="cd")
|
||||
"""cd"""
|
||||
max_count = fields.IntField(null=True, description="最大调用次数")
|
||||
"""最大调用次数"""
|
||||
|
||||
class Meta:
|
||||
table = "plugin_limit"
|
||||
table_description = "插件限制"
|
||||
Reference in New Issue
Block a user