python - 如何简化 python 日志记录代码

Question

我想记录我的任何函数的开始/结束 - 我怎样才能将这段代码简化为包装器或其他东西？

@task()
def srv_dev(destination=development):
    logging.info("Start task " + str(inspect.stack()[0][3]).upper())
    configure(file_server, destination)
    logging.info("End task " + str(inspect.stack()[0][3]).upper()) 

Answer 1

你可以使用 decorator (你已经通过 @task() 做了什么)。这是一个装饰器，它以大写字母的开头和结尾记录任何函数的名称:

import logging
import inspect
import functools

def log_begin_end(func):
    """This is a decorator that logs the name of `func` (in capital letters).

    The name is logged at the beginning and end of the function execution.

    """
    @functools.wraps(func)
    def new_func(*args, **kwargs):

        logging.info("Start task " + func.__name__.upper())
        result = func(*args, **kwargs)
        logging.info("End task " + func.__name__.upper())
        return result

    return new_func

用法如下:

@log_begin_end
def myfunc(x,y,z):
    pass  # or do whatever you want

当然，你可以级联装饰器。因此，在您的情况下，您可以使用:

@task()
@log_begin_end
def srv_dev(destination=development):
    configure(file_server, destination)

现在调用 srv_dev() 将记录:

Start task SRV_DEV

End task SRV_DEV