这段代码定义了一个python装饰器,通过此装饰器可以用来检查指定函数的参数是否是指定的类型,在定义函数时加入此装饰器可以非常清晰的检测函数参数的类型,非常方便,75271.com强烈推荐。
def accepts(exception,**types): def check_accepts(f): assert len(types) == f.func_code.co_argcount, \ 'accept number of arguments not equal with function number of arguments in "%s"' % f.func_name def new_f(*args, **kwds): for i,v in enumerate(args): if types.has_key(f.func_code.co_varnames[i]) and \ not isinstance(v, types[f.func_code.co_varnames[i]]): raise exception("arg '%s'=%r does not match %s" % \ (f.func_code.co_varnames[i],v,types[f.func_code.co_varnames[i]])) del types[f.func_code.co_varnames[i]] for k,v in kwds.iteritems(): if types.has_key(k) and not isinstance(v, types[k]): raise exception("arg '%s'=%r does not match %s" % \ (k,v,types[k])) return f(*args, **kwds) new_f.func_name = f.func_name return new_f return check_accepts def exmaple(): @accepts(Exception,a=int,b=list,c=(str,unicode)) def test(a,b=None,c=None) print 'ok' test(13,c=[],b='df')