source/AID2E-framework/setup_aid2e.sh

0001 #!/bin/bash
0002 # AID2E Framework - Environment Setup Script
0003 # This script activates the virtual environment and sets up PYTHONPATH
0004
0005 # Get the directory where this script is located
0006 current_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
0007
0008 # Activate virtual environment if it exists
0009 if [ -f "$current_dir/.venv/bin/activate" ]; then
0010     echo "Activating virtual environment..."
0011     source "$current_dir/.venv/bin/activate"
0012 else
0013     echo "Warning: Virtual environment not found at $current_dir/.venv"
0014     echo "Please create it with: python3 -m venv .venv"
0015 fi
0016
0017 # Set up PYTHONPATH to include src, tests, examples, local deps, and current directory
0018 if [ -z "$PYTHONPATH" ]; then
0019     # PYTHONPATH is empty or unset, create new one
0020     export PYTHONPATH="$current_dir/src:$current_dir/tests:$current_dir/examples:$current_dir/.local/lib:$current_dir"
0021 else
0022     # PYTHONPATH exists, append to it
0023     export PYTHONPATH="$current_dir/src:$current_dir/tests:$current_dir/examples:$current_dir/.local/lib:$current_dir:$PYTHONPATH"
0024 fi
0025
0026 echo "Environment setup complete!"
0027 echo "PYTHONPATH: $PYTHONPATH"
0028 echo ""
0029 echo "You can now run AID2E commands and scripts."