From a35471c79992b06d19f295caa50c7a22ffcb0656 Mon Sep 17 00:00:00 2001 From: ken-morel Date: Thu, 2 May 2024 01:52:04 +0100 Subject: [PATCH] Update __init__.py --- src/py_overload/__init__.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/src/py_overload/__init__.py b/src/py_overload/__init__.py index 5410415..fe225d7 100644 --- a/src/py_overload/__init__.py +++ b/src/py_overload/__init__.py @@ -35,12 +35,12 @@ def __call__(self, val): def __str__(self): return 'Values(' + ', '.join(map(repr, self)) + ')' - + __repr__ = __str__ def get_name(funcOrCls): - return funcOrCls.__module__+'.'+funcOrCls.__qualname__ + return funcOrCls.__module__+ '.' + funcOrCls.__qualname__ class TypeChecker: @@ -48,6 +48,7 @@ def __init__(self, func): if not callable(func): raise TypeError(self.__class__.__init__.__qualname__) self.func = func + def __call__(self, val): try: return self.func() @@ -122,12 +123,11 @@ def wrapper(*args, **kw): vals.update(kw) except IndexError as e: raise AnnotationError( - f"Was function {get_name(func)} properly annotated?" + f'Was function {get_name(func)} properly annotated?' ) from e errors = [] for k, v in vals.items(): - #(k, v) if isinstance(anno[k], Cast): vals[k] = anno[k](v) #(" casting", k, "to", anno[k]) @@ -162,10 +162,8 @@ def overload(func, name=None): return partial(overload, name=func) if name is None or not isinstance(name, str): name = get_name(func) - if name not in __overloads__: __overloads__[name] = [] - __overloads__[name].append(annotate(func, True)) func.__overloads__ = __overloads__[name] @wraps(func)