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 "" return ""
def get_venv():
try:
return f"({os.path.basename(os.environ['VIRTUAL_ENV'])})─"
except:
return ""
class CustomPrompt(Prompts): class CustomPrompt(Prompts):
def in_prompt_tokens(self, cli=None): def in_prompt_tokens(self, cli=None):
return [ return [
(Token, "\n┌──("), (Token, "\n┌──"),
(Token, get_venv()),
(Token, "("),
(Token.Name.Class, os.getlogin()), (Token.Name.Class, os.getlogin()),
(Token.Prompt, "@"), (Token.Name.Class, "@"),
(Token.Name.Class, "ipython v" + python_version()), (Token.Name.Class, "v" + python_version()),
(Token, ")-["), (Token, ")-["),
(Token.OutPrompt, os.getcwd()), (Token.OutPrompt, os.getcwd()),
(Token, "]"), (Token, "]"),
(Token.Generic.Subheading, get_branch()), (Token.Generic.Subheading, get_branch()),
(Token, " "),
(Token, "\n└─"), (Token, "\n└─"),
( (
Token.Prompt Token.Prompt