matir/skel
1
0
Fork 0
mirror of https://github.com/Matir/skel.git synced 2026-05-25 21:19:09 -07:00
Code Issues Packages Projects Releases Wiki Activity

Fix ipython prompt

Browse Source
This commit is contained in:
David Tomaschik
2026-02-10 15:48:04 -08:00
parent bddd4ed107
commit 0ebb331cb3
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
dotfiles/ipython/profile_default/startup/05-venv-manager.py
View File

@@ -1,6 +1,6 @@
import os
import sys
from IPython import prompts
from IPython.terminal import prompts
def _setup_environment():
ip = get_ipython()
@@ -36,15 +36,18 @@ def _setup_environment():
# 2. PROMPT LOGIC: Inject venv name into the UI
class VenvPrompts(prompts.Prompts):
_prompts = prompts
def in_prompt_tokens(self):
tokens = []
if active_venv_name:
tokens.append((prompts.Token.Prompt, f'({active_venv_name}) '))
tokens.append((self._prompts.Token.Prompt, f'({active_venv_name}) '))
tokens.extend([
(prompts.Token.Prompt, 'In ['),
(prompts.Token.PromptNum, str(self.shell.execution_count)),
(prompts.Token.Prompt, ']: '),
(self._prompts.Token.Prompt, 'In ['),
(self._prompts.Token.PromptNum, str(self.shell.execution_count)),
(self._prompts.Token.Prompt, ']: '),
])
return tokens