如何简化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

您可以使用装潢工 (您已经通过@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()将记录：

启动任务SRV_DEV 终端任务SRV_DEV