import os
|
import sys
|
import django
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'backend.settings')
|
django.setup()
|
|
from django.contrib.auth.models import Group, Permission
|
from django.contrib.contenttypes.models import ContentType
|
from apps.lunarlink.models import Project, API, Case, Config, Variables, Report
|
|
|
def create_default_groups():
|
"""
|
创建默认用户组
|
"""
|
groups_config = [
|
{
|
'name': '管理员组',
|
'permissions': ['all'],
|
'description': '拥有所有权限的管理员组'
|
},
|
{
|
'name': '开发人员组',
|
'permissions': ['view', 'add', 'change'],
|
'description': '开发人员组，可以查看、添加和修改数据'
|
},
|
{
|
'name': '测试人员组',
|
'permissions': ['view', 'add'],
|
'description': '测试人员组，可以查看和添加数据'
|
},
|
{
|
'name': '只读用户组',
|
'permissions': ['view'],
|
'description': '只读用户组，只能查看数据'
|
}
|
]
|
|
created_count = 0
|
for group_config in groups_config:
|
try:
|
group = Group.objects.get(name=group_config['name'])
|
print(f"用户组 '{group_config['name']}' 已存在，跳过创建")
|
continue
|
|
except Group.DoesNotExist:
|
try:
|
group = Group.objects.create(name=group_config['name'])
|
print(f"✓ 成功创建用户组: {group_config['name']}")
|
|
if 'all' in group_config['permissions']:
|
all_permissions = Permission.objects.all()
|
group.permissions.set(all_permissions)
|
print(f" - 已添加所有权限")
|
else:
|
for perm_type in group_config['permissions']:
|
for model in [Project, API, Case, Config, Variables, Report]:
|
content_type = ContentType.objects.get_for_model(model)
|
codename = f"{perm_type}_{model._meta.model_name}"
|
try:
|
perm = Permission.objects.get(
|
content_type=content_type,
|
codename=codename
|
)
|
group.permissions.add(perm)
|
except Permission.DoesNotExist:
|
pass
|
|
print(f" - 已添加权限: {', '.join(group_config['permissions'])}")
|
|
created_count += 1
|
|
except Exception as e:
|
print(f"✗ 创建用户组 '{group_config['name']}' 失败: {e}")
|
|
return created_count
|
|
|
def list_all_groups():
|
"""
|
列出所有用户组
|
"""
|
print("\n" + "=" * 60)
|
print("当前系统用户组列表:")
|
print("=" * 60)
|
groups = Group.objects.all()
|
if not groups:
|
print("暂无用户组")
|
else:
|
for group in groups:
|
print(f" - 组名: {group.name}")
|
print(f" 权限数量: {group.permissions.count()}")
|
print(f" 用户数量: {group.user_set.count()}")
|
print()
|
|
|
if __name__ == '__main__':
|
print("=" * 60)
|
print("开始初始化用户组数据...")
|
print("=" * 60)
|
|
count = create_default_groups()
|
print(f"\n✓ 成功创建 {count} 个用户组")
|
|
list_all_groups()
|
|
print("\n" + "=" * 60)
|
print("用户组初始化完成!")
|
print("=" * 60)
|
