Added more type annotations.

Author: jonathanslenders

prompt_toolkit/contrib/regular_languages/compiler.py

@@ -422,7 +422,7 @@ def get_tuples() -> Iterable[Tuple[str, Tuple[int, int]]]:
             for r, re_match in self._re_matches:
                 for group_name, group_index in r.groupindex.items():
                     if group_name != _INVALID_TRAILING_INPUT:
-                        regs = cast(Tuple[Tuple[int, int], ...], re_match.regs)
+                        regs = re_match.regs
                         reg = regs[group_index]
                         node = self._group_names_to_nodes[group_name]
                         yield (node, reg)

prompt_toolkit/contrib/ssh/server.py

@@ -97,7 +97,9 @@ async def _interact(self) -> None:
                 self._chan.close()
                 self._input.close()
 
-    def terminal_size_changed(self, width, height, pixwidth, pixheight):
+    def terminal_size_changed(
+        self, width: int, height: int, pixwidth, pixheight
+    ) -> None:
         # Send resize event to the current application.
         if self.app_session and self.app_session.app:
             self.app_session.app._on_resize()
@@ -144,7 +146,7 @@ def __init__(
     ) -> None:
         self.interact = interact
 
-    def begin_auth(self, username):
+    def begin_auth(self, username: str) -> bool:
         # No authentication.
         return False
 

prompt_toolkit/filters/__init__.py

@@ -20,3 +20,49 @@
 from .base import Always, Condition, Filter, FilterOrBool, Never
 from .cli import *
 from .utils import is_true, to_filter
+
+__all__ = [
+    # app
+    "has_arg",
+    "has_completions",
+    "completion_is_selected",
+    "has_focus",
+    "buffer_has_focus",
+    "has_selection",
+    "has_validation_error",
+    "is_done",
+    "is_read_only",
+    "is_multiline",
+    "renderer_height_is_known",
+    "in_editing_mode",
+    "in_paste_mode",
+    "vi_mode",
+    "vi_navigation_mode",
+    "vi_insert_mode",
+    "vi_insert_multiple_mode",
+    "vi_replace_mode",
+    "vi_selection_mode",
+    "vi_waiting_for_text_object_mode",
+    "vi_digraph_mode",
+    "vi_recording_macro",
+    "emacs_mode",
+    "emacs_insert_mode",
+    "emacs_selection_mode",
+    "shift_selection_mode",
+    "is_searching",
+    "control_is_searchable",
+    "vi_search_direction_reversed",
+    # base.
+    "Filter",
+    "Never",
+    "Always",
+    "Condition",
+    "FilterOrBool",
+    # utils.
+    "is_true",
+    "to_filter",
+]
+
+from .cli import __all__ as cli_all
+
+__all__.extend(cli_all)

prompt_toolkit/filters/base.py

@@ -118,7 +118,7 @@ def __init__(self, filters: Iterable[Filter]) -> None:
 
         for f in filters:
             if isinstance(f, _AndList):  # Turn nested _AndLists into one.
-                self.filters.extend(cast(_AndList, f).filters)
+                self.filters.extend(f.filters)
             else:
                 self.filters.append(f)
 
@@ -139,7 +139,7 @@ def __init__(self, filters: Iterable[Filter]) -> None:
 
         for f in filters:
             if isinstance(f, _OrList):  # Turn nested _OrLists into one.
-                self.filters.extend(cast(_OrList, f).filters)
+                self.filters.extend(f.filters)
             else:
                 self.filters.append(f)
 
@@ -203,7 +203,7 @@ def feature_is_active():  # `feature_is_active` becomes a Filter.
     :param func: Callable which takes no inputs and returns a boolean.
     """
 
-    def __init__(self, func: Callable[[], bool]):
+    def __init__(self, func: Callable[[], bool]) -> None:
         self.func = func
 
     def __call__(self) -> bool:

prompt_toolkit/formatted_text/base.py

@@ -71,7 +71,7 @@ def to_formatted_text(
     elif isinstance(value, str):
         result = [("", value)]
     elif isinstance(value, list):
-        result = cast(StyleAndTextTuples, value)
+        result = value  # StyleAndTextTuples
     elif hasattr(value, "__pt_formatted_text__"):
         result = cast("MagicFormattedText", value).__pt_formatted_text__()
     elif callable(value):
@@ -146,7 +146,7 @@ def __init__(self, text: str) -> None:
         self.text = text
 
     def format(self, *values: AnyFormattedText) -> AnyFormattedText:
-        def get_result():
+        def get_result() -> AnyFormattedText:
             # Split the template in parts.
             parts = self.text.split("{}")
             assert len(parts) - 1 == len(values)
@@ -166,7 +166,7 @@ def merge_formatted_text(items: Iterable[AnyFormattedText]) -> AnyFormattedText:
     Merge (Concatenate) several pieces of formatted text together.
     """
 
-    def _merge_formatted_text():
+    def _merge_formatted_text() -> AnyFormattedText:
         result = FormattedText()
         for i in items:
             result.extend(to_formatted_text(i))

prompt_toolkit/input/win32_pipe.py

@@ -53,7 +53,7 @@ def __init__(self) -> None:
     def closed(self) -> bool:
         return self._closed
 
-    def fileno(self):
+    def fileno(self) -> int:
         """
         The windows pipe doesn't depend on the file handle.
         """

prompt_toolkit/key_binding/bindings/vi.py

@@ -399,7 +399,7 @@ def load_vi_bindings() -> KeyBindingsBase:
         (
             ("g", "?"),
             Always(),
-            lambda string: cast(str, codecs.encode(string, "rot_13")),
+            lambda string: codecs.encode(string, "rot_13"),
         ),
         # To lowercase
         (("g", "u"), Always(), lambda string: string.lower()),
@@ -1964,7 +1964,7 @@ def _delete_before_multiple_cursors(event: E) -> None:
             event.app.output.bell()
 
     @handle("delete", filter=vi_insert_multiple_mode)
-    def _delete_after_multiple_cursors(event):
+    def _delete_after_multiple_cursors(event: E) -> None:
         """
         Delete, using multiple cursors.
         """

prompt_toolkit/layout/containers.py

@@ -751,7 +751,7 @@ class FloatContainer(Container):
                        floats=[
                            Float(xcursor=True,
                                 ycursor=True,
-                                layout=CompletionMenu(...))
+                                content=CompletionMenu(...))
                        ])
 
     :param z_index: (int or None) When specified, this can be used to bring
@@ -1067,7 +1067,7 @@ def __init__(
         allow_cover_cursor: bool = False,
         z_index: int = 1,
         transparent: bool = False,
-    ):
+    ) -> None:
 
         assert z_index >= 1
 
@@ -2700,7 +2700,7 @@ def to_container(container: AnyContainer) -> Container:
     if isinstance(container, Container):
         return container
     elif hasattr(container, "__pt_container__"):
-        return to_container(cast("MagicContainer", container).__pt_container__())
+        return to_container(container.__pt_container__())
     else:
         raise ValueError("Not a container object: %r" % (container,))
 

prompt_toolkit/layout/layout.py

@@ -217,7 +217,7 @@ def current_window(self) -> Window:
         return self._stack[-1]
 
     @current_window.setter
-    def current_window(self, value: Window):
+    def current_window(self, value: Window) -> None:
         " Set the :class:`.Window` object to be currently focused. "
         self._stack.append(value)
 

prompt_toolkit/layout/processors.py

@@ -849,7 +849,7 @@ def filter_processor(item: Processor) -> Optional[Processor]:
             include_default_input_processors=False,
             lexer=main_control.lexer,
             preview_search=True,
-            search_buffer_control=cast(SearchBufferControl, ti.buffer_control),
+            search_buffer_control=ti.buffer_control,
         )
 
         return buffer_control.create_content(ti.width, ti.height, preview_search=True)