python - 如何在 Python 中以可重用的方式记录函数参数？

Question

我发现自己多次编写这样的代码:

def my_func(a, b, *args, **kwargs):
    saved_args = locals() # Learned about this from http://stackoverflow.com/a/3137022/2829764
    local_var = "This is some other local var that I don't want to log"
    try:
        a/b
    except Exception as e:
        logging.exception("Oh no! My args were: " + str(saved_args))
        raise

运行 my_func(1, 0, "spam", "ham", my_kwarg="eggs") 在 stderr 上给出以下输出:

ERROR:root:Oh no! My args were: {'a': 1, 'args': (u'spam', u'ham'), 'b': 0, 'kwargs': {'my_kwarg': u'eggs'}}
Traceback (most recent call last):
  File "/Users/kuzzooroo/Desktop/question.py", line 17, in my_func
    a/b
ZeroDivisionError: division by zero

我的问题是，我可以写一些可重用的东西，这样我就不必在函数顶部保存 locals() 了吗？它可以用一种很好的 Pythonic 方式完成吗？

编辑:对@mtik00 的另一个请求:理想情况下，我想要某种方式从 my_func 中访问 saved_args 或类似的东西，这样我就可以做一些除了记录未捕获的异常之外的事情(也许我想捕获异常my_func，记录错误，然后继续。

Answer 1

装饰器 正是您要寻找的。这是一个例子:

import logging
from functools import wraps


def arg_logger(func):

    @wraps(func)
    def new_func(*args, **kwargs):
        saved_args = locals()
        try:
            return func(*args, **kwargs)
        except:
            logging.exception("Oh no! My args were: " + str(saved_args))
            raise

    return new_func


@arg_logger
def func(arg1, arg2):
    return 1 / 0

if __name__ == '__main__':
    func(1, 2)

在这里，我使用 arg_logger() 作为装饰器。将装饰器应用于您希望具有此新行为的任何函数。

关于装饰器的讨论很好here .