如何在YAML中指定总是使用dictConfig在项目文件夹中创建日志文件?

时间:2013-03-28 22:49:27

标签: python logging python-3.x yaml configuration-files

在我的Python程序中,我有以下代码:

def main():
    # The file's path
    path = os.path.dirname(os.path.realpath(__file__))
    ...
    # Config file relative to this file
    loggingConf = open('{0}/configs/logging.yml'.format(path), 'r')
    logging.config.dictConfig(yaml.load(loggingConf))
    loggingConf.close()
    logger = logging.getLogger(LOGGER)
    ...

这是我的logging.yml配置文件:

version: 1
formatters:
  default:
    format: '%(asctime)s %(levelname)s %(name)s %(message)s'
handlers:
  console:
    class: logging.StreamHandler
    level: DEBUG
    formatter: default
    stream: ext://sys.stdout
  file:
    class : logging.FileHandler
    formatter: default
    filename: bot.log
loggers:
  cloaked_chatter:
    level: DEBUG
    handlers: [console, file]
    propagate: no

问题是在程序启动的地方创建了bot.log文件。我希望它始终在项目的文件夹中创建,即与我的Python程序在同一文件夹中。

例如,使用./bot.py启动程序会在同一文件夹中创建日志文件。但是使用python3 path/bot.py启动它会将日志文件创建为文件层次结构中Python程序之上的级别。

我应该如何在配置文件中编写文件名来解决这个问题?或者我需要编写自定义处理程序?如果是这样,怎么样?或者使用dictConfig无法解决这个问题?

1 个答案:

答案 0 :(得分:9)

有很多方法可以达到你想要的效果。例如,一种方法是为处理程序创建自定义初始化程序:

import os
import yaml

def logmaker():
    path = os.path.dirname(os.path.realpath(__file__))
    path = os.path.join(path, 'bot.log')
    return logging.FileHandler(path)

def main():
    # The file's path
    path = os.path.dirname(os.path.realpath(__file__))

    # Config file relative to this file
    loggingConf = open('{0}/logging.yml'.format(path), 'r')
    logging.config.dictConfig(yaml.load(loggingConf))
    loggingConf.close()
    logger = logging.getLogger('cloaked_chatter')
    logger.debug('Hello, world!')

if __name__ == '__main__':
    main()

请注意,我将logging.yml移到了脚本旁边。 logmaker是自定义初始化程序。在YAML中指定它,如下所示:

version: 1
formatters:
  default:
    format: '%(asctime)s %(levelname)s %(name)s %(message)s'
handlers:
  console:
    class: logging.StreamHandler
    level: DEBUG
    formatter: default
    stream: ext://sys.stdout
  file:
    () : __main__.logmaker
    formatter: default
loggers:
  cloaked_chatter:
    level: DEBUG
    handlers: [console, file]
    propagate: no

如果您运行Python脚本,您会发现在脚本和YAML文件旁边创建了bot.log。相同的消息将打印到控制台和bot.log

2013-04-16 11:08:11,178 DEBUG cloaked_chatter Hello, world!

N.B。脚本可能有点整洁,但它说明了我的观点。

更新:根据the documentation,使用()作为字典中的键表示该值是可调用的,实际上是处理程序的自定义构造函数。