ovld 0.3.3__tar.gz → 0.3.4__tar.gz

{ovld-0.3.3 → ovld-0.3.4}/PKG-INFO

@@ -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.3 → ovld-0.3.4}/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-0.3.3 → ovld-0.3.4}/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-0.3.4/ovld/version.py

@@ -0,0 +1 @@
+version = "0.3.4"

{ovld-0.3.3 → ovld-0.3.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ovld"
-version = "0.3.3"
+version = "0.3.4"
 description = "Overloading Python functions"
 authors = ["Olivier Breuleux <breuleux@gmail.com>"]
 license = "MIT"

ovld-0.3.3/ovld/version.py

@@ -1 +0,0 @@
-version = "0.3.3"