cursor shape changes with vi editing mode

Author: marskar

ptpython/key_bindings.py

@@ -1,3 +1,5 @@
+import sys
+
 from prompt_toolkit.application import get_app
 from prompt_toolkit.document import Document
 from prompt_toolkit.enums import DEFAULT_BUFFER
@@ -11,6 +13,7 @@
 )
 from prompt_toolkit.key_binding import KeyBindings
 from prompt_toolkit.keys import Keys
+from prompt_toolkit.key_binding.vi_state import InputMode, ViState
 
 from .utils import document_is_multiline_python
 
@@ -201,6 +204,35 @@ def _(event):
         " Abort when Control-C has been pressed. "
         event.app.exit(exception=KeyboardInterrupt, style="class:aborting")
 
+
+
+    def get_input_mode(self):
+        if sys.version_info[0] == 3:
+            app = get_app()
+            app.ttimeoutlen = python_input.ttimeoutlen
+            app.timeoutlen = python_input.timeoutlen
+
+        return self._input_mode
+
+
+    def set_input_mode(self, mode):
+        shape = {InputMode.NAVIGATION: 2, InputMode.REPLACE: 4}.get(mode, 6)
+        cursor = "\x1b[{} q".format(shape)
+
+        if hasattr(sys.stdout, "_cli"):
+            write = sys.stdout._cli.output.write_raw
+        else:
+            write = sys.stdout.write
+
+        write(cursor)
+        sys.stdout.flush()
+
+        self._input_mode = mode
+
+    if python_input.enable_modal_cursor:
+        ViState._input_mode = InputMode.INSERT
+        ViState.input_mode = property(get_input_mode, set_input_mode)
+
     return bindings
 
 

ptpython/python_input.py

@@ -234,6 +234,9 @@ def __init__(
         self.enable_system_bindings: bool = True
         self.enable_input_validation: bool = True
         self.enable_auto_suggest: bool = False
+        self.enable_modal_cursor: bool = True
+        self.ttimeoutlen: float = 0.01
+        self.timeoutlen: float = 0.5
         self.enable_mouse_support: bool = False
         self.enable_history_search: bool = False  # When True, like readline, going
         # back in history will filter the