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.core.exceptions import ValidationError
|
|
|
User = get_user_model()
|
|
|
def create_admin_user(username='admin', password='admin123', email='admin@example.com', name='系统管理员'):
|
"""
|
创建管理员用户
|
|
Args:
|
username: 用户名
|
password: 密码
|
email: 邮箱
|
name: 姓名
|
|
Returns:
|
User: 创建的用户对象或已存在的用户对象
|
"""
|
try:
|
user = User.objects.get(username=username)
|
print(f"用户 '{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"✓ 成功创建管理员用户: {username}")
|
print(f" - 姓名: {name}")
|
print(f" - 邮箱: {email}")
|
print(f" - 状态: {user.status}")
|
return user
|
|
except ValidationError as e:
|
print(f"✗ 创建用户失败，验证错误: {e}")
|
return None
|
except Exception as e:
|
print(f"✗ 创建用户失败: {e}")
|
import traceback
|
traceback.print_exc()
|
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
|
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_all_users():
|
"""
|
列出所有用户
|
"""
|
print("\n" + "=" * 60)
|
print("当前系统用户列表:")
|
print("=" * 60)
|
users = User.objects.all()
|
if not users:
|
print("暂无用户")
|
else:
|
for user in users:
|
print(f" - 用户名: {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()
|
|
|
if __name__ == '__main__':
|
print("=" * 60)
|
print("开始初始化用户数据...")
|
print("=" * 60)
|
|
admin = create_admin_user()
|
if admin:
|
print("\n创建测试用户...")
|
test_count = create_test_users()
|
print(f"\n✓ 成功创建 {test_count} 个测试用户")
|
|
list_all_users()
|
|
print("\n" + "=" * 60)
|
print("用户初始化完成!")
|
print("=" * 60)
|
sys.exit(0)
|
else:
|
print("\n" + "=" * 60)
|
print("用户初始化失败!")
|
print("=" * 60)
|
sys.exit(1)
|
