From 4600f11a0772d826c89ff67643ce04ca3c5b2c38 Mon Sep 17 00:00:00 2001 From: Vinay Sajip Date: Sun, 13 Mar 2005 09:56:36 +0000 Subject: Added optional encoding argument to file handlers. --- Lib/logging/handlers.py | 36 +++++++++++++++++++++++++----------- 1 file changed, 25 insertions(+), 11 deletions(-) diff --git a/Lib/logging/handlers.py b/Lib/logging/handlers.py index 672422b..0801699 100644 --- a/Lib/logging/handlers.py +++ b/Lib/logging/handlers.py @@ -1,4 +1,4 @@ -# Copyright 2001-2004 by Vinay Sajip. All Rights Reserved. +# Copyright 2001-2005 by Vinay Sajip. All Rights Reserved. # # Permission to use, copy, modify, and distribute this software and its # documentation for any purpose and without fee is hereby granted, @@ -29,6 +29,11 @@ To use, simply 'import logging' and log away! import sys, logging, socket, types, os, string, cPickle, struct, time, glob +try: + import codecs +except ImportError: + codecs = None + # # Some constants... # @@ -45,11 +50,15 @@ class BaseRotatingHandler(logging.FileHandler): Not meant to be instantiated directly. Instead, use RotatingFileHandler or TimedRotatingFileHandler. """ - def __init__(self, filename, mode): + def __init__(self, filename, mode, encoding=None): """ Use the specified filename for streamed logging """ - logging.FileHandler.__init__(self, filename, mode) + if codecs is None: + encoding = None + logging.FileHandler.__init__(self, filename, mode, encoding) + self.mode = mode + self.encoding = encoding def emit(self, record): """ @@ -70,7 +79,7 @@ class RotatingFileHandler(BaseRotatingHandler): Handler for logging to a set of files, which switches from one file to the next when the current file reaches a certain size. """ - def __init__(self, filename, mode="a", maxBytes=0, backupCount=0): + def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None): """ Open the specified file and use it as the stream for logging. @@ -91,10 +100,9 @@ class RotatingFileHandler(BaseRotatingHandler): If maxBytes is zero, rollover never occurs. """ - self.mode = mode if maxBytes > 0: - self.mode = "a" # doesn't make sense otherwise! - BaseRotatingHandler.__init__(self, filename, self.mode) + mode = 'a' # doesn't make sense otherwise! + BaseRotatingHandler.__init__(self, filename, mode, encoding) self.maxBytes = maxBytes self.backupCount = backupCount @@ -118,7 +126,10 @@ class RotatingFileHandler(BaseRotatingHandler): os.remove(dfn) os.rename(self.baseFilename, dfn) #print "%s -> %s" % (self.baseFilename, dfn) - self.stream = open(self.baseFilename, "w") + if self.encoding: + self.stream = codecs.open(self.baseFilename, 'w', self.encoding) + else: + self.stream = open(self.baseFilename, 'w') def shouldRollover(self, record): """ @@ -142,8 +153,8 @@ class TimedRotatingFileHandler(BaseRotatingHandler): If backupCount is > 0, when rollover is done, no more than backupCount files are kept - the oldest ones are deleted. """ - def __init__(self, filename, when='h', interval=1, backupCount=0): - BaseRotatingHandler.__init__(self, filename, 'a') + def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None): + BaseRotatingHandler.__init__(self, filename, 'a', encoding) self.when = string.upper(when) self.backupCount = backupCount # Calculate the real rollover interval, which is just the number of @@ -262,7 +273,10 @@ class TimedRotatingFileHandler(BaseRotatingHandler): s.sort() os.remove(s[0]) #print "%s -> %s" % (self.baseFilename, dfn) - self.stream = open(self.baseFilename, "w") + if self.encoding: + self.stream = codecs.open(self.baseFilename, 'w', self.encoding) + else: + self.stream = open(self.baseFilename, 'w') self.rolloverAt = int(time.time()) + self.interval class SocketHandler(logging.Handler): -- cgit v0.12