singularity コンテナ内の venv 環境でプロンプトを便利にする

venv 環境にいるとき、プロンプトの表示がよくない。

export PS1='...'

を venv 環境になると実行するようにする。

OmegaFold を Singularity コンテナ内でインストールするところに以下の作業を追加で行った。

vi ./init_prompt.sh

などとして以下のシェルスクリプトを作成した。

function virtualenv_info(){
    # Get Virtual Env
    if [[ -n "$VIRTUAL_ENV" ]]; then
        # Strip out the path and just leave the env name
        venv="${VIRTUAL_ENV##*/}"
    else
        # In case you don't have one activated
        venv=''
    fi
    [[ -n "$venv" ]] && echo "($venv) "
}
# disable the default virtualenv prompt change
export VIRTUAL_ENV_DISABLE_PROMPT=1
VENV="\$(virtualenv_info)"
export PS1="${VENV}(\s) \u@\H: \w $ ";

このコードを source した。

source ./init_prompt.sh