限制 Python 日志文件

Question

我正在使用 Python3 (Ubuntu 14.04) 写出一个日志文件。我想限制日志文件的大小，所以选择了RotatingFileHandler。此外，我不想轮换日志文件（.log 变为 .log.1 等等） - 我只想限制日志文件的大小。

我试过像这样使用RotatingFileHandler：

filehandler = logging.handlers.RotatingFileHandler( "app.log", "a", 1000, 0, None, True )
logging.basicConfig( format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s", level = logging.DEBUG, handlers = [ filehandler ] )

我发现日志文件继续增长，并且永远不会被截断或“包装”到极限。

我原以为设置backupCount = 0会导致在写入之前检查日志文件大小，如果有空间容纳日志消息，则写入它，否则清空文件并写出消息。

有什么想法吗？

提前致谢，

伯恩迈斯特。

Answer 1

如果您查看source code of the RotatingFileHandler，您会看到backupCount 0 刚刚关闭然后重新打开日志的流。这是一个修改后的版本，可以满足您的要求，丢弃旧文件：

import os
import logging
import logging.handlers


class TruncatedFileHandler(logging.handlers.RotatingFileHandler):
    def __init__(self, filename, mode='a', maxBytes=0, encoding=None, delay=0):
        super(TruncatedFileHandler, self).__init__(
            filename, mode, maxBytes, 0, encoding, delay)

    def doRollover(self):
        """Truncate the file"""
        if self.stream:
            self.stream.close()
        dfn = self.baseFilename + ".1"
        if os.path.exists(dfn):
            os.remove(dfn)
        os.rename(self.baseFilename, dfn)
        os.remove(dfn)
        self.mode = 'w'
        self.stream = self._open()

这里是如何使用它：

filehandler = TruncatedFileHandler("app.log", "w", 1000)
logging.basicConfig(
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
    level=logging.DEBUG, handlers=[filehandler])

logger = logging.getLogger('spam_application')

for i in range(100):
    logger.debug("a log line, number: %s", i)

并证明它有效

❯❯❯ python3.4 main.py
❯❯❯ cat app.log*
2014-06-11 11:50:44,871 - spam_application - DEBUG - a log line, number: 91
2014-06-11 11:50:44,872 - spam_application - DEBUG - a log line, number: 92
2014-06-11 11:50:44,872 - spam_application - DEBUG - a log line, number: 93
2014-06-11 11:50:44,872 - spam_application - DEBUG - a log line, number: 94
2014-06-11 11:50:44,872 - spam_application - DEBUG - a log line, number: 95
2014-06-11 11:50:44,872 - spam_application - DEBUG - a log line, number: 96
2014-06-11 11:50:44,872 - spam_application - DEBUG - a log line, number: 97
2014-06-11 11:50:44,872 - spam_application - DEBUG - a log line, number: 98
2014-06-11 11:50:44,872 - spam_application - DEBUG - a log line, number: 99

Answer 2

受logc的回答启发，我采取了更野蛮的做法：我只是删除了日志文件。

class TruncatedFileHandler( logging.handlers.RotatingFileHandler ):
    def __init__( self, filename, mode = "a", maxBytes = 0, encoding = None, delay = 0 ):
        super( TruncatedFileHandler, self ).__init__( filename, mode, maxBytes, 0, encoding, delay )


    def doRollover( self ):
        if self.stream: self.stream.close()

        if os.path.exists( self.baseFilename ): os.remove( self.baseFilename )

        self.mode = "a" # Not sure why but "w" behaves in the same way as "a".
        self.stream = self._open()

我在我的解决方案和 logc 的解决方案中发现的一点是，一旦调用了 doRollover()，设置 self.mode="a" 或 self.mode="w" 在写入文件方面没有任何区别。也就是说，我希望“w”每次都从文件的开头开始写入，但事实并非如此（它的行为方式与“a”相同，并从它最后写入的位置开始写入）。