Source code for pycopia.logfile

#!/usr/bin/python2.7
# vim:ts=4:sw=4:softtabstop=4:smarttab:expandtab
#
#    Copyright (C) 1999-2006  Keith Dart <keith@kdart.com>
#
#    This library is free software; you can redistribute it and/or
#    modify it under the terms of the GNU Lesser General Public
#    License as published by the Free Software Foundation; either
#    version 2.1 of the License, or (at your option) any later version.
#
#    This library is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
#    Lesser General Public License for more details.

"""
Managing logfile rotation. A ManagedLog object is a file-like object that
rotates itself when a maximum size is reached.

"""

import sys, os

if sys.version_info.major == 3:
    import io
    file = io.FileIO

[docs]class SizeError(IOError): pass
[docs]class LogFile(file): """LogFile(name, [mode="w"], [maxsize=360000]) Opens a new file object. After writing <maxsize> bytes a SizeError will be raised. """ def __init__(self, name, mode="w", maxsize=360000): super(LogFile, self).__init__(name, mode) self.maxsize = maxsize self.written = 0
[docs] def write(self, data): self.written += len(data) super(LogFile, self).write(data) self.flush() if self.written > self.maxsize: raise SizeError
[docs] def rotate(self): return rotate(self)
[docs] def note(self, text): """Writes a specially formated note text to the file.The note starts with the string '\\n#*=' so you can easily filter them. """ self.write("\n#*===== %s =====\n" % (text,))
[docs]class ManagedLog(object): """ManagedLog(name, [maxsize=360000], [maxsave=9]) A ManagedLog instance is a persistent log object. Write data with the write() method. The log size and rotation is handled automatically. """ def __init__(self, name, maxsize=360000, maxsave=9): if os.path.isfile(name): shiftlogs(name, maxsave) self._lf = LogFile(name, "w", maxsize) self.maxsave = maxsave def __repr__(self): return "%s(%r, %r, %r)" % (self.__class__.__name__, self._lf.name, self._lf.maxsize, self.maxsave)
[docs] def write(self, data): try: self._lf.write(data) except SizeError: self._lf = rotate(self._lf, self.maxsave)
[docs] def note(self, data): try: self._lf.note(data) except SizeError: self._lf = rotate(self._lf, self.maxsave)
[docs] def written(self): return self._lf.written
[docs] def rotate(self): self._lf = rotate(self._lf, self.maxsave) # auto-delegate remaining methods (but you should not read or seek an open # log file).
def __getattr__(self, name): return getattr(self._lf, name) # useful for logged stdout for daemon processes
[docs]class ManagedStdio(ManagedLog):
[docs] def write(self, data): try: self._lf.write(data) except SizeError: sys.stdout.flush() sys.stderr.flush() self._lf = rotate(self._lf, self.maxsave) fd = self._lf.fileno() os.dup2(fd, 1) os.dup2(fd, 2) sys.stdout = sys.stderr = self
[docs]def rotate(fileobj, maxsave=9): name = fileobj.name mode = fileobj.mode maxsize = fileobj.maxsize fileobj.close() shiftlogs(name, maxsave) return LogFile(name, mode, maxsize) # assumes basename logfile is closed.
[docs]def shiftlogs(basename, maxsave): topname = "%s.%d" % (basename, maxsave) if os.path.isfile(topname): os.unlink(topname) for i in range(maxsave, 0, -1): oldname = "%s.%d" % (basename, i) newname = "%s.%d" % (basename, i+1) try: os.rename(oldname, newname) except OSError: pass try: os.rename(basename, "%s.1" % (basename)) except OSError: pass
[docs]def open(name, maxsize=360000, maxsave=9): return ManagedLog(name, maxsize, maxsave)
[docs]def writelog(logobj, data): try: logobj.write(data) except SizeError: return rotate(logobj) else: return logobj