python - 如何在不更改参数名称的情况下装饰 Python 函数？

Question

以下 Python 代码定义了一个记录器和一个阶乘函数，然后使用关键字参数调用阶乘函数:

def logger(f):
  def f_(a):
    print("Call", a)
    return f(a)
  return f_

# @logger # uncomment this line to see the problem
def factorial(n):
  return 1 if n == 0 else n * factorial(n-1)

print(factorial(n=5))

产生以下输出(如预期):120。

现在如果我取消注释记录器装饰器，我会得到一个错误，因为参数的名称变成了 a 而不是 n:

如何在不更改参数名称的情况下修饰函数(如阶乘)？

Answer 1

忽略装饰器语法，这是你正在做的:

def logger(f):
    def f_(a):
        print("Call", a)
        return f(a)
    return f_

def factorial(n):
    return 1 if n == 0 else n * factorial(n-1)

# `factorial` is `def f_(a):` now
factorial = logger(factorial)

因此，要简单地解决这个问题，请使用相同的声明，或者不要使用关键字参数 (n=5)

---

更正此问题的更好方法是在内部函数中使用解包:

import functools

def logger(f):
    @functools.wraps(f)
    def f_(*args, **kwargs):
        print("Call", f.__name__, *args, *[f"{k}={v!r}" for k, v in kwargs.items()])
        return f(*args, **kwargs)
    return f_

---

Here's a useful article on the topic.