0

前言

我正在使用带有装饰器的单一调度，例如functools.singledispatch

from functools import singledispatch
from typing import Any


@singledispatch
def serialize(object_: Any) -> str:
    raise TypeError('Serialization for objects '
                    'of type "{type}" is not supported.'
                    .format(type=type(object_)))


@serialize.register(int)
def serialize_int(object_: int) -> str:
    return str(object_)

serialize_int现在如果我想有条件地装饰

log_int_overload = True  # some condition here

if log_int_overload:
    from functools import wraps


    def log(function):
        @wraps(function)
        def wrapped(*args, **kwargs):
            result = function(*args, **kwargs)
            print('For {args}, {kwargs} function returned: {result}.'
                  .format(args=args,
                          kwargs=kwargs,
                          result=result))
            return result

        return wrapped


    serialize_int = log(serialize_int)

然后打电话

>>> serialize(1)
'1'

正如我们所看到的，它调用了“旧”的、未修饰的版本。发生这种情况是因为我们只注册了“旧的”一个重载并且serialize对装饰一无所知，因为它是一个新的函数对象。

问题

如何修改重载，以便“调度程序”功能将获取更改？

我知道serialize_int装修后我们可以重新注册，但是看起来是代码重复的问题，还有其他方法吗？

或更笼统地说：我们能以某种方式就地修改函数对象吗？装饰__call__属性无济于事。

4

0 回答 0