XorA
/
bash-it
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add venv prompt to base theme

pull/2060/head
Petar Nikolovski 2022-01-26 21:55:00 +01:00
parent 1ec8f9df64
commit ec85f44d70
2 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
themes/base.theme.bash
View File

@ -625,3 +625,14 @@ function _save-and-reload-history() {
local autosave=${1:-0} local autosave=${1:-0}
[[ $autosave -eq 1 ]] && history -a && history -c && history -r [[ $autosave -eq 1 ]] && history -a && history -c && history -r
} }
function venv_prompt() {
local python_venv=""
if [[ -n "${CONDA_DEFAULT_ENV:-}" ]]; then
python_venv="${CONDA_DEFAULT_ENV}"
PYTHON_VENV_CHAR=${CONDA_PYTHON_VENV_CHAR}
elif [[ -n "${VIRTUAL_ENV:-}" ]]; then
python_venv="${VIRTUAL_ENV##*/}"
fi
[[ -n "${python_venv}" ]] && echo "${PYTHON_VENV_CHAR}${python_venv}"
}

15
themes/robbyrussell/robbyrussell.theme.bash
View File

@ -18,19 +18,16 @@ function git_prompt_info() {
echo -e "$SCM_PREFIX${bold_red}$SCM_BRANCH$SCM_STATE$SCM_SUFFIX" echo -e "$SCM_PREFIX${bold_red}$SCM_BRANCH$SCM_STATE$SCM_SUFFIX"
} }
function venv_prompt() { function venv_parenthases() {
python_venv="" if [[ $(venv_prompt) != "" ]]; then
# Detect python venv echo "($(venv_prompt)) "
if [[ -n "${CONDA_DEFAULT_ENV}" ]]; then else
python_venv="($PYTHON_VENV_CHAR${CONDA_DEFAULT_ENV}) " echo ""
elif [[ -n "${VIRTUAL_ENV}" ]]; then
python_venv="($PYTHON_VENV_CHAR$(basename "${VIRTUAL_ENV}")) "
fi fi
[[ -n "${python_venv}" ]] && echo "${python_venv}"
} }
function prompt_command() { function prompt_command() {
PS1="$(venv_prompt)${bold_green}➜ ${bold_cyan}\W${reset_color}$(scm_prompt_info)${normal} " PS1="$(venv_parenthases)${bold_green}➜ ${bold_cyan}\W${reset_color}$(scm_prompt_info)${normal} "
} }
PROMPT_COMMAND=prompt_command PROMPT_COMMAND=prompt_command