#!/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()