如何在修饰器中获取参数类型提示？ [英] How to get argument type hint in decorator?

问题描述

如何执行以下操作：

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):
    …

推荐答案

它们存储在函数的__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">}

这篇关于如何在修饰器中获取参数类型提示？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！