Fix the remaining type hint issues.

Author: amjith

litecli/clibuffer.py

@@ -9,8 +9,10 @@
 
 def cli_is_multiline(cli: Any) -> Filter:
     @Condition
-    def cond():
-        doc = get_app().layout.get_buffer_by_name(DEFAULT_BUFFER).document
+    def cond() -> bool:
+        buf = get_app().layout.get_buffer_by_name(DEFAULT_BUFFER)
+        assert buf is not None
+        doc = buf.document
 
         if not cli.multi_line:
             return False

litecli/clistyle.py

@@ -46,7 +46,7 @@
 PROMPT_STYLE_TO_TOKEN: dict[str, Token] = {v: k for k, v in TOKEN_TO_PROMPT_STYLE.items()}
 
 
-def parse_pygments_style(token_name: str, style_object, style_dict: dict[str, str]) -> tuple[Token, str]:
+def parse_pygments_style(token_name: str, style_object: PygmentsStyle | dict, style_dict: dict[str, str]) -> tuple[Token, str]:
     """Parse token type and style string.
 
     :param token_name: str name of Pygments token. Example: "Token.String"
@@ -55,10 +55,10 @@ def parse_pygments_style(token_name: str, style_object, style_dict: dict[str, st
 
     """
     token_type = string_to_tokentype(token_name)
-    try:
+    if isinstance(style_object, PygmentsStyle):
         other_token_type = string_to_tokentype(style_dict[token_name])
         return token_type, style_object.styles[other_token_type]
-    except AttributeError:
+    else:
         return token_type, style_dict[token_name]
 
 
@@ -90,7 +90,7 @@ def style_factory(name: str, cli_style: dict[str, str]) -> _MergedStyle:
     return merge_styles([style_from_pygments_cls(style), override_style, Style(prompt_styles)])
 
 
-def style_factory_output(name: str, cli_style: dict[str, str]):
+def style_factory_output(name: str, cli_style: dict[str, str]) -> PygmentsStyle:
     try:
         style = pygments.styles.get_style_by_name(name).styles
     except ClassNotFound:

litecli/clitoolbar.py

@@ -1,13 +1,13 @@
 from __future__ import annotations
 
-from typing import Callable
+from typing import Callable, Any
 
 from prompt_toolkit.key_binding.vi_state import InputMode
 from prompt_toolkit.enums import EditingMode
 from prompt_toolkit.application import get_app
 
 
-def create_toolbar_tokens_func(cli, show_fish_help: Callable[[], bool]) -> Callable[[], list[tuple[str, str]]]:
+def create_toolbar_tokens_func(cli: Any, show_fish_help: Callable[[], bool]) -> Callable[[], list[tuple[str, str]]]:
     """Return a function that generates the toolbar tokens."""
 
     def get_toolbar_tokens() -> list[tuple[str, str]]:

litecli/completion_refresher.py

@@ -45,6 +45,7 @@ def refresh(
                 # if DB is memory, needed to use same connection
                 # So can't use same connection with different thread
                 self._bg_refresh(executor, callbacks, completer_options)
+                return [(None, None, None, "Auto-completion refresh started in the background.")]
             else:
                 self._completer_thread = threading.Thread(
                     target=self._bg_refresh,
@@ -97,7 +98,7 @@ def _bg_refresh(
             callback(completer)
 
 
-def refresher(name: str, refreshers: dict[str, Callable] = CompletionRefresher.refreshers):
+def refresher(name: str, refreshers: dict[str, Callable] = CompletionRefresher.refreshers) -> Callable:
     """Decorator to add the decorated function to the dictionary of
     refreshers. Any function decorated with a @refresher will be executed as
     part of the completion refresh routine."""

litecli/key_bindings.py

@@ -85,7 +85,7 @@ def _(event: KeyPressEvent) -> None:
         b.complete_state = None
 
     @kb.add("right", filter=completion_is_selected)
-    def _(event):
+    def _(event: KeyPressEvent) -> None:
         """Accept the completion that is selected in the dropdown menu."""
         _logger.debug("Detected right-arrow key.")