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

ovld/core.py

@@ -7,10 +7,27 @@ import textwrap
 import typing
 from types import FunctionType
 
+try:
+    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
 
 
+def is_type_of_type(t):
+    return getattr(t, "__origin__", None) is type
+
+
 class TypeMap(dict):
     """Represents a mapping from types to handlers.
 
@@ -30,8 +47,14 @@ 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()
-        self.types.add(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)
 
@@ -42,7 +65,12 @@ class TypeMap(dict):
         the next time getitem is called.
         """
         results = {}
-        mro = compose_mro(obj_t, self.types)
+        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)):
             handlers = self.entries.get(cls, None)
             if handlers:
@@ -77,9 +105,16 @@ class MultiTypeMap(dict):
     def __init__(self, key_error=KeyError):
         self.maps = {}
         self.empty = MISSING
-        self.transform = type
         self.key_error = key_error
 
+    def transform(self, obj):
+        if isinstance(obj, GenericAlias):
+            return type[obj.__origin__]
+        elif isinstance(obj, type):
+            return type[obj]
+        else:
+            return type(obj)
+
     def register(self, obj_t_tup, nargs, handler):
         """Register a handler for a tuple of argument types.
 
@@ -326,6 +361,9 @@ class _Ovld:
         def clsname(cls):
             if cls is object:
                 return "*"
+            elif is_type_of_type(cls):
+                arg = clsname(cls.__args__[0])
+                return f"type[{arg}]"
             elif hasattr(cls, "__name__"):
                 return cls.__name__
             else:
@@ -489,11 +527,15 @@ class _Ovld:
 
         def _normalize_type(t, force_tuple=False):
             origin = getattr(t, "__origin__", None)
-            if origin is typing.Union:
+            if UnionType and isinstance(t, UnionType):
+                return _normalize_type(t.__args__)
+            elif origin is type:
+                return (t,) if force_tuple else t
+            elif origin is typing.Union:
                 return _normalize_type(t.__args__)
             elif origin is not None:
                 raise TypeError(
-                    "ovld does not accept generic types except Union or Optional"
+                    "ovld does not accept generic types except type, Union or Optional"
                 )
             elif isinstance(t, tuple):
                 return tuple(_normalize_type(t2) for t2 in t)
@@ -923,7 +965,10 @@ class Conformer:
 
 def rename_code(co, newname):  # pragma: no cover
     if hasattr(co, "replace"):
-        return co.replace(co_name=newname)
+        if hasattr(co, "co_qualname"):
+            return co.replace(co_name=newname, co_qualname=newname)
+        else:
+            return co.replace(co_name=newname)
     else:
         return type(co)(
             co.co_argcount,

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.2"
+version = "0.3.4"

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

@@ -1,18 +1,19 @@
 Metadata-Version: 2.1
 Name: ovld
-Version: 0.3.2
+Version: 0.3.4
 Summary: Overloading Python functions
 Home-page: https://github.com/breuleux/ovld
 License: MIT
 Author: Olivier Breuleux
 Author-email: breuleux@gmail.com
-Requires-Python: >=3.6,<4.0
+Requires-Python: >=3.8,<4.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.6
-Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Project-URL: Repository, https://github.com/breuleux/ovld
 Description-Content-Type: text/markdown
 

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.2.dist-info → ovld-0.3.4.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry 1.0.0
+Generator: poetry-core 1.9.0
 Root-Is-Purelib: true
 Tag: py3-none-any

ovld-0.3.2.dist-info/RECORD

@@ -1,9 +0,0 @@
-ovld/__init__.py,sha256=Wa7MHfDw0_AV7jwSwepntCeXzZheELkb8paxVnjp58Q,85
-ovld/core.py,sha256=Ejuu5q6K-OfI_C9xoyLul-P1P0ORTeitwSnu8Ud8wU0,31667
-ovld/mro.py,sha256=Dof3wj-BLcV-T-Jm9tK6zqptp_JGo5lxgyRq99AvY1k,5006
-ovld/utils.py,sha256=JT44u-OiFC2QBUylaubwKlgvvRS0EYPfykxgMbbIrMY,3521
-ovld/version.py,sha256=zjuPzQ2OlxNtrLXb5A3OlPsaRb_b_Ln51AWN6r9VRho,18
-ovld-0.3.2.dist-info/LICENSE,sha256=cSwNTIzd1cbI89xt3PeZZYJP2y3j8Zus4bXgo4svpX8,1066
-ovld-0.3.2.dist-info/WHEEL,sha256=SrtnPGVTMeYWttls9xnWA01eUhCZ3ufFdJUYb1J3r-U,83
-ovld-0.3.2.dist-info/METADATA,sha256=9yq7sErgeIWMU0yRJfx-PskGlg2oZlCmlRA25aUS1X0,9386
-ovld-0.3.2.dist-info/RECORD,,