如何在装饰器中获取参数类型提示?

时间:2017-11-15 03:23:40

标签: python python-3.5

如何执行以下操作:

import typing

def needs_parameter_type(decorated):
    def decorator(*args):
        [do something with the *type* of bar (aka args[0]), some_class]
        decorated(*args)
    return decorator

@needs_parameter_type
def foo(bar: SomeClass):
    …

foo(…)

用例是为了避免以下重复:

@needs_parameter_type(SomeClass)
def foo(bar: SomeClass):
    …

1 个答案:

答案 0 :(得分:2)

这些存储在函数的__annotations__属性中,您可以通过以下方式访问它们:

def needs_parameter_type(decorated):
    def decorator(*args):
        print(decorated.__annotations__)
        decorated(*args)
    return decorator

@needs_parameter_type
def foo(bar: int):
   pass

foo(1)
#  {"bar": <class "int">}