from xonsh.ptk.history import PromptToolkitHistory from xonsh.ptk.key_bindings import load_xonsh_bindings from xonsh.ptk.shortcuts import Prompter from xonsh.events import events from xonsh.shell import transform_command from xonsh.platform import HAS_PYGMENTS, ON_WINDOWS from xonsh.style_tools import ( partial_color_tokenize, _TokenType, DEFAULT_STYLE_DICT as _DEFAULT_STYLE_DICT, ) from xonsh.lazyimps import pygments, pyghooks, winutils from xonsh.pygments_cache import get_all_styles from xonsh.lazyasd import LazyObject Token = _TokenType() events.transmogrify("on_ptk_create", "LoadEvent") events.doc( "on_ptk_create", """ on_ptk_create(prompter: Prompter, history: PromptToolkitHistory, completer: PromptToolkitCompleter, bindings: KeyBindingManager) -> Fired after prompt toolkit has been initialized """, ) # Convert new ansicolor names to names # understood by PTK1 DEFAULT_STYLE_DICT = LazyObject( lambda: ansicolors_to_ptk1_names(_DEFAULT_STYLE_DICT),
from prompt_toolkit.lexers import PygmentsLexer from prompt_toolkit.enums import EditingMode from prompt_toolkit.key_binding import KeyBindings from prompt_toolkit.history import ThreadedHistory from prompt_toolkit.shortcuts import print_formatted_text as ptk_print from prompt_toolkit.shortcuts import CompleteStyle from prompt_toolkit.shortcuts.prompt import PromptSession from prompt_toolkit.formatted_text import PygmentsTokens from prompt_toolkit.styles import merge_styles, Style from prompt_toolkit.styles.pygments import ( style_from_pygments_cls, style_from_pygments_dict, ) Token = _TokenType() events.transmogrify("on_ptk_create", "LoadEvent") events.doc( "on_ptk_create", """ on_ptk_create(prompter: PromptSession, history: PromptToolkitHistory, completer: PromptToolkitCompleter, bindings: KeyBindings) -> Fired after prompt toolkit has been initialized """, ) class PromptToolkit2Shell(BaseShell): """The xonsh shell for prompt_toolkit v2."""