From e3dee2dc678b36a334cba392e0548bd88544e990 Mon Sep 17 00:00:00 2001
From: Bert Peters <bert@bertptrs.nl>
Date: Mon, 10 Sep 2018 14:07:12 +0200
Subject: [PATCH] Handle python virtualenv in prompt.

---
 zsh/.config/zsh/.zshtheme | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/zsh/.config/zsh/.zshtheme b/zsh/.config/zsh/.zshtheme
index 5937c09..aa70820 100644
--- a/zsh/.config/zsh/.zshtheme
+++ b/zsh/.config/zsh/.zshtheme
@@ -12,6 +12,12 @@ function prompt_last_cmd() {
 	fi
 }
 
+function prompt_handle_virtualenv() {
+	if [[ -n $VIRTUAL_ENV ]]; then
+		echo "%{$fg_bold[green]%}🐍 $(basename $VIRTUAL_ENV)%b%f%k "
+	fi
+}
+
 function prompt_ps1() {
 	local ps1 rs='%b%f%k'
     local dir="%{$fg[%~]%}%8~$rs"
@@ -24,6 +30,7 @@ function prompt_ps1() {
 		"$hist"
 		"$dir"
 		$'\n'
+		'$(prompt_handle_virtualenv)'
 		"$prompt"
 	)
     echo "${(j::)ps1}"
@@ -42,3 +49,6 @@ function prompt_rprompt() {
 
 PS1="$(prompt_ps1)"
 RPROMPT="$(prompt_rprompt)"
+
+# We handle virtualenv ourselves, no need to do it twice.
+export VIRTUAL_ENV_DISABLE_PROMPT=1