在装饰器中声明的函数访问变量

Question

我想记录很多指标，例如函数执行所需的时间。我希望解决方案是轻量级的。添加一个装饰器并完成。也许 90% 的时间是这样的：

@LogMetrics
def my_business_logic(var1):
  print 'done'

但是，我还希望能够在 10% 的时间内访问指标对象并执行类似的操作：

@LogMetrics
def my_business_logic(var1):
  metrics.inc_counter('counter_a')
  print 'done'

到目前为止，我有一个处理第一个场景的装饰器：

class LogMetrics(object):

  def __init__(self, module=None):
    self.module = module


  def __call__(self, target):

    def wrapped(*args, **kwargs):
      # set up metrics
      metrics = Metrics(module=module)

      metrics.start_timer()
      try:
        result = target(*args, **kwargs)
      except Exception as e:
        metrics.set_fault()
        raise e
      finally:
        metrics.stop_timer()
        metrics.push()

      return result

    return wrapped

如何支持方案 2？我想做类似answer 的事情，但共识似乎是不这样做。请帮忙...

Answer 1

您可以为函数添加属性。我做了一个非常简单的例子（在 Python3 中）。 add_counter 装饰器将 _counter 属性添加到原始函数（您将添加一个 metrics 对象）。此示例中的计数器在每次调用后递增。最后，一个_orig_func 属性被添加到装饰函数中以允许访问计数器。

def add_counter(func):
    func._counter = 0 
    def decorated(*args, **kwargs):
        result = func(*args, **kwargs)
        func._counter += 1
        return result
    decorated._orig_func = func
    return decorated

@add_counter
def func1():
    print("FUNC1")
    print(func1._orig_func._counter)

@add_counter
def func2():
    print("FUNC2")
    print(func2._orig_func._counter)

func1()
func1()
func1()
func2()
func2()
func1()

Answer 2

不涉及全局对象或禁止魔法的唯一解决方案是将您的指标对象作为可选参数传递给您的函数（我假设如果您可以编辑它，那么您可以更改它的签名）：

class LogMetrics(object):

  ...

  def __call__(self, target):

    def wrapped(*args, **kwargs):
      # set up metrics
      metrics = Metrics(module=module)
      kwargs['metrics'] = metrics  # add metrics object to kwargs
      metrics.start_timer()
      try:
        result = target(*args, **kwargs)
      except Exception as e:
        metrics.set_fault()
        raise e
      finally:
        metrics.stop_timer()
        metrics.push()
      return result
    return wrapped


@LogMetrics
def my_business_logic(var1, metrics=None):
  if metrics is not None:
      metrics.inc_counter('counter_a')
  print 'done'

顺便问一下，您确定要在 wrapped 中而不是在 __call__ 中初始化 Metrics 吗？