Python 3 venv wrapper. Manages all virtual environments under ~/.venv/ .

venv_wrapper

# venv_wrapper, manage all virtual environments under ~/.venv/
# Include the following in .bashrc / .bash_profile / .zshrc
# See https://gist.github.com/dbtek/fb2ddccb18f0cf63a654ea2cc94c8f19
#
# Usage:
# $ mkvenv myvirtualenv   # creates venv under ~/.venv/
# $ venv myvirtualenv     # activates venv
# $ deactivate            # deactivates venv
# $ rmvenv myvirtualenv   # removes venv
# $ rmvenv env1 env2      # removes multiple venvs
# $ lsvenv                # lists all venvs

export VENV_HOME="$HOME/.venv"
[[ -d $VENV_HOME ]] || mkdir -p $VENV_HOME

lsvenv() {
  ls -1 "$VENV_HOME"
}

venv() {
  if [ $# -eq 0 ]; then
    echo "Please provide venv name"
  elif [ -d "$VENV_HOME/$1" ]; then
    source "$VENV_HOME/$1/bin/activate"
  else
    echo "Virtual environment '$1' does not exist."
  fi
}

mkvenv() {
  if [ $# -eq 0 ]; then
    echo "Please provide venv name"
  else
    if [ -d "$VENV_HOME/$1" ]; then
      echo "Virtual environment '$1' already exists."
    else
      echo "Creating venv under $VENV_HOME/$1"
      python3 -m venv "$VENV_HOME/$1"
      echo "Activating $1"
      venv "$1"
    fi
  fi
}

rmvenv() {
  if [ $# -eq 0 ]; then
    echo "Please provide one or more venv names"
  else
    current_venv=$(basename "${VIRTUAL_ENV:-}")
    for env in "$@"; do
      if [ -d "$VENV_HOME/$env" ]; then
        if [ "$env" = "$current_venv" ]; then
          deactivate
          echo "Deactivated current virtual environment '$env'."
        fi
        rm -rf "$VENV_HOME/$env"
        echo "Virtual environment '$env' removed."
      else
        echo "Virtual environment '$env' does not exist."
      fi
    done
  fi
}

complete -C lsvenv venv
complete -C lsvenv rmvenv

Changes and Improvements:

Error Checking:

• Added mkdir -p $VENV_HOME to ensure the directory structure is created if it doesn't exist.
• Added checks to ensure that the virtual environment exists before attempting to activate or remove it.

User Feedback:

• Added feedback messages to inform the user about the creation and deletion of virtual environments.

Multiple Environment Deletion:

• Enhanced rmvenv function to accept multiple arguments, allowing deletion of multiple virtual environments in a single command.
• Added loop in rmvenv to iterate over all provided virtual environment names, checking for their existence and deleting them if they exist.

Deactivation Before Deletion:

• Added logic to deactivate the currently active virtual environment if it is one of the environments being deleted.

Improved Comments:

• Enhanced script comments for better clarity and understanding of usage.