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