feat✨: test hook

This commit is contained in:
HibiKier
2024-02-27 01:14:49 +08:00
parent 7b3793728a
commit 2501a72bb8
6 changed files with 69 additions and 98 deletions
+4 -4
View File
@@ -82,7 +82,7 @@ class LevelUser(Model):
return False
@classmethod
async def check_level(cls, user_id: str, group_id: str, level: int) -> bool:
async def check_level(cls, user_id: str, group_id: str | None, level: int) -> bool:
"""检查用户权限等级是否大于 level
参数:
@@ -97,9 +97,9 @@ class LevelUser(Model):
if user := await cls.get_or_none(user_id=user_id, group_id=group_id):
return user.user_level >= level
else:
user_list = await cls.filter(user_id=user_id).all()
user = max(user_list, key=lambda x: x.user_level)
return user.user_level >= level
if user_list := await cls.filter(user_id=user_id).all():
user = max(user_list, key=lambda x: x.user_level)
return user.user_level >= level
return False
@classmethod