64 lines
1.7 KiB
Bash
Executable File
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 ""
|