python3帶參數的裝飾器 函數參數類型檢查

from inspect import signature#python3纔有的模塊
def typeassert(*args,**kwargs):
    def decorator(fun):
        sig=signature(fun)
        btypes=sig.bind_partial(*args,**kwargs).arguments
        def wrapper(*funargs,**funkwargs):
            for name,stype in sig.bind_partial(*funargs,**funkwargs).arguments.items():
                if name in btypes:
                    if not isinstance(stype,btypes[name]):
                        raise TypeError("'%s' must be '%s'"%(name,btypes[name]))
            return fun(*funargs,**funkwargs)
        return wrapper    
    return decorator

@typeassert(int,str)
def myFun(a,b):
    print(a,b)
myFun(12,"12")
相關文章
相關標籤/搜索