import os
|
import sys
|
import django
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'backend.settings')
|
django.setup()
|
|
from django.contrib.auth import get_user_model
|
from django.contrib.auth.models import Group
|
|
User = get_user_model()
|
|
|
def list_users():
|
"""
|
列出所有用户
|
"""
|
print("=" * 60)
|
print("系统用户列表:")
|
print("=" * 60)
|
users = User.objects.all()
|
if not users:
|
print("暂无用户")
|
else:
|
for user in users:
|
print(f"\n用户名: {user.username}")
|
print(f" 姓名: {user.name}")
|
print(f" 邮箱: {user.email}")
|
print(f" 手机号: {user.phone}")
|
print(f" 状态: {user.status}")
|
print(f" 是否管理员: {'是' if user.is_superuser else '否'}")
|
print(f" 是否员工: {'是' if user.is_staff else '否'}")
|
print(f" 是否激活: {'是' if user.is_active else '否'}")
|
print(f" 所属用户组: {', '.join([g.name for g in user.groups.all()]) or '无'}")
|
|
|
def create_admin_user(username='admin', password='admin123', email='admin@example.com', name='系统管理员'):
|
"""
|
创建管理员用户
|
"""
|
try:
|
user = User.objects.get(username=username)
|
print(f"\n用户 '{username}' 已存在，跳过创建")
|
return user
|
|
except User.DoesNotExist:
|
try:
|
user = User.objects.create_superuser(
|
username=username,
|
email=email,
|
password=password
|
)
|
user.name = name
|
user.phone = None
|
user.show_hosts = False
|
user.status = 'approved'
|
user.save()
|
print(f"\n✓ 成功创建管理员用户: {username}")
|
print(f" - 姓名: {name}")
|
print(f" - 邮箱: {email}")
|
print(f" - 密码: {password}")
|
print(f" - 状态: {user.status}")
|
return user
|
|
except Exception as e:
|
print(f"\n✗ 创建用户失败: {e}")
|
return None
|
|
|
def create_test_users():
|
"""
|
创建测试用户
|
"""
|
test_users = [
|
{
|
'username': 'testuser',
|
'password': 'test123',
|
'email': 'test@example.com',
|
'name': '测试用户',
|
'is_staff': False,
|
'is_superuser': False
|
},
|
{
|
'username': 'developer',
|
'password': 'dev123',
|
'email': 'dev@example.com',
|
'name': '开发人员',
|
'is_staff': True,
|
'is_superuser': False
|
}
|
]
|
|
created_count = 0
|
print("\n" + "=" * 60)
|
print("创建测试用户...")
|
print("=" * 60)
|
|
for user_data in test_users:
|
try:
|
user = User.objects.get(username=user_data['username'])
|
print(f"用户 '{user_data['username']}' 已存在，跳过创建")
|
continue
|
|
except User.DoesNotExist:
|
try:
|
user = User.objects.create_user(
|
username=user_data['username'],
|
email=user_data['email'],
|
password=user_data['password']
|
)
|
user.name = user_data['name']
|
user.is_staff = user_data['is_staff']
|
user.is_superuser = user_data['is_superuser']
|
user.phone = None
|
user.show_hosts = False
|
user.status = 'approved'
|
user.save()
|
print(f"✓ 成功创建用户: {user_data['username']} ({user_data['name']})")
|
created_count += 1
|
|
except Exception as e:
|
print(f"✗ 创建用户 '{user_data['username']}' 失败: {e}")
|
|
return created_count
|
|
|
def list_groups():
|
"""
|
列出所有用户组
|
"""
|
print("\n" + "=" * 60)
|
print("系统用户组列表:")
|
print("=" * 60)
|
groups = Group.objects.all()
|
if not groups:
|
print("暂无用户组")
|
else:
|
for group in groups:
|
print(f"\n组名: {group.name}")
|
print(f" 权限数量: {group.permissions.count()}")
|
print(f" 用户数量: {group.user_set.count()}")
|
if group.user_set.exists():
|
print(f" 成员: {', '.join([u.username for u in group.user_set.all()])}")
|
|
|
def create_default_groups():
|
"""
|
创建默认用户组
|
"""
|
groups_config = [
|
{'name': '管理员组', 'description': '拥有所有权限的管理员组'},
|
{'name': '开发人员组', 'description': '开发人员组'},
|
{'name': '测试人员组', 'description': '测试人员组'},
|
{'name': '只读用户组', 'description': '只读用户组'},
|
]
|
|
created_count = 0
|
print("\n" + "=" * 60)
|
print("创建默认用户组...")
|
print("=" * 60)
|
|
for group_config in groups_config:
|
try:
|
Group.objects.get(name=group_config['name'])
|
print(f"用户组 '{group_config['name']}' 已存在")
|
except Group.DoesNotExist:
|
try:
|
Group.objects.create(name=group_config['name'])
|
print(f"✓ 创建用户组: {group_config['name']}")
|
created_count += 1
|
except Exception as e:
|
print(f"✗ 创建用户组 '{group_config['name']}' 失败: {e}")
|
|
return created_count
|
|
|
if __name__ == '__main__':
|
import argparse
|
|
parser = argparse.ArgumentParser(description='用户管理脚本')
|
parser.add_argument('--list', action='store_true', help='列出所有用户')
|
parser.add_argument('--create-admin', action='store_true', help='创建管理员用户')
|
parser.add_argument('--create-test', action='store_true', help='创建测试用户')
|
parser.add_argument('--create-groups', action='store_true', help='创建默认用户组')
|
parser.add_argument('--list-groups', action='store_true', help='列出所有用户组')
|
parser.add_argument('--init', action='store_true', help='初始化所有数据（创建管理员、测试用户和用户组）')
|
|
args = parser.parse_args()
|
|
if args.list:
|
list_users()
|
elif args.create_admin:
|
create_admin_user()
|
elif args.create_test:
|
create_test_users()
|
elif args.create_groups:
|
create_default_groups()
|
elif args.list_groups:
|
list_groups()
|
elif args.init:
|
print("=" * 60)
|
print("初始化用户数据...")
|
print("=" * 60)
|
|
admin = create_admin_user()
|
if admin:
|
test_count = create_test_users()
|
group_count = create_default_groups()
|
|
list_users()
|
list_groups()
|
|
print("\n" + "=" * 60)
|
print("初始化完成!")
|
print("=" * 60)
|
print(f"✓ 创建了 {test_count} 个测试用户")
|
print(f"✓ 创建了 {group_count} 个用户组")
|
print("\n默认管理员账号:")
|
print(" 用户名: admin")
|
print(" 密码: admin123")
|
print("\n⚠️ 请在首次登录后修改默认密码!")
|
else:
|
parser.print_help()
|
