mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1359 lines
41 KiB
Bash
1359 lines
41 KiB
Bash
4 years ago
|
# -*- mode: shell-script -*-
|
||
|
#
|
||
|
# Shell functions to act as wrapper for Ian Bicking's virtualenv
|
||
|
# (http://pypi.python.org/pypi/virtualenv)
|
||
|
#
|
||
|
#
|
||
|
# Copyright Doug Hellmann, All Rights Reserved
|
||
|
#
|
||
|
# Permission to use, copy, modify, and distribute this software and its
|
||
|
# documentation for any purpose and without fee is hereby granted,
|
||
|
# provided that the above copyright notice appear in all copies and that
|
||
|
# both that copyright notice and this permission notice appear in
|
||
|
# supporting documentation, and that the name of Doug Hellmann not be used
|
||
|
# in advertising or publicity pertaining to distribution of the software
|
||
|
# without specific, written prior permission.
|
||
|
#
|
||
|
# DOUG HELLMANN DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
|
||
|
# INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO
|
||
|
# EVENT SHALL DOUG HELLMANN BE LIABLE FOR ANY SPECIAL, INDIRECT OR
|
||
|
# CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF
|
||
|
# USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
||
|
# OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
||
|
# PERFORMANCE OF THIS SOFTWARE.
|
||
|
#
|
||
|
#
|
||
|
# Project home page: http://www.doughellmann.com/projects/virtualenvwrapper/
|
||
|
#
|
||
|
#
|
||
|
# Setup:
|
||
|
#
|
||
|
# 1. Create a directory to hold the virtual environments.
|
||
|
# (mkdir $HOME/.virtualenvs).
|
||
|
# 2. Add a line like "export WORKON_HOME=$HOME/.virtualenvs"
|
||
|
# to your .bashrc.
|
||
|
# 3. Add a line like "source /path/to/this/file/virtualenvwrapper.sh"
|
||
|
# to your .bashrc.
|
||
|
# 4. Run: source ~/.bashrc
|
||
|
# 5. Run: workon
|
||
|
# 6. A list of environments, empty, is printed.
|
||
|
# 7. Run: mkvirtualenv temp
|
||
|
# 8. Run: workon
|
||
|
# 9. This time, the "temp" environment is included.
|
||
|
# 10. Run: workon temp
|
||
|
# 11. The virtual environment is activated.
|
||
|
#
|
||
|
|
||
|
# Locate the global Python where virtualenvwrapper is installed.
|
||
|
if [ "${VIRTUALENVWRAPPER_PYTHON:-}" = "" ]
|
||
|
then
|
||
|
VIRTUALENVWRAPPER_PYTHON="$(command \which python)"
|
||
|
fi
|
||
|
|
||
|
# Set the name of the virtualenv app to use.
|
||
|
if [ "${VIRTUALENVWRAPPER_VIRTUALENV:-}" = "" ]
|
||
|
then
|
||
|
VIRTUALENVWRAPPER_VIRTUALENV="virtualenv"
|
||
|
fi
|
||
|
|
||
|
# Set the name of the virtualenv-clone app to use.
|
||
|
if [ "${VIRTUALENVWRAPPER_VIRTUALENV_CLONE:-}" = "" ]
|
||
|
then
|
||
|
VIRTUALENVWRAPPER_VIRTUALENV_CLONE="virtualenv-clone"
|
||
|
fi
|
||
|
|
||
|
# Define script folder depending on the platorm (Win32/Unix)
|
||
|
VIRTUALENVWRAPPER_ENV_BIN_DIR="bin"
|
||
|
if [ "${OS:-}" = "Windows_NT" ] && ([ "${MSYSTEM:-}" = "MINGW32" ] || [ "${MSYSTEM:-}" = "MINGW64" ])
|
||
|
then
|
||
|
# Only assign this for msys, cygwin use standard Unix paths
|
||
|
# and its own python installation
|
||
|
VIRTUALENVWRAPPER_ENV_BIN_DIR="Scripts"
|
||
|
fi
|
||
|
|
||
|
# Let the user override the name of the file that holds the project
|
||
|
# directory name.
|
||
|
if [ "${VIRTUALENVWRAPPER_PROJECT_FILENAME:-}" = "" ]
|
||
|
then
|
||
|
export VIRTUALENVWRAPPER_PROJECT_FILENAME=".project"
|
||
|
fi
|
||
|
|
||
|
# Let the user tell us they never want to cd to projects
|
||
|
# automatically.
|
||
|
export VIRTUALENVWRAPPER_WORKON_CD=${VIRTUALENVWRAPPER_WORKON_CD:-1}
|
||
|
|
||
|
# Remember where we are running from.
|
||
|
if [ -z "${VIRTUALENVWRAPPER_SCRIPT:-}" ]
|
||
|
then
|
||
|
if [ -n "$BASH" ]
|
||
|
then
|
||
|
export VIRTUALENVWRAPPER_SCRIPT="$BASH_SOURCE"
|
||
|
elif [ -n "$ZSH_VERSION" ]
|
||
|
then
|
||
|
export VIRTUALENVWRAPPER_SCRIPT="$0"
|
||
|
else
|
||
|
export VIRTUALENVWRAPPER_SCRIPT="${.sh.file}"
|
||
|
fi
|
||
|
fi
|
||
|
|
||
|
# Portable shell scripting is hard, let's go shopping.
|
||
|
#
|
||
|
# People insist on aliasing commands like 'cd', either with a real
|
||
|
# alias or even a shell function. Under bash and zsh, "builtin" forces
|
||
|
# the use of a command that is part of the shell itself instead of an
|
||
|
# alias, function, or external command, while "command" does something
|
||
|
# similar but allows external commands. Under ksh "builtin" registers
|
||
|
# a new command from a shared library, but "command" will pick up
|
||
|
# existing builtin commands. We need to use a builtin for cd because
|
||
|
# we are trying to change the state of the current shell, so we use
|
||
|
# "builtin" for bash and zsh but "command" under ksh.
|
||
|
function virtualenvwrapper_cd {
|
||
|
if [ -n "${BASH:-}" ]
|
||
|
then
|
||
|
builtin \cd "$@"
|
||
|
elif [ -n "${ZSH_VERSION:-}" ]
|
||
|
then
|
||
|
builtin \cd -q "$@"
|
||
|
else
|
||
|
command \cd "$@"
|
||
|
fi
|
||
|
}
|
||
|
|
||
|
function virtualenvwrapper_expandpath {
|
||
|
if [ "$1" = "" ]; then
|
||
|
return 1
|
||
|
else
|
||
|
"$VIRTUALENVWRAPPER_PYTHON" -c "import os,sys; sys.stdout.write(os.path.normpath(os.path.expanduser(os.path.expandvars(\"$1\")))+'\n')"
|
||
|
return 0
|
||
|
fi
|
||
|
}
|
||
|
|
||
|
function virtualenvwrapper_absolutepath {
|
||
|
if [ "$1" = "" ]; then
|
||
|
return 1
|
||
|
else
|
||
|
"$VIRTUALENVWRAPPER_PYTHON" -c "import os,sys; sys.stdout.write(os.path.abspath(\"$1\")+'\n')"
|
||
|
return 0
|
||
|
fi
|
||
|
}
|
||
|
|
||
|
function virtualenvwrapper_derive_workon_home {
|
||
|
typeset workon_home_dir="$WORKON_HOME"
|
||
|
|
||
|
# Make sure there is a default value for WORKON_HOME.
|
||
|
# You can override this setting in your .bashrc.
|
||
|
if [ "$workon_home_dir" = "" ]
|
||
|
then
|
||
|
workon_home_dir="$HOME/.virtualenvs"
|
||
|
fi
|
||
|
|
||
|
# If the path is relative, prefix it with $HOME
|
||
|
# (note: for compatibility)
|
||
|
if echo "$workon_home_dir" | (unset GREP_OPTIONS; command \grep '^[^/~]' > /dev/null)
|
||
|
then
|
||
|
workon_home_dir="$HOME/$WORKON_HOME"
|
||
|
fi
|
||
|
|
||
|
# Only call on Python to fix the path if it looks like the
|
||
|
# path might contain stuff to expand.
|
||
|
# (it might be possible to do this in shell, but I don't know a
|
||
|
# cross-shell-safe way of doing it -wolever)
|
||
|
if echo "$workon_home_dir" | (unset GREP_OPTIONS; command \egrep '([\$~]|//)' >/dev/null)
|
||
|
then
|
||
|
# This will normalize the path by:
|
||
|
# - Removing extra slashes (e.g., when TMPDIR ends in a slash)
|
||
|
# - Expanding variables (e.g., $foo)
|
||
|
# - Converting ~s to complete paths (e.g., ~/ to /home/brian/ and ~arthur to /home/arthur)
|
||
|
workon_home_dir="$(virtualenvwrapper_expandpath "$workon_home_dir")"
|
||
|
fi
|
||
|
|
||
|
echo "$workon_home_dir"
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
# Check if the WORKON_HOME directory exists,
|
||
|
# create it if it does not
|
||
|
# seperate from creating the files in it because this used to just error
|
||
|
# and maybe other things rely on the dir existing before that happens.
|
||
|
function virtualenvwrapper_verify_workon_home {
|
||
|
RC=0
|
||
|
if [ ! -d "$WORKON_HOME/" ]
|
||
|
then
|
||
|
if [ "$1" != "-q" ]
|
||
|
then
|
||
|
echo "NOTE: Virtual environments directory $WORKON_HOME does not exist. Creating..." 1>&2
|
||
|
fi
|
||
|
mkdir -p "$WORKON_HOME"
|
||
|
RC=$?
|
||
|
fi
|
||
|
return $RC
|
||
|
}
|
||
|
|
||
|
#HOOK_VERBOSE_OPTION="-q"
|
||
|
|
||
|
# Function to wrap mktemp so tests can replace it for error condition
|
||
|
# testing.
|
||
|
function virtualenvwrapper_mktemp {
|
||
|
command \mktemp "$@"
|
||
|
}
|
||
|
|
||
|
# Expects 1 argument, the suffix for the new file.
|
||
|
function virtualenvwrapper_tempfile {
|
||
|
# Note: the 'X's must come last
|
||
|
typeset suffix=${1:-hook}
|
||
|
typeset file
|
||
|
|
||
|
file="$(virtualenvwrapper_mktemp -t virtualenvwrapper-$suffix-XXXXXXXXXX)"
|
||
|
touch "$file"
|
||
|
if [ $? -ne 0 ] || [ -z "$file" ] || [ ! -f "$file" ]
|
||
|
then
|
||
|
echo "ERROR: virtualenvwrapper could not create a temporary file name." 1>&2
|
||
|
return 1
|
||
|
fi
|
||
|
echo $file
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
# Run the hooks
|
||
|
function virtualenvwrapper_run_hook {
|
||
|
typeset hook_script
|
||
|
typeset result
|
||
|
|
||
|
hook_script="$(virtualenvwrapper_tempfile ${1}-hook)" || return 1
|
||
|
|
||
|
# Use a subshell to run the python interpreter with hook_loader so
|
||
|
# we can change the working directory. This avoids having the
|
||
|
# Python 3 interpreter decide that its "prefix" is the virtualenv
|
||
|
# if we happen to be inside the virtualenv when we start.
|
||
|
( \
|
||
|
virtualenvwrapper_cd "$WORKON_HOME" &&
|
||
|
"$VIRTUALENVWRAPPER_PYTHON" -m 'virtualenvwrapper.hook_loader' \
|
||
|
${HOOK_VERBOSE_OPTION:-} --script "$hook_script" "$@" \
|
||
|
)
|
||
|
result=$?
|
||
|
|
||
|
if [ $result -eq 0 ]
|
||
|
then
|
||
|
if [ ! -f "$hook_script" ]
|
||
|
then
|
||
|
echo "ERROR: virtualenvwrapper_run_hook could not find temporary file $hook_script" 1>&2
|
||
|
command \rm -f "$hook_script"
|
||
|
return 2
|
||
|
fi
|
||
|
# cat "$hook_script"
|
||
|
source "$hook_script"
|
||
|
elif [ "${1}" = "initialize" ]
|
||
|
then
|
||
|
cat - 1>&2 <<EOF
|
||
|
virtualenvwrapper.sh: There was a problem running the initialization hooks.
|
||
|
|
||
|
If Python could not import the module virtualenvwrapper.hook_loader,
|
||
|
check that virtualenvwrapper has been installed for
|
||
|
VIRTUALENVWRAPPER_PYTHON=$VIRTUALENVWRAPPER_PYTHON and that PATH is
|
||
|
set properly.
|
||
|
EOF
|
||
|
fi
|
||
|
command \rm -f "$hook_script"
|
||
|
return $result
|
||
|
}
|
||
|
|
||
|
# Set up tab completion. (Adapted from Arthur Koziel's version at
|
||
|
# http://arthurkoziel.com/2008/10/11/virtualenvwrapper-bash-completion/)
|
||
|
function virtualenvwrapper_setup_tab_completion {
|
||
|
if [ -n "${BASH:-}" ] ; then
|
||
|
_virtualenvs () {
|
||
|
local cur="${COMP_WORDS[COMP_CWORD]}"
|
||
|
COMPREPLY=( $(compgen -W "`virtualenvwrapper_show_workon_options`" -- ${cur}) )
|
||
|
}
|
||
|
_cdvirtualenv_complete () {
|
||
|
local cur="$2"
|
||
|
COMPREPLY=( $(cdvirtualenv && compgen -d -- "${cur}" ) )
|
||
|
}
|
||
|
_cdsitepackages_complete () {
|
||
|
local cur="$2"
|
||
|
COMPREPLY=( $(cdsitepackages && compgen -d -- "${cur}" ) )
|
||
|
}
|
||
|
complete -o nospace -F _cdvirtualenv_complete -S/ cdvirtualenv
|
||
|
complete -o nospace -F _cdsitepackages_complete -S/ cdsitepackages
|
||
|
complete -o default -o nospace -F _virtualenvs workon
|
||
|
complete -o default -o nospace -F _virtualenvs rmvirtualenv
|
||
|
complete -o default -o nospace -F _virtualenvs cpvirtualenv
|
||
|
complete -o default -o nospace -F _virtualenvs showvirtualenv
|
||
|
elif [ -n "$ZSH_VERSION" ] ; then
|
||
|
_virtualenvs () {
|
||
|
reply=( $(virtualenvwrapper_show_workon_options) )
|
||
|
}
|
||
|
_cdvirtualenv_complete () {
|
||
|
reply=( $(cdvirtualenv && ls -d ${1}*) )
|
||
|
}
|
||
|
_cdsitepackages_complete () {
|
||
|
reply=( $(cdsitepackages && ls -d ${1}*) )
|
||
|
}
|
||
|
compctl -K _virtualenvs workon rmvirtualenv cpvirtualenv showvirtualenv
|
||
|
compctl -K _cdvirtualenv_complete cdvirtualenv
|
||
|
compctl -K _cdsitepackages_complete cdsitepackages
|
||
|
fi
|
||
|
}
|
||
|
|
||
|
# Set up virtualenvwrapper properly
|
||
|
function virtualenvwrapper_initialize {
|
||
|
export WORKON_HOME="$(virtualenvwrapper_derive_workon_home)"
|
||
|
|
||
|
virtualenvwrapper_verify_workon_home -q || return 1
|
||
|
|
||
|
# Set the location of the hook scripts
|
||
|
if [ "$VIRTUALENVWRAPPER_HOOK_DIR" = "" ]
|
||
|
then
|
||
|
VIRTUALENVWRAPPER_HOOK_DIR="$WORKON_HOME"
|
||
|
fi
|
||
|
export VIRTUALENVWRAPPER_HOOK_DIR
|
||
|
|
||
|
mkdir -p "$VIRTUALENVWRAPPER_HOOK_DIR"
|
||
|
|
||
|
virtualenvwrapper_run_hook "initialize"
|
||
|
|
||
|
virtualenvwrapper_setup_tab_completion
|
||
|
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
# Verify that the passed resource is in path and exists
|
||
|
function virtualenvwrapper_verify_resource {
|
||
|
typeset exe_path="$(command \which "$1" | (unset GREP_OPTIONS; command \grep -v "not found"))"
|
||
|
if [ "$exe_path" = "" ]
|
||
|
then
|
||
|
echo "ERROR: virtualenvwrapper could not find $1 in your path" >&2
|
||
|
return 1
|
||
|
fi
|
||
|
if [ ! -e "$exe_path" ]
|
||
|
then
|
||
|
echo "ERROR: Found $1 in path as \"$exe_path\" but that does not exist" >&2
|
||
|
return 1
|
||
|
fi
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
|
||
|
# Verify that virtualenv is installed and visible
|
||
|
function virtualenvwrapper_verify_virtualenv {
|
||
|
virtualenvwrapper_verify_resource $VIRTUALENVWRAPPER_VIRTUALENV
|
||
|
}
|
||
|
|
||
|
|
||
|
function virtualenvwrapper_verify_virtualenv_clone {
|
||
|
virtualenvwrapper_verify_resource $VIRTUALENVWRAPPER_VIRTUALENV_CLONE
|
||
|
}
|
||
|
|
||
|
|
||
|
# Verify that the requested environment exists
|
||
|
function virtualenvwrapper_verify_workon_environment {
|
||
|
typeset env_name="$1"
|
||
|
if [ ! -d "$WORKON_HOME/$env_name" ]
|
||
|
then
|
||
|
echo "ERROR: Environment '$env_name' does not exist. Create it with 'mkvirtualenv $env_name'." >&2
|
||
|
return 1
|
||
|
fi
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
# Verify that the active environment exists
|
||
|
function virtualenvwrapper_verify_active_environment {
|
||
|
if [ ! -n "${VIRTUAL_ENV}" ] || [ ! -d "${VIRTUAL_ENV}" ]
|
||
|
then
|
||
|
echo "ERROR: no virtualenv active, or active virtualenv is missing" >&2
|
||
|
return 1
|
||
|
fi
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
# Help text for mkvirtualenv
|
||
|
function virtualenvwrapper_mkvirtualenv_help {
|
||
|
echo "Usage: mkvirtualenv [-a project_path] [-i package] [-r requirements_file] [virtualenv options] env_name"
|
||
|
echo
|
||
|
echo " -a project_path"
|
||
|
echo
|
||
|
echo " Provide a full path to a project directory to associate with"
|
||
|
echo " the new environment."
|
||
|
echo
|
||
|
echo " -i package"
|
||
|
echo
|
||
|
echo " Install a package after the environment is created."
|
||
|
echo " This option may be repeated."
|
||
|
echo
|
||
|
echo " -r requirements_file"
|
||
|
echo
|
||
|
echo " Provide a pip requirements file to install a base set of packages"
|
||
|
echo " into the new environment."
|
||
|
echo;
|
||
|
echo 'virtualenv help:';
|
||
|
echo;
|
||
|
"$VIRTUALENVWRAPPER_VIRTUALENV" $@;
|
||
|
}
|
||
|
|
||
|
# Create a new environment, in the WORKON_HOME.
|
||
|
#
|
||
|
# Usage: mkvirtualenv [options] ENVNAME
|
||
|
# (where the options are passed directly to virtualenv)
|
||
|
#
|
||
|
#:help:mkvirtualenv: Create a new virtualenv in $WORKON_HOME
|
||
|
function mkvirtualenv {
|
||
|
typeset -a in_args
|
||
|
typeset -a out_args
|
||
|
typeset -i i
|
||
|
typeset tst
|
||
|
typeset a
|
||
|
typeset envname
|
||
|
typeset requirements
|
||
|
typeset packages
|
||
|
typeset interpreter
|
||
|
typeset project
|
||
|
|
||
|
in_args=( "$@" )
|
||
|
|
||
|
if [ -n "$ZSH_VERSION" ]
|
||
|
then
|
||
|
i=1
|
||
|
tst="-le"
|
||
|
else
|
||
|
i=0
|
||
|
tst="-lt"
|
||
|
fi
|
||
|
while [ $i $tst $# ]
|
||
|
do
|
||
|
a="${in_args[$i]}"
|
||
|
# echo "arg $i : $a"
|
||
|
case "$a" in
|
||
|
-a)
|
||
|
i=$(( $i + 1 ))
|
||
|
project="${in_args[$i]}"
|
||
|
if [ ! -d "$project" ]
|
||
|
then
|
||
|
echo "Cannot associate project with $project, it is not a directory" 1>&2
|
||
|
return 1
|
||
|
fi
|
||
|
project="$(virtualenvwrapper_absolutepath ${project})";;
|
||
|
-h|--help)
|
||
|
virtualenvwrapper_mkvirtualenv_help $a;
|
||
|
return;;
|
||
|
-i)
|
||
|
i=$(( $i + 1 ));
|
||
|
packages="$packages ${in_args[$i]}";;
|
||
|
-p|--python*)
|
||
|
if echo "$a" | grep -q "="
|
||
|
then
|
||
|
interpreter="$(echo "$a" | cut -f2 -d=)"
|
||
|
else
|
||
|
i=$(( $i + 1 ))
|
||
|
interpreter="${in_args[$i]}"
|
||
|
fi;;
|
||
|
-r)
|
||
|
i=$(( $i + 1 ));
|
||
|
requirements="${in_args[$i]}";
|
||
|
requirements="$(virtualenvwrapper_expandpath "$requirements")";;
|
||
|
*)
|
||
|
if [ ${#out_args} -gt 0 ]
|
||
|
then
|
||
|
out_args=( "${out_args[@]-}" "$a" )
|
||
|
else
|
||
|
out_args=( "$a" )
|
||
|
fi;;
|
||
|
esac
|
||
|
i=$(( $i + 1 ))
|
||
|
done
|
||
|
|
||
|
if [ ! -z $interpreter ]
|
||
|
then
|
||
|
out_args=( "--python=$interpreter" ${out_args[@]} )
|
||
|
fi;
|
||
|
|
||
|
set -- "${out_args[@]}"
|
||
|
|
||
|
eval "envname=\$$#"
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_virtualenv || return 1
|
||
|
(
|
||
|
[ -n "$ZSH_VERSION" ] && setopt SH_WORD_SPLIT
|
||
|
virtualenvwrapper_cd "$WORKON_HOME" &&
|
||
|
"$VIRTUALENVWRAPPER_VIRTUALENV" $VIRTUALENVWRAPPER_VIRTUALENV_ARGS "$@" &&
|
||
|
[ -d "$WORKON_HOME/$envname" ] && \
|
||
|
virtualenvwrapper_run_hook "pre_mkvirtualenv" "$envname"
|
||
|
)
|
||
|
typeset RC=$?
|
||
|
[ $RC -ne 0 ] && return $RC
|
||
|
|
||
|
# If they passed a help option or got an error from virtualenv,
|
||
|
# the environment won't exist. Use that to tell whether
|
||
|
# we should switch to the environment and run the hook.
|
||
|
[ ! -d "$WORKON_HOME/$envname" ] && return 0
|
||
|
|
||
|
# If they gave us a project directory, set it up now
|
||
|
# so the activate hooks can find it.
|
||
|
if [ ! -z "$project" ]
|
||
|
then
|
||
|
setvirtualenvproject "$WORKON_HOME/$envname" "$project"
|
||
|
RC=$?
|
||
|
[ $RC -ne 0 ] && return $RC
|
||
|
fi
|
||
|
|
||
|
# Now activate the new environment
|
||
|
workon "$envname"
|
||
|
|
||
|
if [ ! -z "$requirements" ]
|
||
|
then
|
||
|
pip install -r "$requirements"
|
||
|
fi
|
||
|
|
||
|
for a in $packages
|
||
|
do
|
||
|
pip install $a
|
||
|
done
|
||
|
|
||
|
virtualenvwrapper_run_hook "post_mkvirtualenv"
|
||
|
}
|
||
|
|
||
|
#:help:rmvirtualenv: Remove a virtualenv
|
||
|
function rmvirtualenv {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
if [ ${#@} = 0 ]
|
||
|
then
|
||
|
echo "Please specify an environment." >&2
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
# support to remove several environments
|
||
|
typeset env_name
|
||
|
# Must quote the parameters, as environments could have spaces in their names
|
||
|
for env_name in "$@"
|
||
|
do
|
||
|
echo "Removing $env_name..."
|
||
|
typeset env_dir="$WORKON_HOME/$env_name"
|
||
|
if [ "$VIRTUAL_ENV" = "$env_dir" ]
|
||
|
then
|
||
|
echo "ERROR: You cannot remove the active environment ('$env_name')." >&2
|
||
|
echo "Either switch to another environment, or run 'deactivate'." >&2
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
if [ ! -d "$env_dir" ]; then
|
||
|
echo "Did not find environment $env_dir to remove." >&2
|
||
|
fi
|
||
|
|
||
|
# Move out of the current directory to one known to be
|
||
|
# safe, in case we are inside the environment somewhere.
|
||
|
typeset prior_dir="$(pwd)"
|
||
|
virtualenvwrapper_cd "$WORKON_HOME"
|
||
|
|
||
|
virtualenvwrapper_run_hook "pre_rmvirtualenv" "$env_name"
|
||
|
command \rm -rf "$env_dir"
|
||
|
virtualenvwrapper_run_hook "post_rmvirtualenv" "$env_name"
|
||
|
|
||
|
# If the directory we used to be in still exists, move back to it.
|
||
|
if [ -d "$prior_dir" ]
|
||
|
then
|
||
|
virtualenvwrapper_cd "$prior_dir"
|
||
|
fi
|
||
|
done
|
||
|
}
|
||
|
|
||
|
# List the available environments.
|
||
|
function virtualenvwrapper_show_workon_options {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
# NOTE: DO NOT use ls or cd here because colorized versions spew control
|
||
|
# characters into the output list.
|
||
|
# echo seems a little faster than find, even with -depth 3.
|
||
|
# Note that this is a little tricky, as there may be spaces in the path.
|
||
|
#
|
||
|
# 1. Look for environments by finding the activate scripts.
|
||
|
# Use a subshell so we can suppress the message printed
|
||
|
# by zsh if the glob pattern fails to match any files.
|
||
|
# This yields a single, space-separated line containing all matches.
|
||
|
# 2. Replace the trailing newline with a space, so every
|
||
|
# possible env has a space following it.
|
||
|
# 3. Strip the bindir/activate script suffix, replacing it with
|
||
|
# a slash, as that is an illegal character in a directory name.
|
||
|
# This yields a slash-separated list of possible env names.
|
||
|
# 4. Replace each slash with a newline to show the output one name per line.
|
||
|
# 5. Eliminate any lines with * on them because that means there
|
||
|
# were no envs.
|
||
|
(virtualenvwrapper_cd "$WORKON_HOME" && echo */$VIRTUALENVWRAPPER_ENV_BIN_DIR/activate) 2>/dev/null \
|
||
|
| command \tr "\n" " " \
|
||
|
| command \sed "s|/$VIRTUALENVWRAPPER_ENV_BIN_DIR/activate |/|g" \
|
||
|
| command \tr "/" "\n" \
|
||
|
| command \sed "/^\s*$/d" \
|
||
|
| (unset GREP_OPTIONS; command \egrep -v '^\*$') 2>/dev/null
|
||
|
}
|
||
|
|
||
|
function _lsvirtualenv_usage {
|
||
|
echo "lsvirtualenv [-blh]"
|
||
|
echo " -b -- brief mode"
|
||
|
echo " -l -- long mode"
|
||
|
echo " -h -- this help message"
|
||
|
}
|
||
|
|
||
|
#:help:lsvirtualenv: list virtualenvs
|
||
|
function lsvirtualenv {
|
||
|
|
||
|
typeset long_mode=true
|
||
|
if command -v "getopts" >/dev/null 2>&1
|
||
|
then
|
||
|
# Use getopts when possible
|
||
|
OPTIND=1
|
||
|
while getopts ":blh" opt "$@"
|
||
|
do
|
||
|
case "$opt" in
|
||
|
l) long_mode=true;;
|
||
|
b) long_mode=false;;
|
||
|
h) _lsvirtualenv_usage;
|
||
|
return 1;;
|
||
|
?) echo "Invalid option: -$OPTARG" >&2;
|
||
|
_lsvirtualenv_usage;
|
||
|
return 1;;
|
||
|
esac
|
||
|
done
|
||
|
else
|
||
|
# fallback on getopt for other shell
|
||
|
typeset -a args
|
||
|
args=($(getopt blh "$@"))
|
||
|
if [ $? != 0 ]
|
||
|
then
|
||
|
_lsvirtualenv_usage
|
||
|
return 1
|
||
|
fi
|
||
|
for opt in $args
|
||
|
do
|
||
|
case "$opt" in
|
||
|
-l) long_mode=true;;
|
||
|
-b) long_mode=false;;
|
||
|
-h) _lsvirtualenv_usage;
|
||
|
return 1;;
|
||
|
esac
|
||
|
done
|
||
|
fi
|
||
|
|
||
|
if $long_mode
|
||
|
then
|
||
|
allvirtualenv showvirtualenv "$env_name"
|
||
|
else
|
||
|
virtualenvwrapper_show_workon_options
|
||
|
fi
|
||
|
}
|
||
|
|
||
|
#:help:showvirtualenv: show details of a single virtualenv
|
||
|
function showvirtualenv {
|
||
|
typeset env_name="$1"
|
||
|
if [ -z "$env_name" ]
|
||
|
then
|
||
|
if [ -z "$VIRTUAL_ENV" ]
|
||
|
then
|
||
|
echo "showvirtualenv [env]"
|
||
|
return 1
|
||
|
fi
|
||
|
env_name=$(basename "$VIRTUAL_ENV")
|
||
|
fi
|
||
|
|
||
|
virtualenvwrapper_run_hook "get_env_details" "$env_name"
|
||
|
echo
|
||
|
}
|
||
|
|
||
|
# Show help for workon
|
||
|
function virtualenvwrapper_workon_help {
|
||
|
echo "Usage: workon env_name"
|
||
|
echo ""
|
||
|
echo " Deactivate any currently activated virtualenv"
|
||
|
echo " and activate the named environment, triggering"
|
||
|
echo " any hooks in the process."
|
||
|
echo ""
|
||
|
echo " workon"
|
||
|
echo ""
|
||
|
echo " Print a list of available environments."
|
||
|
echo " (See also lsvirtualenv -b)"
|
||
|
echo ""
|
||
|
echo " workon (-h|--help)"
|
||
|
echo ""
|
||
|
echo " Show this help message."
|
||
|
echo ""
|
||
|
echo " workon (-c|--cd) envname"
|
||
|
echo ""
|
||
|
echo " After activating the environment, cd to the associated"
|
||
|
echo " project directory if it is set."
|
||
|
echo ""
|
||
|
echo " workon (-n|--no-cd) envname"
|
||
|
echo ""
|
||
|
echo " After activating the environment, do not cd to the"
|
||
|
echo " associated project directory."
|
||
|
echo ""
|
||
|
}
|
||
|
|
||
|
#:help:workon: list or change working virtualenvs
|
||
|
function workon {
|
||
|
typeset -a in_args
|
||
|
typeset -a out_args
|
||
|
|
||
|
in_args=( "$@" )
|
||
|
|
||
|
if [ -n "$ZSH_VERSION" ]
|
||
|
then
|
||
|
i=1
|
||
|
tst="-le"
|
||
|
else
|
||
|
i=0
|
||
|
tst="-lt"
|
||
|
fi
|
||
|
typeset cd_after_activate=$VIRTUALENVWRAPPER_WORKON_CD
|
||
|
while [ $i $tst $# ]
|
||
|
do
|
||
|
a="${in_args[$i]}"
|
||
|
case "$a" in
|
||
|
-h|--help)
|
||
|
virtualenvwrapper_workon_help;
|
||
|
return 0;;
|
||
|
-n|--no-cd)
|
||
|
cd_after_activate=0;;
|
||
|
-c|--cd)
|
||
|
cd_after_activate=1;;
|
||
|
*)
|
||
|
if [ ${#out_args} -gt 0 ]
|
||
|
then
|
||
|
out_args=( "${out_args[@]-}" "$a" )
|
||
|
else
|
||
|
out_args=( "$a" )
|
||
|
fi;;
|
||
|
esac
|
||
|
i=$(( $i + 1 ))
|
||
|
done
|
||
|
|
||
|
set -- "${out_args[@]}"
|
||
|
|
||
|
typeset env_name="$1"
|
||
|
if [ "$env_name" = "" ]
|
||
|
then
|
||
|
lsvirtualenv -b
|
||
|
return 1
|
||
|
elif [ "$env_name" = "." ]
|
||
|
then
|
||
|
# The IFS default of breaking on whitespace causes issues if there
|
||
|
# are spaces in the env_name, so change it.
|
||
|
IFS='%'
|
||
|
env_name="$(basename $(pwd))"
|
||
|
unset IFS
|
||
|
fi
|
||
|
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_workon_environment "$env_name" || return 1
|
||
|
|
||
|
activate="$WORKON_HOME/$env_name/$VIRTUALENVWRAPPER_ENV_BIN_DIR/activate"
|
||
|
if [ ! -f "$activate" ]
|
||
|
then
|
||
|
echo "ERROR: Environment '$WORKON_HOME/$env_name' does not contain an activate script." >&2
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
# Deactivate any current environment "destructively"
|
||
|
# before switching so we use our override function,
|
||
|
# if it exists, but make sure it's the deactivate function
|
||
|
# we set up
|
||
|
type deactivate >/dev/null 2>&1
|
||
|
if [ $? -eq 0 ]
|
||
|
then
|
||
|
typeset -f deactivate | grep 'typeset env_postdeactivate_hook' >/dev/null 2>&1
|
||
|
if [ $? -eq 0 ]
|
||
|
then
|
||
|
deactivate
|
||
|
unset -f deactivate >/dev/null 2>&1
|
||
|
fi
|
||
|
fi
|
||
|
|
||
|
virtualenvwrapper_run_hook "pre_activate" "$env_name"
|
||
|
|
||
|
source "$activate"
|
||
|
|
||
|
# Save the deactivate function from virtualenv under a different name
|
||
|
virtualenvwrapper_original_deactivate=`typeset -f deactivate | sed 's/deactivate/virtualenv_deactivate/g'`
|
||
|
eval "$virtualenvwrapper_original_deactivate"
|
||
|
unset -f deactivate >/dev/null 2>&1
|
||
|
|
||
|
# Replace the deactivate() function with a wrapper.
|
||
|
eval 'deactivate () {
|
||
|
typeset env_postdeactivate_hook
|
||
|
typeset old_env
|
||
|
|
||
|
# Call the local hook before the global so we can undo
|
||
|
# any settings made by the local postactivate first.
|
||
|
virtualenvwrapper_run_hook "pre_deactivate"
|
||
|
|
||
|
env_postdeactivate_hook="$VIRTUAL_ENV/$VIRTUALENVWRAPPER_ENV_BIN_DIR/postdeactivate"
|
||
|
old_env=$(basename "$VIRTUAL_ENV")
|
||
|
|
||
|
# Call the original function.
|
||
|
virtualenv_deactivate $1
|
||
|
|
||
|
virtualenvwrapper_run_hook "post_deactivate" "$old_env"
|
||
|
|
||
|
if [ ! "$1" = "nondestructive" ]
|
||
|
then
|
||
|
# Remove this function
|
||
|
unset -f virtualenv_deactivate >/dev/null 2>&1
|
||
|
unset -f deactivate >/dev/null 2>&1
|
||
|
fi
|
||
|
|
||
|
}'
|
||
|
|
||
|
VIRTUALENVWRAPPER_PROJECT_CD=$cd_after_activate virtualenvwrapper_run_hook "post_activate"
|
||
|
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
|
||
|
# Prints the Python version string for the current interpreter.
|
||
|
function virtualenvwrapper_get_python_version {
|
||
|
# Uses the Python from the virtualenv rather than
|
||
|
# VIRTUALENVWRAPPER_PYTHON because we're trying to determine the
|
||
|
# version installed there so we can build up the path to the
|
||
|
# site-packages directory.
|
||
|
"$VIRTUAL_ENV/$VIRTUALENVWRAPPER_ENV_BIN_DIR/python" -V 2>&1 | cut -f2 -d' ' | cut -f-2 -d.
|
||
|
}
|
||
|
|
||
|
# Prints the path to the site-packages directory for the current environment.
|
||
|
function virtualenvwrapper_get_site_packages_dir {
|
||
|
"$VIRTUAL_ENV/$VIRTUALENVWRAPPER_ENV_BIN_DIR/python" -c "import distutils.sysconfig; print(distutils.sysconfig.get_python_lib())"
|
||
|
}
|
||
|
|
||
|
# Path management for packages outside of the virtual env.
|
||
|
# Based on a contribution from James Bennett and Jannis Leidel.
|
||
|
#
|
||
|
# add2virtualenv directory1 directory2 ...
|
||
|
#
|
||
|
# Adds the specified directories to the Python path for the
|
||
|
# currently-active virtualenv. This will be done by placing the
|
||
|
# directory names in a path file named
|
||
|
# "virtualenv_path_extensions.pth" inside the virtualenv's
|
||
|
# site-packages directory; if this file does not exist, it will be
|
||
|
# created first.
|
||
|
#
|
||
|
#:help:add2virtualenv: add directory to the import path
|
||
|
function add2virtualenv {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_active_environment || return 1
|
||
|
|
||
|
site_packages="`virtualenvwrapper_get_site_packages_dir`"
|
||
|
|
||
|
if [ ! -d "${site_packages}" ]
|
||
|
then
|
||
|
echo "ERROR: currently-active virtualenv does not appear to have a site-packages directory" >&2
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
# Prefix with _ to ensure we are loaded as early as possible,
|
||
|
# and at least before easy_install.pth.
|
||
|
path_file="$site_packages/_virtualenv_path_extensions.pth"
|
||
|
|
||
|
if [ "$*" = "" ]
|
||
|
then
|
||
|
echo "Usage: add2virtualenv dir [dir ...]"
|
||
|
if [ -f "$path_file" ]
|
||
|
then
|
||
|
echo
|
||
|
echo "Existing paths:"
|
||
|
cat "$path_file" | grep -v "^import"
|
||
|
fi
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
remove=0
|
||
|
if [ "$1" = "-d" ]
|
||
|
then
|
||
|
remove=1
|
||
|
shift
|
||
|
fi
|
||
|
|
||
|
if [ ! -f "$path_file" ]
|
||
|
then
|
||
|
echo "import sys; sys.__plen = len(sys.path)" > "$path_file" || return 1
|
||
|
echo "import sys; new=sys.path[sys.__plen:]; del sys.path[sys.__plen:]; p=getattr(sys,'__egginsert',0); sys.path[p:p]=new; sys.__egginsert = p+len(new)" >> "$path_file" || return 1
|
||
|
fi
|
||
|
|
||
|
for pydir in "$@"
|
||
|
do
|
||
|
absolute_path="$(virtualenvwrapper_absolutepath "$pydir")"
|
||
|
if [ "$absolute_path" != "$pydir" ]
|
||
|
then
|
||
|
echo "Warning: Converting \"$pydir\" to \"$absolute_path\"" 1>&2
|
||
|
fi
|
||
|
|
||
|
if [ $remove -eq 1 ]
|
||
|
then
|
||
|
sed -i.tmp "\:^$absolute_path$: d" "$path_file"
|
||
|
else
|
||
|
sed -i.tmp '1 a\
|
||
|
'"$absolute_path"'
|
||
|
' "$path_file"
|
||
|
fi
|
||
|
rm -f "${path_file}.tmp"
|
||
|
done
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
# Does a ``cd`` to the site-packages directory of the currently-active
|
||
|
# virtualenv.
|
||
|
#:help:cdsitepackages: change to the site-packages directory
|
||
|
function cdsitepackages {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_active_environment || return 1
|
||
|
typeset site_packages="`virtualenvwrapper_get_site_packages_dir`"
|
||
|
virtualenvwrapper_cd "$site_packages/$1"
|
||
|
}
|
||
|
|
||
|
# Does a ``cd`` to the root of the currently-active virtualenv.
|
||
|
#:help:cdvirtualenv: change to the $VIRTUAL_ENV directory
|
||
|
function cdvirtualenv {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_active_environment || return 1
|
||
|
virtualenvwrapper_cd "$VIRTUAL_ENV/$1"
|
||
|
}
|
||
|
|
||
|
# Shows the content of the site-packages directory of the currently-active
|
||
|
# virtualenv
|
||
|
#:help:lssitepackages: list contents of the site-packages directory
|
||
|
function lssitepackages {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_active_environment || return 1
|
||
|
typeset site_packages="`virtualenvwrapper_get_site_packages_dir`"
|
||
|
ls $@ "$site_packages"
|
||
|
|
||
|
path_file="$site_packages/_virtualenv_path_extensions.pth"
|
||
|
if [ -f "$path_file" ]
|
||
|
then
|
||
|
echo
|
||
|
echo "_virtualenv_path_extensions.pth:"
|
||
|
cat "$path_file"
|
||
|
fi
|
||
|
}
|
||
|
|
||
|
# Toggles the currently-active virtualenv between having and not having
|
||
|
# access to the global site-packages.
|
||
|
#:help:toggleglobalsitepackages: turn access to global site-packages on/off
|
||
|
function toggleglobalsitepackages {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_active_environment || return 1
|
||
|
typeset no_global_site_packages_file="`virtualenvwrapper_get_site_packages_dir`/../no-global-site-packages.txt"
|
||
|
if [ -f $no_global_site_packages_file ]; then
|
||
|
rm $no_global_site_packages_file
|
||
|
[ "$1" = "-q" ] || echo "Enabled global site-packages"
|
||
|
else
|
||
|
touch $no_global_site_packages_file
|
||
|
[ "$1" = "-q" ] || echo "Disabled global site-packages"
|
||
|
fi
|
||
|
}
|
||
|
|
||
|
#:help:cpvirtualenv: duplicate the named virtualenv to make a new one
|
||
|
function cpvirtualenv {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_virtualenv_clone || return 1
|
||
|
|
||
|
typeset src_name="$1"
|
||
|
typeset trg_name="$2"
|
||
|
typeset src
|
||
|
typeset trg
|
||
|
|
||
|
# without a source there is nothing to do
|
||
|
if [ "$src_name" = "" ]; then
|
||
|
echo "Please provide a valid virtualenv to copy."
|
||
|
return 1
|
||
|
else
|
||
|
# see if it's already in workon
|
||
|
if [ ! -e "$WORKON_HOME/$src_name" ]; then
|
||
|
# so it's a virtualenv we are importing
|
||
|
# make sure we have a full path
|
||
|
# and get the name
|
||
|
src="$(virtualenvwrapper_expandpath "$src_name")"
|
||
|
# final verification
|
||
|
if [ ! -e "$src" ]; then
|
||
|
echo "Please provide a valid virtualenv to copy."
|
||
|
return 1
|
||
|
fi
|
||
|
src_name="$(basename "$src")"
|
||
|
else
|
||
|
src="$WORKON_HOME/$src_name"
|
||
|
fi
|
||
|
fi
|
||
|
|
||
|
if [ "$trg_name" = "" ]; then
|
||
|
# target not given, assume
|
||
|
# same as source
|
||
|
trg="$WORKON_HOME/$src_name"
|
||
|
trg_name="$src_name"
|
||
|
else
|
||
|
trg="$WORKON_HOME/$trg_name"
|
||
|
fi
|
||
|
trg="$(virtualenvwrapper_expandpath "$trg")"
|
||
|
|
||
|
# validate trg does not already exist
|
||
|
# catch copying virtualenv in workon home
|
||
|
# to workon home
|
||
|
if [ -e "$trg" ]; then
|
||
|
echo "$trg_name virtualenv already exists."
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
echo "Copying $src_name as $trg_name..."
|
||
|
(
|
||
|
[ -n "$ZSH_VERSION" ] && setopt SH_WORD_SPLIT
|
||
|
virtualenvwrapper_cd "$WORKON_HOME" &&
|
||
|
"$VIRTUALENVWRAPPER_VIRTUALENV_CLONE" "$src" "$trg"
|
||
|
[ -d "$trg" ] &&
|
||
|
virtualenvwrapper_run_hook "pre_cpvirtualenv" "$src" "$trg_name" &&
|
||
|
virtualenvwrapper_run_hook "pre_mkvirtualenv" "$trg_name"
|
||
|
)
|
||
|
typeset RC=$?
|
||
|
[ $RC -ne 0 ] && return $RC
|
||
|
|
||
|
[ ! -d "$WORKON_HOME/$trg_name" ] && return 1
|
||
|
|
||
|
# Now activate the new environment
|
||
|
workon "$trg_name"
|
||
|
|
||
|
virtualenvwrapper_run_hook "post_mkvirtualenv"
|
||
|
virtualenvwrapper_run_hook "post_cpvirtualenv"
|
||
|
}
|
||
|
|
||
|
#
|
||
|
# virtualenvwrapper project functions
|
||
|
#
|
||
|
|
||
|
# Verify that the PROJECT_HOME directory exists
|
||
|
function virtualenvwrapper_verify_project_home {
|
||
|
if [ -z "$PROJECT_HOME" ]
|
||
|
then
|
||
|
echo "ERROR: Set the PROJECT_HOME shell variable to the name of the directory where projects should be created." >&2
|
||
|
return 1
|
||
|
fi
|
||
|
if [ ! -d "$PROJECT_HOME" ]
|
||
|
then
|
||
|
[ "$1" != "-q" ] && echo "ERROR: Project directory '$PROJECT_HOME' does not exist. Create it or set PROJECT_HOME to an existing directory." >&2
|
||
|
return 1
|
||
|
fi
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
# Given a virtualenv directory and a project directory,
|
||
|
# set the virtualenv up to be associated with the
|
||
|
# project
|
||
|
#:help:setvirtualenvproject: associate a project directory with a virtualenv
|
||
|
function setvirtualenvproject {
|
||
|
typeset venv="$1"
|
||
|
typeset prj="$2"
|
||
|
if [ -z "$venv" ]
|
||
|
then
|
||
|
venv="$VIRTUAL_ENV"
|
||
|
fi
|
||
|
if [ -z "$prj" ]
|
||
|
then
|
||
|
prj="$(pwd)"
|
||
|
else
|
||
|
prj=$(virtualenvwrapper_absolutepath "${prj}")
|
||
|
fi
|
||
|
|
||
|
# If what we were given isn't a directory, see if it is under
|
||
|
# $WORKON_HOME.
|
||
|
if [ ! -d "$venv" ]
|
||
|
then
|
||
|
venv="$WORKON_HOME/$venv"
|
||
|
fi
|
||
|
if [ ! -d "$venv" ]
|
||
|
then
|
||
|
echo "No virtualenv $(basename $venv)" 1>&2
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
# Make sure we have a valid project setting
|
||
|
if [ ! -d "$prj" ]
|
||
|
then
|
||
|
echo "Cannot associate virtualenv with \"$prj\", it is not a directory" 1>&2
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
echo "Setting project for $(basename $venv) to $prj"
|
||
|
echo "$prj" > "$venv/$VIRTUALENVWRAPPER_PROJECT_FILENAME"
|
||
|
}
|
||
|
|
||
|
# Show help for mkproject
|
||
|
function virtualenvwrapper_mkproject_help {
|
||
|
echo "Usage: mkproject [-f|--force] [-t template] [virtualenv options] project_name"
|
||
|
echo
|
||
|
echo "-f, --force Create the virtualenv even if the project directory"
|
||
|
echo " already exists"
|
||
|
echo
|
||
|
echo "Multiple templates may be selected. They are applied in the order"
|
||
|
echo "specified on the command line."
|
||
|
echo
|
||
|
echo "mkvirtualenv help:"
|
||
|
echo
|
||
|
mkvirtualenv -h
|
||
|
echo
|
||
|
echo "Available project templates:"
|
||
|
echo
|
||
|
"$VIRTUALENVWRAPPER_PYTHON" -c 'from virtualenvwrapper.hook_loader import main; main()' -l project.template
|
||
|
}
|
||
|
|
||
|
#:help:mkproject: create a new project directory and its associated virtualenv
|
||
|
function mkproject {
|
||
|
typeset -a in_args
|
||
|
typeset -a out_args
|
||
|
typeset -i i
|
||
|
typeset tst
|
||
|
typeset a
|
||
|
typeset t
|
||
|
typeset force
|
||
|
typeset templates
|
||
|
|
||
|
in_args=( "$@" )
|
||
|
force=0
|
||
|
|
||
|
if [ -n "$ZSH_VERSION" ]
|
||
|
then
|
||
|
i=1
|
||
|
tst="-le"
|
||
|
else
|
||
|
i=0
|
||
|
tst="-lt"
|
||
|
fi
|
||
|
while [ $i $tst $# ]
|
||
|
do
|
||
|
a="${in_args[$i]}"
|
||
|
case "$a" in
|
||
|
-h|--help)
|
||
|
virtualenvwrapper_mkproject_help;
|
||
|
return;;
|
||
|
-f|--force)
|
||
|
force=1;;
|
||
|
-t)
|
||
|
i=$(( $i + 1 ));
|
||
|
templates="$templates ${in_args[$i]}";;
|
||
|
*)
|
||
|
if [ ${#out_args} -gt 0 ]
|
||
|
then
|
||
|
out_args=( "${out_args[@]-}" "$a" )
|
||
|
else
|
||
|
out_args=( "$a" )
|
||
|
fi;;
|
||
|
esac
|
||
|
i=$(( $i + 1 ))
|
||
|
done
|
||
|
|
||
|
set -- "${out_args[@]}"
|
||
|
|
||
|
# echo "templates $templates"
|
||
|
# echo "remainder $@"
|
||
|
# return 0
|
||
|
|
||
|
eval "typeset envname=\$$#"
|
||
|
virtualenvwrapper_verify_project_home || return 1
|
||
|
|
||
|
if [ -d "$PROJECT_HOME/$envname" -a $force -eq 0 ]
|
||
|
then
|
||
|
echo "Project $envname already exists." >&2
|
||
|
return 1
|
||
|
fi
|
||
|
|
||
|
mkvirtualenv "$@" || return 1
|
||
|
|
||
|
virtualenvwrapper_cd "$PROJECT_HOME"
|
||
|
|
||
|
virtualenvwrapper_run_hook "project.pre_mkproject" $envname
|
||
|
|
||
|
echo "Creating $PROJECT_HOME/$envname"
|
||
|
mkdir -p "$PROJECT_HOME/$envname"
|
||
|
setvirtualenvproject "$VIRTUAL_ENV" "$PROJECT_HOME/$envname"
|
||
|
|
||
|
virtualenvwrapper_cd "$PROJECT_HOME/$envname"
|
||
|
|
||
|
for t in $templates
|
||
|
do
|
||
|
echo
|
||
|
echo "Applying template $t"
|
||
|
# For some reason zsh insists on prefixing the template
|
||
|
# names with a space, so strip them out before passing
|
||
|
# the value to the hook loader.
|
||
|
virtualenvwrapper_run_hook --name $(echo $t | sed 's/^ //') "project.template" "$envname" "$PROJECT_HOME/$envname"
|
||
|
done
|
||
|
|
||
|
virtualenvwrapper_run_hook "project.post_mkproject"
|
||
|
}
|
||
|
|
||
|
#:help:cdproject: change directory to the active project
|
||
|
function cdproject {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_active_environment || return 1
|
||
|
if [ -f "$VIRTUAL_ENV/$VIRTUALENVWRAPPER_PROJECT_FILENAME" ]
|
||
|
then
|
||
|
typeset project_dir="$(cat "$VIRTUAL_ENV/$VIRTUALENVWRAPPER_PROJECT_FILENAME")"
|
||
|
if [ ! -z "$project_dir" ]
|
||
|
then
|
||
|
virtualenvwrapper_cd "$project_dir"
|
||
|
else
|
||
|
echo "Project directory $project_dir does not exist" 1>&2
|
||
|
return 1
|
||
|
fi
|
||
|
else
|
||
|
echo "No project set in $VIRTUAL_ENV/$VIRTUALENVWRAPPER_PROJECT_FILENAME" 1>&2
|
||
|
return 1
|
||
|
fi
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
#
|
||
|
# Temporary virtualenv
|
||
|
#
|
||
|
# Originally part of virtualenvwrapper.tmpenv plugin
|
||
|
#
|
||
|
#:help:mktmpenv: create a temporary virtualenv
|
||
|
function mktmpenv {
|
||
|
typeset tmpenvname
|
||
|
typeset RC
|
||
|
typeset -a in_args
|
||
|
typeset -a out_args
|
||
|
|
||
|
in_args=( "$@" )
|
||
|
|
||
|
if [ -n "$ZSH_VERSION" ]
|
||
|
then
|
||
|
i=1
|
||
|
tst="-le"
|
||
|
else
|
||
|
i=0
|
||
|
tst="-lt"
|
||
|
fi
|
||
|
typeset cd_after_activate=$VIRTUALENVWRAPPER_WORKON_CD
|
||
|
while [ $i $tst $# ]
|
||
|
do
|
||
|
a="${in_args[$i]}"
|
||
|
case "$a" in
|
||
|
-n|--no-cd)
|
||
|
cd_after_activate=0;;
|
||
|
-c|--cd)
|
||
|
cd_after_activate=1;;
|
||
|
*)
|
||
|
if [ ${#out_args} -gt 0 ]
|
||
|
then
|
||
|
out_args=( "${out_args[@]-}" "$a" )
|
||
|
else
|
||
|
out_args=( "$a" )
|
||
|
fi;;
|
||
|
esac
|
||
|
i=$(( $i + 1 ))
|
||
|
done
|
||
|
|
||
|
set -- "${out_args[@]}"
|
||
|
|
||
|
# Generate a unique temporary name
|
||
|
tmpenvname=$("$VIRTUALENVWRAPPER_PYTHON" -c 'import uuid,sys; sys.stdout.write(uuid.uuid4()+"\n")' 2>/dev/null)
|
||
|
if [ -z "$tmpenvname" ]
|
||
|
then
|
||
|
# This python does not support uuid
|
||
|
tmpenvname=$("$VIRTUALENVWRAPPER_PYTHON" -c 'import random,sys; sys.stdout.write(hex(random.getrandbits(64))[2:-1]+"\n")' 2>/dev/null)
|
||
|
fi
|
||
|
tmpenvname="tmp-$tmpenvname"
|
||
|
|
||
|
# Create the environment
|
||
|
mkvirtualenv "$@" "$tmpenvname"
|
||
|
RC=$?
|
||
|
if [ $RC -ne 0 ]
|
||
|
then
|
||
|
return $RC
|
||
|
fi
|
||
|
|
||
|
# Change working directory
|
||
|
[ "$cd_after_activate" = "1" ] && cdvirtualenv
|
||
|
|
||
|
# Create the tmpenv marker file
|
||
|
echo "This is a temporary environment. It will be deleted when you run 'deactivate'." | tee "$VIRTUAL_ENV/README.tmpenv"
|
||
|
|
||
|
# Update the postdeactivate script
|
||
|
cat - >> "$VIRTUAL_ENV/$VIRTUALENVWRAPPER_ENV_BIN_DIR/postdeactivate" <<EOF
|
||
|
if [ -f "$VIRTUAL_ENV/README.tmpenv" ]
|
||
|
then
|
||
|
echo "Removing temporary environment:" $(basename "$VIRTUAL_ENV")
|
||
|
rmvirtualenv $(basename "$VIRTUAL_ENV")
|
||
|
fi
|
||
|
EOF
|
||
|
}
|
||
|
|
||
|
#
|
||
|
# Remove all installed packages from the env
|
||
|
#
|
||
|
#:help:wipeenv: remove all packages installed in the current virtualenv
|
||
|
function wipeenv {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
virtualenvwrapper_verify_active_environment || return 1
|
||
|
|
||
|
typeset req_file="$(virtualenvwrapper_tempfile "requirements.txt")"
|
||
|
pip freeze | egrep -v '(distribute|wsgiref|appdirs|packaging|pyparsing|six)' > "$req_file"
|
||
|
if [ -n "$(cat "$req_file")" ]
|
||
|
then
|
||
|
echo "Uninstalling packages:"
|
||
|
cat "$req_file"
|
||
|
echo
|
||
|
pip uninstall -y $(cat "$req_file" | grep -v '^-f' | sed 's/>/=/g' | cut -f1 -d=)
|
||
|
else
|
||
|
echo "Nothing to remove."
|
||
|
fi
|
||
|
rm -f "$req_file"
|
||
|
}
|
||
|
|
||
|
#
|
||
|
# Run a command in each virtualenv
|
||
|
#
|
||
|
#:help:allvirtualenv: run a command in all virtualenvs
|
||
|
function allvirtualenv {
|
||
|
virtualenvwrapper_verify_workon_home || return 1
|
||
|
typeset d
|
||
|
|
||
|
# The IFS default of breaking on whitespace causes issues if there
|
||
|
# are spaces in the env_name, so change it.
|
||
|
IFS='%'
|
||
|
virtualenvwrapper_show_workon_options | while read d
|
||
|
do
|
||
|
[ ! -d "$WORKON_HOME/$d" ] && continue
|
||
|
echo "$d"
|
||
|
echo "$d" | sed 's/./=/g'
|
||
|
# Activate the environment, but not with workon
|
||
|
# because we don't want to trigger any hooks.
|
||
|
(source "$WORKON_HOME/$d/$VIRTUALENVWRAPPER_ENV_BIN_DIR/activate";
|
||
|
virtualenvwrapper_cd "$VIRTUAL_ENV";
|
||
|
"$@")
|
||
|
echo
|
||
|
done
|
||
|
unset IFS
|
||
|
}
|
||
|
|
||
|
#:help:virtualenvwrapper: show this help message
|
||
|
function virtualenvwrapper {
|
||
|
cat <<EOF
|
||
|
|
||
|
virtualenvwrapper is a set of extensions to Ian Bicking's virtualenv
|
||
|
tool. The extensions include wrappers for creating and deleting
|
||
|
virtual environments and otherwise managing your development workflow,
|
||
|
making it easier to work on more than one project at a time without
|
||
|
introducing conflicts in their dependencies.
|
||
|
|
||
|
For more information please refer to the documentation:
|
||
|
|
||
|
http://virtualenvwrapper.readthedocs.org/en/latest/command_ref.html
|
||
|
|
||
|
Commands available:
|
||
|
|
||
|
EOF
|
||
|
|
||
|
typeset helpmarker="#:help:"
|
||
|
cat "$VIRTUALENVWRAPPER_SCRIPT" \
|
||
|
| grep "^$helpmarker" \
|
||
|
| sed -e "s/^$helpmarker/ /g" \
|
||
|
| sort \
|
||
|
| sed -e 's/$/\'$'\n/g'
|
||
|
}
|
||
|
|
||
|
#
|
||
|
# Invoke the initialization functions
|
||
|
#
|
||
|
virtualenvwrapper_initialize
|