Add __all__ (#13704)

donBarbos · web-flow · commit 9234985c3df5 · 2025-03-25T14:09:32.000+01:00
diff --git a/stubs/PyYAML/@tests/stubtest_allowlist.txt b/stubs/PyYAML/@tests/stubtest_allowlist.txt
@@ -1,17 +1,3 @@
-# TODO: missing from stub
-yaml.composer.__all__
-yaml.constructor.__all__
-yaml.dumper.__all__
-yaml.emitter.__all__
-yaml.error.__all__
-yaml.loader.__all__
-yaml.parser.__all__
-yaml.reader.__all__
-yaml.representer.__all__
-yaml.resolver.__all__
-yaml.scanner.__all__
-yaml.serializer.__all__
-
 # yaml._yaml is for backwards compatibility so none of it matters anyway
 yaml._yaml.__test__
 
diff --git a/stubs/PyYAML/yaml/composer.pyi b/stubs/PyYAML/yaml/composer.pyi
@@ -16,3 +16,5 @@ class Composer:
     def compose_scalar_node(self, anchor: dict[Any, Node]) -> ScalarNode: ...
     def compose_sequence_node(self, anchor: dict[Any, Node]) -> SequenceNode: ...
     def compose_mapping_node(self, anchor: dict[Any, Node]) -> MappingNode: ...
+
+__all__ = ["Composer", "ComposerError"]
diff --git a/stubs/PyYAML/yaml/constructor.pyi b/stubs/PyYAML/yaml/constructor.pyi
@@ -101,3 +101,5 @@ class Constructor(SafeConstructor):
     def construct_python_object(self, suffix, node): ...
     def construct_python_object_apply(self, suffix, node, newobj=False): ...
     def construct_python_object_new(self, suffix, node): ...
+
+__all__ = ["BaseConstructor", "SafeConstructor", "FullConstructor", "UnsafeConstructor", "Constructor", "ConstructorError"]
diff --git a/stubs/PyYAML/yaml/dumper.pyi b/stubs/PyYAML/yaml/dumper.pyi
@@ -69,3 +69,5 @@ class Dumper(Emitter, Serializer, Representer, Resolver):
         tags: Mapping[str, str] | None = None,
         sort_keys: bool = True,
     ) -> None: ...
+
+__all__ = ["BaseDumper", "SafeDumper", "Dumper"]
diff --git a/stubs/PyYAML/yaml/emitter.pyi b/stubs/PyYAML/yaml/emitter.pyi
@@ -126,3 +126,5 @@ class Emitter:
     def write_folded(self, text: str) -> None: ...
     def write_literal(self, text: str) -> None: ...
     def write_plain(self, text: str, split: bool = ...) -> None: ...
+
+__all__ = ["Emitter", "EmitterError"]
diff --git a/stubs/PyYAML/yaml/error.pyi b/stubs/PyYAML/yaml/error.pyi
@@ -24,3 +24,5 @@ class MarkedYAMLError(YAMLError):
         problem_mark: Mark | None = None,
         note: str | None = None,
     ) -> None: ...
+
+__all__ = ["Mark", "YAMLError", "MarkedYAMLError"]
diff --git a/stubs/PyYAML/yaml/loader.pyi b/stubs/PyYAML/yaml/loader.pyi
@@ -25,3 +25,5 @@ class Loader(Reader, Scanner, Parser, Composer, Constructor, Resolver):
 
 class UnsafeLoader(Reader, Scanner, Parser, Composer, Constructor, Resolver):
     def __init__(self, stream: _ReadStream) -> None: ...
+
+__all__ = ["BaseLoader", "FullLoader", "SafeLoader", "Loader", "UnsafeLoader"]
diff --git a/stubs/PyYAML/yaml/parser.pyi b/stubs/PyYAML/yaml/parser.pyi
@@ -43,3 +43,5 @@ class Parser:
     def parse_flow_mapping_value(self): ...
     def parse_flow_mapping_empty_value(self): ...
     def process_empty_scalar(self, mark): ...
+
+__all__ = ["Parser", "ParserError"]
diff --git a/stubs/PyYAML/yaml/reader.pyi b/stubs/PyYAML/yaml/reader.pyi
@@ -37,3 +37,5 @@ class Reader:
     def check_printable(self, data): ...
     def update(self, length): ...
     def update_raw(self, size=4096): ...
+
+__all__ = ["Reader", "ReaderError"]
diff --git a/stubs/PyYAML/yaml/representer.pyi b/stubs/PyYAML/yaml/representer.pyi
@@ -59,3 +59,5 @@ class Representer(SafeRepresenter):
     def represent_module(self, data: ModuleType) -> ScalarNode: ...
     def represent_object(self, data) -> SequenceNode | MappingNode: ...
     def represent_ordered_dict(self, data: Mapping[Any, Any]) -> SequenceNode: ...
+
+__all__ = ["BaseRepresenter", "SafeRepresenter", "Representer", "RepresenterError"]
diff --git a/stubs/PyYAML/yaml/resolver.pyi b/stubs/PyYAML/yaml/resolver.pyi
@@ -23,3 +23,5 @@ class BaseResolver:
     def resolve(self, kind, value, implicit): ...
 
 class Resolver(BaseResolver): ...
+
+__all__ = ["BaseResolver", "Resolver"]
diff --git a/stubs/PyYAML/yaml/scanner.pyi b/stubs/PyYAML/yaml/scanner.pyi
@@ -95,3 +95,5 @@ class Scanner:
     def scan_tag_uri(self, name, start_mark): ...
     def scan_uri_escapes(self, name, start_mark): ...
     def scan_line_break(self): ...
+
+__all__ = ["Scanner", "ScannerError"]
diff --git a/stubs/PyYAML/yaml/serializer.pyi b/stubs/PyYAML/yaml/serializer.pyi
@@ -23,3 +23,5 @@ class Serializer:
     def anchor_node(self, node): ...
     def generate_anchor(self, node): ...
     def serialize_node(self, node, parent, index): ...
+
+__all__ = ["Serializer", "SerializerError"]
diff --git a/stubs/aiofiles/@tests/stubtest_allowlist.txt b/stubs/aiofiles/@tests/stubtest_allowlist.txt
@@ -1,8 +1,3 @@
-# TODO: missing from stub
-aiofiles.__all__
-aiofiles.tempfile.__all__
-aiofiles.threadpool.__all__
-
 # These all delegate using *args,**kwargs, but stubs use signature of
 # method they are being delegated to.
 aiofiles.threadpool.binary.AsyncBufferedIOBase.close
diff --git a/stubs/aiofiles/aiofiles/__init__.pyi b/stubs/aiofiles/aiofiles/__init__.pyi
@@ -8,3 +8,5 @@ from .threadpool import (
     stdout as stdout,
     stdout_bytes as stdout_bytes,
 )
+
+__all__ = ["open", "tempfile", "stdin", "stdout", "stderr", "stdin_bytes", "stdout_bytes", "stderr_bytes"]
diff --git a/stubs/aiofiles/aiofiles/tempfile/__init__.pyi b/stubs/aiofiles/aiofiles/tempfile/__init__.pyi
@@ -321,3 +321,5 @@ def TemporaryDirectory(
 
 class AiofilesContextManagerTempDir(AiofilesContextManager[AsyncTemporaryDirectory]):
     async def __aenter__(self) -> str: ...  # type: ignore[override]
+
+__all__ = ["NamedTemporaryFile", "TemporaryFile", "SpooledTemporaryFile", "TemporaryDirectory"]
diff --git a/stubs/aiofiles/aiofiles/threadpool/__init__.pyi b/stubs/aiofiles/aiofiles/threadpool/__init__.pyi
@@ -104,3 +104,5 @@ stderr: AsyncTextIndirectIOWrapper
 stdin_bytes: AsyncIndirectBufferedIOBase
 stdout_bytes: AsyncIndirectBufferedIOBase
 stderr_bytes: AsyncIndirectBufferedIOBase
+
+__all__ = ("open", "stdin", "stdout", "stderr", "stdin_bytes", "stdout_bytes", "stderr_bytes")
diff --git a/stubs/boltons/@tests/stubtest_allowlist.txt b/stubs/boltons/@tests/stubtest_allowlist.txt
@@ -1,19 +1,2 @@
-# TODO: missing from stub
-boltons.debugutils.__all__
-boltons.dictutils.__all__
-boltons.excutils.__all__
-boltons.fileutils.__all__
-boltons.formatutils.__all__
-boltons.gcutils.__all__
-boltons.jsonutils.__all__
-boltons.listutils.__all__
-boltons.namedutils.__all__
-boltons.pathutils.__all__
-boltons.queueutils.__all__
-boltons.setutils.__all__
-boltons.strutils.__all__
-boltons.tableutils.__all__
-boltons.tbutils.__all__
-
 boltons.funcutils.CachedInstancePartial.__partialmethod__
 boltons.funcutils.InstancePartial.__partialmethod__
diff --git a/stubs/boltons/boltons/debugutils.pyi b/stubs/boltons/boltons/debugutils.pyi
@@ -6,3 +6,5 @@ def pdb_on_exception(limit: int = 100) -> None: ...
 def wrap_trace(
     obj, hook: Callable[..., Any] = ..., which: str | None = None, events: str | None = None, label: str | None = None
 ): ...
+
+__all__ = ["pdb_on_signal", "pdb_on_exception", "wrap_trace"]
diff --git a/stubs/boltons/boltons/dictutils.pyi b/stubs/boltons/boltons/dictutils.pyi
@@ -99,3 +99,5 @@ class FrozenDict(dict[_KT, _VT]):
     def popitem(self, *a, **kw) -> NoReturn: ...
     def setdefault(self, *a, **kw) -> NoReturn: ...
     def clear(self, *a, **kw) -> NoReturn: ...
+
+__all__ = ["MultiDict", "OMD", "OrderedMultiDict", "OneToOne", "ManyToMany", "subdict", "FrozenDict"]
diff --git a/stubs/boltons/boltons/excutils.pyi b/stubs/boltons/boltons/excutils.pyi
@@ -7,3 +7,5 @@ class ExceptionCauseMixin(Exception):
     def get_str(self) -> str: ...
 
 class MathError(ExceptionCauseMixin, ValueError): ...
+
+__all__ = ["ExceptionCauseMixin"]
diff --git a/stubs/boltons/boltons/fileutils.pyi b/stubs/boltons/boltons/fileutils.pyi
@@ -91,3 +91,5 @@ class DummyFile:
     def __next__(self) -> NoReturn: ...
     def __enter__(self) -> None: ...
     def __exit__(self, exc_type, exc_val, exc_tb) -> None: ...
+
+__all__ = ["mkdir_p", "atomic_save", "AtomicSaver", "FilePerms", "iter_find_files", "copytree"]
diff --git a/stubs/boltons/boltons/formatutils.pyi b/stubs/boltons/boltons/formatutils.pyi
@@ -35,3 +35,12 @@ class DeferredValue(Generic[_T]):
     def __float__(self) -> float: ...
     def __unicode__(self) -> str: ...
     def __format__(self, fmt: str) -> str: ...
+
+__all__ = [
+    "DeferredValue",
+    "get_format_args",
+    "tokenize_format_str",
+    "construct_format_field_str",
+    "infer_positional_format_args",
+    "BaseFormatField",
+]
diff --git a/stubs/boltons/boltons/gcutils.pyi b/stubs/boltons/boltons/gcutils.pyi
@@ -12,3 +12,5 @@ class GCToggler:
 
 toggle_gc: GCToggler
 toggle_gc_postcollect: GCToggler
+
+__all__ = ["get_all", "GCToggler", "toggle_gc", "toggle_gc_postcollect"]
diff --git a/stubs/boltons/boltons/jsonutils.pyi b/stubs/boltons/boltons/jsonutils.pyi
@@ -23,3 +23,5 @@ class JSONLIterator:
     def __iter__(self) -> Self: ...
     def next(self) -> Any: ...
     __next__ = next
+
+__all__ = ["JSONLIterator", "reverse_iter_lines"]
diff --git a/stubs/boltons/boltons/listutils.pyi b/stubs/boltons/boltons/listutils.pyi
@@ -28,3 +28,5 @@ BList: TypeAlias = BarrelList[_T]
 class SplayList(list[_T]):
     def shift(self, item_index: int, dest_index: int = 0) -> None: ...
     def swap(self, item_index: SupportsIndex, dest_index: SupportsIndex) -> None: ...
+
+__all__ = ["BList", "BarrelList"]
diff --git a/stubs/boltons/boltons/namedutils.pyi b/stubs/boltons/boltons/namedutils.pyi
@@ -2,3 +2,5 @@ from collections.abc import Iterable
 
 def namedtuple(typename: str, field_names: str | Iterable[str], verbose: bool = False, rename: bool = False): ...
 def namedlist(typename: str, field_names: str | Iterable[str], verbose: bool = False, rename: bool = False): ...
+
+__all__ = ["namedlist", "namedtuple"]
diff --git a/stubs/boltons/boltons/pathutils.pyi b/stubs/boltons/boltons/pathutils.pyi
@@ -11,3 +11,5 @@ def augpath(
 ) -> str: ...
 def shrinkuser(path: StrPath, home: str = "~") -> str: ...
 def expandpath(path: StrPath) -> str: ...
+
+__all__ = ["augpath", "shrinkuser", "expandpath"]
diff --git a/stubs/boltons/boltons/queueutils.pyi b/stubs/boltons/boltons/queueutils.pyi
@@ -12,3 +12,5 @@ class HeapPriorityQueue(BasePriorityQueue): ...
 class SortedPriorityQueue(BasePriorityQueue): ...
 
 PriorityQueue: TypeAlias = SortedPriorityQueue
+
+__all__ = ["PriorityQueue", "BasePriorityQueue", "HeapPriorityQueue", "SortedPriorityQueue"]
diff --git a/stubs/boltons/boltons/setutils.pyi b/stubs/boltons/boltons/setutils.pyi
@@ -97,3 +97,5 @@ class _ComplementSet:
     def __len__(self) -> int: ...
     def __iter__(self) -> Iterator[Any]: ...
     def __bool__(self) -> bool: ...
+
+__all__ = ["IndexedSet", "complement"]
diff --git a/stubs/boltons/boltons/strutils.pyi b/stubs/boltons/boltons/strutils.pyi
@@ -64,3 +64,38 @@ class MultiReplace:
 def multi_replace(text: str, sub_map: dict[str, str], **kwargs) -> str: ...
 def unwrap_text(text: str, ending: str | None = "\n\n") -> str: ...
 def removeprefix(text: str, prefix: str) -> str: ...
+
+__all__ = [
+    "camel2under",
+    "under2camel",
+    "slugify",
+    "split_punct_ws",
+    "unit_len",
+    "ordinalize",
+    "cardinalize",
+    "pluralize",
+    "singularize",
+    "asciify",
+    "is_ascii",
+    "is_uuid",
+    "html2text",
+    "strip_ansi",
+    "bytes2human",
+    "find_hashtags",
+    "a10n",
+    "gzip_bytes",
+    "gunzip_bytes",
+    "iter_splitlines",
+    "indent",
+    "escape_shell_args",
+    "args2cmd",
+    "args2sh",
+    "parse_int_list",
+    "format_int_list",
+    "complement_int_list",
+    "int_ranges_from_int_list",
+    "MultiReplace",
+    "multi_replace",
+    "unwrap_text",
+    "removeprefix",
+]
diff --git a/stubs/boltons/boltons/tableutils.pyi b/stubs/boltons/boltons/tableutils.pyi
@@ -61,3 +61,5 @@ class Table:
     ): ...
     def get_cell_html(self, value): ...
     def to_text(self, with_headers: bool = True, maxlen: Incomplete | None = None): ...
+
+__all__ = ["Table"]
diff --git a/stubs/boltons/boltons/tbutils.pyi b/stubs/boltons/boltons/tbutils.pyi
@@ -93,3 +93,14 @@ class ParsedException:
     def from_string(cls, tb_str: str) -> Self: ...
 
 ParsedTB = ParsedException
+
+__all__ = [
+    "ExceptionInfo",
+    "TracebackInfo",
+    "Callpoint",
+    "ContextualExceptionInfo",
+    "ContextualTracebackInfo",
+    "ContextualCallpoint",
+    "print_exception",
+    "ParsedException",
+]
diff --git a/stubs/console-menu/@tests/stubtest_allowlist.txt b/stubs/console-menu/@tests/stubtest_allowlist.txt
diff --git a/stubs/console-menu/consolemenu/__init__.pyi b/stubs/console-menu/consolemenu/__init__.pyi
@@ -4,3 +4,14 @@ from .menu_formatter import MenuFormatBuilder as MenuFormatBuilder
 from .multiselect_menu import MultiSelectMenu as MultiSelectMenu
 from .prompt_utils import PromptUtils as PromptUtils
 from .selection_menu import SelectionMenu as SelectionMenu
+
+__all__ = [
+    "ConsoleMenu",
+    "SelectionMenu",
+    "MultiSelectMenu",
+    "MenuFormatBuilder",
+    "PromptUtils",
+    "Screen",
+    "items",
+    "clear_terminal",
+]
diff --git a/stubs/console-menu/consolemenu/format/__init__.pyi b/stubs/console-menu/consolemenu/format/__init__.pyi
@@ -12,3 +12,18 @@ from .menu_borders import (
 from .menu_margins import MenuMargins as MenuMargins
 from .menu_padding import MenuPadding as MenuPadding
 from .menu_style import MenuStyle as MenuStyle
+
+__all__ = [
+    "MenuBorderStyle",
+    "MenuBorderStyleType",
+    "MenuBorderStyleFactory",
+    "MenuMargins",
+    "MenuPadding",
+    "MenuStyle",
+    "AsciiBorderStyle",
+    "LightBorderStyle",
+    "HeavyBorderStyle",
+    "DoubleLineBorderStyle",
+    "DoubleLineOuterLightInnerBorderStyle",
+    "HeavyOuterLightInnerBorderStyle",
+]
diff --git a/stubs/console-menu/consolemenu/items/__init__.pyi b/stubs/console-menu/consolemenu/items/__init__.pyi
@@ -4,3 +4,5 @@ from .external_item import ExternalItem as ExternalItem
 from .function_item import FunctionItem as FunctionItem
 from .selection_item import SelectionItem as SelectionItem
 from .submenu_item import SubmenuItem as SubmenuItem
+
+__all__ = ["CommandItem", "ExitItem", "ExternalItem", "FunctionItem", "MenuItem", "SelectionItem", "SubmenuItem"]
diff --git a/stubs/mysqlclient/@tests/stubtest_allowlist.txt b/stubs/mysqlclient/@tests/stubtest_allowlist.txt
@@ -1,5 +1 @@
-# TODO: missing from stub
-MySQLdb.__all__
-MySQLdb.constants.__all__
-
 MySQLdb.Connection
diff --git a/stubs/mysqlclient/MySQLdb/__init__.pyi b/stubs/mysqlclient/MySQLdb/__init__.pyi
@@ -49,3 +49,47 @@ def Binary(x): ...
 def Connect(*args, **kwargs) -> Connection: ...
 
 connect = Connect
+
+__all__ = [
+    "BINARY",
+    "Binary",
+    "Connect",
+    "Connection",
+    "DATE",
+    "Date",
+    "Time",
+    "Timestamp",
+    "DateFromTicks",
+    "TimeFromTicks",
+    "TimestampFromTicks",
+    "DataError",
+    "DatabaseError",
+    "Error",
+    "FIELD_TYPE",
+    "IntegrityError",
+    "InterfaceError",
+    "InternalError",
+    "MySQLError",
+    "NUMBER",
+    "NotSupportedError",
+    "DBAPISet",
+    "OperationalError",
+    "ProgrammingError",
+    "ROWID",
+    "STRING",
+    "TIME",
+    "TIMESTAMP",
+    "Warning",
+    "apilevel",
+    "connect",
+    "connections",
+    "constants",
+    "converters",
+    "cursors",
+    "debug",
+    "get_client_info",
+    "paramstyle",
+    "string_literal",
+    "threadsafety",
+    "version_info",
+]
diff --git a/stubs/mysqlclient/MySQLdb/constants/__init__.pyi b/stubs/mysqlclient/MySQLdb/constants/__init__.pyi
@@ -1 +1,3 @@
 from . import CLIENT as CLIENT, CR as CR, ER as ER, FIELD_TYPE as FIELD_TYPE, FLAG as FLAG
+
+__all__ = ["CR", "FIELD_TYPE", "CLIENT", "ER", "FLAG"]
diff --git a/stubs/pyserial/@tests/stubtest_allowlist_win32.txt b/stubs/pyserial/@tests/stubtest_allowlist_win32.txt
@@ -1,6 +1,3 @@
-# TODO: missing from stub
-serial.win32.__all__
-
 # Error: failed to import
 # =======================
 serial.serialposix  # Posix only
diff --git a/stubs/pyserial/serial/win32.pyi b/stubs/pyserial/serial/win32.pyi
diff --git a/stubs/pytz/@tests/stubtest_allowlist.txt b/stubs/pytz/@tests/stubtest_allowlist.txt
diff --git a/stubs/pytz/pytz/__init__.pyi b/stubs/pytz/pytz/__init__.pyi
diff --git a/stubs/pytz/pytz/reference.pyi b/stubs/pytz/pytz/reference.pyi
diff --git a/stubs/pytz/pytz/tzinfo.pyi b/stubs/pytz/pytz/tzinfo.pyi
diff --git a/stubs/shapely/@tests/stubtest_allowlist.txt b/stubs/shapely/@tests/stubtest_allowlist.txt
diff --git a/stubs/shapely/shapely/_ragged_array.pyi b/stubs/shapely/shapely/_ragged_array.pyi
diff --git a/stubs/simplejson/@tests/stubtest_allowlist.txt b/stubs/simplejson/@tests/stubtest_allowlist.txt
diff --git a/stubs/simplejson/simplejson/__init__.pyi b/stubs/simplejson/simplejson/__init__.pyi
diff --git a/stubs/simplejson/simplejson/decoder.pyi b/stubs/simplejson/simplejson/decoder.pyi
diff --git a/stubs/simplejson/simplejson/scanner.pyi b/stubs/simplejson/simplejson/scanner.pyi
diff --git a/stubs/singledispatch/@tests/stubtest_allowlist.txt b/stubs/singledispatch/@tests/stubtest_allowlist.txt
diff --git a/stubs/singledispatch/singledispatch.pyi b/stubs/singledispatch/singledispatch.pyi
