feat(number): auto adjust cursor position for negative toggle

kazhala · Dec 7, 2021 · 318b2bf · 318b2bf
1 parent 4df3fbb
commit 318b2bf
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/InquirerPy/prompts/number.py b/InquirerPy/prompts/number.py
@@ -396,9 +396,17 @@ def _handle_negative_toggle(self, _) -> None:
             self._whole_buffer.text = "0"
             return
         if self._whole_buffer.text.startswith("-"):
+            move_cursor = self._whole_buffer.cursor_position < len(
+                self._whole_buffer.text
+            )
             self._whole_buffer.text = self._whole_buffer.text[1:]
+            if move_cursor:
+                self._whole_buffer.cursor_position -= 1
         else:
+            move_cursor = self._whole_buffer.cursor_position != 0
             self._whole_buffer.text = f"-{self._whole_buffer.text}"
+            if move_cursor:
+                self._whole_buffer.cursor_position += 1
 
     def _on_whole_text_change(self, buffer: Buffer) -> None:
         self._whole_width = len(buffer.text) + 1