python通过装饰器检查函数参数数据类型的方法_Python

python通过装饰器检查函数参数数据类型的方法

2019-11-28 17:17niuniu Python

这篇文章主要介绍了python通过装饰器检查函数参数数据类型的方法,涉及Python装饰器的使用技巧,具有一定参考借鉴价值,需要的朋友可以参考下

本文实例讲述了python通过装饰器检查函数参数数据类型的方法。分享给大家供大家参考。具体分析如下：

这段代码定义了一个python装饰器，通过此装饰器可以用来检查指定函数的参数是否是指定的类型，在定义函数时加入此装饰器可以非常清晰的检测函数参数的类型，非常方便

	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')