【Python】venvを使いながら環境変数PS1を設定してプロンプトを見やすくしたい

1
# This file must be used with "source bin/activate" *from bash*
2
# You cannot run it directly
3

4
deactivate () {
5
  # reset old environment variables
6
  if [ -n "${_OLD_VIRTUAL_PATH:-}" ] ; then    # <--- deactivateする時に通る
7
    PATH="${_OLD_VIRTUAL_PATH:-}"
8
    export PATH
9
    unset _OLD_VIRTUAL_PATH
10
  fi
11
  if [ -n "${_OLD_VIRTUAL_PYTHONHOME:-}" ] ; then
12
    PYTHONHOME="${_OLD_VIRTUAL_PYTHONHOME:-}"
13
    export PYTHONHOME
14
    unset _OLD_VIRTUAL_PYTHONHOME
15
  fi
16

17
  # Call hash to forget past commands. Without forgetting
18
  # past commands the $PATH changes we made may not be respected
19
  hash -r 2> /dev/null
20

21
  if [ -n "${_OLD_VIRTUAL_PS1:-}" ] ; then
22
    PS1="${_OLD_VIRTUAL_PS1:-}"
23
    export PS1
24
    unset _OLD_VIRTUAL_PS1
25
  fi
26

27
  unset VIRTUAL_ENV
28
  unset VIRTUAL_ENV_PROMPT
29
  if [ ! "${1:-}" = "nondestructive" ] ; then
30
  # Self destruct!
31
    unset -f deactivate
32
  fi
33
}
34

35
# unset irrelevant variables
36
deactivate nondestructive
37

38
# on Windows, a path can contain colons and backslashes and has to be converted:
39
if [ "${OSTYPE:-}" = "cygwin" ] || [ "${OSTYPE:-}" = "msys" ] ; then
40
  # transform D:\path\to\venv to /d/path/to/venv on MSYS
41
  # and to /cygdrive/d/path/to/venv on Cygwin
42
  export VIRTUAL_ENV=$(cygpath /home/user/NotionSynchronizer/.venv)
43
else
44
  # use the path as-is
45
  export VIRTUAL_ENV=/home/user/NotionSynchronizer/.venv
46
fi
47

48
_OLD_VIRTUAL_PATH="$PATH"
49
PATH="$VIRTUAL_ENV/"bin":$PATH"
50
export PATH
51

52
# unset PYTHONHOME if set
53
# this will fail if PYTHONHOME is set to the empty string (which is bad anyway)
54
# could use `if (set -u; : $PYTHONHOME) ;` in bash
55
if [ -n "${PYTHONHOME:-}" ] ; then
56
  _OLD_VIRTUAL_PYTHONHOME="${PYTHONHOME:-}"
57
  unset PYTHONHOME
58
fi
59

60
if [ -z "${VIRTUAL_ENV_DISABLE_PROMPT:-}" ] ; then
61
  _OLD_VIRTUAL_PS1="${PS1:-}"    # <--- activateする時に通る
62
  PS1='(.venv) '"${PS1:-}"
63
  export PS1
64
  VIRTUAL_ENV_PROMPT='(.venv) '
65
  export VIRTUAL_ENV_PROMPT
66
fi
67

68
# Call hash to forget past commands. Without forgetting
69
# past commands the $PATH changes we made may not be respected
70
hash -r 2> /dev/null

1
function _current_branch() {
2
  _git_branch=$(git branch --show-current 2>/dev/null) && echo "[branch: $_git_branch] "
3
}
4

5
function edit-ps1-env() {
6
  _my_ps1='\[\e]0;\u@\h: \w\a\]${debian_chroot:+($debian_chroot)}\[\033[01;32m\]\u@\h\[\033[00m\]:\[\033[01;34m\]\w\[\033[00m\] \[\033[01;33m\]$(_current_branch)\[\033[00m\]\[\033[01;36m\]\$\[\033[00m\] '
7
  case ${1} in
8
    --activate | -a)
9
      export _OLD_VIRTUAL_PS1="$_my_ps1"
10
      _my_venv_dir='.venv'
11
      PS1="\[\033[01;31m\]($_my_venv_dir)\[\033[00m\] ""${_my_ps1:-}"
12
      return 0
13
      ;;
14
    --deactivate | -d)
15
      # nothing to do
16
      return 0
17
      ;;
18
    *)
19
      echo "[ERROR] invalid options: '${1}'"
20
      return 1
21
      ;;
22
  esac
23
}