activate_env.sh

#!/bin/bash
# 虚拟环境激活脚本 / Virtual Environment Activation Script

show_usage() {
    echo "使用方法 / Usage: $0 [level]"
    echo ""
    echo "可用级别 / Available levels:"
    echo "  01 或 beginner    - 激活初级RAG环境"
    echo "  02 或 intermediate - 激活中级RAG环境"
    echo ""
    echo "示例 / Examples:"
    echo "  $0 01              # 激活初级环境"
    echo "  $0 beginner        # 激活初级环境"
    echo "  $0 02              # 激活中级环境"
    echo "  $0 intermediate    # 激活中级环境"
    echo ""
    echo "注意：此脚本需要使用 source 命令运行"
    echo "Note: This script needs to be run with 'source' command"
    echo "例如 / Example: source activate_env.sh 01"
}

activate_environment() {
    local level=$1
    local env_path=""
    local level_name=""
    
    case $level in
        "01"|"beginner")
            env_path="01-beginner/rag_01_env"
            level_name="01-Beginner"
            ;;
        "02"|"intermediate")
            env_path="02-intermediate/rag_02_env"
            level_name="02-Intermediate"
            ;;
        *)
            echo "❌ 未知级别: $level"
            echo "❌ Unknown level: $level"
            show_usage
            return 1
            ;;
    esac
    
    # 检查环境是否存在
    if [ ! -d "$env_path" ]; then
        echo "❌ 虚拟环境不存在: $env_path"
        echo "❌ Virtual environment not found: $env_path"
        echo ""
        echo "请先创建环境 / Please create the environment first:"
        echo "cd $(dirname $env_path) && python -m venv $(basename $env_path)"
        return 1
    fi
    
    # 检查激活脚本是否存在
    if [ ! -f "$env_path/bin/activate" ]; then
        echo "❌ 激活脚本不存在: $env_path/bin/activate"
        echo "❌ Activation script not found: $env_path/bin/activate"
        return 1
    fi
    
    # 激活环境
    echo "🚀 激活 $level_name 环境..."
    echo "🚀 Activating $level_name environment..."
    source "$env_path/bin/activate"
    
    # 验证激活是否成功
    if [ "$VIRTUAL_ENV" = "$(pwd)/$env_path" ]; then
        echo "✅ 成功激活: $level_name"
        echo "✅ Successfully activated: $level_name"
        echo "📍 环境路径: $VIRTUAL_ENV"
        echo "📍 Environment path: $VIRTUAL_ENV"
        echo "🐍 Python版本: $(python --version)"
        echo ""
        echo "💡 提示 / Tips:"
        echo "   - 使用 'deactivate' 退出虚拟环境"
        echo "   - Use 'deactivate' to exit the virtual environment"
        echo "   - 使用 'pip list' 查看已安装的包"
        echo "   - Use 'pip list' to see installed packages"
    else
        echo "❌ 激活失败"
        echo "❌ Activation failed"
        return 1
    fi
}

# 主逻辑
if [ $# -eq 0 ]; then
    echo "❌ 请指定要激活的环境级别"
    echo "❌ Please specify the environment level to activate"
    show_usage
    exit 1
fi

# 检查是否使用source运行
if [ "${BASH_SOURCE[0]}" = "${0}" ]; then
    echo "⚠️  警告: 请使用 'source' 命令运行此脚本"
    echo "⚠️  Warning: Please run this script with 'source' command"
    echo ""
    echo "正确用法 / Correct usage:"
    echo "source $0 $1"
    exit 1
fi

activate_environment $1