add filters + formatters __init__.pyi

Author: lev-blit

stubs/Pygments/pygments/filters/__init__.pyi

@@ -1,58 +1,58 @@
-from collections.abc import Generator, Iterable, Iterator
+from collections.abc import Callable, Generator, Iterable, Iterator
 from typing import Any
 
 from pygments.filter import Filter
 from pygments.lexer import Lexer
 from pygments.token import _TokenType
 
-def find_filter_class(filtername): ...
-def get_filter_by_name(filtername, **options): ...
+def find_filter_class(filtername: str) -> type[Filter]: ...
+def get_filter_by_name(filtername: str, **options: Any) -> Filter: ...
 def get_all_filters() -> Generator[str, None, None]: ...
 
 class CodeTagFilter(Filter):
     tag_re: Any
-    def __init__(self, **options) -> None: ...
+    def __init__(self, **options: Any) -> None: ...
     def filter(self, lexer: Lexer, stream: Iterable[tuple[_TokenType, str]]) -> Iterator[tuple[_TokenType, str]]: ...
 
 class SymbolFilter(Filter):
-    latex_symbols: Any
-    isabelle_symbols: Any
-    lang_map: Any
-    symbols: Any
-    def __init__(self, **options) -> None: ...
+    latex_symbols: dict[str, str]
+    isabelle_symbols: dict[str, str]
+    lang_map: dict[str, dict[str, str]]
+    symbols: dict[str, str]
+    def __init__(self, **options: Any) -> None: ...
     def filter(self, lexer: Lexer, stream: Iterable[tuple[_TokenType, str]]) -> Iterator[tuple[_TokenType, str]]: ...
 
 class KeywordCaseFilter(Filter):
-    convert: Any
-    def __init__(self, **options) -> None: ...
+    convert: Callable[[str], str]
+    def __init__(self, **options: Any) -> None: ...
     def filter(self, lexer: Lexer, stream: Iterable[tuple[_TokenType, str]]) -> Iterator[tuple[_TokenType, str]]: ...
 
 class NameHighlightFilter(Filter):
-    names: Any
-    tokentype: Any
-    def __init__(self, **options) -> None: ...
+    names: set[str]
+    tokentype: _TokenType
+    def __init__(self, **options: Any) -> None: ...
     def filter(self, lexer: Lexer, stream: Iterable[tuple[_TokenType, str]]) -> Iterator[tuple[_TokenType, str]]: ...
 
 class ErrorToken(Exception): ...
 
 class RaiseOnErrorTokenFilter(Filter):
-    exception: Any
-    def __init__(self, **options) -> None: ...
+    exception: type[Exception]
+    def __init__(self, **options: Any) -> None: ...
     def filter(self, lexer: Lexer, stream: Iterable[tuple[_TokenType, str]]) -> Iterator[tuple[_TokenType, str]]: ...
 
 class VisibleWhitespaceFilter(Filter):
-    wstt: Any
-    def __init__(self, **options) -> None: ...
+    wstt: bool
+    def __init__(self, **options: Any) -> None: ...
     def filter(self, lexer: Lexer, stream: Iterable[tuple[_TokenType, str]]) -> Iterator[tuple[_TokenType, str]]: ...
 
 class GobbleFilter(Filter):
-    n: Any
-    def __init__(self, **options) -> None: ...
-    def gobble(self, value, left): ...
+    n: int
+    def __init__(self, **options: Any) -> None: ...
+    def gobble(self, value: str, left: int) -> tuple[str, int]: ...
     def filter(self, lexer: Lexer, stream: Iterable[tuple[_TokenType, str]]) -> Iterator[tuple[_TokenType, str]]: ...
 
 class TokenMergeFilter(Filter):
-    def __init__(self, **options) -> None: ...
+    def __init__(self, **options: Any) -> None: ...
     def filter(self, lexer: Lexer, stream: Iterable[tuple[_TokenType, str]]) -> Iterator[tuple[_TokenType, str]]: ...
 
-FILTERS: Any
+FILTERS: dict[str, type[Filter]]

stubs/Pygments/pygments/formatters/__init__.pyi

@@ -20,6 +20,7 @@ from .terminal import TerminalFormatter as TerminalFormatter
 from .terminal256 import Terminal256Formatter as Terminal256Formatter, TerminalTrueColorFormatter as TerminalTrueColorFormatter
 
 def get_all_formatters() -> Generator[type[Formatter[Any]], None, None]: ...
-def get_formatter_by_name(_alias, **options): ...
-def load_formatter_from_file(filename, formattername: str = "CustomFormatter", **options): ...
-def get_formatter_for_filename(fn, **options): ...
+def find_formatter_class(alias: str) -> type[Formatter[Any]]: ...
+def get_formatter_by_name(_alias: str, **options: Any) -> Formatter[Any]: ...
+def load_formatter_from_file(filename: str, formattername: str = "CustomFormatter", **options: Any) -> Formatter[Any]: ...
+def get_formatter_for_filename(fn: str, **options: Any) -> Formatter[Any]: ...

stubs/Pygments/pygments/formatters/_mapping.pyi

@@ -1,3 +1 @@
-from typing import Any
-
-FORMATTERS: Any
+FORMATTERS: dict[str, tuple[str, str, tuple[str, ...], tuple[str, ...], str]]

stubs/Pygments/pygments/lexer.pyi

@@ -31,7 +31,7 @@ class Lexer(metaclass=LexerMeta):
     encoding: str
     filters: list[Filter]
     def __init__(self, **options: Any) -> None: ...
-    def add_filter(self, filter_: Filter, **options: Any) -> None: ...
+    def add_filter(self, filter_: str | Filter, **options: Any) -> None: ...
     def get_tokens(self, text: str, unfiltered: bool = False) -> Iterator[tuple[_TokenType, str]]: ...
     def get_tokens_unprocessed(self, text: str) -> Iterator[tuple[int, _TokenType, str]]: ...