145 lines
4.1 KiB
Bash
145 lines
4.1 KiB
Bash
#!/usr/bin/env bash
|
|
# tabstop: 4
|
|
_VENV_BASE=$HOME/.venvs
|
|
## get the name of the branch, within the following revision control systems
|
|
function __fetch_repo_details() {
|
|
if [ -d ".git" ]; then
|
|
BRANCHNAME=`git rev-parse --abbrev-ref HEAD`
|
|
# create an SSH wrapper if the tree we're in is "special"
|
|
# then use that special wrapper
|
|
BBNAME=`cat .git/config|grep bitbucket|cut -d "@" -f 2|cut -d "." -f 1`
|
|
if [ $? -eq 0 ]; then
|
|
if [ -f "~/bin/${BBNAME}bitbucketwrapper.sh" ]; then
|
|
GIT_SSH="~/bin/${BBNAME}bitbucketwrapper.sh"
|
|
fi
|
|
fi
|
|
elif [ -d ".hg" ]; then
|
|
BRANCHNAME=`hg branch`
|
|
elif [ -d ".bzr" ]; then
|
|
BRANCHNAME=`cat .bzr/branch/branch.conf|awk -F/ '{print $(NF-1)}'`
|
|
fi
|
|
if [ -z $BRANCHNAME ]; then
|
|
return
|
|
else
|
|
REPONAME=`basename "$PWD"`
|
|
fi
|
|
# figure out our virtual env type
|
|
if [ -f requirements.txt ]; then
|
|
VENV_TYPE="python"
|
|
_VENV_ROOT=$_VENV_BASE/python
|
|
fi
|
|
if [ ! -d $_VENV_ROOT ]; then
|
|
mkdir -p $_VENV_ROOT
|
|
fi
|
|
# pyenv users get an environment prefixed with python version
|
|
if [ -f .python-version ]; then
|
|
PYTHON_VERSION=`cat .python-version`
|
|
VENV_NAME=$PYTHON_VERSION.$REPONAME.$BRANCHNAME
|
|
else
|
|
VENV_NAME=$REPONAME.$BRANCHNAME
|
|
fi
|
|
VENV_PATH=$_VENV_ROOT/$VENV_NAME
|
|
}
|
|
# create the pythonbrew venv combination
|
|
# if there's a requirements.txt we'll pip that puppy
|
|
function bruisemake() {
|
|
if [ ! -z $2 ]; then
|
|
export PATH=$2:$PATH
|
|
fi
|
|
if [ -z $VENV_TYPE ]; then
|
|
echo "You are not in a repository root"
|
|
return 1
|
|
fi
|
|
if [ $VENV_TYPE == "python" ]; then
|
|
if [ ! -z $1 ]; then
|
|
VENV_PATH=$_VENV_BASE/python/$1
|
|
fi
|
|
if [ ! -d $VENV_PATH ]; then
|
|
virtualenv --no-site-packages $VENV_PATH
|
|
fi
|
|
source $VENV_PATH/bin/activate
|
|
pip install -r requirements.txt
|
|
fi
|
|
}
|
|
function __bruise() {
|
|
# pyenv support or not ;)
|
|
if [ -z $PYTHON_VERSION ]; then
|
|
VENV_PREFIX="${PYTHON_VERSION}"
|
|
else
|
|
VENV_PREFIX="${PYTHON_VERSION}."
|
|
fi
|
|
# if for some reason we're testing in a local venv, it wins
|
|
if [ -d `pwd`/venv ]; then
|
|
BRUISE_DIR=`pwd`/venv
|
|
else
|
|
bruisebase="${_VENV_ROOT}/${VENV_PREFIX}${REPONAME}"
|
|
if [ -d ${bruisebase}.${BRANCHNAME} ]; then
|
|
BRUISE_DIR=${bruisebase}.${BRANCHNAME}
|
|
elif [ -d ${bruisebase}.master ]; then
|
|
BRUISE_DIR=${bruisebase}.master
|
|
elif [ -d ${bruisebase} ]; then
|
|
BRUISE_DIR=${bruisebase}
|
|
fi
|
|
fi
|
|
if [ -f .python-version ]; then
|
|
pyenv versions|grep `cat .python-version`
|
|
if [ $? -ne 0 ]; then
|
|
echo "Looks like you need pyenv to install your version, I'll go do that now."
|
|
pyenv install `cat .python-version`
|
|
pyenv rehash
|
|
pip install virtualenv
|
|
cd .
|
|
fi
|
|
fi
|
|
if [ -z $BRUISE_DIR ]; then
|
|
echo "No environment exists, try running bruisemake"
|
|
return 1
|
|
fi
|
|
if [ $VENV_TYPE == "python" ]; then
|
|
echo "Environment loaded from ${BRUISE_DIR}"
|
|
source $BRUISE_DIR/bin/activate
|
|
fi
|
|
# todo, compare modified times on requirements.txt and the $VENV_PATH
|
|
if [ requirements.txt -nt $BRUISE_DIR ]; then
|
|
pip install -r requirements.txt
|
|
fi
|
|
}
|
|
function bruiselist() {
|
|
for i in `ls $_VENV_BASE`; do
|
|
ls $_VENV_BASE/$i
|
|
done
|
|
}
|
|
function bruiseuse() {
|
|
if [ -z $1 ]; then
|
|
__bruise
|
|
else
|
|
source $_VENV_BASE/python/$1/bin/activate
|
|
fi
|
|
}
|
|
function bruisedelete() {
|
|
for i in $*; do
|
|
_bruisedelete $i
|
|
done
|
|
}
|
|
function _bruisedelete() {
|
|
if [ -z $1 ]; then
|
|
echo "bruisedelete bruise_to_delete"
|
|
return 1
|
|
fi
|
|
for i in `ls $_VENV_BASE`; do
|
|
envs=`ls $_VENV_BASE/$i`
|
|
for e in $envs; do
|
|
if [ "X$e" == "X$1" ]; then
|
|
rm -rf $_VENV_BASE/$i/$e
|
|
fi
|
|
done
|
|
done
|
|
}
|
|
cd() {
|
|
builtin cd "$@"
|
|
__fetch_repo_details
|
|
if [ ! -z $VENV_TYPE ]; then
|
|
__bruise
|
|
fi
|
|
}
|