def loop(cmd, history_file):
from prompt_toolkit import CommandLineInterface, AbortAction
from prompt_toolkit import Exit
from prompt_toolkit.layout import Layout
from prompt_toolkit.line import Line
from prompt_toolkit.renderer import Output
cli_line = Line(completer=SQLCompleter(cmd.connection, cmd.lines),
history=TruncatedFileHistory(
history_file, max_length=MAX_HISTORY_LENGTH))
layout = Layout(
before_input=CrashPrompt(cmd.lines),
menus=[],
lexer=SqlLexer,
bottom_toolbars=[],
show_tildes=False,
)
key_binding_factories = _detect_key_bindings()
cli = CommandLineInterface(style=MonokaiStyle,
layout=layout,
line=cli_line,
key_binding_factories=key_binding_factories)
output = Output(cli.renderer.stdout)
global get_num_columns
def get_num_columns():
return output.get_size().columns
try:
while True:
doc = cli.read_input(on_exit=AbortAction.RAISE_EXCEPTION)
cmd.process(doc.text)
except Exit: # Quit on Ctrl-D keypress
cmd.logger.warn(u'Bye!')
return
def __init__(
self, stdout: TextIO, default_color_depth: Optional[ColorDepth] = None
) -> None:
self.win32_output = Win32Output(stdout, default_color_depth=default_color_depth)
self.vt100_output = Vt100_Output(
stdout, lambda: Size(0, 0), default_color_depth=default_color_depth
)
@property
def responds_to_cpr(self) -> bool:
return False # We don't need this on Windows.
def __getattr__(self, name: str) -> Any:
if name in (
"get_size",
"get_rows_below_cursor_position",
"enable_mouse_support",
"disable_mouse_support",
"scroll_buffer_to_prompt",
"get_win32_screen_buffer_info",
"enable_bracketed_paste",
"disable_bracketed_paste",
):
return getattr(self.win32_output, name)
else:
return getattr(self.vt100_output, name)
Output.register(ConEmuOutput)
class ConEmuOutput(object):
"""
ConEmu (Windows) output abstraction.
ConEmu is a Windows console application, but it also supports ANSI escape
sequences. This output class is actually a proxy to both `Win32Output` and
`Vt100_Output`. It uses `Win32Output` for console sizing and scrolling, but
all cursor movements and scrolling happens through the `Vt100_Output`.
This way, we can have 256 colors in ConEmu and Cmder. Rendering will be
even a little faster as well.
http://conemu.github.io/
http://gooseberrycreative.com/cmder/
"""
def __init__(self, stdout):
self.win32_output = Win32Output(stdout)
self.vt100_output = Vt100_Output(stdout, lambda: None)
def __getattr__(self, name):
if name in ('get_size', 'get_rows_below_cursor_position',
'enable_mouse_support', 'disable_mouse_support',
'scroll_buffer_to_prompt', 'get_win32_screen_buffer_info',
'enable_bracketed_paste', 'disable_bracketed_paste'):
return getattr(self.win32_output, name)
else:
return getattr(self.vt100_output, name)
Output.register(ConEmuOutput)
self.vt100_output.flush()
finally:
# Restore console mode.
windll.kernel32.SetConsoleMode(self._hconsole, original_mode)
def __getattr__(self, name: str) -> Any:
if name in ('get_size', 'get_rows_below_cursor_position',
'enable_mouse_support', 'disable_mouse_support',
'scroll_buffer_to_prompt', 'get_win32_screen_buffer_info',
'enable_bracketed_paste', 'disable_bracketed_paste'):
return getattr(self.win32_output, name)
else:
return getattr(self.vt100_output, name)
Output.register(Windows10_Output)
def is_win_vt100_enabled() -> bool:
"""
Returns True when we're running Windows and VT100 escape sequences are
supported.
"""
if not is_windows():
return False
hconsole = windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE)
# Get original console mode.
original_mode = DWORD(0)
windll.kernel32.GetConsoleMode(hconsole, byref(original_mode))
self.vt100_output.flush()
finally:
# Restore console mode.
windll.kernel32.SetConsoleMode(self._hconsole, original_mode)
def __getattr__(self, name: str) -> Any:
if name in ('get_size', 'get_rows_below_cursor_position',
'enable_mouse_support', 'disable_mouse_support',
'scroll_buffer_to_prompt', 'get_win32_screen_buffer_info',
'enable_bracketed_paste', 'disable_bracketed_paste'):
return getattr(self.win32_output, name)
else:
return getattr(self.vt100_output, name)
Output.register(Windows10_Output)
def is_win_vt100_enabled() -> bool:
"""
Returns True when we're running Windows and VT100 escape sequences are
supported.
"""
if not is_windows():
return False
hconsole = windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE)
# Get original console mode.
original_mode = DWORD(0)
windll.kernel32.GetConsoleMode(hconsole, byref(original_mode))
