venvworkon.sh

workon() {
    # Sets WORKON_HOME to default if not set already
    if [[ -z $WORKON_HOME ]]; then
        WORKON_HOME="$HOME/.venv"
    fi

    # Creates WORKON_HOME directory if not exists
    if [[ ! -e $WORKON_HOME ]]; then
        mkdir -p $WORKON_HOME
        echo "Virtual env directory created in $WORKON_HOME"
    fi

    # Quit if there is no arguments, while showing available venvs
    if [[ "$#" -lt "1" ]]; then
        echo "Usage: workon ENV_NAME PYTHON_VERSION"
        echo "Available envs:"
        ls "$WORKON_HOME"
        return
    fi

    envdir="$WORKON_HOME/$1"
    python_version="3"
    projectdir=""

    if [[ "$1" == "." ]]; then
        target=$(basename `pwd`)
        envdir="$WORKON_HOME/$target"
        projectdir=$(pwd)
    fi

    if [[ "$#" -gt "1" ]]; then
        python_version=$2
    fi

    python_exec="python$python_version"

    # Creates virtual env if doesn't exists
    if [[ ! -e "$envdir" ]]; then
        $python_exec -m venv $envdir

        if [[ "$projectdir" != "" ]]; then
            echo "projectdir=$projectdir" > "$envdir/postactivate.sh"
        fi

        echo "Virtual env created in $envdir"
    fi

    source "$envdir/bin/activate"
    unset envdir  # VIRTUAL_ENV is now set by bin/activate

    # Source postactivate.sh if exists in the virtual env
    if [[ -e "$VIRTUAL_ENV/postactivate.sh" ]]; then
        source "$VIRTUAL_ENV/postactivate.sh"
    fi

    if [[ -e "$projectdir/.env" ]]; then
        . "$projectdir/.env"
    fi

    # Jumps to project directory if exists and PROJECT_HOME is set
    projecthome="$PROJECT_HOME/$(basename $VIRTUAL_ENV)"
    if [[ -d "$projecthome" ]]; then
        cd "$projecthome"

        if [[ -e "$projecthome/.env" ]]; then
            . "$projecthome/.env"
        fi
    fi
}