python - 装饰器创建类

Question

我正在编写一堆可能具有可变输出的代码，例如一个算术函数，其中我可以让输出为 float 或整数。基本上我的问题是，如果我为我需要的每种对象类型(可能是七个或八个)创建一个装饰器，我会因为不断重复以下内容而发疯:

def Int(fn):
    def wrapper():
        return int(fn())
    return wrapper

我想要的是一个像下面这样的类，它将根据实例化的名称创建一个装饰器，并且它将是上面函数的副本，但进行了适当的类型修改。

class Decorator(object):
    def __init__(self):
        ...
    ...


Int = Decorator()

# Then I can use @Int

任何帮助将不胜感激。谢谢。

Answer 1

您无法让Decorator知道它将被分配给什么名称。分配发生在实例化之后，因此对象在被分配名称时就已经被创建了。

但是，您可以创建一个动态创建装饰器的装饰器:

from functools import wraps
def set_return_type(typeobj):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            return typeobj(func(*args, **kwargs))
        return wrapper
    return decorator

然后，您可以通过为所需类型提供类型对象参数来使用此装饰器:

@set_return_type(int)   # Causes decorated function to return ints
@set_return_type(float) # Causes decorated function to return floats

下面是一个演示:

>>> from functools import wraps
>>> def set_return_type(typeobj):
...     def decorator(func):
...         @wraps(func)
...         def wrapper(*args, **kwargs):
...             return typeobj(func(*args, **kwargs))
...         return wrapper
...     return decorator
...
>>> @set_return_type(float)
... def test():
...     return 1
...
>>> test()
1.0
>>>