From bc69ac6b7ef3fc39bc2ba35dbd4ac71056c9c808 Mon Sep 17 00:00:00 2001
From: TheK4n <djvlad967891@gmail.com>
Date: Mon, 30 May 2022 14:26:11 +0300
Subject: [PATCH] add venv into ipython prompt

---
 sub/ipython/ipython_config.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/sub/ipython/ipython_config.py b/sub/ipython/ipython_config.py
index 3457c46..4dc2970 100644
--- a/sub/ipython/ipython_config.py
+++ b/sub/ipython/ipython_config.py
@@ -18,19 +18,27 @@ def get_branch():
         return ""
 
 
+def get_venv():
+    try:
+        return f"({os.path.basename(os.environ['VIRTUAL_ENV'])})─"
+    except:
+        return ""
+
+
 class CustomPrompt(Prompts):
 
     def in_prompt_tokens(self, cli=None):
         return [
-            (Token, "\n┌──("),
+            (Token, "\n┌──"),
+            (Token, get_venv()),
+            (Token, "("),
             (Token.Name.Class, os.getlogin()),
-            (Token.Prompt, "@"),
-            (Token.Name.Class, "ipython v" + python_version()),
+            (Token.Name.Class, "@"),
+            (Token.Name.Class, "v" + python_version()),
             (Token, ")-["),
             (Token.OutPrompt, os.getcwd()),
             (Token, "]"),
             (Token.Generic.Subheading, get_branch()),
-            (Token, " "),
             (Token, "\n└─"),
             (
                 Token.Prompt