# dubugly.py

# Mohammad Usman

#

# A collection of useful decorators

from inspect import signature

from functools import wraps

# Should only be used on functions/methods that don't have keyword arguments

def typeassert(*ty_args, **ty_kwargs):

def decorate(func):

# Map function argument names to supplied types

sig = signature(func)

bound_types = sig.bind_partial(*ty_args, **ty_kwargs).arguments

@wraps(func)

def wrapper(*args, **kwargs):

bound_values = sig.bind(*args, **kwargs)

# Enforce type assertions across supplied arguments

for name, value in bound_values.arguments.items():

if name in bound_types:

if not isinstance(value, bound_types[name]):

raise TypeError(

'Argument {} must be {}'.format(name, bound_types[name])

)

return func(*args, **kwargs)

return wrapper

return decorate