Files
jiangmingzhao-daily-report/backend/create_superuser.py

86 lines
2.2 KiB
Python
Raw Normal View History

#!/usr/bin/env python
"""
创建超级用户脚本
"""
import os
import sys
import django
# 设置Django环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings')
django.setup()
from django.contrib.auth import get_user_model
User = get_user_model()
def create_superuser():
"""创建超级用户"""
username = 'admin'
email = 'admin@example.com'
password = 'admin123456'
if User.objects.filter(username=username).exists():
print(f'超级用户 {username} 已存在')
return
user = User.objects.create_superuser(
username=username,
email=email,
password=password,
first_name='管理员',
last_name='',
department='系统管理部',
position='系统管理员'
)
print(f'超级用户创建成功!')
print(f'用户名: {username}')
print(f'密码: {password}')
print(f'邮箱: {email}')
# 创建一些测试用户
create_test_users()
def create_test_users():
"""创建测试用户"""
test_users = [
{
'username': 'zhangsan',
'email': 'zhangsan@example.com',
'password': 'test123456',
'first_name': '',
'last_name': '',
'department': '技术部',
'position': '前端工程师'
},
{
'username': 'lisi',
'email': 'lisi@example.com',
'password': 'test123456',
'first_name': '',
'last_name': '',
'department': '技术部',
'position': '后端工程师'
},
{
'username': 'wangwu',
'email': 'wangwu@example.com',
'password': 'test123456',
'first_name': '',
'last_name': '',
'department': '产品部',
'position': '产品经理'
}
]
for user_data in test_users:
if not User.objects.filter(username=user_data['username']).exists():
User.objects.create_user(**user_data)
print(f'测试用户 {user_data["username"]} 创建成功')
else:
print(f'测试用户 {user_data["username"]} 已存在')
if __name__ == '__main__':
create_superuser()