python-info: provide python version info for theming

Closes sorin-ionescu#1173 Closes sorin-ionescu#958
ayr-ton · May 24, 2018 · d9a936b · d9a936b
1 parent 911e155
commit d9a936b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/modules/python/README.md b/modules/python/README.md
@@ -135,6 +135,10 @@ following style in the `prompt_name_setup` function.
 Then add `$python_info[virtualenv]` to `$PROMPT` or `$RPROMPT` and call
 `python-info` in the `prompt_name_preexec` hook function.
 
+Similarly, you can use `:prezto:module:python:info:version:format` with `%v` for
+the version and add `$python_info[version]` to your prompt for the current
+python version/
+
 Authors
 -------
 

diff --git a/modules/python/functions/python-info b/modules/python/functions/python-info
@@ -11,6 +11,8 @@
 
 local virtualenv_format
 local virtualenv_formatted
+local version_format
+local version_formatted
 
 # Clean up previous $python_info.
 unset python_info
@@ -30,4 +32,10 @@ if [[ -n "$CONDA_DEFAULT_ENV" ]]; then
   python_info[virtualenv]="$virtualenv_formatted"
 fi
 
+zstyle -s ':prezto:module:python:info:version' format 'version_format'
+if [[ -n "$version_format" ]]; then
+  zformat -f version_formatted "$version_format" "v:${$(python3 --version)#Python }"
+  python_info[version]="$version_formatted"
+fi
+
 # }