python - 如何使用python引用可调用装饰器对象中的对象实例？

Question

上下文:

我希望能够装饰函数，以便我可以跟踪它们的统计数据。使用this作为引用，我开始尝试制作自己的可调用装饰器对象。

这是我最终得到的结果:

def Stats(fn):
    Class StatsObject(object):
        def __init__(self, fn):
            self.fn = fn
            self.stats = {}

        def __call__(self, obj, *args, **kwargs):
            self.stats['times_called'] = self.stats.get('times_called', 0) + 1
            return self.fn(obj, *args, **kwargs)

    function = StatsObject(fn)
    def wrapper(self, *args **kwargs):
        return function(self, *args, **kwargs)
    return wrapper

Class MockClass(object):
    @Stats
    def mock_fn(self, *args, **kwargs):
        # do things

问题:

这实际上正确地调用了mock_fn函数，但我没有对包装函数之外的统计对象的引用。即我不能这样做:

mc = MockClass()
mc.mock_fn()
mc.mock_fn.stats
# HasNoAttribute Exception

然后我尝试更改以下代码，认识到这是一个范围问题:

来自:

    function = StatsObject(fn)
    def wrapper(self, *args **kwargs):
        return function(self, *args, **kwargs)
    return wrapper

致:

    function = StatsObject(fn)
    return function

但是我当然丢失了 self 引用(self 成为 StatsObject 实例，obj 成为第一个参数，并且 MockClass 对象自引用丢失)。

所以我明白为什么会发生第一个问题，但不会发生第二个问题。有什么方法可以将 MockClass 的自引用传递给 StatsObject __call__ 函数吗？

Answer 1

在 Python 中，函数本身实际上可以具有属性。

def Stats(fn):
    class StatsObject(object):
        def __init__(self, fn):
            self.fn = fn
            self.stats = {}

        def __call__(self, obj, *args, **kwargs):
            self.stats['times_called'] = self.stats.get('times_called', 0) + 1
            return self.fn(obj, *args, **kwargs)

    function = StatsObject(fn)
    def wrapper(self, *args **kwargs):
        return function(self, *args, **kwargs)

    # KEY LINE BELOW: make the StatsObject available outside as "stats_fn"
    wrapper.stats_fn = function

    return wrapper

class MockClass(object):
    @Stats
    def mock_fn(self, *args, **kwargs):
        # do things

关键行是将 StatsObject 实例(您可能会误导，在本地命名为 function)分配为您返回的函数的属性来自装饰器。

一旦执行此操作，self.mock_fn.stats_fn.stats(不是 self.mock_fn()!该属性位于函数上，而不是其返回值)将在 MockClass 的实例内工作，并且 MockClass.mock_fn.stats_fn.stats 将在外部可用。统计信息将在 MockClass 的所有实例中是全局的(因为装饰器被调用一次，而不是每个实例调用一次)，这可能是也可能不是您想要的。