add venv into ipython prompt
This commit is contained in:
parent
67b1e6c9bf
commit
bc69ac6b7e
@ -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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user