mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
135 lines
3.8 KiB
Python
135 lines
3.8 KiB
Python
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);",
|
|
]
|