Siarhei Siniak
1530fcf108
1. 7f51a0332edd0c675c2d314ca3e62df7ef041281 deps/ipython (8.4.0-33-g7f51a0332)
27 lines
616 B
Python
27 lines
616 B
Python
"""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):
|
|
|
|
def in_prompt_tokens(self, cli=None):
|
|
return [(Token, os.getcwd()),
|
|
(Token.Prompt, '>>>')]
|
|
|
|
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
|
|
|
|
|
|
|
|
|