247 lines
8.8 KiB
Python
247 lines
8.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
CCA 部署脚本 — 通过 paramiko 上传文件到腾讯云服务器
|
||
|
|
"""
|
||
|
|
import sys
|
||
|
|
sys.stdout.reconfigure(encoding='utf-8')
|
||
|
|
|
||
|
|
import os
|
||
|
|
import paramiko
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
HOST = '101.42.29.217'
|
||
|
|
PORT = 22
|
||
|
|
USER = 'root'
|
||
|
|
PASS = 'liutong65'
|
||
|
|
|
||
|
|
CCA_ROOT = Path(__file__).resolve().parent.parent
|
||
|
|
SRC_DIR = CCA_ROOT / 'src'
|
||
|
|
DEPLOY_DIR = CCA_ROOT / 'deploy'
|
||
|
|
|
||
|
|
# 服务器目标路径
|
||
|
|
SERVER_CCA_SRC = '/workspace/military_tech_voice/backend/cca_src'
|
||
|
|
SERVER_FRONTEND = '/workspace/military_tech_voice/frontend'
|
||
|
|
SERVER_WWW = '/var/www/voice'
|
||
|
|
SERVER_BACKEND = '/workspace/military_tech_voice/backend'
|
||
|
|
SERVER_ROUTES = f'{SERVER_BACKEND}/app/routes'
|
||
|
|
|
||
|
|
# 需要上传的 src 模块
|
||
|
|
SRC_MODULES = [
|
||
|
|
'asr_client.py',
|
||
|
|
'line_breaker.py',
|
||
|
|
'ai_line_breaker.py',
|
||
|
|
'ai_proofreader.py',
|
||
|
|
'term_normalizer.py',
|
||
|
|
'hotword_extractor.py',
|
||
|
|
'srt_writer.py',
|
||
|
|
'segment_splitter.py',
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def connect():
|
||
|
|
ssh = paramiko.SSHClient()
|
||
|
|
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
|
|
ssh.connect(HOST, PORT, USER, PASS)
|
||
|
|
sftp = ssh.open_sftp()
|
||
|
|
return ssh, sftp
|
||
|
|
|
||
|
|
|
||
|
|
def run(ssh, cmd):
|
||
|
|
print(f' $ {cmd}')
|
||
|
|
_, stdout, stderr = ssh.exec_command(cmd)
|
||
|
|
out = stdout.read().decode('utf-8', errors='replace').strip()
|
||
|
|
err = stderr.read().decode('utf-8', errors='replace').strip()
|
||
|
|
if out:
|
||
|
|
print(f' {out[:500]}')
|
||
|
|
if err:
|
||
|
|
print(f' [stderr] {err[:500]}')
|
||
|
|
return out
|
||
|
|
|
||
|
|
|
||
|
|
def upload(sftp, local_path, remote_path):
|
||
|
|
print(f' ↑ {Path(local_path).name} → {remote_path}')
|
||
|
|
sftp.put(str(local_path), remote_path)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
print('=== CCA 部署开始 ===\n')
|
||
|
|
ssh, sftp = connect()
|
||
|
|
print('[1/7] 连接成功\n')
|
||
|
|
|
||
|
|
# Step 2: 创建 cca_src 目录 + 上传源码
|
||
|
|
print('[2/7] 上传 CCA 源码模块...')
|
||
|
|
run(ssh, f'mkdir -p {SERVER_CCA_SRC}')
|
||
|
|
for module in SRC_MODULES:
|
||
|
|
local = SRC_DIR / module
|
||
|
|
if local.exists():
|
||
|
|
upload(sftp, local, f'{SERVER_CCA_SRC}/{module}')
|
||
|
|
else:
|
||
|
|
print(f' ⚠ 跳过（不存在）: {module}')
|
||
|
|
print()
|
||
|
|
|
||
|
|
# Step 3: 上传 cca_route.py 到 app/routes/
|
||
|
|
print('[3/7] 上传 cca_route.py...')
|
||
|
|
run(ssh, f'mkdir -p {SERVER_ROUTES}')
|
||
|
|
upload(sftp, DEPLOY_DIR / 'cca_route.py', f'{SERVER_ROUTES}/cca.py')
|
||
|
|
# 确保 __init__.py 存在
|
||
|
|
run(ssh, f'touch {SERVER_ROUTES}/__init__.py')
|
||
|
|
print()
|
||
|
|
|
||
|
|
# Step 4: 上传 cca.html 到前端目录
|
||
|
|
print('[4/7] 上传 cca.html...')
|
||
|
|
upload(sftp, DEPLOY_DIR / 'cca.html', f'{SERVER_FRONTEND}/cca.html')
|
||
|
|
upload(sftp, DEPLOY_DIR / 'cca.html', f'{SERVER_WWW}/cca.html')
|
||
|
|
print()
|
||
|
|
|
||
|
|
# Step 5: 配置 .env（追加 CCA 相关变量）
|
||
|
|
print('[5/7] 配置 .env...')
|
||
|
|
env_path = f'{SERVER_BACKEND}/.env'
|
||
|
|
existing_env = ''
|
||
|
|
try:
|
||
|
|
with sftp.open(env_path, 'r') as f:
|
||
|
|
existing_env = f.read().decode('utf-8', errors='replace')
|
||
|
|
except FileNotFoundError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
env_additions = []
|
||
|
|
if 'XFYUN_APP_ID' not in existing_env:
|
||
|
|
env_additions.append('# === CCA 唱词助手凭证 ===')
|
||
|
|
env_additions.append('# 讯飞录音文件转写（账号1-默认）')
|
||
|
|
env_additions.append('XFYUN_APP_ID=4c423e35')
|
||
|
|
env_additions.append('XFYUN_SECRET_KEY=b9e0b97d5dda072c9b4b8fb59e7e3d22')
|
||
|
|
env_additions.append('# 讯飞备用账号2')
|
||
|
|
env_additions.append('# XFYUN_APP_ID=52ae3024')
|
||
|
|
env_additions.append('# XFYUN_SECRET_KEY=d65de0eb282a4339e2b1e14fd119e42e')
|
||
|
|
if 'DEEPSEEK_API_KEY' not in existing_env:
|
||
|
|
env_additions.append('# DeepSeek（校对+折行+热词）')
|
||
|
|
env_additions.append('DEEPSEEK_API_KEY=sk-01a7868a88a04ab494e4f05c1f3f06e2')
|
||
|
|
|
||
|
|
if env_additions:
|
||
|
|
with sftp.open(env_path, 'a') as f:
|
||
|
|
f.write('\n' + '\n'.join(env_additions) + '\n')
|
||
|
|
print(' .env 已追加 CCA 凭证')
|
||
|
|
else:
|
||
|
|
print(' .env 已有 CCA 凭证，跳过')
|
||
|
|
print()
|
||
|
|
|
||
|
|
# Step 6: 注册 CCA 蓝图到 Flask main.py
|
||
|
|
print('[6/7] 注册 CCA 蓝图...')
|
||
|
|
main_py_path = f'{SERVER_BACKEND}/app/main.py'
|
||
|
|
with sftp.open(main_py_path, 'r') as f:
|
||
|
|
main_content = f.read().decode('utf-8', errors='replace')
|
||
|
|
|
||
|
|
if 'cca' not in main_content.lower():
|
||
|
|
# 找到最后一个 register_blueprint 的位置，在其后追加
|
||
|
|
lines = main_content.split('\n')
|
||
|
|
insert_idx = -1
|
||
|
|
for i, line in enumerate(lines):
|
||
|
|
if 'register_blueprint' in line:
|
||
|
|
insert_idx = i
|
||
|
|
|
||
|
|
if insert_idx >= 0:
|
||
|
|
indent = ' ' # 匹配现有缩进
|
||
|
|
# 检查现有蓝图注册的缩进
|
||
|
|
existing_line = lines[insert_idx]
|
||
|
|
indent = existing_line[:len(existing_line) - len(existing_line.lstrip())]
|
||
|
|
|
||
|
|
cca_lines = [
|
||
|
|
'',
|
||
|
|
f'{indent}# CCA 唱词助手',
|
||
|
|
f'{indent}from app.routes.cca import bp as cca_bp',
|
||
|
|
f'{indent}app.register_blueprint(cca_bp)',
|
||
|
|
]
|
||
|
|
for j, cca_line in enumerate(cca_lines):
|
||
|
|
lines.insert(insert_idx + 1 + j, cca_line)
|
||
|
|
|
||
|
|
new_content = '\n'.join(lines)
|
||
|
|
with sftp.open(main_py_path, 'w') as f:
|
||
|
|
f.write(new_content)
|
||
|
|
print(' 已注册 CCA 蓝图')
|
||
|
|
else:
|
||
|
|
print(' ⚠ 未找到 register_blueprint，请手动注册')
|
||
|
|
else:
|
||
|
|
print(' CCA 蓝图已注册，跳过')
|
||
|
|
print()
|
||
|
|
|
||
|
|
# Step 7: 安装依赖 + 修改 index.html + 重启
|
||
|
|
print('[7/7] 安装依赖、修改首页、重启服务...')
|
||
|
|
|
||
|
|
# 安装 Python 依赖
|
||
|
|
run(ssh, f'{SERVER_BACKEND}/venv/bin/pip install openai python-docx 2>&1 | tail -5')
|
||
|
|
|
||
|
|
# 修改 index.html 添加 CCA 入口按钮
|
||
|
|
index_path = f'{SERVER_FRONTEND}/index.html'
|
||
|
|
with sftp.open(index_path, 'r') as f:
|
||
|
|
index_content = f.read().decode('utf-8', errors='replace')
|
||
|
|
|
||
|
|
if '唱词助手' not in index_content and 'cca.html' not in index_content:
|
||
|
|
# 在导航栏中找到合适位置插入按钮
|
||
|
|
# 典型位置：header 或 nav 区域的最后一个链接之后
|
||
|
|
if '<header' in index_content or '<nav' in index_content:
|
||
|
|
# 找 </header> 或 </nav> 前插入
|
||
|
|
for marker in ['</nav>', '</header>']:
|
||
|
|
if marker in index_content:
|
||
|
|
btn_html = ' <a href="cca.html" class="nav-link">唱词助手</a>\n'
|
||
|
|
index_content = index_content.replace(marker, btn_html + ' ' + marker)
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
# 备用：在 body 开头加一个浮动按钮
|
||
|
|
btn_html = '<div style="position:fixed;top:10px;right:10px;z-index:9999"><a href="cca.html" style="background:#4a9eff;color:#fff;padding:8px 16px;border-radius:6px;text-decoration:none;font-size:14px">唱词助手</a></div>\n'
|
||
|
|
index_content = index_content.replace('<body>', '<body>\n' + btn_html)
|
||
|
|
|
||
|
|
with sftp.open(index_path, 'w') as f:
|
||
|
|
f.write(index_content)
|
||
|
|
|
||
|
|
# 同步到 /var/www/voice/
|
||
|
|
run(ssh, f'cp {SERVER_FRONTEND}/index.html {SERVER_WWW}/index.html')
|
||
|
|
print(' index.html 已添加唱词助手入口')
|
||
|
|
else:
|
||
|
|
print(' index.html 已有唱词助手入口，跳过')
|
||
|
|
|
||
|
|
# 增大 Nginx 上传限制（音频文件可能较大）
|
||
|
|
nginx_conf = '/etc/nginx/nginx.conf'
|
||
|
|
with sftp.open(nginx_conf, 'r') as f:
|
||
|
|
nginx_content = f.read().decode('utf-8', errors='replace')
|
||
|
|
|
||
|
|
if 'client_max_body_size' not in nginx_content:
|
||
|
|
nginx_content = nginx_content.replace(
|
||
|
|
'http {',
|
||
|
|
'http {\n client_max_body_size 200m;'
|
||
|
|
)
|
||
|
|
with sftp.open(nginx_conf, 'w') as f:
|
||
|
|
f.write(nginx_content)
|
||
|
|
run(ssh, 'nginx -t && nginx -s reload')
|
||
|
|
print(' Nginx 已增大上传限制至 200MB')
|
||
|
|
else:
|
||
|
|
print(' Nginx 上传限制已配置')
|
||
|
|
|
||
|
|
# 重启 Flask 服务
|
||
|
|
print('\n 重启 Flask...')
|
||
|
|
# 查找并重启 Flask 进程
|
||
|
|
run(ssh, 'pkill -f "flask run" || pkill -f "gunicorn" || pkill -f "python.*app" || true')
|
||
|
|
# 给进程时间退出
|
||
|
|
import time
|
||
|
|
time.sleep(2)
|
||
|
|
|
||
|
|
# 检查服务启动方式
|
||
|
|
service_out = run(ssh, 'systemctl list-units --type=service | grep -i voice || systemctl list-units --type=service | grep -i flask || true')
|
||
|
|
if 'voice' in service_out.lower() or 'flask' in service_out.lower():
|
||
|
|
service_name = service_out.split()[0] if service_out else ''
|
||
|
|
if service_name:
|
||
|
|
run(ssh, f'systemctl restart {service_name}')
|
||
|
|
else:
|
||
|
|
# 直接后台启动
|
||
|
|
run(ssh, f'cd {SERVER_BACKEND} && source venv/bin/activate && nohup python -m flask run --host=0.0.0.0 --port=5000 > /tmp/flask_cca.log 2>&1 &')
|
||
|
|
|
||
|
|
print()
|
||
|
|
print('=== CCA 部署完成 ===')
|
||
|
|
print(f'访问地址: http://{HOST}/cca.html')
|
||
|
|
print(f'API 地址: http://{HOST}/api/cca/')
|
||
|
|
|
||
|
|
sftp.close()
|
||
|
|
ssh.close()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|