首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何使用Python装饰器检查函数参数?

如何使用Python装饰器检查函数参数?
EN

Stack Overflow用户
提问于 2013-03-08 17:29:50
回答 10查看 49.4K关注 0票数 48

在调用某些函数之前,我想定义一些泛型装饰器来检查参数。

类似于:

代码语言:javascript
复制
@checkArguments(types = ['int', 'float'])
def myFunction(thisVarIsAnInt, thisVarIsAFloat)
    ''' Here my code '''
    pass

边注:

  1. 类型检查只是为了给出一个例子
  2. 我正在使用Python2.7,但是Python3.0也可能很有趣

编辑2021:有趣的是,类型检查并没有从长远来看与类型暗示形象化合作。

EN

回答 10

Stack Overflow用户

回答已采纳

发布于 2013-03-08 17:49:24

来自函数和方法的装饰器

Python 2

代码语言:javascript
复制
def accepts(*types):
    def check_accepts(f):
        assert len(types) == f.func_code.co_argcount
        def new_f(*args, **kwds):
            for (a, t) in zip(args, types):
                assert isinstance(a, t), \
                       "arg %r does not match %s" % (a,t)
            return f(*args, **kwds)
        new_f.func_name = f.func_name
        return new_f
    return check_accepts

Python 3

在Python3中,func_code已改为__code__func_name已更改为__name__

代码语言:javascript
复制
def accepts(*types):
    def check_accepts(f):
        assert len(types) == f.__code__.co_argcount
        def new_f(*args, **kwds):
            for (a, t) in zip(args, types):
                assert isinstance(a, t), \
                       "arg %r does not match %s" % (a,t)
            return f(*args, **kwds)
        new_f.__name__ = f.__name__
        return new_f
    return check_accepts

用法:

代码语言:javascript
复制
@accepts(int, (int,float))
def func(arg1, arg2):
    return arg1 * arg2

func(3, 2) # -> 6
func('3', 2) # -> AssertionError: arg '3' does not match <type 'int'>

arg2可以是int,也可以是float

票数 53
EN

Stack Overflow用户

发布于 2013-03-08 17:32:41

在Python3.3上,您可以使用函数注释并检查:

代码语言:javascript
复制
import inspect

def validate(f):
    def wrapper(*args):
        fname = f.__name__
        fsig = inspect.signature(f)
        vars = ', '.join('{}={}'.format(*pair) for pair in zip(fsig.parameters, args))
        params={k:v for k,v in zip(fsig.parameters, args)}
        print('wrapped call to {}({})'.format(fname, params))
        for k, v in fsig.parameters.items():
            p=params[k]
            msg='call to {}({}): {} failed {})'.format(fname, vars, k, v.annotation.__name__)
            assert v.annotation(params[k]), msg
        ret = f(*args)
        print('  returning {} with annotation: "{}"'.format(ret, fsig.return_annotation))
        return ret
    return wrapper

@validate
def xXy(x: lambda _x: 10<_x<100, y: lambda _y: isinstance(_y,float)) -> ('x times y','in X and Y units'):
    return x*y

xy = xXy(10,3)
print(xy)

如果存在验证错误,请打印:

代码语言:javascript
复制
AssertionError: call to xXy(x=12, y=3): y failed <lambda>)

如果没有验证错误,请打印:

代码语言:javascript
复制
wrapped call to xXy({'y': 3.0, 'x': 12})
  returning 36.0 with annotation: "('x times y', 'in X and Y units')"

您可以使用函数而不是lambda来获得断言失败中的名称。

票数 18
EN

Stack Overflow用户

发布于 2016-02-29 14:58:24

正如你所知道的,仅仅根据一个论点的类型来拒绝它并不是节奏曲。

Pythonic的方法更像是“先处理它”。

这就是为什么我宁愿做一个装潢师来转换论点

代码语言:javascript
复制
def enforce(*types):
    def decorator(f):
        def new_f(*args, **kwds):
            #we need to convert args into something mutable   
            newargs = []        
            for (a, t) in zip(args, types):
               newargs.append( t(a)) #feel free to have more elaborated convertion
            return f(*newargs, **kwds)
        return new_f
    return decorator

这样,您的函数就会得到您所期望的类型,但是如果参数可以像浮点一样嘎嘎地运行,则它将被接受。

代码语言:javascript
复制
@enforce(int, float)
def func(arg1, arg2):
    return arg1 * arg2

print (func(3, 2)) # -> 6.0
print (func('3', 2)) # -> 6.0
print (func('three', 2)) # -> ValueError: invalid literal for int() with base 10: 'three'

我使用这个技巧(使用适当的转换方法)来处理向量

我编写的许多方法都期望使用MyVector类,因为它具有大量的功能;但是有时您只想编写

代码语言:javascript
复制
transpose ((2,4))
票数 11
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/15299878

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档