使用setup_logging()函数进行日志设置

在Python中，使用logging模块进行日志记录是非常常见的。logging模块提供了一组用于创建和管理日志记录的函数和类。

在使用logging模块之前，我们通常需要进行一些基本的设置，例如定义日志的格式、选择日志记录级别等。为了简化这些设置步骤，我们可以编写一个setup_logging函数来封装这些设置过程。

下面是一个使用setup_logging函数进行日志设置的简单示例：

import logging

def setup_logging(logfile='app.log', level=logging.INFO):
    # 创建一个logger实例
    logger = logging.getLogger()
    logger.setLevel(level)

    # 创建一个文件处理器，用于将日志写入文件
    file_handler = logging.FileHandler(logfile)
    file_handler.setLevel(level)

    # 创建一个控制台处理器，用于将日志输出到控制台
    console_handler = logging.StreamHandler()
    console_handler.setLevel(level)

    # 定义日志格式
    formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
    file_handler.setFormatter(formatter)
    console_handler.setFormatter(formatter)

    # 将处理器添加到logger实例
    logger.addHandler(file_handler)
    logger.addHandler(console_handler)

    # 返回logger实例，以便在代码中进行日志记录
    return logger

def main():
    # 设置日志
    logger = setup_logging()

    # 记录一条日志信息
    logger.info('This is a log message')

if __name__ == '__main__':
    main()

在上面的示例中，setup_logging函数接受两个可选参数：logfile和level。logfile用于指定日志文件的路径和名称，默认为app.log。level用于指定日志记录的级别，默认为logging.INFO。

在setup_logging函数内部，我们首先创建了一个logger实例，并设置了日志级别。接下来，我们创建了一个文件处理器和一个控制台处理器，并为它们设置了相同的日志级别。然后，我们定义了日志的格式，并将其应用到处理器上。最后，我们将处理器添加到logger实例中，以便可以进行日志记录。

在main函数中，我们首先调用setup_logging函数来设置日志。然后，我们使用logger.info方法记录一条日志信息。

运行上述代码后，日志将被写入app.log文件中，并且会同时在控制台上显示。日志的格式为YYYY-MM-DD HH:MM:SS - LEVEL - MESSAGE。

使用logging模块进行日志记录是一种非常灵活和强大的方式。通过编写一个setup_logging函数，我们可以将常见的日志设置封装起来，使其在多个代码文件中重复使用变得更加方便。这样，我们就可以更轻松地进行日志记录，从而更好地监控和调试我们的应用程序。