#!/usr/bin/env python
|
"""
|
首次部署初始化脚本
|
|
该脚本用于首次部署系统时执行所有必要的初始化操作，包括：
|
1. 数据库迁移
|
2. 创建管理员用户
|
3. 创建默认用户组
|
4. 收集静态文件
|
|
使用方法:
|
python scripts/deploy_init.py
|
|
注意事项:
|
- 确保已配置好数据库连接信息 (conf/env.py)
|
- 确保数据库已创建
|
- 确保已安装所有依赖 (pip install -r requirements.txt)
|
"""
|
|
import os
|
import sys
|
import django
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'backend.settings')
|
django.setup()
|
|
from django.core.management import call_command
|
|
|
def check_database_connection():
|
"""
|
检查数据库连接
|
"""
|
print("=" * 60)
|
print("检查数据库连接...")
|
print("=" * 60)
|
|
try:
|
from django.db import connection
|
with connection.cursor() as cursor:
|
cursor.execute("SELECT 1")
|
print("✓ 数据库连接正常")
|
return True
|
except Exception as e:
|
print(f"✗ 数据库连接失败: {e}")
|
print("\n请检查 conf/env.py 中的数据库配置:")
|
print(" - DATABASE_NAME")
|
print(" - DATABASE_USER")
|
print(" - DATABASE_PASSWORD")
|
print(" - DATABASE_HOST")
|
print(" - DATABASE_PORT")
|
return False
|
|
|
def run_migrations():
|
"""
|
执行数据库迁移
|
"""
|
print("\n" + "=" * 60)
|
print("执行数据库迁移...")
|
print("=" * 60)
|
|
try:
|
print("\n[1/2] 生成迁移文件...")
|
call_command('makemigrations', verbosity=1)
|
print("✓ 迁移文件生成成功")
|
|
print("\n[2/2] 应用迁移...")
|
call_command('migrate', verbosity=1)
|
print("✓ 数据库迁移成功")
|
return True
|
|
except Exception as e:
|
print(f"✗ 数据库迁移失败: {e}")
|
import traceback
|
traceback.print_exc()
|
return False
|
|
|
def create_admin_user():
|
"""
|
创建管理员用户
|
"""
|
print("\n" + "=" * 60)
|
print("创建管理员用户...")
|
print("=" * 60)
|
|
from django.contrib.auth import get_user_model
|
User = get_user_model()
|
|
default_username = 'admin'
|
default_password = 'admin123'
|
default_email = 'admin@example.com'
|
default_name = '系统管理员'
|
|
try:
|
user = User.objects.get(username=default_username)
|
print(f"管理员用户 '{default_username}' 已存在，跳过创建")
|
print(f" 用户名: {user.username}")
|
print(f" 姓名: {user.name}")
|
print(f" 邮箱: {user.email}")
|
return True
|
|
except User.DoesNotExist:
|
try:
|
user = User.objects.create_superuser(
|
username=default_username,
|
email=default_email,
|
password=default_password
|
)
|
user.name = default_name
|
user.phone = None
|
user.show_hosts = False
|
user.status = 'approved'
|
user.save()
|
|
print("✓ 管理员用户创建成功")
|
print(f" 用户名: {default_username}")
|
print(f" 密码: {default_password}")
|
print(f" 邮箱: {default_email}")
|
print(f" 姓名: {default_name}")
|
print("\n⚠️ 重要提示: 请在首次登录后修改默认密码!")
|
return True
|
|
except Exception as e:
|
print(f"✗ 创建管理员用户失败: {e}")
|
import traceback
|
traceback.print_exc()
|
return False
|
|
|
def create_default_groups():
|
"""
|
创建默认用户组
|
"""
|
print("\n" + "=" * 60)
|
print("创建默认用户组...")
|
print("=" * 60)
|
|
from django.contrib.auth.models import Group
|
|
groups = [
|
{'name': '管理员组', 'description': '拥有所有权限的管理员组'},
|
{'name': '开发人员组', 'description': '开发人员组'},
|
{'name': '测试人员组', 'description': '测试人员组'},
|
{'name': '只读用户组', 'description': '只读用户组'},
|
]
|
|
created_count = 0
|
for group_info in groups:
|
try:
|
Group.objects.get(name=group_info['name'])
|
print(f" 用户组 '{group_info['name']}' 已存在")
|
except Group.DoesNotExist:
|
try:
|
Group.objects.create(name=group_info['name'])
|
print(f" ✓ 创建用户组: {group_info['name']}")
|
created_count += 1
|
except Exception as e:
|
print(f" ✗ 创建用户组 '{group_info['name']}' 失败: {e}")
|
|
print(f"\n✓ 成功创建 {created_count} 个用户组")
|
return True
|
|
|
def collect_static_files():
|
"""
|
收集静态文件
|
"""
|
print("\n" + "=" * 60)
|
print("收集静态文件...")
|
print("=" * 60)
|
|
try:
|
call_command('collectstatic', '--noinput', verbosity=1)
|
print("✓ 静态文件收集成功")
|
return True
|
|
except Exception as e:
|
print(f"✗ 静态文件收集失败: {e}")
|
import traceback
|
traceback.print_exc()
|
return False
|
|
|
def display_summary():
|
"""
|
显示部署摘要
|
"""
|
print("\n" + "=" * 60)
|
print("部署摘要")
|
print("=" * 60)
|
|
from django.contrib.auth import get_user_model
|
from django.contrib.auth.models import Group
|
from apps.lunarlink.models import Project
|
|
User = get_user_model()
|
|
print(f"\n用户数量: {User.objects.count()}")
|
print(f"用户组数量: {Group.objects.count()}")
|
print(f"项目数量: {Project.objects.count()}")
|
|
print("\n默认管理员账号:")
|
print(" 用户名: admin")
|
print(" 密码: admin123")
|
print("\n⚠️ 请在首次登录后立即修改默认密码!")
|
|
|
def main():
|
"""
|
主函数
|
"""
|
print("\n")
|
print("╔" + "=" * 58 + "╗")
|
print("║" + " " * 10 + "接口自动化平台 - 首次部署初始化" + " " * 13 + "║")
|
print("╚" + "=" * 58 + "╝")
|
print()
|
|
steps = [
|
("检查数据库连接", check_database_connection),
|
("执行数据库迁移", run_migrations),
|
("创建管理员用户", create_admin_user),
|
("创建默认用户组", create_default_groups),
|
("收集静态文件", collect_static_files),
|
]
|
|
failed_steps = []
|
|
for step_name, step_func in steps:
|
if not step_func():
|
failed_steps.append(step_name)
|
print(f"\n✗ '{step_name}' 失败，终止部署")
|
break
|
|
if not failed_steps:
|
display_summary()
|
print("\n" + "=" * 60)
|
print("✓ 部署初始化完成!")
|
print("=" * 60)
|
print("\n系统已准备就绪，可以启动服务:")
|
print(" python manage.py runserver")
|
print("\n或使用 gunicorn 启动:")
|
print(" gunicorn backend.wsgi:application -c gunicorn_conf.py")
|
print()
|
return 0
|
else:
|
print("\n" + "=" * 60)
|
print("✗ 部署初始化失败!")
|
print("=" * 60)
|
print(f"\n失败的步骤: {', '.join(failed_steps)}")
|
print("\n请检查错误信息并修复后重试")
|
print()
|
return 1
|
|
|
if __name__ == '__main__':
|
sys.exit(main())
|
