Implement.

Author: dnwpark

typemap/type_eval/_apply_generic.py

@@ -281,7 +281,14 @@ def _resolved_function_signature(func, args):
     return sig
 
 
-def get_local_defns(boxed: Boxed) -> tuple[dict[str, Any], dict[str, Any]]:
+def get_local_defns(
+    boxed: Boxed,
+) -> tuple[
+    dict[str, Any],
+    dict[
+        str, types.FunctionType | classmethod | staticmethod | WrappedOverloaded
+    ],
+]:
     from typemap.typing import GenericCallable
 
     annos: dict[str, Any] = {}
@@ -315,6 +322,8 @@ def get_local_defns(boxed: Boxed) -> tuple[dict[str, Any], dict[str, Any]]:
                 # XXX: This is totally wrong; we still need to do
                 # substitute in class vars
                 local_fn = stuff
+            elif overloaded := _is_overloaded_function(stuff):
+                local_fn = overloaded
 
             # If we got stuck, we build a GenericCallable that
             # computes the type once it has been given type
@@ -358,6 +367,23 @@ def lam(*vs):
     return annos, dct
 
 
+@dataclasses.dataclass(frozen=True)
+class WrappedOverloaded:
+    functions: tuple[types.FunctionType, ...]
+
+
+def _is_overloaded_function(func):
+    module_overload_registry = typing._overload_registry[func.__module__]
+    if not module_overload_registry:
+        return None
+
+    func_overload_registry = module_overload_registry[func.__qualname__]
+    if not func_overload_registry:
+        return
+
+    return WrappedOverloaded(tuple(func_overload_registry.values()))
+
+
 def flatten_class_new_proto(cls: type) -> type:
     # This is a hacky version of flatten_class that works by using
     # NewProtocol on Members!

typemap/type_eval/_eval_operators.py

@@ -39,6 +39,7 @@
     Member,
     Members,
     NewProtocol,
+    Overloaded,
     Param,
     RaiseError,
     Slice,
@@ -169,6 +170,17 @@ def get_annotated_method_hints(cls, *, ctx):
                     object,
                     acls,
                 )
+            elif isinstance(attr, _apply_generic.WrappedOverloaded):
+                overloads = [
+                    _function_type(_eval_types(of, ctx), receiver_type=acls)
+                    for of in attr.functions
+                ]
+                hints[name] = (
+                    Overloaded[*overloads],
+                    ("ClassVar",),
+                    object,
+                    acls,
+                )
 
     return hints