Merge remote-tracking branch 'origin/master' into async

python · AlexWaygood · Mar 13, 2022 · Mar 13, 2022 · Mar 14, 2022 · Mar 22, 2022
commit c79ad7f681244a5d3340b4b355fe1d6b252aa82a
diff --git a/mypy/types.py b/mypy/types.py
@@ -1318,6 +1318,21 @@ def has_readable_member(self, name: str) -> bool:
     def is_awaitable(self) -> bool:
         return self.type.get("__await__") is not None
 
+    def is_singleton_type(self) -> bool:
+        # TODO:
+        # Also make this return True if the type corresponds to NotImplemented?
+        return (
+            self.type.is_enum
+            and len(self.get_enum_values()) == 1
+            or self.type.fullname == "builtins.ellipsis"
+        )
+
+    def get_enum_values(self) -> List[str]:
+        """Return the list of values for an Enum."""
+        return [
+            name for name, sym in self.type.names.items() if isinstance(sym.node, mypy.nodes.Var)
+        ]
+
 
 class FunctionLike(ProperType):
     """Abstract base class for function types."""