initial-import: 2026-04-11 21:18:55

scripts/utils/setup_env.sh

@@ -0,0 +1,63 @@
+#!/bin/bash
+# 量化交易系统环境设置脚本
+# 使用方法: source setup_env.sh
+
+# 获取脚本所在目录
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+cd "$SCRIPT_DIR"
+
+echo "=========================================="
+echo "  量化交易系统 - 环境初始化"
+echo "=========================================="
+echo ""
+
+# 检查虚拟环境是否存在
+if [ ! -d "vnpy_env" ]; then
+    echo "⚠️  虚拟环境不存在，正在创建..."
+    python3 -m venv vnpy_env
+    echo "✅ 虚拟环境创建成功"
+fi
+
+# 激活虚拟环境
+echo "🔧 激活虚拟环境..."
+source vnpy_env/bin/activate
+
+# 升级 pip
+echo "🔧 升级 pip..."
+pip install --upgrade pip -q
+
+# 检查依赖是否安装
+if [ ! -f "vnpy_env/.dependencies_installed" ]; then
+    echo "📦 安装项目依赖..."
+    pip install -r requirements.txt
+    touch vnpy_env/.dependencies_installed
+    echo "✅ 依赖安装完成"
+else
+    echo "✅ 依赖已安装"
+fi
+
+# 检查必要的目录结构
+echo "📂 检查目录结构..."
+mkdir -p vnpy_project/{logs,data,strategies,backup}
+echo "✅ 目录结构检查完成"
+
+# 设置环境变量
+export QUANT_ENV=development
+export QUANT_DB_TYPE=sqlite
+export PYTHONPATH="$SCRIPT_DIR:$PYTHONPATH"
+
+echo ""
+echo "=========================================="
+echo "  ✅ 环境初始化完成！"
+echo "=========================================="
+echo ""
+echo "📌 环境变量已设置："
+echo "   - QUANT_ENV: $QUANT_ENV"
+echo "   - QUANT_DB_TYPE: $QUANT_DB_TYPE"
+echo "   - PYTHONPATH: $PYTHONPATH"
+echo ""
+echo "📌 常用命令："
+echo "   - 运行系统: python main.py"
+echo "   - 测试数据库: python test_database.py"
+echo "   - 退出虚拟环境: deactivate"
+echo ""