#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 管理后台部署脚本 - 将构建好的管理后台文件上传到服务器 使用方法: python deploy.py """ import os import sys import shutil import subprocess from pathlib import Path # 配置变量 SERVER_IP = "101.200.208.45" USERNAME = "root" REMOTE_PATH = "/data/www/emotion-museum-admin" # 本地路径 SCRIPT_DIR = Path(__file__).parent.absolute() DIST_DIR = SCRIPT_DIR / "dist" class Colors: """终端颜色""" GREEN = '\033[32m' RED = '\033[31m' YELLOW = '\033[33m' RESET = '\033[0m' def log_info(msg): """打印信息日志""" print(f"{Colors.GREEN}{Colors.RESET} {msg}") def log_error(msg): """打印错误日志""" print(f"{Colors.RED}{Colors.RESET} {msg}") def log_step(msg): """打印步骤日志""" print(f" {msg}") def run_command(cmd, cwd=None, shell=True, capture=True, timeout=60): """执行本地命令""" try: if capture: result = subprocess.run(cmd, cwd=cwd, shell=shell, capture_output=True, text=True, timeout=timeout) return result.returncode == 0, result.stdout, result.stderr else: result = subprocess.run(cmd, cwd=cwd, shell=shell, 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_command_realtime(cmd, cwd=None, shell=True, timeout=120): """执行命令并实时输出（用于大文件上传进度）""" try: result = subprocess.run(cmd, cwd=cwd, shell=shell, timeout=timeout) return result.returncode == 0 except subprocess.TimeoutExpired: log_error(f"命令执行超时 ({timeout}s)") return False except Exception as e: log_error(f"执行失败: {e}") return False def check_npm(): """检查npm是否安装""" success, _, _ = run_command("npm --version") if not success: log_error("错误: 未找到npm命令，请先安装Node.js") sys.exit(1) def check_ssh(): """检查SSH连接""" log_step(" 检查SSH连接...") cmd = f'ssh -o BatchMode=yes -o ConnectTimeout=10 -o StrictHostKeyChecking=no {USERNAME}@{SERVER_IP} "echo ok"' success, stdout, stderr = run_command(cmd, timeout=15) if not success: log_error("SSH连接失败，请检查:") print(f" 1. 服务器IP: {SERVER_IP}") print(f" 2. SSH密钥是否配置正确") print(f" 3. 服务器防火墙是否允许22端口") if "Host key verification failed" in stderr: log_error("主机密钥验证失败，请先手动执行: ssh root@101.200.208.45") sys.exit(1) log_info("SSH连接正常") def clean_dist(): """清理旧的构建文件""" log_step(" 清理旧的构建文件...") if DIST_DIR.exists(): shutil.rmtree(DIST_DIR) def build_project(): """构建项目""" log_step(" 开始构建管理后台项目（生产环境）...") os.chdir(SCRIPT_DIR) env = os.environ.copy() env["NODE_ENV"] = "production" try: result = subprocess.run( "npm run build", shell=True, cwd=SCRIPT_DIR, env=env, timeout=300 ) if result.returncode != 0: log_error("管理后台项目构建失败，请检查代码") sys.exit(1) except Exception as e: log_error(f"构建失败: {e}") sys.exit(1) log_info("管理后台项目构建成功") def verify_dist(): """验证dist目录是否存在""" if not DIST_DIR.exists(): log_error("错误: 构建后dist目录仍不存在，请检查构建配置") sys.exit(1) index_file = DIST_DIR / "index.html" assets_dir = DIST_DIR / "assets" if not index_file.exists(): log_error("错误: dist/index.html 不存在") sys.exit(1) if not assets_dir.exists(): log_error("错误: dist/assets 目录不存在") sys.exit(1) def create_remote_dir(): """创建远程目录""" log_step(" 创建远程目录...") cmd = f'ssh -o BatchMode=yes -o ConnectTimeout=10 -o StrictHostKeyChecking=no {USERNAME}@{SERVER_IP} "mkdir -p {REMOTE_PATH}"' success, _, stderr = run_command(cmd, timeout=15) if not success: log_error(f"创建远程目录失败: {stderr}") sys.exit(1) def upload_files(): """上传文件到服务器""" log_step(" 上传文件到服务器...") print(f" 正在上传到 {SERVER_IP}:{REMOTE_PATH}") # 使用 rsync 替代 scp，支持断点续传和进度显示 assets_dir = DIST_DIR / "assets" # 上传 index.html cmd1 = f'scp -o BatchMode=yes -o ConnectTimeout=10 -o StrictHostKeyChecking=no "{DIST_DIR / "index.html"}" {USERNAME}@{SERVER_IP}:{REMOTE_PATH}/' success1 = run_command_realtime(cmd1, timeout=60) if not success1: log_error("上传 index.html 失败") return False log_info("index.html 上传成功") # 上传 assets 目录（使用 rsync 更高效） cmd2 = f'rsync -avz --progress -e "ssh -o BatchMode=yes -o ConnectTimeout=10 -o StrictHostKeyChecking=no" "{assets_dir}/" {USERNAME}@{SERVER_IP}:{REMOTE_PATH}/assets/' print(f" 正在上传 assets 目录...") success2 = run_command_realtime(cmd2, timeout=300) if not success2: # rsync 不可用则回退到 scp log_info("rsync不可用，使用scp上传...") cmd2_fallback = f'scp -o BatchMode=yes -o ConnectTimeout=10 -o StrictHostKeyChecking=no -r "{assets_dir}" {USERNAME}@{SERVER_IP}:{REMOTE_PATH}/' success2 = run_command_realtime(cmd2_fallback, timeout=300) if not success2: log_error("上传 assets 目录失败") return False log_info("assets 目录上传成功") return True def clean_remote_old_files(): """清理远程服务器上旧的assets目录（全量替换策略）""" log_step(" 清理远程旧文件...") # 直接删除远程 assets 目录，后续会重新上传 cmd = f'ssh -o BatchMode=yes -o ConnectTimeout=10 -o StrictHostKeyChecking=no {USERNAME}@{SERVER_IP} "rm -rf {REMOTE_PATH}/assets"' success, _, stderr = run_command(cmd, timeout=15) if success: log_info("远程旧文件已清理") else: log_info(f"清理远程旧文件跳过（可能目录不存在）: {stderr}") def set_permissions(): """设置文件权限""" log_step(" 设置文件权限...") cmd = f'ssh -o BatchMode=yes -o ConnectTimeout=10 -o StrictHostKeyChecking=no {USERNAME}@{SERVER_IP} "find {REMOTE_PATH} -type f -exec chmod 644 {{}} \\; && find {REMOTE_PATH} -type d -exec chmod 755 {{}} \\;"' success, _, stderr = run_command(cmd, timeout=30) if not success: log_error(f"设置权限失败: {stderr}") else: log_info("文件权限设置成功") def deploy(): """执行部署""" print("开始部署管理后台应用到服务器...") # 检查依赖 check_npm() check_ssh() # 清理旧构建 clean_dist() # 构建项目 build_project() # 验证构建结果 verify_dist() # 创建远程目录 create_remote_dir() # 清理远程旧assets目录 clean_remote_old_files() # 上传文件（全量上传） if upload_files(): # 设置权限 set_permissions() log_info("管理后台部署完成！") print(f" 访问地址: http://{SERVER_IP}/emotion-museum-admin/") else: log_error("部署失败，请检查:") print(" 1. 服务器IP地址是否正确") print(" 2. SSH密钥是否配置正确") print(" 3. 服务器目录权限是否正确") sys.exit(1) def main(): """主函数""" deploy() if __name__ == "__main__": main()