cwd_prompt.py
22 lines
| 584 B
| text/x-python
|
PythonLexer
Matthias Bussonnier
|
r22430 | """This is an example that shows how to create new prompts for IPython | ||
""" | ||||
from IPython.terminal.prompts import Prompts, Token | ||||
import os | ||||
class MyPrompt(Prompts): | ||||
Matthias Bussonnier
|
r28557 | def in_prompt_tokens(self): | ||
Matthias Bussonnier
|
r28566 | return [(Token, os.getcwd()), (Token.Prompt, ">>>")] | ||
Matthias Bussonnier
|
r28557 | |||
Matthias Bussonnier
|
r22430 | |||
def load_ipython_extension(shell): | ||||
new_prompts = MyPrompt(shell) | ||||
new_prompts.old_prompts = shell.prompts | ||||
shell.prompts = new_prompts | ||||
def unload_ipython_extension(shell): | ||||
if not hasattr(shell.prompts, 'old_prompts'): | ||||
print("cannot unload") | ||||
else: | ||||
shell.prompts = shell.prompts.old_prompts | ||||