Fix various __all__ bugs and omissions (#7618)

python · Apr 14, 2022 · 40985b4 · 40985b4
1 parent f87e811
commit 40985b4
Show file tree

Hide file tree

Showing 5 changed files with 204 additions and 40 deletions.
diff --git a/stdlib/_pydecimal.pyi b/stdlib/_pydecimal.pyi
@@ -1,3 +1,85 @@
+import sys
+
 # This is a slight lie, the implementations aren't exactly identical
 # However, in all likelihood, the differences are inconsequential
 from decimal import *
+
+if sys.version_info >= (3, 7):
+    __all__ = [
+        "Decimal",
+        "Context",
+        "DecimalTuple",
+        "DefaultContext",
+        "BasicContext",
+        "ExtendedContext",
+        "DecimalException",
+        "Clamped",
+        "InvalidOperation",
+        "DivisionByZero",
+        "Inexact",
+        "Rounded",
+        "Subnormal",
+        "Overflow",
+        "Underflow",
+        "FloatOperation",
+        "DivisionImpossible",
+        "InvalidContext",
+        "ConversionSyntax",
+        "DivisionUndefined",
+        "ROUND_DOWN",
+        "ROUND_HALF_UP",
+        "ROUND_HALF_EVEN",
+        "ROUND_CEILING",
+        "ROUND_FLOOR",
+        "ROUND_UP",
+        "ROUND_HALF_DOWN",
+        "ROUND_05UP",
+        "setcontext",
+        "getcontext",
+        "localcontext",
+        "MAX_PREC",
+        "MAX_EMAX",
+        "MIN_EMIN",
+        "MIN_ETINY",
+        "HAVE_THREADS",
+        "HAVE_CONTEXTVAR",
+    ]
+else:
+    __all__ = [
+        "Decimal",
+        "Context",
+        "DecimalTuple",
+        "DefaultContext",
+        "BasicContext",
+        "ExtendedContext",
+        "DecimalException",
+        "Clamped",
+        "InvalidOperation",
+        "DivisionByZero",
+        "Inexact",
+        "Rounded",
+        "Subnormal",
+        "Overflow",
+        "Underflow",
+        "FloatOperation",
+        "DivisionImpossible",
+        "InvalidContext",
+        "ConversionSyntax",
+        "DivisionUndefined",
+        "ROUND_DOWN",
+        "ROUND_HALF_UP",
+        "ROUND_HALF_EVEN",
+        "ROUND_CEILING",
+        "ROUND_FLOOR",
+        "ROUND_UP",
+        "ROUND_HALF_DOWN",
+        "ROUND_05UP",
+        "setcontext",
+        "getcontext",
+        "localcontext",
+        "MAX_PREC",
+        "MAX_EMAX",
+        "MIN_EMIN",
+        "MIN_ETINY",
+        "HAVE_THREADS",
+    ]
diff --git a/stdlib/calendar.pyi b/stdlib/calendar.pyi
@@ -4,32 +4,67 @@ from collections.abc import Iterable, Sequence
 from time import struct_time
 from typing_extensions import Literal
 
-__all__ = [
-    "IllegalMonthError",
-    "IllegalWeekdayError",
-    "setfirstweekday",
-    "firstweekday",
-    "isleap",
-    "leapdays",
-    "weekday",
-    "monthrange",
-    "monthcalendar",
-    "prmonth",
-    "month",
-    "prcal",
-    "calendar",
-    "timegm",
-    "month_name",
-    "month_abbr",
-    "day_name",
-    "day_abbr",
-    "Calendar",
-    "TextCalendar",
-    "HTMLCalendar",
-    "LocaleTextCalendar",
-    "LocaleHTMLCalendar",
-    "weekheader",
-]
+if sys.version_info >= (3, 10):
+    __all__ = [
+        "IllegalMonthError",
+        "IllegalWeekdayError",
+        "setfirstweekday",
+        "firstweekday",
+        "isleap",
+        "leapdays",
+        "weekday",
+        "monthrange",
+        "monthcalendar",
+        "prmonth",
+        "month",
+        "prcal",
+        "calendar",
+        "timegm",
+        "month_name",
+        "month_abbr",
+        "day_name",
+        "day_abbr",
+        "Calendar",
+        "TextCalendar",
+        "HTMLCalendar",
+        "LocaleTextCalendar",
+        "LocaleHTMLCalendar",
+        "weekheader",
+        "FRIDAY",
+        "MONDAY",
+        "SATURDAY",
+        "SUNDAY",
+        "THURSDAY",
+        "TUESDAY",
+        "WEDNESDAY",
+    ]
+else:
+    __all__ = [
+        "IllegalMonthError",
+        "IllegalWeekdayError",
+        "setfirstweekday",
+        "firstweekday",
+        "isleap",
+        "leapdays",
+        "weekday",
+        "monthrange",
+        "monthcalendar",
+        "prmonth",
+        "month",
+        "prcal",
+        "calendar",
+        "timegm",
+        "month_name",
+        "month_abbr",
+        "day_name",
+        "day_abbr",
+        "Calendar",
+        "TextCalendar",
+        "HTMLCalendar",
+        "LocaleTextCalendar",
+        "LocaleHTMLCalendar",
+        "weekheader",
+    ]
 
 _LocaleType = tuple[str | None, str | None]
 

diff --git a/stdlib/poplib.pyi b/stdlib/poplib.pyi
@@ -1,13 +1,9 @@
 import socket
 import ssl
-import sys
 from typing import Any, BinaryIO, NoReturn, Pattern, overload
 from typing_extensions import Literal
 
-if sys.version_info >= (3, 10):
-    __all__ = ["POP3", "error_proto", "POP3_SSL"]
-else:
-    __all__ = ["POP3", "error_proto"]
+__all__ = ["POP3", "error_proto", "POP3_SSL"]
 
 _LongResp = tuple[bytes, list[bytes], int]
 

diff --git a/stdlib/tty.pyi b/stdlib/tty.pyi
@@ -2,6 +2,8 @@ import sys
 from typing import IO
 
 if sys.platform != "win32":
+    __all__ = ["setraw", "setcbreak"]
+
     _FD = int | IO[str]
 
     # XXX: Undocumented integer constants

diff --git a/stdlib/typing_extensions.pyi b/stdlib/typing_extensions.pyi
@@ -30,6 +30,56 @@ from typing import (  # noqa: Y022,Y027
     overload as overload,
 )
 
+__all__ = [
+    "ClassVar",
+    "Concatenate",
+    "Final",
+    "LiteralString",
+    "ParamSpec",
+    "Self",
+    "Type",
+    "TypeVarTuple",
+    "Unpack",
+    "Awaitable",
+    "AsyncIterator",
+    "AsyncIterable",
+    "Coroutine",
+    "AsyncGenerator",
+    "AsyncContextManager",
+    "ChainMap",
+    "ContextManager",
+    "Counter",
+    "Deque",
+    "DefaultDict",
+    "OrderedDict",
+    "TypedDict",
+    "SupportsIndex",
+    "Annotated",
+    "assert_never",
+    "dataclass_transform",
+    "final",
+    "IntVar",
+    "is_typeddict",
+    "Literal",
+    "NewType",
+    "overload",
+    "Protocol",
+    "reveal_type",
+    "runtime",
+    "runtime_checkable",
+    "Text",
+    "TypeAlias",
+    "TypeGuard",
+    "TYPE_CHECKING",
+    "Never",
+    "NoReturn",
+    "Required",
+    "NotRequired",
+    "get_args",
+    "get_origin",
+    "get_type_hints",
+]
+
 _T = TypeVar("_T")
 _F = TypeVar("_F", bound=Callable[..., Any])
 _TC = TypeVar("_TC", bound=Type[object])
@@ -82,15 +132,14 @@ TypedDict: object
 
 OrderedDict = _Alias()
 
-if sys.version_info >= (3, 7):
-    def get_type_hints(
-        obj: Callable[..., Any],
-        globalns: dict[str, Any] | None = ...,
-        localns: dict[str, Any] | None = ...,
-        include_extras: bool = ...,
-    ) -> dict[str, Any]: ...
-    def get_args(tp: Any) -> tuple[Any, ...]: ...
-    def get_origin(tp: Any) -> Any | None: ...
+def get_type_hints(
+    obj: Callable[..., Any],
+    globalns: dict[str, Any] | None = ...,
+    localns: dict[str, Any] | None = ...,
+    include_extras: bool = ...,
+) -> dict[str, Any]: ...
+def get_args(tp: Any) -> tuple[Any, ...]: ...
+def get_origin(tp: Any) -> Any | None: ...
 
 Annotated: _SpecialForm
 _AnnotatedAlias: Any  # undocumented