修改说明
This commit is contained in:
zhaojie
266
scripts/deploy.sh
Normal file
266
scripts/deploy.sh
Normal file
@@ -0,0 +1,266 @@
|
||||
#!/bin/bash
|
||||
# TSP智能助手部署脚本
|
||||
|
||||
set -e # 遇到错误立即退出
|
||||
|
||||
# 颜色定义
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# 日志函数
|
||||
log_info() {
|
||||
echo -e "${GREEN}[INFO]${NC} $1"
|
||||
}
|
||||
|
||||
log_warn() {
|
||||
echo -e "${YELLOW}[WARN]${NC} $1"
|
||||
}
|
||||
|
||||
log_error() {
|
||||
echo -e "${RED}[ERROR]${NC} $1"
|
||||
}
|
||||
|
||||
# 检查依赖
|
||||
check_dependencies() {
|
||||
log_info "检查系统依赖..."
|
||||
|
||||
# 检查Python
|
||||
if ! command -v python3 &> /dev/null; then
|
||||
log_error "Python3 未安装"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# 检查pip
|
||||
if ! command -v pip3 &> /dev/null; then
|
||||
log_error "pip3 未安装"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# 检查Git
|
||||
if ! command -v git &> /dev/null; then
|
||||
log_error "Git 未安装"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "依赖检查完成"
|
||||
}
|
||||
|
||||
# 创建虚拟环境
|
||||
setup_venv() {
|
||||
local venv_path=$1
|
||||
log_info "创建虚拟环境: $venv_path"
|
||||
|
||||
if [ ! -d "$venv_path" ]; then
|
||||
python3 -m venv "$venv_path"
|
||||
fi
|
||||
|
||||
source "$venv_path/bin/activate"
|
||||
pip install --upgrade pip
|
||||
log_info "虚拟环境设置完成"
|
||||
}
|
||||
|
||||
# 安装依赖
|
||||
install_dependencies() {
|
||||
log_info "安装Python依赖..."
|
||||
pip install -r requirements.txt
|
||||
log_info "依赖安装完成"
|
||||
}
|
||||
|
||||
# 数据库迁移
|
||||
run_migrations() {
|
||||
log_info "运行数据库迁移..."
|
||||
|
||||
# 检查数据库文件
|
||||
if [ ! -f "tsp_assistant.db" ]; then
|
||||
log_info "初始化数据库..."
|
||||
python init_database.py
|
||||
fi
|
||||
|
||||
log_info "数据库迁移完成"
|
||||
}
|
||||
|
||||
# 创建systemd服务文件
|
||||
create_systemd_service() {
|
||||
local service_name=$1
|
||||
local app_path=$2
|
||||
local service_file="/etc/systemd/system/${service_name}.service"
|
||||
|
||||
log_info "创建systemd服务文件: $service_file"
|
||||
|
||||
sudo tee "$service_file" > /dev/null <<EOF
|
||||
[Unit]
|
||||
Description=TSP智能助手服务
|
||||
After=network.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
User=www-data
|
||||
Group=www-data
|
||||
WorkingDirectory=$app_path
|
||||
Environment=PATH=$app_path/venv/bin
|
||||
ExecStart=$app_path/venv/bin/python start_dashboard.py
|
||||
Restart=always
|
||||
RestartSec=10
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
EOF
|
||||
|
||||
sudo systemctl daemon-reload
|
||||
sudo systemctl enable "$service_name"
|
||||
log_info "systemd服务创建完成"
|
||||
}
|
||||
|
||||
# 创建nginx配置
|
||||
create_nginx_config() {
|
||||
local domain=$1
|
||||
local app_port=$2
|
||||
local config_file="/etc/nginx/sites-available/tsp_assistant"
|
||||
|
||||
log_info "创建nginx配置: $config_file"
|
||||
|
||||
sudo tee "$config_file" > /dev/null <<EOF
|
||||
server {
|
||||
listen 80;
|
||||
server_name $domain;
|
||||
|
||||
location / {
|
||||
proxy_pass http://127.0.0.1:$app_port;
|
||||
proxy_set_header Host \$host;
|
||||
proxy_set_header X-Real-IP \$remote_addr;
|
||||
proxy_set_header X-Forwarded-For \$proxy_add_x_forwarded_for;
|
||||
proxy_set_header X-Forwarded-Proto \$scheme;
|
||||
}
|
||||
|
||||
location /static {
|
||||
alias $app_path/src/web/static;
|
||||
expires 1y;
|
||||
add_header Cache-Control "public, immutable";
|
||||
}
|
||||
}
|
||||
EOF
|
||||
|
||||
# 启用站点
|
||||
sudo ln -sf "$config_file" /etc/nginx/sites-enabled/
|
||||
sudo nginx -t
|
||||
sudo systemctl reload nginx
|
||||
log_info "nginx配置完成"
|
||||
}
|
||||
|
||||
# 主部署函数
|
||||
deploy() {
|
||||
local environment=${1:-production}
|
||||
local domain=${2:-localhost}
|
||||
local app_port=${3:-5000}
|
||||
|
||||
log_info "开始部署TSP智能助手到 $environment 环境"
|
||||
|
||||
# 设置部署路径
|
||||
case $environment in
|
||||
development)
|
||||
DEPLOY_PATH="./dev_deploy"
|
||||
SERVICE_NAME=""
|
||||
;;
|
||||
staging)
|
||||
DEPLOY_PATH="/opt/tsp_assistant_staging"
|
||||
SERVICE_NAME="tsp_assistant_staging"
|
||||
;;
|
||||
production)
|
||||
DEPLOY_PATH="/opt/tsp_assistant"
|
||||
SERVICE_NAME="tsp_assistant"
|
||||
;;
|
||||
*)
|
||||
log_error "未知环境: $environment"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
# 检查依赖
|
||||
check_dependencies
|
||||
|
||||
# 创建部署目录
|
||||
log_info "创建部署目录: $DEPLOY_PATH"
|
||||
sudo mkdir -p "$DEPLOY_PATH"
|
||||
sudo chown $USER:$USER "$DEPLOY_PATH"
|
||||
|
||||
# 复制文件
|
||||
log_info "复制应用文件..."
|
||||
cp -r . "$DEPLOY_PATH/"
|
||||
cd "$DEPLOY_PATH"
|
||||
|
||||
# 设置虚拟环境
|
||||
setup_venv "venv"
|
||||
|
||||
# 安装依赖
|
||||
install_dependencies
|
||||
|
||||
# 运行迁移
|
||||
run_migrations
|
||||
|
||||
# 创建服务文件（非开发环境）
|
||||
if [ "$environment" != "development" ] && [ -n "$SERVICE_NAME" ]; then
|
||||
create_systemd_service "$SERVICE_NAME" "$DEPLOY_PATH"
|
||||
create_nginx_config "$domain" "$app_port"
|
||||
fi
|
||||
|
||||
log_info "部署完成！"
|
||||
|
||||
if [ "$environment" != "development" ]; then
|
||||
log_info "启动服务..."
|
||||
sudo systemctl start "$SERVICE_NAME"
|
||||
sudo systemctl status "$SERVICE_NAME"
|
||||
else
|
||||
log_info "开发环境部署完成，使用以下命令启动："
|
||||
log_info "cd $DEPLOY_PATH && source venv/bin/activate && python start_dashboard.py"
|
||||
fi
|
||||
}
|
||||
|
||||
# 回滚函数
|
||||
rollback() {
|
||||
local backup_name=$1
|
||||
|
||||
if [ -z "$backup_name" ]; then
|
||||
log_error "请指定备份名称"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "回滚到备份: $backup_name"
|
||||
|
||||
# 停止服务
|
||||
sudo systemctl stop tsp_assistant
|
||||
|
||||
# 恢复备份
|
||||
if [ -d "backups/$backup_name" ]; then
|
||||
sudo rm -rf /opt/tsp_assistant
|
||||
sudo cp -r "backups/$backup_name" /opt/tsp_assistant
|
||||
sudo chown -R www-data:www-data /opt/tsp_assistant
|
||||
|
||||
# 重启服务
|
||||
sudo systemctl start tsp_assistant
|
||||
log_info "回滚完成"
|
||||
else
|
||||
log_error "备份不存在: $backup_name"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# 主函数
|
||||
main() {
|
||||
case ${1:-deploy} in
|
||||
deploy)
|
||||
deploy "$2" "$3" "$4"
|
||||
;;
|
||||
rollback)
|
||||
rollback "$2"
|
||||
;;
|
||||
*)
|
||||
echo "用法: $0 {deploy|rollback} [environment] [domain] [port]"
|
||||
echo "环境: development, staging, production"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
main "$@"
|
||||
Reference in New Issue
Block a user