ovld 0.3.3__py3-none-any.whl → 0.3.4__py3-none-any.whl

ovld/core.py

@@ -8,10 +8,18 @@ import typing
 from types import FunctionType
 
 try:
-    from types import UnionType
+    from types import GenericAlias, UnionType
 except ImportError:  # pragma: no cover
     UnionType = None
 
+    class GenericAliasMC(type):
+        def __instancecheck__(cls, obj):
+            return hasattr(obj, "__origin__")
+
+    class GenericAlias(metaclass=GenericAliasMC):
+        pass
+
+
 from .mro import compose_mro
 from .utils import BOOTSTRAP, MISSING, keyword_decorator
 
@@ -39,8 +47,13 @@ class TypeMap(dict):
 
     def register(self, obj_t, handler):
         """Register a handler for the given object type."""
+        if isinstance(obj_t, str):
+            obj_t = eval(obj_t, getattr(handler[0], "__globals__", {}))
+
         self.clear()
-        if not is_type_of_type(obj_t):
+        if is_type_of_type(obj_t):
+            self.types.add(obj_t.__args__[0])
+        else:
             self.types.add(obj_t)
         s = self.entries.setdefault(obj_t, set())
         s.add(handler)
@@ -54,6 +67,8 @@ class TypeMap(dict):
         results = {}
         if is_type_of_type(obj_t):
             mro = [type[t] for t in compose_mro(obj_t.__args__[0], self.types)]
+            mro.append(type)
+            mro.append(object)
         else:
             mro = compose_mro(obj_t, self.types)
         for lvl, cls in enumerate(reversed(mro)):
@@ -93,7 +108,9 @@ class MultiTypeMap(dict):
         self.key_error = key_error
 
     def transform(self, obj):
-        if isinstance(obj, type):
+        if isinstance(obj, GenericAlias):
+            return type[obj.__origin__]
+        elif isinstance(obj, type):
             return type[obj]
         else:
             return type(obj)

ovld/utils.py

@@ -2,6 +2,8 @@
 
 import functools
 import sys
+from dataclasses import dataclass
+from typing import Protocol, runtime_checkable
 
 
 class Named:
@@ -143,9 +145,20 @@ def has_attribute(*attrs):
     return check
 
 
+@runtime_checkable
+@dataclass
+class Dataclass(Protocol):
+    @classmethod
+    def __subclasshook__(cls, subclass):
+        return hasattr(subclass, "__dataclass_fields__") and hasattr(
+            subclass, "__dataclass_params__"
+        )
+
+
 __all__ = [
     "BOOTSTRAP",
     "MISSING",
+    "Dataclass",
     "Named",
     "deferred",
     "exactly",

ovld/version.py

@@ -1 +1 @@
-version = "0.3.3"
+version = "0.3.4"

{ovld-0.3.3.dist-info → ovld-0.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ovld
-Version: 0.3.3
+Version: 0.3.4
 Summary: Overloading Python functions
 Home-page: https://github.com/breuleux/ovld
 License: MIT

ovld-0.3.4.dist-info/RECORD

@@ -0,0 +1,9 @@
+ovld/__init__.py,sha256=Wa7MHfDw0_AV7jwSwepntCeXzZheELkb8paxVnjp58Q,85
+ovld/core.py,sha256=Iz9VMQAqRGsgZqwBkR8uNoBdnC0vp4RYhVDcMydcPlo,33078
+ovld/mro.py,sha256=Dof3wj-BLcV-T-Jm9tK6zqptp_JGo5lxgyRq99AvY1k,5006
+ovld/utils.py,sha256=8kINtHrWdC0bNVccxYIkHBaGFk_YFpjtnjAa9GyXNmA,3861
+ovld/version.py,sha256=gFnDcL_GoJXrglbZytpscjJFX7Ge1G2_exIvsJP20D4,18
+ovld-0.3.4.dist-info/LICENSE,sha256=cSwNTIzd1cbI89xt3PeZZYJP2y3j8Zus4bXgo4svpX8,1066
+ovld-0.3.4.dist-info/METADATA,sha256=1JtepDt_ZRKv07KGQb0DhymtG5OnYXXUyy1UEg4XHy4,9439
+ovld-0.3.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+ovld-0.3.4.dist-info/RECORD,,

ovld-0.3.3.dist-info/RECORD

@@ -1,9 +0,0 @@
-ovld/__init__.py,sha256=Wa7MHfDw0_AV7jwSwepntCeXzZheELkb8paxVnjp58Q,85
-ovld/core.py,sha256=uLRvhoXb0d8ide648_e1EsYdqDZZZtz3rLqVAsOdisw,32571
-ovld/mro.py,sha256=Dof3wj-BLcV-T-Jm9tK6zqptp_JGo5lxgyRq99AvY1k,5006
-ovld/utils.py,sha256=JT44u-OiFC2QBUylaubwKlgvvRS0EYPfykxgMbbIrMY,3521
-ovld/version.py,sha256=nlGXcobmeAmNzRkhqDLZxGFLcidRwZSZPlWcdlVyxvw,18
-ovld-0.3.3.dist-info/LICENSE,sha256=cSwNTIzd1cbI89xt3PeZZYJP2y3j8Zus4bXgo4svpX8,1066
-ovld-0.3.3.dist-info/METADATA,sha256=vf9bZaiRGKrZzreiu_jvSFLkIzd4cyP6Agz4qarM4Zc,9439
-ovld-0.3.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-ovld-0.3.3.dist-info/RECORD,,