Question

我有一个功能

import logging

def print_logs():
    logging.info('info log 1.')
    logging.warning('warning log 1.')
    logging.error('error log 1.')
    logging.error('error log 2.')

我想调用此函数，然后获取指定日志记录级别的所有日志。所以我希望能够做类似的事情：

print_logs()
error_logs = get_logs_by_level('error')

或

error_logs = get_logs_by_level(print_logs, 'error')

并且函数print_logs运行，并且error_logs将是['error log 1.', 'error log 2']

这可能吗？

Answer 1

以下是一个安装处理程序以捕获日志的示例。

这在Python 2.7中有效。对于Python 3，将cStringIO.StringIO替换为io.StringIO或io.BytesIO。或使用不需要的ListHander。

如果您希望保留日志记录的格式，然后再对每条消息的行之间进行分隔。可以对此进行修改，但需要编写Handler。

import logging
import cStringIO

class Handler(object):
    def __init__(self, handler):
        self.handler = handler
    def __enter__(self):
        logging.root.handlers.append(self.handler)
        return self.handler
    def __exit__(self, type, value, tb):
        logging.root.handlers.remove(self.handler)

def print_logs():
    logging.info('info log 1.')
    logging.warning('warning log 1.')
    logging.error('error log 1.')
    logging.error('error log 2.')

h = logging.StreamHandler(stream=cStringIO.StringIO())
h.setLevel('ERROR')

with Handler(h):
    print_logs()

error_logs = h.stream.getvalue().splitlines()
error_logs

class RecordListHandler(logging.Handler):
    def __init__(self):
        logging.Handler.__init__(self)
        self.log = []

    def emit(self, record):
        self.log.append(record)

class ListHandler(logging.Handler):

    def __init__(self):
        logging.Handler.__init__(self)
        self.log = []

    def emit(self, record):
        msg = self.format(record)
        self.log.append(msg)

h = RecordListHandler()
h.setLevel('ERROR')

with Handler(h):
    print_logs()

error_logs = h.log
error_logs

h = ListHandler()
h.setLevel('ERROR')

with Handler(h):
    print_logs()

error_logs = h.log
error_logs

在python中获取某个级别的所有日志？

1 个答案: