peanut
37fbd6671d
io.TextIOWrapper 默认使用块缓冲，子脚本长时间输出时被缓冲不显示。 添加 line_buffering=True 确保每行实时刷新到终端。 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
413 lines
13 KiB
Python
413 lines
13 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
情绪博物馆 - 统一部署脚本
|
||
合并 deploy-all.sh / deploy-to-prod.sh / deploy-domain.sh
|
||
|
||
使用方法:
|
||
python deploy.py # 部署所有服务
|
||
python deploy.py backend # 仅部署后端
|
||
python deploy.py frontend # 仅部署前端
|
||
python deploy.py admin # 仅部署管理后台
|
||
python deploy.py life-script # 仅部署 Life-Script
|
||
python deploy.py ssl # 仅申请 SSL 证书
|
||
python deploy.py nginx # 仅部署 Nginx 配置
|
||
python deploy.py verify # 仅验证部署结果
|
||
python deploy.py all # 部署所有服务（同无参数）
|
||
|
||
Author: Peanut
|
||
Created: 2026-05-17
|
||
Purpose: 统一部署所有服务到生产服务器，替代原有的多个 shell 脚本
|
||
"""
|
||
|
||
import io
|
||
import os
|
||
import sys
|
||
import time
|
||
import subprocess
|
||
from pathlib import Path
|
||
|
||
# 强制 stdout/stderr 使用 UTF-8 编码，避免 Windows GBK 编码错误
|
||
if hasattr(sys.stdout, 'buffer'):
|
||
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', errors='replace', line_buffering=True)
|
||
sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8', errors='replace', line_buffering=True)
|
||
|
||
# ============================================================================
|
||
# 配置
|
||
# ============================================================================
|
||
SERVER_IP = "101.200.208.45"
|
||
USERNAME = "root"
|
||
DOMAIN = "lifescript.happylifeos.com"
|
||
|
||
# 项目根目录（脚本所在目录）
|
||
PROJECT_DIR = Path(__file__).parent.absolute()
|
||
|
||
SSH_OPTS = f"-o BatchMode=yes -o ConnectTimeout=10 -o StrictHostKeyChecking=no"
|
||
|
||
# ============================================================================
|
||
# 日志
|
||
# ============================================================================
|
||
|
||
|
||
class Colors:
|
||
RED = '\033[0;31m'
|
||
GREEN = '\033[0;32m'
|
||
YELLOW = '\033[1;33m'
|
||
BLUE = '\033[0;34m'
|
||
NC = '\033[0m'
|
||
|
||
|
||
def log_info(msg):
|
||
print(f"{Colors.GREEN}[INFO]{Colors.NC} {msg}")
|
||
|
||
|
||
def log_warn(msg):
|
||
print(f"{Colors.YELLOW}[WARN]{Colors.NC} {msg}")
|
||
|
||
|
||
def log_error(msg):
|
||
print(f"{Colors.RED}[ERROR]{Colors.NC} {msg}")
|
||
|
||
|
||
def log_section(msg):
|
||
print(f"\n{Colors.BLUE}{'=' * 60}{Colors.NC}")
|
||
print(f"{Colors.BLUE}{msg}{Colors.NC}")
|
||
print(f"{Colors.BLUE}{'=' * 60}{Colors.NC}\n")
|
||
|
||
|
||
# ============================================================================
|
||
# 工具函数
|
||
# ============================================================================
|
||
|
||
|
||
def run_command(cmd, cwd=None, timeout=120, capture=True):
|
||
"""执行本地命令"""
|
||
try:
|
||
if capture:
|
||
result = subprocess.run(
|
||
cmd, shell=True, cwd=cwd,
|
||
capture_output=True, text=True, encoding='utf-8', errors='replace',
|
||
timeout=timeout
|
||
)
|
||
return result.returncode == 0, result.stdout.strip(), result.stderr.strip()
|
||
else:
|
||
result = subprocess.run(
|
||
cmd, shell=True, cwd=cwd, timeout=timeout
|
||
)
|
||
return result.returncode == 0, "", ""
|
||
except subprocess.TimeoutExpired:
|
||
return False, "", f"命令执行超时 ({timeout}s): {cmd}"
|
||
except Exception as e:
|
||
return False, "", str(e)
|
||
|
||
|
||
def run_ssh_args(args, timeout=30, capture=True):
|
||
"""执行 ssh/scp 命令（使用列表参数，避免 Windows cmd.exe 找不到 ssh）"""
|
||
try:
|
||
if capture:
|
||
result = subprocess.run(
|
||
args, shell=False, cwd=PROJECT_DIR,
|
||
capture_output=True, text=True, encoding='utf-8', errors='replace',
|
||
timeout=timeout
|
||
)
|
||
return result.returncode == 0, result.stdout.strip(), result.stderr.strip()
|
||
else:
|
||
result = subprocess.run(
|
||
args, shell=False, cwd=PROJECT_DIR, timeout=timeout
|
||
)
|
||
return result.returncode == 0, "", ""
|
||
except subprocess.TimeoutExpired:
|
||
return False, "", f"命令执行超时 ({timeout}s): {' '.join(args)}"
|
||
except Exception as e:
|
||
return False, "", str(e)
|
||
|
||
|
||
def ssh_command(cmd, timeout=30):
|
||
"""在远程服务器执行命令"""
|
||
return run_ssh_args([
|
||
"ssh", "-o", "BatchMode=yes", "-o", "ConnectTimeout=10",
|
||
"-o", "StrictHostKeyChecking=no",
|
||
f"{USERNAME}@{SERVER_IP}", cmd
|
||
], timeout=timeout)
|
||
|
||
|
||
def scp_file(local, remote, timeout=120):
|
||
"""上传文件到远程服务器"""
|
||
return run_ssh_args([
|
||
"scp", "-o", "BatchMode=yes", "-o", "ConnectTimeout=10",
|
||
"-o", "StrictHostKeyChecking=no",
|
||
local, f"{USERNAME}@{SERVER_IP}:{remote}"
|
||
], timeout=timeout)
|
||
|
||
|
||
def check_ssh():
|
||
"""检查 SSH 连接"""
|
||
log_info("检查 SSH 连接...")
|
||
success, stdout, stderr = ssh_command("echo ok", timeout=15)
|
||
if not success:
|
||
log_error(f"SSH 连接失败，请先配置免密登录: ssh {USERNAME}@{SERVER_IP}")
|
||
if stderr:
|
||
log_error(f"错误详情: {stderr}")
|
||
sys.exit(1)
|
||
log_info("SSH 连接正常")
|
||
|
||
|
||
# ============================================================================
|
||
# Nginx 配置
|
||
# ============================================================================
|
||
|
||
|
||
def deploy_nginx():
|
||
"""部署 Nginx 配置"""
|
||
log_section("部署 Nginx 配置")
|
||
nginx_conf = PROJECT_DIR / "conf" / "emotion-museum.conf"
|
||
if not nginx_conf.exists():
|
||
log_error(f"Nginx 配置文件不存在: {nginx_conf}")
|
||
return False
|
||
|
||
log_info("上传 Nginx 配置文件...")
|
||
remote_conf = f"/etc/nginx/sites-available/{DOMAIN}.conf"
|
||
# 先创建远程目录
|
||
ssh_command("mkdir -p /etc/nginx/sites-available /etc/nginx/sites-enabled", timeout=15)
|
||
ok, _, err = scp_file(str(nginx_conf), remote_conf, timeout=60)
|
||
if not ok:
|
||
log_error(f"上传 Nginx 配置失败: {err}")
|
||
return False
|
||
|
||
log_info("启用站点配置...")
|
||
ok, _, err = ssh_command(
|
||
f"ln -snf {remote_conf} /etc/nginx/sites-enabled/{DOMAIN}.conf "
|
||
f"&& rm -f /etc/nginx/sites-enabled/default"
|
||
)
|
||
if not ok:
|
||
log_error(f"启用站点失败: {err}")
|
||
return False
|
||
|
||
log_info("验证 Nginx 配置...")
|
||
ok, _, err = ssh_command("nginx -t", timeout=15)
|
||
if ok:
|
||
ssh_command("systemctl reload nginx", timeout=30)
|
||
log_info("Nginx 配置重载完成")
|
||
return True
|
||
else:
|
||
log_error(f"Nginx 配置验证失败: {err}")
|
||
return False
|
||
|
||
|
||
# ============================================================================
|
||
# 后端 - 调用 backend-single/deploy.sh remote
|
||
# ============================================================================
|
||
|
||
|
||
def deploy_backend():
|
||
"""部署后端服务"""
|
||
log_section("部署后端服务")
|
||
deploy_script = PROJECT_DIR / "backend-single" / "deploy.py"
|
||
if not deploy_script.exists():
|
||
log_error(f"后端部署脚本不存在: {deploy_script}")
|
||
return False
|
||
|
||
log_info("执行后端部署...")
|
||
ok, _, err = run_command(
|
||
"python3 deploy.py remote",
|
||
cwd=str(PROJECT_DIR / "backend-single"),
|
||
timeout=600,
|
||
capture=False
|
||
)
|
||
if ok:
|
||
log_info("后端部署完成")
|
||
return True
|
||
else:
|
||
log_error(f"后端部署失败: {err}")
|
||
return False
|
||
|
||
|
||
# ============================================================================
|
||
# 前端 - 调用 web/deploy.sh
|
||
# ============================================================================
|
||
|
||
|
||
def deploy_frontend():
|
||
"""部署用户前端"""
|
||
log_section("部署用户前端")
|
||
deploy_script = PROJECT_DIR / "web" / "deploy.py"
|
||
if not deploy_script.exists():
|
||
log_error(f"前端部署脚本不存在: {deploy_script}")
|
||
return False
|
||
|
||
log_info("执行前端部署...")
|
||
ok, _, err = run_command(
|
||
"python3 deploy.py",
|
||
cwd=str(PROJECT_DIR / "web"),
|
||
timeout=600,
|
||
capture=False
|
||
)
|
||
if ok:
|
||
log_info("前端部署完成")
|
||
return True
|
||
else:
|
||
log_error(f"前端部署失败: {err}")
|
||
return False
|
||
|
||
|
||
# ============================================================================
|
||
# 管理后台 - 调用 web-admin/deploy.py
|
||
# ============================================================================
|
||
|
||
|
||
def deploy_admin():
|
||
"""部署管理后台"""
|
||
log_section("部署管理后台")
|
||
deploy_script = PROJECT_DIR / "web-admin" / "deploy.py"
|
||
if not deploy_script.exists():
|
||
log_error(f"管理后台部署脚本不存在: {deploy_script}")
|
||
return False
|
||
|
||
log_info("执行管理后台部署...")
|
||
ok, _, err = run_command(
|
||
"python3 deploy.py",
|
||
cwd=str(PROJECT_DIR / "web-admin"),
|
||
timeout=600,
|
||
capture=False
|
||
)
|
||
if ok:
|
||
log_info("管理后台部署完成")
|
||
return True
|
||
else:
|
||
log_error(f"管理后台部署失败: {err}")
|
||
return False
|
||
|
||
|
||
# ============================================================================
|
||
# Life-Script - 调用 life-script/deploy.sh
|
||
# ============================================================================
|
||
|
||
|
||
def deploy_life_script():
|
||
"""部署 Life-Script"""
|
||
log_section("部署 Life-Script")
|
||
deploy_script = PROJECT_DIR / "life-script" / "deploy.py"
|
||
if not deploy_script.exists():
|
||
log_error(f"Life-Script 部署脚本不存在: {deploy_script}")
|
||
return False
|
||
|
||
log_info("执行 Life-Script 部署...")
|
||
ok, _, err = run_command(
|
||
"python3 deploy.py",
|
||
cwd=str(PROJECT_DIR / "life-script"),
|
||
timeout=600,
|
||
capture=False
|
||
)
|
||
if ok:
|
||
log_info("Life-Script 部署完成")
|
||
return True
|
||
else:
|
||
log_error(f"Life-Script 部署失败: {err}")
|
||
return False
|
||
|
||
|
||
# ============================================================================
|
||
# 验证部署
|
||
# ============================================================================
|
||
|
||
|
||
def verify_deploy():
|
||
"""验证部署结果"""
|
||
log_section("验证部署")
|
||
log_info("检查 HTTPS 访问...")
|
||
|
||
endpoints = [
|
||
(f"https://{DOMAIN}/", "前端页面"),
|
||
(f"https://{DOMAIN}/emotion-museum-admin/", "管理后台"),
|
||
(f"https://{DOMAIN}/api/", "API 代理"),
|
||
(f"http://{DOMAIN}/", "HTTP 跳转"),
|
||
]
|
||
|
||
for url, label in endpoints:
|
||
ok, stdout, _ = run_command(
|
||
f'curl -k -s -o /dev/null -w "HTTP %{{http_code}}" {url}',
|
||
timeout=30
|
||
)
|
||
if ok and stdout:
|
||
log_info(f" {label}: {stdout}")
|
||
else:
|
||
log_warn(f" {label}: 访问异常")
|
||
|
||
log_info("验证完成")
|
||
return True
|
||
|
||
|
||
# ============================================================================
|
||
# 主程序
|
||
# ============================================================================
|
||
|
||
|
||
def print_usage():
|
||
"""打印使用说明"""
|
||
print("情绪博物馆 - 统一部署脚本")
|
||
print()
|
||
print("使用方法:")
|
||
print(" python deploy.py # 部署所有服务")
|
||
print(" python deploy.py backend # 仅部署后端")
|
||
print(" python deploy.py frontend # 仅部署前端")
|
||
print(" python deploy.py admin # 仅部署管理后台")
|
||
print(" python deploy.py life-script # 仅部署 Life-Script")
|
||
print(" python deploy.py ssl # 仅申请 SSL 证书")
|
||
print(" python deploy.py nginx # 仅部署 Nginx 配置")
|
||
print(" python deploy.py verify # 仅验证部署结果")
|
||
print(" python deploy.py all # 部署所有服务（同无参数）")
|
||
|
||
|
||
def main():
|
||
deploy_type = sys.argv[1] if len(sys.argv) > 1 else "all"
|
||
|
||
if deploy_type in ("--help", "-h", "help"):
|
||
print_usage()
|
||
return
|
||
|
||
start_time = time.time()
|
||
|
||
log_section("情绪博物馆 - 统一部署")
|
||
log_info(f"部署类型: {deploy_type}")
|
||
log_info(f"部署时间: {time.strftime('%Y-%m-%d %H:%M:%S')}")
|
||
|
||
actions = {
|
||
"backend": lambda: check_ssh() or True and deploy_backend(),
|
||
"frontend": lambda: check_ssh() or True and deploy_frontend(),
|
||
"admin": lambda: check_ssh() or True and deploy_admin(),
|
||
"life-script": lambda: check_ssh() or True and deploy_life_script(),
|
||
"nginx": lambda: check_ssh() or True and deploy_nginx(),
|
||
"verify": lambda: verify_deploy(),
|
||
"all": None, # handled separately
|
||
}
|
||
|
||
if deploy_type == "all":
|
||
check_ssh()
|
||
deploy_nginx() or True
|
||
deploy_backend()
|
||
deploy_frontend()
|
||
deploy_admin()
|
||
deploy_life_script()
|
||
verify_deploy()
|
||
elif deploy_type in actions:
|
||
actions[deploy_type]()
|
||
else:
|
||
log_error(f"无效的部署类型: {deploy_type}")
|
||
print("使用方法: python deploy.py [backend|frontend|admin|life-script|nginx|verify|all]")
|
||
sys.exit(1)
|
||
|
||
duration = int(time.time() - start_time)
|
||
log_section(f"部署完成 (耗时 {duration}s)")
|
||
log_info(f"用户前端: https://{DOMAIN}/")
|
||
log_info(f"管理后台: https://{DOMAIN}/emotion-museum-admin/")
|
||
log_info(f"Life-Script: https://{DOMAIN}/life-script/")
|
||
log_info(f"API 地址: https://{DOMAIN}/api")
|
||
log_info(f"WebSocket: wss://{DOMAIN}/ws")
|
||
log_info(f"API 文档: https://{DOMAIN}/doc.html")
|
||
log_info(f"Swagger UI: https://{DOMAIN}/swagger-ui/index.html")
|
||
log_info(f"API Specs: https://{DOMAIN}/v3/api-docs")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|