临时调整权限模块,备份
This commit is contained in:
98
base/auth/test_permission.py
Normal file
98
base/auth/test_permission.py
Normal file
@@ -0,0 +1,98 @@
|
||||
from .permission import PermissionManager, PermissionStatus
|
||||
from .group_permission import GroupPermissionManager
|
||||
|
||||
|
||||
def test_permission_system():
|
||||
# 获取权限管理器实例
|
||||
perm_manager = PermissionManager()
|
||||
group_manager = GroupPermissionManager()
|
||||
|
||||
# 测试注册权限
|
||||
print("测试注册权限...")
|
||||
robot_perm = perm_manager.register_permission(
|
||||
"ROBOT",
|
||||
"群机器人",
|
||||
"🔧 群机器人 [总开关]"
|
||||
)
|
||||
print(f"注册权限: {robot_perm.key} = {robot_perm.value} - {robot_perm.description}")
|
||||
|
||||
# 测试重复注册
|
||||
print("\n测试重复注册...")
|
||||
same_perm = perm_manager.register_permission(
|
||||
"ROBOT",
|
||||
"群机器人",
|
||||
"🔧 群机器人 [总开关]"
|
||||
)
|
||||
print(f"重复注册结果: {same_perm.key} = {same_perm.value} - {same_perm.description}")
|
||||
|
||||
# 测试注册新权限
|
||||
print("\n测试注册新权限...")
|
||||
news_perm = perm_manager.register_permission(
|
||||
"NEWS",
|
||||
"新闻播报",
|
||||
"📰 每日新闻播报",
|
||||
plugin_id="news_plugin"
|
||||
)
|
||||
print(f"新权限: {news_perm.key} = {news_perm.value} - {news_perm.description}")
|
||||
|
||||
# 测试获取权限
|
||||
print("\n测试获取权限...")
|
||||
retrieved_perm = perm_manager.get_permission("ROBOT")
|
||||
print(f"获取权限: {retrieved_perm.key} = {retrieved_perm.value} - {retrieved_perm.description}")
|
||||
|
||||
# 测试通过值获取权限
|
||||
print("\n测试通过值获取权限...")
|
||||
perm_by_value = perm_manager.get_permission_by_value(1)
|
||||
print(f"通过值获取: {perm_by_value.key} = {perm_by_value.value} - {perm_by_value.description}")
|
||||
|
||||
# 测试获取所有权限
|
||||
print("\n测试获取所有权限...")
|
||||
all_perms = perm_manager.get_all_permissions()
|
||||
print("所有权限:")
|
||||
for perm in all_perms:
|
||||
print(f" {perm.key} = {perm.value} - {perm.description}")
|
||||
|
||||
# 测试获取插件权限
|
||||
print("\n测试获取插件权限...")
|
||||
plugin_perms = perm_manager.get_plugin_permissions("news_plugin")
|
||||
print("插件权限:")
|
||||
for perm in plugin_perms:
|
||||
print(f" {perm.key} = {perm.value} - {perm.description}")
|
||||
|
||||
# 测试群组权限
|
||||
print("\n测试群组权限...")
|
||||
group_id = "test_group"
|
||||
|
||||
# 添加群组
|
||||
group_manager.add_group(group_id)
|
||||
print(f"添加群组: {group_id}")
|
||||
|
||||
# 设置群组权限
|
||||
group_manager.set_group_permission(group_id, robot_perm, PermissionStatus.ENABLED)
|
||||
print(f"设置权限: {group_id} - {robot_perm.key} = {PermissionStatus.ENABLED.value}")
|
||||
|
||||
# 获取群组权限
|
||||
status = group_manager.get_group_permission(group_id, robot_perm)
|
||||
print(f"获取权限状态: {group_id} - {robot_perm.key} = {status.value}")
|
||||
|
||||
# 检查权限
|
||||
result = group_manager.check_permission(group_id, robot_perm)
|
||||
print(f"检查权限: {group_id} - {robot_perm.key} = {result}")
|
||||
|
||||
# 列出群组权限
|
||||
print("\n列出群组权限...")
|
||||
perms = group_manager.list_group_permissions(group_id)
|
||||
print(f"群组 {group_id} 的权限:")
|
||||
for perm, status in perms.items():
|
||||
print(f" {perm.key} = {status.value}")
|
||||
|
||||
# 移除群组
|
||||
print("\n测试移除群组...")
|
||||
if group_manager.remove_group(group_id):
|
||||
print(f"成功移除群组: {group_id}")
|
||||
else:
|
||||
print(f"移除群组失败: {group_id}")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
test_permission_system()
|
||||
Reference in New Issue
Block a user