aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/prompt-toolkit/py2/prompt_toolkit/terminal/vt100_input.py
diff options
context:
space:
mode:
authormonster <monster@ydb.tech>2022-07-07 14:41:37 +0300
committermonster <monster@ydb.tech>2022-07-07 14:41:37 +0300
commit06e5c21a835c0e923506c4ff27929f34e00761c2 (patch)
tree75efcbc6854ef9bd476eb8bf00cc5c900da436a2 /contrib/python/prompt-toolkit/py2/prompt_toolkit/terminal/vt100_input.py
parent03f024c4412e3aa613bb543cf1660176320ba8f4 (diff)
downloadydb-06e5c21a835c0e923506c4ff27929f34e00761c2.tar.gz
fix ya.make
Diffstat (limited to 'contrib/python/prompt-toolkit/py2/prompt_toolkit/terminal/vt100_input.py')
-rw-r--r--contrib/python/prompt-toolkit/py2/prompt_toolkit/terminal/vt100_input.py520
1 files changed, 0 insertions, 520 deletions
diff --git a/contrib/python/prompt-toolkit/py2/prompt_toolkit/terminal/vt100_input.py b/contrib/python/prompt-toolkit/py2/prompt_toolkit/terminal/vt100_input.py
deleted file mode 100644
index 74841312fad..00000000000
--- a/contrib/python/prompt-toolkit/py2/prompt_toolkit/terminal/vt100_input.py
+++ /dev/null
@@ -1,520 +0,0 @@
-"""
-Parser for VT100 input stream.
-"""
-from __future__ import unicode_literals
-
-import os
-import re
-import six
-import termios
-import tty
-
-from six.moves import range
-
-from ..keys import Keys
-from ..key_binding.input_processor import KeyPress
-
-__all__ = (
- 'InputStream',
- 'raw_mode',
- 'cooked_mode',
-)
-
-_DEBUG_RENDERER_INPUT = False
-_DEBUG_RENDERER_INPUT_FILENAME = 'prompt-toolkit-render-input.log'
-
-
-# Regex matching any CPR response
-# (Note that we use '\Z' instead of '$', because '$' could include a trailing
-# newline.)
-_cpr_response_re = re.compile('^' + re.escape('\x1b[') + r'\d+;\d+R\Z')
-
-# Mouse events:
-# Typical: "Esc[MaB*" Urxvt: "Esc[96;14;13M" and for Xterm SGR: "Esc[<64;85;12M"
-_mouse_event_re = re.compile('^' + re.escape('\x1b[') + r'(<?[\d;]+[mM]|M...)\Z')
-
-# Regex matching any valid prefix of a CPR response.
-# (Note that it doesn't contain the last character, the 'R'. The prefix has to
-# be shorter.)
-_cpr_response_prefix_re = re.compile('^' + re.escape('\x1b[') + r'[\d;]*\Z')
-
-_mouse_event_prefix_re = re.compile('^' + re.escape('\x1b[') + r'(<?[\d;]*|M.{0,2})\Z')
-
-
-class _Flush(object):
- """ Helper object to indicate flush operation to the parser. """
- pass
-
-
-# Mapping of vt100 escape codes to Keys.
-ANSI_SEQUENCES = {
- '\x1b': Keys.Escape,
-
- '\x00': Keys.ControlSpace, # Control-Space (Also for Ctrl-@)
- '\x01': Keys.ControlA, # Control-A (home)
- '\x02': Keys.ControlB, # Control-B (emacs cursor left)
- '\x03': Keys.ControlC, # Control-C (interrupt)
- '\x04': Keys.ControlD, # Control-D (exit)
- '\x05': Keys.ControlE, # Contrel-E (end)
- '\x06': Keys.ControlF, # Control-F (cursor forward)
- '\x07': Keys.ControlG, # Control-G
- '\x08': Keys.ControlH, # Control-H (8) (Identical to '\b')
- '\x09': Keys.ControlI, # Control-I (9) (Identical to '\t')
- '\x0a': Keys.ControlJ, # Control-J (10) (Identical to '\n')
- '\x0b': Keys.ControlK, # Control-K (delete until end of line; vertical tab)
- '\x0c': Keys.ControlL, # Control-L (clear; form feed)
- '\x0d': Keys.ControlM, # Control-M (13) (Identical to '\r')
- '\x0e': Keys.ControlN, # Control-N (14) (history forward)
- '\x0f': Keys.ControlO, # Control-O (15)
- '\x10': Keys.ControlP, # Control-P (16) (history back)
- '\x11': Keys.ControlQ, # Control-Q
- '\x12': Keys.ControlR, # Control-R (18) (reverse search)
- '\x13': Keys.ControlS, # Control-S (19) (forward search)
- '\x14': Keys.ControlT, # Control-T
- '\x15': Keys.ControlU, # Control-U
- '\x16': Keys.ControlV, # Control-V
- '\x17': Keys.ControlW, # Control-W
- '\x18': Keys.ControlX, # Control-X
- '\x19': Keys.ControlY, # Control-Y (25)
- '\x1a': Keys.ControlZ, # Control-Z
-
- '\x1c': Keys.ControlBackslash, # Both Control-\ and Ctrl-|
- '\x1d': Keys.ControlSquareClose, # Control-]
- '\x1e': Keys.ControlCircumflex, # Control-^
- '\x1f': Keys.ControlUnderscore, # Control-underscore (Also for Ctrl-hypen.)
- '\x7f': Keys.Backspace, # (127) Backspace
- '\x1b[A': Keys.Up,
- '\x1b[B': Keys.Down,
- '\x1b[C': Keys.Right,
- '\x1b[D': Keys.Left,
- '\x1b[H': Keys.Home,
- '\x1bOH': Keys.Home,
- '\x1b[F': Keys.End,
- '\x1bOF': Keys.End,
- '\x1b[3~': Keys.Delete,
- '\x1b[3;2~': Keys.ShiftDelete, # xterm, gnome-terminal.
- '\x1b[3;5~': Keys.ControlDelete, # xterm, gnome-terminal.
- '\x1b[1~': Keys.Home, # tmux
- '\x1b[4~': Keys.End, # tmux
- '\x1b[5~': Keys.PageUp,
- '\x1b[6~': Keys.PageDown,
- '\x1b[7~': Keys.Home, # xrvt
- '\x1b[8~': Keys.End, # xrvt
- '\x1b[Z': Keys.BackTab, # shift + tab
- '\x1b[2~': Keys.Insert,
-
- '\x1bOP': Keys.F1,
- '\x1bOQ': Keys.F2,
- '\x1bOR': Keys.F3,
- '\x1bOS': Keys.F4,
- '\x1b[[A': Keys.F1, # Linux console.
- '\x1b[[B': Keys.F2, # Linux console.
- '\x1b[[C': Keys.F3, # Linux console.
- '\x1b[[D': Keys.F4, # Linux console.
- '\x1b[[E': Keys.F5, # Linux console.
- '\x1b[11~': Keys.F1, # rxvt-unicode
- '\x1b[12~': Keys.F2, # rxvt-unicode
- '\x1b[13~': Keys.F3, # rxvt-unicode
- '\x1b[14~': Keys.F4, # rxvt-unicode
- '\x1b[15~': Keys.F5,
- '\x1b[17~': Keys.F6,
- '\x1b[18~': Keys.F7,
- '\x1b[19~': Keys.F8,
- '\x1b[20~': Keys.F9,
- '\x1b[21~': Keys.F10,
- '\x1b[23~': Keys.F11,
- '\x1b[24~': Keys.F12,
- '\x1b[25~': Keys.F13,
- '\x1b[26~': Keys.F14,
- '\x1b[28~': Keys.F15,
- '\x1b[29~': Keys.F16,
- '\x1b[31~': Keys.F17,
- '\x1b[32~': Keys.F18,
- '\x1b[33~': Keys.F19,
- '\x1b[34~': Keys.F20,
-
- # Xterm
- '\x1b[1;2P': Keys.F13,
- '\x1b[1;2Q': Keys.F14,
- # '\x1b[1;2R': Keys.F15, # Conflicts with CPR response.
- '\x1b[1;2S': Keys.F16,
- '\x1b[15;2~': Keys.F17,
- '\x1b[17;2~': Keys.F18,
- '\x1b[18;2~': Keys.F19,
- '\x1b[19;2~': Keys.F20,
- '\x1b[20;2~': Keys.F21,
- '\x1b[21;2~': Keys.F22,
- '\x1b[23;2~': Keys.F23,
- '\x1b[24;2~': Keys.F24,
-
- '\x1b[1;5A': Keys.ControlUp, # Cursor Mode
- '\x1b[1;5B': Keys.ControlDown, # Cursor Mode
- '\x1b[1;5C': Keys.ControlRight, # Cursor Mode
- '\x1b[1;5D': Keys.ControlLeft, # Cursor Mode
-
- '\x1b[1;2A': Keys.ShiftUp,
- '\x1b[1;2B': Keys.ShiftDown,
- '\x1b[1;2C': Keys.ShiftRight,
- '\x1b[1;2D': Keys.ShiftLeft,
-
- # Tmux sends following keystrokes when control+arrow is pressed, but for
- # Emacs ansi-term sends the same sequences for normal arrow keys. Consider
- # it a normal arrow press, because that's more important.
- '\x1bOA': Keys.Up,
- '\x1bOB': Keys.Down,
- '\x1bOC': Keys.Right,
- '\x1bOD': Keys.Left,
-
- '\x1b[5A': Keys.ControlUp,
- '\x1b[5B': Keys.ControlDown,
- '\x1b[5C': Keys.ControlRight,
- '\x1b[5D': Keys.ControlLeft,
-
- '\x1bOc': Keys.ControlRight, # rxvt
- '\x1bOd': Keys.ControlLeft, # rxvt
-
- '\x1b[200~': Keys.BracketedPaste, # Start of bracketed paste.
-
- # Meta + arrow keys. Several terminals handle this differently.
- # The following sequences are for xterm and gnome-terminal.
- # (Iterm sends ESC followed by the normal arrow_up/down/left/right
- # sequences, and the OSX Terminal sends ESCb and ESCf for "alt
- # arrow_left" and "alt arrow_right." We don't handle these
- # explicitely, in here, because would could not distinguesh between
- # pressing ESC (to go to Vi navigation mode), followed by just the
- # 'b' or 'f' key. These combinations are handled in
- # the input processor.)
- '\x1b[1;3D': (Keys.Escape, Keys.Left),
- '\x1b[1;3C': (Keys.Escape, Keys.Right),
- '\x1b[1;3A': (Keys.Escape, Keys.Up),
- '\x1b[1;3B': (Keys.Escape, Keys.Down),
-
- # Sequences generated by numpad 5. Not sure what it means. (It doesn't
- # appear in 'infocmp'. Just ignore.
- '\x1b[E': Keys.Ignore, # Xterm.
- '\x1b[G': Keys.Ignore, # Linux console.
-}
-
-
-class _IsPrefixOfLongerMatchCache(dict):
- """
- Dictiory that maps input sequences to a boolean indicating whether there is
- any key that start with this characters.
- """
- def __missing__(self, prefix):
- # (hard coded) If this could be a prefix of a CPR response, return
- # True.
- if (_cpr_response_prefix_re.match(prefix) or _mouse_event_prefix_re.match(prefix)):
- result = True
- else:
- # If this could be a prefix of anything else, also return True.
- result = any(v for k, v in ANSI_SEQUENCES.items() if k.startswith(prefix) and k != prefix)
-
- self[prefix] = result
- return result
-
-
-_IS_PREFIX_OF_LONGER_MATCH_CACHE = _IsPrefixOfLongerMatchCache()
-
-
-class InputStream(object):
- """
- Parser for VT100 input stream.
-
- Feed the data through the `feed` method and the correct callbacks of the
- `input_processor` will be called.
-
- ::
-
- def callback(key):
- pass
- i = InputStream(callback)
- i.feed('data\x01...')
-
- :attr input_processor: :class:`~prompt_toolkit.key_binding.InputProcessor` instance.
- """
- # Lookup table of ANSI escape sequences for a VT100 terminal
- # Hint: in order to know what sequences your terminal writes to stdin, run
- # "od -c" and start typing.
- def __init__(self, feed_key_callback):
- assert callable(feed_key_callback)
-
- self.feed_key_callback = feed_key_callback
- self.reset()
-
- if _DEBUG_RENDERER_INPUT:
- self.LOG = open(_DEBUG_RENDERER_INPUT_FILENAME, 'ab')
-
- def reset(self, request=False):
- self._in_bracketed_paste = False
- self._start_parser()
-
- def _start_parser(self):
- """
- Start the parser coroutine.
- """
- self._input_parser = self._input_parser_generator()
- self._input_parser.send(None)
-
- def _get_match(self, prefix):
- """
- Return the key that maps to this prefix.
- """
- # (hard coded) If we match a CPR response, return Keys.CPRResponse.
- # (This one doesn't fit in the ANSI_SEQUENCES, because it contains
- # integer variables.)
- if _cpr_response_re.match(prefix):
- return Keys.CPRResponse
-
- elif _mouse_event_re.match(prefix):
- return Keys.Vt100MouseEvent
-
- # Otherwise, use the mappings.
- try:
- return ANSI_SEQUENCES[prefix]
- except KeyError:
- return None
-
- def _input_parser_generator(self):
- """
- Coroutine (state machine) for the input parser.
- """
- prefix = ''
- retry = False
- flush = False
-
- while True:
- flush = False
-
- if retry:
- retry = False
- else:
- # Get next character.
- c = yield
-
- if c == _Flush:
- flush = True
- else:
- prefix += c
-
- # If we have some data, check for matches.
- if prefix:
- is_prefix_of_longer_match = _IS_PREFIX_OF_LONGER_MATCH_CACHE[prefix]
- match = self._get_match(prefix)
-
- # Exact matches found, call handlers..
- if (flush or not is_prefix_of_longer_match) and match:
- self._call_handler(match, prefix)
- prefix = ''
-
- # No exact match found.
- elif (flush or not is_prefix_of_longer_match) and not match:
- found = False
- retry = True
-
- # Loop over the input, try the longest match first and
- # shift.
- for i in range(len(prefix), 0, -1):
- match= self._get_match(prefix[:i])
- if match:
- self._call_handler(match, prefix[:i])
- prefix = prefix[i:]
- found = True
-
- if not found:
- self._call_handler(prefix[0], prefix[0])
- prefix = prefix[1:]
-
- def _call_handler(self, key, insert_text):
- """
- Callback to handler.
- """
- if isinstance(key, tuple):
- for k in key:
- self._call_handler(k, insert_text)
- else:
- if key == Keys.BracketedPaste:
- self._in_bracketed_paste = True
- self._paste_buffer = ''
- else:
- self.feed_key_callback(KeyPress(key, insert_text))
-
- def feed(self, data):
- """
- Feed the input stream.
-
- :param data: Input string (unicode).
- """
- assert isinstance(data, six.text_type)
-
- if _DEBUG_RENDERER_INPUT:
- self.LOG.write(repr(data).encode('utf-8') + b'\n')
- self.LOG.flush()
-
- # Handle bracketed paste. (We bypass the parser that matches all other
- # key presses and keep reading input until we see the end mark.)
- # This is much faster then parsing character by character.
- if self._in_bracketed_paste:
- self._paste_buffer += data
- end_mark = '\x1b[201~'
-
- if end_mark in self._paste_buffer:
- end_index = self._paste_buffer.index(end_mark)
-
- # Feed content to key bindings.
- paste_content = self._paste_buffer[:end_index]
- self.feed_key_callback(KeyPress(Keys.BracketedPaste, paste_content))
-
- # Quit bracketed paste mode and handle remaining input.
- self._in_bracketed_paste = False
- remaining = self._paste_buffer[end_index + len(end_mark):]
- self._paste_buffer = ''
-
- self.feed(remaining)
-
- # Handle normal input character by character.
- else:
- for i, c in enumerate(data):
- if self._in_bracketed_paste:
- # Quit loop and process from this position when the parser
- # entered bracketed paste.
- self.feed(data[i:])
- break
- else:
- # Replace \r by \n. (Some clients send \r instead of \n
- # when enter is pressed. E.g. telnet and some other
- # terminals.)
-
- # XXX: We should remove this in a future version. It *is*
- # now possible to recognise the difference.
- # (We remove ICRNL/INLCR/IGNCR below.)
- # However, this breaks IPython and maybe other applications,
- # because they bind ControlJ (\n) for handling the Enter key.
-
- # When this is removed, replace Enter=ControlJ by
- # Enter=ControlM in keys.py.
- if c == '\r':
- c = '\n'
- self._input_parser.send(c)
-
- def flush(self):
- """
- Flush the buffer of the input stream.
-
- This will allow us to handle the escape key (or maybe meta) sooner.
- The input received by the escape key is actually the same as the first
- characters of e.g. Arrow-Up, so without knowing what follows the escape
- sequence, we don't know whether escape has been pressed, or whether
- it's something else. This flush function should be called after a
- timeout, and processes everything that's still in the buffer as-is, so
- without assuming any characters will folow.
- """
- self._input_parser.send(_Flush)
-
- def feed_and_flush(self, data):
- """
- Wrapper around ``feed`` and ``flush``.
- """
- self.feed(data)
- self.flush()
-
-
-class raw_mode(object):
- """
- ::
-
- with raw_mode(stdin):
- ''' the pseudo-terminal stdin is now used in raw mode '''
-
- We ignore errors when executing `tcgetattr` fails.
- """
- # There are several reasons for ignoring errors:
- # 1. To avoid the "Inappropriate ioctl for device" crash if somebody would
- # execute this code (In a Python REPL, for instance):
- #
- # import os; f = open(os.devnull); os.dup2(f.fileno(), 0)
- #
- # The result is that the eventloop will stop correctly, because it has
- # to logic to quit when stdin is closed. However, we should not fail at
- # this point. See:
- # https://github.com/jonathanslenders/python-prompt-toolkit/pull/393
- # https://github.com/jonathanslenders/python-prompt-toolkit/issues/392
-
- # 2. Related, when stdin is an SSH pipe, and no full terminal was allocated.
- # See: https://github.com/jonathanslenders/python-prompt-toolkit/pull/165
- def __init__(self, fileno):
- self.fileno = fileno
- try:
- self.attrs_before = termios.tcgetattr(fileno)
- except termios.error:
- # Ignore attribute errors.
- self.attrs_before = None
-
- def __enter__(self):
- # NOTE: On os X systems, using pty.setraw() fails. Therefor we are using this:
- try:
- newattr = termios.tcgetattr(self.fileno)
- except termios.error:
- pass
- else:
- newattr[tty.LFLAG] = self._patch_lflag(newattr[tty.LFLAG])
- newattr[tty.IFLAG] = self._patch_iflag(newattr[tty.IFLAG])
-
- # VMIN defines the number of characters read at a time in
- # non-canonical mode. It seems to default to 1 on Linux, but on
- # Solaris and derived operating systems it defaults to 4. (This is
- # because the VMIN slot is the same as the VEOF slot, which
- # defaults to ASCII EOT = Ctrl-D = 4.)
- newattr[tty.CC][termios.VMIN] = 1
-
- termios.tcsetattr(self.fileno, termios.TCSANOW, newattr)
-
- # Put the terminal in cursor mode. (Instead of application mode.)
- os.write(self.fileno, b'\x1b[?1l')
-
- @classmethod
- def _patch_lflag(cls, attrs):
- return attrs & ~(termios.ECHO | termios.ICANON | termios.IEXTEN | termios.ISIG)
-
- @classmethod
- def _patch_iflag(cls, attrs):
- return attrs & ~(
- # Disable XON/XOFF flow control on output and input.
- # (Don't capture Ctrl-S and Ctrl-Q.)
- # Like executing: "stty -ixon."
- termios.IXON | termios.IXOFF |
-
- # Don't translate carriage return into newline on input.
- termios.ICRNL | termios.INLCR | termios.IGNCR
- )
-
- def __exit__(self, *a, **kw):
- if self.attrs_before is not None:
- try:
- termios.tcsetattr(self.fileno, termios.TCSANOW, self.attrs_before)
- except termios.error:
- pass
-
- # # Put the terminal in application mode.
- # self._stdout.write('\x1b[?1h')
-
-
-class cooked_mode(raw_mode):
- """
- The opposide of ``raw_mode``, used when we need cooked mode inside a
- `raw_mode` block. Used in `CommandLineInterface.run_in_terminal`.::
-
- with cooked_mode(stdin):
- ''' the pseudo-terminal stdin is now used in cooked mode. '''
- """
- @classmethod
- def _patch_lflag(cls, attrs):
- return attrs | (termios.ECHO | termios.ICANON | termios.IEXTEN | termios.ISIG)
-
- @classmethod
- def _patch_iflag(cls, attrs):
- # Turn the ICRNL flag back on. (Without this, calling `input()` in
- # run_in_terminal doesn't work and displays ^M instead. Ptpython
- # evaluates commands using `run_in_terminal`, so it's important that
- # they translate ^M back into ^J.)
- return attrs | termios.ICRNL