Bring back type annotation support of dunder methods in stub generator (

#12828) Fixes #12717
python · May 21, 2022 · f19a711 · f19a711
1 parent 927a9ba
commit f19a711
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/mypy/stubgenc.py b/mypy/stubgenc.py
@@ -288,7 +288,7 @@ def infer_prop_type(docstr: Optional[str]) -> Optional[str]:
             return None
 
     # Ignore special properties/attributes.
-    if name.startswith('__') and name.endswith('__'):
+    if is_skipped_attribute(name):
         return
 
     inferred = infer_prop_type(getattr(obj, '__doc__', None))

diff --git a/test-data/pybind11_mypy_demo/stubgen/pybind11_mypy_demo/basics.pyi b/test-data/pybind11_mypy_demo/stubgen/pybind11_mypy_demo/basics.pyi
@@ -5,6 +5,7 @@ PI: float
 
 class Point:
     class AngleUnit:
+        __members__: ClassVar[dict] = ...  # read-only
         __entries: ClassVar[dict] = ...
         degree: ClassVar[Point.AngleUnit] = ...
         radian: ClassVar[Point.AngleUnit] = ...
@@ -22,6 +23,7 @@ class Point:
         def value(self) -> int: ...
 
     class LengthUnit:
+        __members__: ClassVar[dict] = ...  # read-only
         __entries: ClassVar[dict] = ...
         inch: ClassVar[Point.LengthUnit] = ...
         mm: ClassVar[Point.LengthUnit] = ...