#!/bin/bash
# 交易系统依赖安装脚本

cd "$(dirname "$0")"

# 检查虚拟环境是否存在
if [ -d "../.venv" ]; then
    VENV_PATH="../.venv"
    echo "使用项目根目录的虚拟环境: $VENV_PATH"
elif [ -d ".venv" ]; then
    VENV_PATH=".venv"
    echo "使用trading_system目录的虚拟环境: $VENV_PATH"
else
    # 创建虚拟环境（优先在项目根目录）
    if [ -d ".." ]; then
        VENV_PATH="../.venv"
        echo "在项目根目录创建虚拟环境: $VENV_PATH"
        python3 -m venv "$VENV_PATH"
    else
        VENV_PATH=".venv"
        echo "在trading_system目录创建虚拟环境: $VENV_PATH"
        python3 -m venv "$VENV_PATH"
    fi
fi

# 激活虚拟环境
source "$VENV_PATH/bin/activate"

# 升级pip
pip install --upgrade pip

# 安装依赖
echo "安装交易系统依赖..."
pip install -r requirements.txt

echo ""
echo "✓ 依赖安装完成！"
echo ""
echo "使用虚拟环境运行："
echo "  source $VENV_PATH/bin/activate"
echo "  python main.py"