Python Virtual Environment Helpers

1# Add to ~/.bashrc or ~/.zshrc
2
3# Create and activate a Python virtual environment
4venv() {
5    local env_name="${1:-venv}"
6    
7    if [ -d "$env_name" ]; then
8        echo "Activating existing environment: $env_name"
9        source "$env_name/bin/activate"
10    else
11        echo "Creating new environment: $env_name"
12        python3 -m venv "$env_name"
13        source "$env_name/bin/activate"
14        pip install --upgrade pip
15    fi
16}
17
18# Deactivate and remove virtual environment
19venv-remove() {
20    local env_name="${1:-venv}"
21    
22    if [ -n "$VIRTUAL_ENV" ]; then
23        deactivate
24    fi
25    
26    if [ -d "$env_name" ]; then
27        rm -rf "$env_name"
28        echo "Removed environment: $env_name"
29    else
30        echo "Environment not found: $env_name"
31    fi
32}