add venv into ipython prompt

This commit is contained in:
TheK4n 2022-05-30 14:26:11 +03:00
parent 67b1e6c9bf
commit bc69ac6b7e

View File

@ -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