127 lines
3.2 KiB
Bash
127 lines
3.2 KiB
Bash
#!/bin/bash
|
||
# FastAPI 应用启动脚本 (使用 uv 包管理)
|
||
|
||
set -e
|
||
|
||
echo "========================================"
|
||
echo "启动 FastAPI 时间序列分析系统 v2.0"
|
||
echo "========================================"
|
||
echo ""
|
||
|
||
# 检查 uv
|
||
if ! command -v /home/syy/.local/bin/uv &> /dev/null; then
|
||
echo "错误: 未找到 uv"
|
||
exit 1
|
||
fi
|
||
|
||
echo "✓ uv 已安装"
|
||
echo ""
|
||
|
||
# 检查虚拟环境，如果不存在则创建
|
||
if [ ! -d ".venv" ]; then
|
||
echo "创建虚拟环境..."
|
||
/home/syy/.local/bin/uv venv --python 3.10
|
||
fi
|
||
|
||
# 激活虚拟环境
|
||
echo "激活虚拟环境..."
|
||
source .venv/bin/activate
|
||
|
||
# 加载 .env（不覆盖已存在的环境变量）
|
||
if [ -f ".env" ]; then
|
||
echo "加载 .env..."
|
||
while IFS=$'\t' read -r key quoted_value; do
|
||
[ -z "$key" ] && continue
|
||
# 仅允许合法的环境变量名
|
||
if [[ ! "$key" =~ ^[A-Za-z_][A-Za-z0-9_]*$ ]]; then
|
||
continue
|
||
fi
|
||
# 已在环境中显式设置的变量优先
|
||
if [ -z "${!key+x}" ]; then
|
||
eval "export ${key}=${quoted_value}"
|
||
fi
|
||
done < <(python - <<'PY'
|
||
import os
|
||
import shlex
|
||
|
||
try:
|
||
from dotenv import dotenv_values
|
||
except Exception:
|
||
dotenv_values = None
|
||
|
||
if dotenv_values is None:
|
||
raise SystemExit(0)
|
||
|
||
values = dotenv_values('.env')
|
||
for k, v in values.items():
|
||
if k is None or v is None:
|
||
continue
|
||
# 输出：KEY<TAB>shell_quoted_value
|
||
print(f"{k}\t{shlex.quote(str(v))}")
|
||
PY
|
||
)
|
||
echo "✓ .env 加载完成"
|
||
echo ""
|
||
fi
|
||
|
||
# 检查并安装依赖
|
||
echo "检查依赖..."
|
||
python -c "import fastapi; import uvicorn; import pandas; import numpy" 2>/dev/null || {
|
||
echo "安装依赖..."
|
||
/home/syy/.local/bin/uv pip install \
|
||
'fastapi>=0.104.1' \
|
||
'uvicorn[standard]>=0.24.0' \
|
||
'python-multipart>=0.0.6' \
|
||
'python-dotenv>=1.0.0' \
|
||
'pandas>=2.2.2' \
|
||
'numpy>=1.26.4' \
|
||
'scipy>=1.13.0' \
|
||
'scikit-learn>=1.3.0' \
|
||
'statsmodels>=0.14.0' \
|
||
'matplotlib>=3.7.2' \
|
||
'seaborn>=0.12.2' \
|
||
'openai>=1.3.0' \
|
||
'gradio_client>=0.9.0' \
|
||
'requests>=2.31.0' \
|
||
'psutil>=5.9.5'
|
||
}
|
||
|
||
echo "✓ 依赖检查完成"
|
||
echo ""
|
||
|
||
# 创建必要的目录
|
||
mkdir -p uploads logs temp resource/fonts
|
||
|
||
# 设置环境变量（如果没有设置）
|
||
export ENV=${ENV:-"development"}
|
||
export DEBUG=${DEBUG:-"False"}
|
||
export HOST=${HOST:-"0.0.0.0"}
|
||
export PORT=${PORT:-"60201"}
|
||
export LOG_LEVEL=${LOG_LEVEL:-"INFO"}
|
||
|
||
echo "环境配置:"
|
||
echo " ENV=$ENV"
|
||
echo " DEBUG=$DEBUG"
|
||
echo " HOST=$HOST"
|
||
echo " PORT=$PORT"
|
||
echo " LOG_LEVEL=$LOG_LEVEL"
|
||
echo ""
|
||
|
||
# 启动应用
|
||
echo "启动应用..."
|
||
echo ""
|
||
echo "=================================="
|
||
echo "✓ 访问地址: http://localhost:$PORT"
|
||
echo "✓ API 文档: http://localhost:$PORT/docs"
|
||
echo "✓ ReDoc: http://localhost:$PORT/redoc"
|
||
echo "=================================="
|
||
echo ""
|
||
echo "按 Ctrl+C 停止应用"
|
||
echo ""
|
||
|
||
# 使用 uvicorn 运行
|
||
python -m uvicorn app.main:app \
|
||
--host $HOST \
|
||
--port $PORT \
|
||
--log-level $(echo $LOG_LEVEL | tr '[:upper:]' '[:lower:]')
|