logging.error() 被调用了多少次？

Question

也许它根本不存在，因为我找不到它。 但是使用 python 的日志记录包，有没有办法查询记录器以找出特定函数被调用的次数？ 例如，报告了多少错误/警告？

Answer 1

日志记录模块似乎不支持此功能。 从长远来看，您可能最好创建一个新模块，并通过对现有日志记录模块中的项目进行子类化来添加此功能，以添加您需要的功能，但您也可以使用装饰器轻松实现此行为：

class CallCounted:
    """Decorator to determine number of calls for a method"""

    def __init__(self,method):
        self.method=method
        self.counter=0

    def __call__(self,*args,**kwargs):
        self.counter+=1
        return self.method(*args,**kwargs)


import logging
logging.error = CallCounted(logging.error)
logging.error('one')
logging.error('two')
print(logging.error.counter)

输出：

ERROR:root:one
ERROR:root:two
2

Answer 2

您还可以向记录器添加一个新的处理程序来计算所有调用：

class MsgCounterHandler(logging.Handler):
    level2count = None

    def __init__(self, *args, **kwargs):
        super(MsgCounterHandler, self).__init__(*args, **kwargs)
        self.level2count = {}

    def emit(self, record):
        l = record.levelname
        if (l not in self.level2count):
            self.level2count[l] = 0
        self.level2count[l] += 1

然后您可以使用字典来输出调用数。

Answer 3

有一个 warnings 模块 -在某种程度上——做了一些这样的事情。

您可能希望将此计数功能添加到自定义 Handler 。 问题是有一百万个处理程序，您可能希望将其添加到多个类型中。

您可能想将其添加到过滤器，因为那是独立于正在使用的处理程序。

Answer 4

基于罗尔夫的回答和如何将字典写入文件，这里是另一个将计数存储在 json 文件中的解决方案。 如果 json 文件存在并且 continue_counts=True，它将恢复初始化时的计数。

import json
import logging
import logging.handlers
import os


class MsgCounterHandler(logging.Handler):
    """
    A handler class which counts the logging records by level and periodically writes the counts to a json file.
    """

    level2count_dict = None

    def __init__(self, filename, continue_counts=True, *args, **kwargs):
        """
        Initialize the handler.

        PARAMETER
        ---------
        continue_counts: bool, optional
            defines if the counts should be loaded and restored if the json file exists already.
        """
        super(MsgCounterHandler, self).__init__(*args, **kwargs)

        filename = os.fspath(filename)
        self.baseFilename = os.path.abspath(filename)

        self.continue_counts = continue_counts

        # if another instance of this class is created, get the actual counts
        if self.level2count_dict is None:
            self.level2count_dict = self.load_counts_from_file()

    def emit(self, record):
        """
        Counts a record.
        In case, create add the level to the dict.
        If the time has come, update the json file.
        """
        level = record.levelname
        if level not in self.level2count_dict:
            self.level2count_dict[level] = 0
        self.level2count_dict[level] += 1

        self.flush()

    def flush(self):
        """
        Flushes the dictionary.
        """
        self.acquire()
        try:
            json.dump(self.level2count_dict, open(self.baseFilename, 'w'))
        finally:
            self.release()

    def load_counts_from_file(self):
        """
        Load the dictionary from a json file or create an empty dictionary
        """
        if os.path.exists(self.baseFilename) and self.continue_counts:
            try:
                level2count_dict = json.load(open(self.baseFilename))
            except Exception as a:
                logging.warning(f'Failed to load counts with: {a}')
                level2count_dict = {}
        else:
            level2count_dict = {}
        return level2count_dict