python-decorator.md

Python decorator example

def typecheck(type_):
    def decorator_(func):
    
        def wrapper_(name):
           if not type(name) == type_:
               raise TypeError
           else:
               return func(name)
        
        return wrapper_
        
    return decorator_

@typecheck(str)
def f_str(v):
    print('str ' + v)

@typecheck(int)
def f_int(v):
    print('int ' + str(v))


f_str('a string')
f_int(1)
f_str(1)