Enhance type safety in md2term.py by adding type annotations for code_block_lines and asserting token types after markdown rendering. This improves code clarity and ensures expected data structures are used throughout the rendering process.

Author: statico

md2term.py

@@ -32,7 +32,7 @@ class TerminalRenderer:
     def __init__(self, console: Console):
         self.console = console
         self.in_code_block = False
-        self.code_block_lines = []
+        self.code_block_lines: List[str] = []
         self.code_block_lang = None
 
     def render(self, tokens: List[Dict[str, Any]]) -> None:
@@ -269,6 +269,8 @@ def _render_callout(self, content: str, callout_info: Dict[str, str]) -> None:
             markdown = mistune.create_markdown(renderer=None)
             try:
                 tokens = markdown(callout_content)
+                # Type assertion since we know renderer=None returns tokens
+                assert isinstance(tokens, list)
                 temp_renderer = TerminalRenderer(temp_console)
                 temp_renderer.render(tokens)
             except Exception:
@@ -495,6 +497,8 @@ def _render_and_count(self, content: str) -> None:
         # Parse and render to temp console
         markdown = mistune.create_markdown(renderer=None)
         tokens = markdown(content)
+        # Type assertion since we know renderer=None returns tokens
+        assert isinstance(tokens, list)
         temp_renderer = TerminalRenderer(temp_console)
         temp_renderer.render(tokens)
 
@@ -529,6 +533,8 @@ def _render_final(self) -> None:
             try:
                 markdown = mistune.create_markdown(renderer=None)
                 tokens = markdown(self.buffer)
+                # Type assertion since we know renderer=None returns tokens
+                assert isinstance(tokens, list)
                 renderer = TerminalRenderer(self.console)
                 renderer.render(tokens)
             except Exception:
@@ -544,6 +550,8 @@ def finalize(self) -> None:
             try:
                 markdown = mistune.create_markdown(renderer=None)
                 tokens = markdown(self.buffer)
+                # Type assertion since we know renderer=None returns tokens
+                assert isinstance(tokens, list)
                 renderer = TerminalRenderer(self.console)
                 renderer.render(tokens)
             except Exception: