如何使用函数注释来验证函数调用类型

时间:2017-02-27 00:28:59

标签: python python-3.x python-decorators type-hinting

我最近刚发现有一个叫做函数注释的东西,但我不太清楚如何使用它。这就是我到目前为止所做的:

def check_type(f):
    def decorated(*args, **kwargs):
        counter=0
        for arg, type in zip(args, f.__annotations__.items()):
            if not isinstance(arg, type[1]):
                msg = 'Not the valid type'
                raise ValueError(msg)
            counter+=1

        return f(*args, **kwargs)
    return decorated

@check_type
def foo(a: int, b: list, c: str): #a must be int, b must be list, c must be str
    print(a,b,c)

foo(12, [1,2], '12') #This works

foo(12, 12, 12) #This raises a value error just as I wanted to

foo(a=12, b=12, c=12) #But this works too:(

正如您所看到的,我正在尝试使用注释和装饰器检查abc的类型,如果不是ValueError则会引发def check_type(f): def decorated(*args, **kwargs): for name, type in f.__annotations__.items(): if not isinstance(kwargs[name], type): msg = 'Not the valid type' raise ValueError(msg) return f(*args, **kwargs) return decorated #But now they have to be assigned using keyword args #so only foo(a=3,b=[],c='a') works foo(3,[],'a') results in a keyerror #How can I combine them? 正确的类型。当我在调用时不使用关键字参数时,它工作正常。但是,如果我使用关键字参数,则不会检查类型。我试图让它运作但我没有运气。

我的代码不支持关键字参数。因为我没有任何检查的东西。我也不知道如何检查它。这是我需要帮助的地方。

我也是这样做的:

[NSURLConnection sendAsynchronousRequest:request queue:myQueue completionHandler:^(NSURLResponse *response, NSData *data, NSError *error) {
        NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *) response;
        NSLog(@"response status code: %ld, error status : %@", (long)[httpResponse statusCode], error.description);

        if ((long)[httpResponse statusCode] >= 200 && (long)[httpResponse statusCode]< 400)
            {
               // do stuff
                [self requestFunction]; //Web Service
            }
}];

1 个答案:

答案 0 :(得分:4)

正如Paul所建议的那样,最好使用bind Signature对象(位于inspect)的*args方法来绑定**kwargsf将被提供给from inspect import signature from typing import get_type_hints def check_range(f): def decorated(*args, **kwargs): counter=0 # use get_type_hints instead of __annotations__ annotations = get_type_hints(f) # bind signature to arguments and get an # ordered dictionary of the arguments b = signature(f).bind(*args, **kwargs).arguments for name, value in b.items(): if not isinstance(value, annotations[name]): msg = 'Not the valid type' raise ValueError(msg) counter+=1 return f(*args, **kwargs) return decorated ,然后检查类型是否匹配:

dict

你的第一个案子实际上是随机成功的。 < 3.6在Python zip中有一个随机顺序,当您再次启动Python解释器时很可能会改变,这意味着您执行的f.__annotations__ ping不具有确定性。

而不是遍历get_type_hints,而不是通过b.items()抓取它,然后通过name(这是一个OrderedDict并保证顺序)获取名称和值,用{{{ 1}}。