typing(): more typing

flashback/debugging/filters/call_highlight_filter.py

@@ -1,5 +1,10 @@
+from __future__ import annotations
+
+from collections.abc import Generator
+
 from pygments.filters import Filter
-from pygments.token import Name
+from pygments.token import Name, _TokenType
+from pygments.lexer import Lexer
 
 
 class CallHighlightFilter(Filter):
@@ -15,7 +20,7 @@ def __init__(self, **kwargs) -> None:
         """
         Filter.__init__(self, **kwargs)
 
-    def filter(self, _lexer, stream):
+    def filter(self, _lexer: Lexer, stream: Generator) -> Generator[tuple[_TokenType, str], None, None]:
         """
         Iterates over the stream of tokens and searches for a name followed by an opening paren to
         change its type to Name.Function.
@@ -31,11 +36,11 @@ def filter(self, _lexer, stream):
         the stack.
 
         Params:
-            lexer (pygments.lexer.Lexer): the lexer instance
-            stream (generator): the stream of couples tokentype-value
+            lexer: the lexer instance
+            stream: the stream of couples tokentype-value
 
         Yields:
-            tuple<pygments.token._TokenType, str>: the token type and token value
+            the token type and token value
         """
         try:
             stack = [next(stream)]

flashback/debugging/filters/decorator_operator_filter.py

@@ -1,5 +1,10 @@
+from __future__ import annotations
+
+from collections.abc import Generator
+
 from pygments.filters import Filter
-from pygments.token import Name, Operator
+from pygments.token import Name, Operator, _TokenType
+from pygments.lexer import Lexer
 
 
 class DecoratorOperatorFilter(Filter):
@@ -15,7 +20,7 @@ def __init__(self, **kwargs):
         """
         Filter.__init__(self, **kwargs)
 
-    def filter(self, _lexer, stream):
+    def filter(self, _lexer: Lexer, stream: Generator) -> Generator[tuple[_TokenType, str], None, None]:
         """
         Iterates over the stream of tokens and splits a `pygments.token.Name.Decorator: into two
         components.
@@ -24,11 +29,11 @@ def filter(self, _lexer, stream):
         but pygments treat the whole thing as a decorator. This filter fixes this behaviour.
 
         Params:
-            lexer (pygments.lexer.Lexer): the lexer instance
-            stream (generator): the stream of couples tokentype-value
+            lexer: the lexer instance
+            stream: the stream of couples tokentype-value
 
         Yields:
-            tuple<pygments.token._TokenType, str>: the token type and token value
+            the token type and token value
         """
         for ttype, value in stream:
             if ttype is Name.Decorator:

flashback/debugging/filters/type_highlight_filter.py

@@ -1,33 +1,38 @@
+from __future__ import annotations
+
+from collections.abc import Generator
+
+from pygments.lexer import Lexer
 from pygments.filters import Filter
-from pygments.token import Name, Keyword
+from pygments.token import Name, Keyword, _TokenType
 
 
 class TypeHighlightFilter(Filter):
     """
     Modifies the token type of a Name token to Keyword.Type if its value appears in a list of values.
     """
 
-    def __init__(self, names, **kwargs):
+    def __init__(self, names: list[str], **kwargs) -> None:
         """
         Params:
-            names (Iterable<str>): the list of names to change the token type
-            kwargs (dict): every additional keyword parameters
+            names: the list of names to change the token type
+            kwargs: every additional keyword parameters
         """
         Filter.__init__(self, **kwargs)
 
         self.names = set(names)
 
-    def filter(self, _lexer, stream):
+    def filter(self, _lexer: Lexer, stream: Generator) -> Generator[tuple[_TokenType, str], None, None]:
         """
         Iterates over the stream of tokens and modifies a token's type if its value appears in a
         list of names.
 
         Params:
-            lexer (pygments.lexer.Lexer): the lexer instance
-            stream (generator): the stream of couples tokentype-value
+            lexer: the lexer instance
+            stream: the stream of couples tokentype-value
 
         Yields:
-            tuple<pygments.token._TokenType, str>: the token type and token value
+            the token type and token value
         """
         for ttype, value in stream:
             if ttype in Name and value in self.names:

flashback/debugging/styles/jellybeans.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import ClassVar
 
 from pygments.style import Style

flashback/sampled.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Callable
 from queue import Queue
 import functools