From 9efc0c0dca93613fe692b8d61d6d8a2b6178bc91 Mon Sep 17 00:00:00 2001 From: Akuli Date: Sun, 26 Dec 2021 14:53:41 +0200 Subject: [PATCH] pygments: Annotate several generator functions (#6695) --- stubs/Pygments/pygments/filters/__init__.pyi | 4 ++-- stubs/Pygments/pygments/plugin.pyi | 17 ++++++++++++----- 2 files changed, 14 insertions(+), 7 deletions(-) diff --git a/stubs/Pygments/pygments/filters/__init__.pyi b/stubs/Pygments/pygments/filters/__init__.pyi index 15f60f43d..05325c8c6 100644 --- a/stubs/Pygments/pygments/filters/__init__.pyi +++ b/stubs/Pygments/pygments/filters/__init__.pyi @@ -1,4 +1,4 @@ -from collections.abc import Iterable, Iterator +from collections.abc import Generator, Iterable, Iterator from typing import Any from pygments.filter import Filter @@ -7,7 +7,7 @@ from pygments.token import _TokenType def find_filter_class(filtername): ... def get_filter_by_name(filtername, **options): ... -def get_all_filters(): ... +def get_all_filters() -> Generator[str, None, None]: ... class CodeTagFilter(Filter): tag_re: Any diff --git a/stubs/Pygments/pygments/plugin.pyi b/stubs/Pygments/pygments/plugin.pyi index e2115ee9d..b79b75ab4 100644 --- a/stubs/Pygments/pygments/plugin.pyi +++ b/stubs/Pygments/pygments/plugin.pyi @@ -1,10 +1,17 @@ +from typing import Any, Generator, Iterable + +from pygments.filter import Filter +from pygments.formatter import Formatter +from pygments.lexer import Lexer +from pygments.style import Style + LEXER_ENTRY_POINT: str FORMATTER_ENTRY_POINT: str STYLE_ENTRY_POINT: str FILTER_ENTRY_POINT: str -def iter_entry_points(group_name): ... -def find_plugin_lexers() -> None: ... -def find_plugin_formatters() -> None: ... -def find_plugin_styles() -> None: ... -def find_plugin_filters() -> None: ... +def iter_entry_points(group_name: str) -> Iterable[Any]: ... +def find_plugin_lexers() -> Generator[type[Lexer], None, None]: ... +def find_plugin_formatters() -> Generator[tuple[str, type[Formatter]], None, None]: ... +def find_plugin_styles() -> Generator[tuple[str, type[Style]], None, None]: ... +def find_plugin_filters() -> Generator[tuple[str, type[Filter]], None, None]: ...