Various typing fixes:

- Accept `AnyFormattedText` in `create_confirm_session`.
- Fixed incorrect import (in TYPE_CHECKING block).
- Removed type:ignore comments and casts that are no longer needed.

Author: jonathanslenders

src/prompt_toolkit/contrib/telnet/server.py

@@ -244,7 +244,7 @@ def _run_in_terminal(self, func: Callable[[], None]) -> None:
         # Make sure that when an application was active for this connection,
         # that we print the text above the application.
         if self.context:
-            self.context.run(run_in_terminal, func)  # type: ignore
+            self.context.run(run_in_terminal, func)
         else:
             raise RuntimeError("Called _run_in_terminal outside `run_application`.")
 

src/prompt_toolkit/document.py

@@ -54,15 +54,15 @@ class _ImmutableLineList(List[str]):
     def _error(self, *a: object, **kw: object) -> NoReturn:
         raise NotImplementedError("Attempt to modify an immutable list.")
 
-    __setitem__ = _error  # type: ignore
+    __setitem__ = _error
     append = _error
     clear = _error
     extend = _error
     insert = _error
     pop = _error
     remove = _error
     reverse = _error
-    sort = _error  # type: ignore
+    sort = _error
 
 
 class _DocumentCache:

src/prompt_toolkit/filters/app.py

@@ -4,7 +4,7 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, cast
+from typing import TYPE_CHECKING
 
 from prompt_toolkit.application.current import get_app
 from prompt_toolkit.cache import memoized
@@ -61,7 +61,7 @@ def has_focus(value: FocusableElement) -> Condition:
     """
     from prompt_toolkit.buffer import Buffer
     from prompt_toolkit.layout import walk
-    from prompt_toolkit.layout.containers import Container, Window, to_container
+    from prompt_toolkit.layout.containers import Window, to_container
     from prompt_toolkit.layout.controls import UIControl
 
     if isinstance(value, str):
@@ -94,7 +94,7 @@ def test() -> bool:
                 # focused.
                 current_window = get_app().layout.current_window
 
-                for c in walk(cast(Container, value)):
+                for c in walk(value):
                     if isinstance(c, Window) and c == current_window:
                         return True
                 return False

src/prompt_toolkit/key_binding/vi_state.py

@@ -6,7 +6,7 @@
 from prompt_toolkit.clipboard import ClipboardData
 
 if TYPE_CHECKING:
-    from .key_bindings.vi import TextObject
+    from .bindings.vi import TextObject
     from .key_processor import KeyPressEvent
 
 __all__ = [

src/prompt_toolkit/shortcuts/prompt.py

@@ -1474,7 +1474,7 @@ def prompt(
 
 
 def create_confirm_session(
-    message: str, suffix: str = " (y/n) "
+    message: AnyFormattedText, suffix: str = " (y/n) "
 ) -> PromptSession[bool]:
     """
     Create a `PromptSession` object for the 'confirm' function.
@@ -1505,7 +1505,7 @@ def _(event: E) -> None:
     return session
 
 
-def confirm(message: str = "Confirm?", suffix: str = " (y/n) ") -> bool:
+def confirm(message: AnyFormattedText = "Confirm?", suffix: str = " (y/n) ") -> bool:
     """
     Display a confirmation prompt that returns True/False.
     """