Files

64 lines
1.7 KiB
Bash
Executable File

#!/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 ""