Skip to content

Commit

Permalink
Update __init__.py
Browse files Browse the repository at this point in the history
  • Loading branch information
ken-morel committed May 2, 2024
1 parent 00f6fd7 commit a35471c
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions src/py_overload/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,19 +35,20 @@ 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:
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()
Expand Down Expand Up @@ -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])
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit a35471c

Please sign in to comment.