ovld 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

ovld/codegen.py

@@ -6,7 +6,7 @@ from ast import _splitlines_no_ff as splitlines
 from itertools import count
 from types import FunctionType
 
-from .utils import MISSING, NameDatabase, sigstring
+from .utils import MISSING, NameDatabase, keyword_decorator, sigstring
 
 _current = count()
 
@@ -298,6 +298,9 @@ def codegen_specializer(typemap, fn, tup):
     return func
 
 
-def code_generator(fn):
+@keyword_decorator
+def code_generator(fn, priority=0):
     fn.specializer = codegen_specializer
+    if priority:
+        fn.priority = priority
     return fn

ovld/core.py

@@ -18,6 +18,7 @@ from .signatures import ArgumentAnalyzer, LazySignature, Signature
 from .typemap import MultiTypeMap
 from .utils import (
     MISSING,
+    ResolutionError,
     UsageError,
     keyword_decorator,
     sigstring,
@@ -51,6 +52,7 @@ def bootstrap_dispatch(ov, name):
     dispatch.add_mixins = ov.add_mixins
     dispatch.unregister = ov.unregister
     dispatch.next = ov.next
+    dispatch.first_entry = first_entry
     return dispatch
 
 
@@ -95,6 +97,7 @@ class Ovld:
         self._locked = False
         self.mixins = []
         self.argument_analysis = ArgumentAnalyzer()
+        self.dispatch = bootstrap_dispatch(self, name=self.shortname)
         self.add_mixins(*mixins)
 
     @property
@@ -162,12 +165,12 @@ class Ovld:
     def _key_error(self, key, possibilities=None):
         typenames = sigstring(key)
         if not possibilities:
-            return TypeError(f"No method in {self} for argument types [{typenames}]")
+            return ResolutionError(f"No method in {self} for argument types [{typenames}]")
         else:
             hlp = ""
             for c in possibilities:
                 hlp += f"* {c.handler.__name__}  (priority: {c.priority}, specificity: {list(c.specificity)})\n"
-            return TypeError(
+            return ResolutionError(
                 f"Ambiguous resolution in {self} for"
                 f" argument types [{typenames}]\n"
                 f"Candidates are:\n{hlp}"
@@ -176,10 +179,11 @@ class Ovld:
 
     def rename(self, name, shortname=None):
         """Rename this Ovld."""
-        self.name = name
-        self.shortname = shortname or name
-        self.__name__ = self.shortname
-        self.dispatch = bootstrap_dispatch(self, name=self.shortname)
+        if name != self.name:
+            self.name = name
+            self.shortname = shortname or name
+            self.__name__ = self.shortname
+            self.dispatch = bootstrap_dispatch(self, name=self.shortname)
 
     def __set_name__(self, inst, name):
         self.rename(name)
@@ -217,8 +221,6 @@ class Ovld:
 
         self.analyze_arguments()
         dispatch = generate_dispatch(self, self.argument_analysis)
-        if not hasattr(self, "dispatch"):
-            self.dispatch = bootstrap_dispatch(self, name=self.shortname)
         self.dispatch.__code__ = rename_code(dispatch.__code__, self.shortname)
         self.dispatch.__kwdefaults__ = dispatch.__kwdefaults__
         self.dispatch.__annotations__ = dispatch.__annotations__
@@ -258,6 +260,7 @@ class Ovld:
         """Register a function."""
         if fn is None:
             return partial(self._register, priority=priority)
+        priority = getattr(fn, "priority", priority)
         return self._register(fn, priority)
 
     def _register(self, fn, priority):
@@ -290,13 +293,16 @@ class Ovld:
         self._update()
 
     def _update(self):
-        if self._compiled:
-            self.compile()
+        self.reset()
         for child in self.children:
             child._update()
         if hasattr(self, "dispatch"):
             self.dispatch.__doc__ = self.mkdoc()
 
+    def reset(self):
+        self._compiled = False
+        self.dispatch.__code__ = self.dispatch.first_entry.__code__
+
     def copy(self, mixins=[], linkback=False):
         """Create a copy of this Ovld.
 
@@ -319,8 +325,6 @@ class Ovld:
             return ov
 
     def __get__(self, obj, cls):
-        if not self._compiled:
-            self.compile()
         return self.dispatch.__get__(obj, cls)
 
     def __call__(self, *args, **kwargs):  # pragma: no cover
@@ -328,8 +332,6 @@ class Ovld:
 
         This should be replaced by an auto-generated function.
         """
-        if not self._compiled:
-            self.compile()
         return self.dispatch(*args, **kwargs)
 
     def next(self, *args):

ovld/medley.py

@@ -110,7 +110,7 @@ class ChainAll(ImplList):
 class BuildOvld(Combiner):
     def __init__(self, field=None, ovld=None):
         super().__init__(field)
-        self.ovld = ovld or Ovld(linkback=True)
+        self.ovld = ovld or Ovld(name=field, linkback=True)
         self.pending = []
         if field is not None:
             self.__set_name__(None, field)
@@ -125,7 +125,7 @@ class BuildOvld(Combiner):
         self.pending.clear()
         if not self.ovld.defns:
             return ABSENT
-        self.ovld.compile()
+        self.ovld.reset()
         return self.ovld.dispatch
 
     def copy(self):
@@ -144,10 +144,14 @@ class BuildOvld(Combiner):
 
 
 class medley_cls_dict(dict):
-    def __init__(self, bases):
+    def __init__(self, bases, default_combiner=None):
+        if default_combiner is None:
+            (default_combiner,) = {b._ovld_default_combiner for b in bases}
         super().__init__()
         self._combiners = {}
+        self._default_combiner = default_combiner
         self.set_direct("_ovld_combiners", self._combiners)
+        self.set_direct("_ovld_default_combiner", default_combiner)
         self._basic = set()
         for base in bases:
             for attr, combiner in getattr(base, "_ovld_combiners", {}).items():
@@ -175,7 +179,7 @@ class medley_cls_dict(dict):
         combiner = self._combiners.get(attr, None)
         if combiner is None:
             if inspect.isfunction(value) or isinstance(value, Ovld):
-                combiner = BuildOvld(attr)
+                combiner = self._default_combiner(attr)
             else:
                 combiner = KeepLast(attr)
             self._combiners[attr] = combiner
@@ -207,6 +211,19 @@ def specialize(cls, key):
     return new_t
 
 
+def remap_field(dc_field, require_default=False):
+    if require_default:
+        if dc_field.default is MISSING:
+            # NOTE: we do not accept default_factory, because we need the default value to be set
+            # in the class so that existing instances of classes[0] can see it.
+            raise TypeError(
+                f"Dataclass field '{dc_field.name}' must have a default value (not a default_factory) in order to be melded in."
+            )
+    dc_field = copy(dc_field)
+    dc_field.kw_only = True
+    return dc_field
+
+
 class MedleyMC(type):
     def __subclasscheck__(cls, subclass):
         if getattr(cls, "_ovld_medleys", None):
@@ -214,10 +231,10 @@ class MedleyMC(type):
         return super().__subclasscheck__(subclass)
 
     @classmethod
-    def __prepare__(mcls, name, bases):
-        return medley_cls_dict(bases)
+    def __prepare__(mcls, name, bases, default_combiner=None):
+        return medley_cls_dict(bases, default_combiner=default_combiner)
 
-    def __new__(mcls, name, bases, namespace):
+    def __new__(mcls, name, bases, namespace, default_combiner=None):
         result = super().__new__(mcls, name, bases, namespace)
         for attr, combiner in result._ovld_combiners.items():
             if (value := combiner.get(result)) is not ABSENT:
@@ -236,13 +253,16 @@ class MedleyMC(type):
         ]
         return dc
 
-    def extend(cls, *others):
-        if not others:
+    def extend(cls, *others, extend_subclasses=True):
+        if not others:  # pragma: no cover
             return cls
-        melded = meld_classes((cls, *others), require_defaults=True)
+        all_fields = [(f.name, f.type, f) for f in fields(cls)]
+        for other in others:
+            all_fields += [(f.name, f.type, remap_field(f, True)) for f in fields(other)]
+        melded = make_dataclass("_", fields=all_fields)
         for other in others:
             for k, v in vars(other).items():
-                if k in ["__module__", "__firstlineno__"]:
+                if k in ["__module__", "__firstlineno__", "__static_attributes__"]:
                     continue
                 elif comb := cls._ovld_combiners.get(k):
                     comb.juxtapose(v)
@@ -250,9 +270,10 @@ class MedleyMC(type):
                 elif not k.startswith("_ovld_") and not k.startswith("__"):
                     setattr(cls, k, v)
         cls.__init__ = melded.__init__
-        for subcls in cls.__subclasses__():
-            subothers = [o for o in others if not issubclass(subcls, o)]
-            subcls.extend(*subothers)
+        if extend_subclasses:
+            for subcls in cls.__subclasses__():
+                subothers = [o for o in others if not issubclass(subcls, o)]
+                subcls.extend(*subothers, extend_subclasses=False)
         return cls
 
     def __add__(cls, other):
@@ -290,7 +311,7 @@ def use_combiner(combiner):
     return deco
 
 
-class Medley(metaclass=MedleyMC):
+class Medley(metaclass=MedleyMC, default_combiner=BuildOvld):
     __post_init__ = RunAll()
     __add__ = KeepLast()
     __sub__ = KeepLast()
@@ -313,13 +334,10 @@ def unmeld_classes(main: type, exclude: type):
 _meld_classes_cache = {}
 
 
-def meld_classes(classes, require_defaults=False):
+def meld_classes(classes):
     medleys = {}
-    for i, cls in enumerate(classes):
-        if require_defaults and i == 0:
-            medleys[cls] = True
-        else:
-            medleys.update({x: True for x in getattr(cls, "_ovld_medleys", [cls])})
+    for cls in classes:
+        medleys.update({x: True for x in getattr(cls, "_ovld_medleys", [cls])})
     for cls in classes:
         if not hasattr(cls, "_ovld_medleys"):
             for base in cls.mro():
@@ -329,36 +347,26 @@ def meld_classes(classes, require_defaults=False):
     if len(medleys) == 1:
         return medleys[0]
 
-    cache_key = (medleys, require_defaults)
+    cache_key = medleys
     if cache_key in _meld_classes_cache:
         return _meld_classes_cache[cache_key]
 
-    def remap_field(dc_field, require_default):
-        if require_default:
-            if dc_field.default is MISSING:
-                # NOTE: we do not accept default_factory, because we need the default value to be set
-                # in the class so that existing instances of classes[0] can see it.
-                raise TypeError(
-                    f"Dataclass field '{dc_field.name}' must have a default value (not a default_factory) in order to be melded in."
-                )
-        dc_field = copy(dc_field)
-        dc_field.kw_only = True
-        return dc_field
-
     cg_fields = set()
     dc_fields = []
 
     for base in medleys:
-        rqdef = require_defaults and base is not medleys[0]
         cg_fields.update(base._ovld_codegen_fields)
         dc_fields.extend(
-            (f.name, f.type, remap_field(f, rqdef)) for f in base.__dataclass_fields__.values()
+            (f.name, f.type, remap_field(f)) for f in base.__dataclass_fields__.values()
         )
 
     merged = medley_cls_dict(medleys)
     merged.set_direct("_ovld_codegen_fields", tuple(cg_fields))
     merged.set_direct("_ovld_medleys", tuple(medleys))
 
+    if "__qualname__" in merged._combiners:
+        del merged._combiners["__qualname__"]
+
     result = make_dataclass(
         cls_name="+".join(sorted(c.__name__ for c in medleys)),
         bases=tuple(medleys),

ovld/recode.py

@@ -405,11 +405,8 @@ def _search_names(co, specials, glb, closure=None):
         if isinstance(co, CodeType):
             if closure is not None:
                 for varname, cell in zip(co.co_freevars, closure):
-                    try:
-                        if any(cell.cell_contents is v for v in values):
-                            yield varname
-                    except ValueError:  # cell is empty
-                        pass
+                    if any(cell.cell_contents is v for v in values):
+                        yield varname
             for name in co.co_names:
                 if any(glb.get(name, None) is v for v in values):
                     yield name

ovld/utils.py

@@ -65,6 +65,10 @@ class UsageError(Exception):
     pass
 
 
+class ResolutionError(TypeError):
+    pass
+
+
 class SpecialForm:
     def __init__(self, name, message=None):
         self.__name = name

ovld/version.py

@@ -1 +1 @@
-version = "0.5.0"
+version = "0.5.1"

{ovld-0.5.0.dist-info → ovld-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ovld
-Version: 0.5.0
+Version: 0.5.1
 Summary: Overloading Python functions
 Project-URL: Homepage, https://ovld.readthedocs.io/en/latest/
 Project-URL: Documentation, https://ovld.readthedocs.io/en/latest/

ovld-0.5.1.dist-info/RECORD

@@ -0,0 +1,18 @@
+ovld/__init__.py,sha256=JuCM8Sj65gobV0KYyLr95cSI23Pi6RYZ7X3_F3fdsSw,1821
+ovld/abc.py,sha256=4qpZyYwI8dWgY1Oiv5FhdKg2uzNcyWxIpGmGJVcjXrs,1177
+ovld/codegen.py,sha256=oHfQHlVJUB1Y0PO--rhhkfAmgOhIhKNZyBcodrS1GEs,9210
+ovld/core.py,sha256=bnp2LIVBUCvNhI53csWaAs7hlxinOBOXhPeoAw_GD3s,17443
+ovld/dependent.py,sha256=h3j4oQYTQfGqMzggWlLV6TpojX_GtYRFWAO0GcMB0Zs,9085
+ovld/medley.py,sha256=bp4e-6qH5_Qq05poPu7AcuWf-0mqNCqKMrdK44K2hSs,12667
+ovld/mro.py,sha256=nqe9jHvsaLu-CfXoP2f7tHg5_8IkcESWMnBmvr4XGIo,4605
+ovld/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ovld/recode.py,sha256=YrftCClBcP_luOdqeeQ_EQJddRjikLpY5RmYGRCQrOs,16008
+ovld/signatures.py,sha256=Q8JucSOun0ESGx14aWtHtBzLEiM6FxY5HP3imyqXoDo,8984
+ovld/typemap.py,sha256=5Pro1Ee60fH4L7NW7k5nbN5EfDygA0LFHcI6o3mCagI,13596
+ovld/types.py,sha256=0hkhAR5_5793NABdrM-fP1dSJBhYof85FILKqVP2YMg,13392
+ovld/utils.py,sha256=wO9IQqPVkcS1Opd8cl79WMMpWdCbMQY66PuWF2hMkvc,4272
+ovld/version.py,sha256=-u_JHy_NHod_JgjQcxN3CYo6TWCz6XQgbb-Z1FmF2Fg,18
+ovld-0.5.1.dist-info/METADATA,sha256=Lp8FkGbpbaMftXuCYLX3yeCdNTjV67U4ov0h9GV7tCI,9276
+ovld-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ovld-0.5.1.dist-info/licenses/LICENSE,sha256=cSwNTIzd1cbI89xt3PeZZYJP2y3j8Zus4bXgo4svpX8,1066
+ovld-0.5.1.dist-info/RECORD,,

ovld-0.5.0.dist-info/RECORD

@@ -1,18 +0,0 @@
-ovld/__init__.py,sha256=JuCM8Sj65gobV0KYyLr95cSI23Pi6RYZ7X3_F3fdsSw,1821
-ovld/abc.py,sha256=4qpZyYwI8dWgY1Oiv5FhdKg2uzNcyWxIpGmGJVcjXrs,1177
-ovld/codegen.py,sha256=0ZZdXw-nn9t52BX93GX6zwTeBB_wCmc5OxSk37sBrVU,9112
-ovld/core.py,sha256=uAnzuxa3hZjZSbCs3FW_0uR9qQlY-HZhgKpDTyvp59Y,17346
-ovld/dependent.py,sha256=h3j4oQYTQfGqMzggWlLV6TpojX_GtYRFWAO0GcMB0Zs,9085
-ovld/medley.py,sha256=Gin0EYeuQ7qZdDKNFFIwBY4o8W-VT5pl2bP8GCSopU4,12136
-ovld/mro.py,sha256=nqe9jHvsaLu-CfXoP2f7tHg5_8IkcESWMnBmvr4XGIo,4605
-ovld/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ovld/recode.py,sha256=AoNgJw75RAlfSBeFL5-UeB_T7wN_Xqzuahr9VhqC_yI,16126
-ovld/signatures.py,sha256=Q8JucSOun0ESGx14aWtHtBzLEiM6FxY5HP3imyqXoDo,8984
-ovld/typemap.py,sha256=5Pro1Ee60fH4L7NW7k5nbN5EfDygA0LFHcI6o3mCagI,13596
-ovld/types.py,sha256=0hkhAR5_5793NABdrM-fP1dSJBhYof85FILKqVP2YMg,13392
-ovld/utils.py,sha256=q2zFrK3OifhDu4SsVyLP4WKIp8FDDcJmLwMEUoWHS6c,4227
-ovld/version.py,sha256=-beU6vQRolzltfD0fQR6KSzR5yUgbfwYBlKbvzktqbg,18
-ovld-0.5.0.dist-info/METADATA,sha256=HzSeUKkJrJRUbPJZqw1XZpkW58EKhgl6GOi32CaPPcw,9276
-ovld-0.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ovld-0.5.0.dist-info/licenses/LICENSE,sha256=cSwNTIzd1cbI89xt3PeZZYJP2y3j8Zus4bXgo4svpX8,1066
-ovld-0.5.0.dist-info/RECORD,,