blob: b53262e99634ee282071e19f8763447e287bed6a [file] [log] [blame]
Thomas Woutersb2137042007-02-01 18:02:27 +00001# Copyright 2001-2007 by Vinay Sajip. All Rights Reserved.
Guido van Rossum57102f82002-11-13 16:15:58 +00002#
3# Permission to use, copy, modify, and distribute this software and its
4# documentation for any purpose and without fee is hereby granted,
5# provided that the above copyright notice appear in all copies and that
6# both that copyright notice and this permission notice appear in
7# supporting documentation, and that the name of Vinay Sajip
8# not be used in advertising or publicity pertaining to distribution
9# of the software without specific, written prior permission.
10# VINAY SAJIP DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING
11# ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
12# VINAY SAJIP BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR
13# ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER
14# IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
15# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
Guido van Rossum57102f82002-11-13 16:15:58 +000016
17"""
Vinay Sajip3f742842004-02-28 16:07:46 +000018Additional handlers for the logging package for Python. The core package is
19based on PEP 282 and comments thereto in comp.lang.python, and influenced by
20Apache's log4j system.
Guido van Rossum57102f82002-11-13 16:15:58 +000021
Thomas Woutersb2137042007-02-01 18:02:27 +000022Copyright (C) 2001-2007 Vinay Sajip. All Rights Reserved.
Guido van Rossum57102f82002-11-13 16:15:58 +000023
24To use, simply 'import logging' and log away!
25"""
26
Guido van Rossum13257902007-06-07 23:15:56 +000027import sys, logging, socket, os, struct, time, glob
Guido van Rossum99603b02007-07-20 00:22:32 +000028import pickle
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +000029from stat import ST_DEV, ST_INO
Guido van Rossum57102f82002-11-13 16:15:58 +000030
Vinay Sajip4600f112005-03-13 09:56:36 +000031try:
32 import codecs
33except ImportError:
34 codecs = None
35
Guido van Rossum57102f82002-11-13 16:15:58 +000036#
37# Some constants...
38#
39
40DEFAULT_TCP_LOGGING_PORT = 9020
41DEFAULT_UDP_LOGGING_PORT = 9021
42DEFAULT_HTTP_LOGGING_PORT = 9022
43DEFAULT_SOAP_LOGGING_PORT = 9023
44SYSLOG_UDP_PORT = 514
45
Thomas Wouters477c8d52006-05-27 19:21:47 +000046_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
47
Vinay Sajip17c52d82004-07-03 11:48:34 +000048class BaseRotatingHandler(logging.FileHandler):
49 """
50 Base class for handlers that rotate log files at a certain point.
51 Not meant to be instantiated directly. Instead, use RotatingFileHandler
52 or TimedRotatingFileHandler.
53 """
Vinay Sajip4600f112005-03-13 09:56:36 +000054 def __init__(self, filename, mode, encoding=None):
Vinay Sajip17c52d82004-07-03 11:48:34 +000055 """
56 Use the specified filename for streamed logging
57 """
Vinay Sajip4600f112005-03-13 09:56:36 +000058 if codecs is None:
59 encoding = None
60 logging.FileHandler.__init__(self, filename, mode, encoding)
61 self.mode = mode
62 self.encoding = encoding
Guido van Rossum57102f82002-11-13 16:15:58 +000063
Vinay Sajip17c52d82004-07-03 11:48:34 +000064 def emit(self, record):
65 """
66 Emit a record.
67
68 Output the record to the file, catering for rollover as described
69 in doRollover().
70 """
Vinay Sajip3970c112004-07-08 10:24:04 +000071 try:
72 if self.shouldRollover(record):
73 self.doRollover()
74 logging.FileHandler.emit(self, record)
Vinay Sajip85c19092005-10-31 13:14:19 +000075 except (KeyboardInterrupt, SystemExit):
76 raise
Vinay Sajip3970c112004-07-08 10:24:04 +000077 except:
78 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000079
80class RotatingFileHandler(BaseRotatingHandler):
81 """
82 Handler for logging to a set of files, which switches from one file
83 to the next when the current file reaches a certain size.
84 """
Vinay Sajip4600f112005-03-13 09:56:36 +000085 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None):
Guido van Rossum57102f82002-11-13 16:15:58 +000086 """
87 Open the specified file and use it as the stream for logging.
88
89 By default, the file grows indefinitely. You can specify particular
90 values of maxBytes and backupCount to allow the file to rollover at
91 a predetermined size.
92
93 Rollover occurs whenever the current log file is nearly maxBytes in
94 length. If backupCount is >= 1, the system will successively create
95 new files with the same pathname as the base file, but with extensions
96 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
97 and a base file name of "app.log", you would get "app.log",
98 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
99 written to is always "app.log" - when it gets filled up, it is closed
100 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
101 exist, then they are renamed to "app.log.2", "app.log.3" etc.
102 respectively.
103
104 If maxBytes is zero, rollover never occurs.
105 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000106 if maxBytes > 0:
Vinay Sajip4600f112005-03-13 09:56:36 +0000107 mode = 'a' # doesn't make sense otherwise!
108 BaseRotatingHandler.__init__(self, filename, mode, encoding)
Guido van Rossum57102f82002-11-13 16:15:58 +0000109 self.maxBytes = maxBytes
110 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000111
112 def doRollover(self):
113 """
114 Do a rollover, as described in __init__().
115 """
116
117 self.stream.close()
118 if self.backupCount > 0:
119 for i in range(self.backupCount - 1, 0, -1):
120 sfn = "%s.%d" % (self.baseFilename, i)
121 dfn = "%s.%d" % (self.baseFilename, i + 1)
122 if os.path.exists(sfn):
123 #print "%s -> %s" % (sfn, dfn)
124 if os.path.exists(dfn):
125 os.remove(dfn)
126 os.rename(sfn, dfn)
127 dfn = self.baseFilename + ".1"
128 if os.path.exists(dfn):
129 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000130 os.rename(self.baseFilename, dfn)
Guido van Rossum57102f82002-11-13 16:15:58 +0000131 #print "%s -> %s" % (self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000132 self.mode = 'w'
133 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000134
Vinay Sajip17c52d82004-07-03 11:48:34 +0000135 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000136 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000137 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000138
Vinay Sajip17c52d82004-07-03 11:48:34 +0000139 Basically, see if the supplied record would cause the file to exceed
140 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000141 """
142 if self.maxBytes > 0: # are we rolling over?
143 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000144 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000145 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000146 return 1
147 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000148
Vinay Sajip17c52d82004-07-03 11:48:34 +0000149class TimedRotatingFileHandler(BaseRotatingHandler):
150 """
151 Handler for logging to a file, rotating the log file at certain timed
152 intervals.
153
154 If backupCount is > 0, when rollover is done, no more than backupCount
155 files are kept - the oldest ones are deleted.
156 """
Vinay Sajip4600f112005-03-13 09:56:36 +0000157 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None):
158 BaseRotatingHandler.__init__(self, filename, 'a', encoding)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000159 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000160 self.backupCount = backupCount
161 # Calculate the real rollover interval, which is just the number of
162 # seconds between rollovers. Also set the filename suffix used when
163 # a rollover occurs. Current 'when' events supported:
164 # S - Seconds
165 # M - Minutes
166 # H - Hours
167 # D - Days
168 # midnight - roll over at midnight
169 # W{0-6} - roll over on a certain day; 0 - Monday
170 #
171 # Case of the 'when' specifier is not important; lower or upper case
172 # will work.
173 currentTime = int(time.time())
174 if self.when == 'S':
175 self.interval = 1 # one second
176 self.suffix = "%Y-%m-%d_%H-%M-%S"
177 elif self.when == 'M':
178 self.interval = 60 # one minute
179 self.suffix = "%Y-%m-%d_%H-%M"
180 elif self.when == 'H':
181 self.interval = 60 * 60 # one hour
182 self.suffix = "%Y-%m-%d_%H"
183 elif self.when == 'D' or self.when == 'MIDNIGHT':
184 self.interval = 60 * 60 * 24 # one day
185 self.suffix = "%Y-%m-%d"
186 elif self.when.startswith('W'):
187 self.interval = 60 * 60 * 24 * 7 # one week
188 if len(self.when) != 2:
189 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
190 if self.when[1] < '0' or self.when[1] > '6':
191 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
192 self.dayOfWeek = int(self.when[1])
193 self.suffix = "%Y-%m-%d"
194 else:
195 raise ValueError("Invalid rollover interval specified: %s" % self.when)
196
Vinay Sajipe7d40662004-10-03 19:12:07 +0000197 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip17c52d82004-07-03 11:48:34 +0000198 self.rolloverAt = currentTime + self.interval
199
200 # If we are rolling over at midnight or weekly, then the interval is already known.
201 # What we need to figure out is WHEN the next interval is. In other words,
202 # if you are rolling over at midnight, then your base interval is 1 day,
203 # but you want to start that one day clock at midnight, not now. So, we
204 # have to fudge the rolloverAt value in order to trigger the first rollover
205 # at the right time. After that, the regular interval will take care of
206 # the rest. Note that this code doesn't care about leap seconds. :)
207 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
208 # This could be done with less code, but I wanted it to be clear
209 t = time.localtime(currentTime)
210 currentHour = t[3]
211 currentMinute = t[4]
212 currentSecond = t[5]
213 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000214 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
215 currentSecond)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000216 self.rolloverAt = currentTime + r
217 # If we are rolling over on a certain day, add in the number of days until
218 # the next rollover, but offset by 1 since we just calculated the time
219 # until the next day starts. There are three cases:
220 # Case 1) The day to rollover is today; in this case, do nothing
221 # Case 2) The day to rollover is further in the interval (i.e., today is
222 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
223 # next rollover is simply 6 - 2 - 1, or 3.
224 # Case 3) The day to rollover is behind us in the interval (i.e., today
225 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
226 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
227 # number of days left in the current week (1) plus the number
228 # of days in the next week until the rollover day (3).
229 if when.startswith('W'):
230 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000231 if day != self.dayOfWeek:
232 if day < self.dayOfWeek:
233 daysToWait = self.dayOfWeek - day - 1
234 else:
235 daysToWait = 6 - day + self.dayOfWeek
Vinay Sajipe7d40662004-10-03 19:12:07 +0000236 self.rolloverAt = self.rolloverAt + (daysToWait * (60 * 60 * 24))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000237
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000238 #print "Will rollover at %d, %d seconds from now" % (self.rolloverAt, self.rolloverAt - currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000239
240 def shouldRollover(self, record):
241 """
242 Determine if rollover should occur
243
244 record is not used, as we are just comparing times, but it is needed so
245 the method siguratures are the same
246 """
247 t = int(time.time())
248 if t >= self.rolloverAt:
249 return 1
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000250 #print "No need to rollover: %d, %d" % (t, self.rolloverAt)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000251 return 0
252
253 def doRollover(self):
254 """
255 do a rollover; in this case, a date/time stamp is appended to the filename
256 when the rollover happens. However, you want the file to be named for the
257 start of the interval, not the current time. If there is a backup count,
258 then we have to get a list of matching filenames, sort them and remove
259 the one with the oldest suffix.
260 """
261 self.stream.close()
262 # get the time that this sequence started at and make it a TimeTuple
263 t = self.rolloverAt - self.interval
264 timeTuple = time.localtime(t)
265 dfn = self.baseFilename + "." + time.strftime(self.suffix, timeTuple)
266 if os.path.exists(dfn):
267 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000268 os.rename(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000269 if self.backupCount > 0:
270 # find the oldest log file and delete it
271 s = glob.glob(self.baseFilename + ".20*")
272 if len(s) > self.backupCount:
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000273 s.sort()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000274 os.remove(s[0])
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000275 #print "%s -> %s" % (self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000276 self.mode = 'w'
277 self.stream = self._open()
Vinay Sajipd9520412006-01-16 09:13:58 +0000278 self.rolloverAt = self.rolloverAt + self.interval
Guido van Rossum57102f82002-11-13 16:15:58 +0000279
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000280class WatchedFileHandler(logging.FileHandler):
281 """
282 A handler for logging to a file, which watches the file
283 to see if it has changed while in use. This can happen because of
284 usage of programs such as newsyslog and logrotate which perform
285 log file rotation. This handler, intended for use under Unix,
286 watches the file to see if it has changed since the last emit.
287 (A file has changed if its device or inode have changed.)
288 If it has changed, the old file stream is closed, and the file
289 opened to get a new stream.
290
291 This handler is not appropriate for use under Windows, because
292 under Windows open files cannot be moved or renamed - logging
293 opens the files with exclusive locks - and so there is no need
294 for such a handler. Furthermore, ST_INO is not supported under
295 Windows; stat always returns zero for this value.
296
297 This handler is based on a suggestion and patch by Chad J.
298 Schroeder.
299 """
300 def __init__(self, filename, mode='a', encoding=None):
301 logging.FileHandler.__init__(self, filename, mode, encoding)
302 stat = os.stat(self.baseFilename)
303 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
304
305 def emit(self, record):
306 """
307 Emit a record.
308
309 First check if the underlying file has changed, and if it
310 has, close the old stream and reopen the file to get the
311 current stream.
312 """
313 if not os.path.exists(self.baseFilename):
314 stat = None
315 changed = 1
316 else:
317 stat = os.stat(self.baseFilename)
318 changed = (stat[ST_DEV] != self.dev) or (stat[ST_INO] != self.ino)
319 if changed:
320 self.stream.flush()
321 self.stream.close()
322 self.stream = self._open()
323 if stat is None:
324 stat = os.stat(self.baseFilename)
325 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
326 logging.FileHandler.emit(self, record)
327
Guido van Rossum57102f82002-11-13 16:15:58 +0000328class SocketHandler(logging.Handler):
329 """
330 A handler class which writes logging records, in pickle format, to
331 a streaming socket. The socket is kept open across logging calls.
332 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000333 The pickle which is sent is that of the LogRecord's attribute dictionary
334 (__dict__), so that the receiver does not need to have the logging module
335 installed in order to process the logging event.
336
337 To unpickle the record at the receiving end into a LogRecord, use the
338 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000339 """
340
341 def __init__(self, host, port):
342 """
343 Initializes the handler with a specific host address and port.
344
345 The attribute 'closeOnError' is set to 1 - which means that if
346 a socket error occurs, the socket is silently closed and then
347 reopened on the next logging call.
348 """
349 logging.Handler.__init__(self)
350 self.host = host
351 self.port = port
352 self.sock = None
353 self.closeOnError = 0
Vinay Sajip48cfe382004-02-20 13:17:27 +0000354 self.retryTime = None
355 #
356 # Exponential backoff parameters.
357 #
358 self.retryStart = 1.0
359 self.retryMax = 30.0
360 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000361
Guido van Rossumd8faa362007-04-27 19:54:29 +0000362 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000363 """
364 A factory method which allows subclasses to define the precise
365 type of socket they want.
366 """
367 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000368 if hasattr(s, 'settimeout'):
369 s.settimeout(timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000370 s.connect((self.host, self.port))
371 return s
372
Vinay Sajip48cfe382004-02-20 13:17:27 +0000373 def createSocket(self):
374 """
375 Try to create a socket, using an exponential backoff with
376 a max retry time. Thanks to Robert Olson for the original patch
377 (SF #815911) which has been slightly refactored.
378 """
379 now = time.time()
380 # Either retryTime is None, in which case this
381 # is the first time back after a disconnect, or
382 # we've waited long enough.
383 if self.retryTime is None:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000384 attempt = 1
Vinay Sajip48cfe382004-02-20 13:17:27 +0000385 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000386 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000387 if attempt:
388 try:
389 self.sock = self.makeSocket()
390 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000391 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000392 #Creation failed, so set the retry time and return.
393 if self.retryTime is None:
394 self.retryPeriod = self.retryStart
395 else:
396 self.retryPeriod = self.retryPeriod * self.retryFactor
397 if self.retryPeriod > self.retryMax:
398 self.retryPeriod = self.retryMax
399 self.retryTime = now + self.retryPeriod
400
Guido van Rossum57102f82002-11-13 16:15:58 +0000401 def send(self, s):
402 """
403 Send a pickled string to the socket.
404
405 This function allows for partial sends which can happen when the
406 network is busy.
407 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000408 if self.sock is None:
409 self.createSocket()
410 #self.sock can be None either because we haven't reached the retry
411 #time yet, or because we have reached the retry time and retried,
412 #but are still unable to connect.
413 if self.sock:
414 try:
415 if hasattr(self.sock, "sendall"):
416 self.sock.sendall(s)
417 else:
418 sentsofar = 0
419 left = len(s)
420 while left > 0:
421 sent = self.sock.send(s[sentsofar:])
422 sentsofar = sentsofar + sent
423 left = left - sent
424 except socket.error:
425 self.sock.close()
426 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000427
428 def makePickle(self, record):
429 """
430 Pickles the record in binary format with a length prefix, and
431 returns it ready for transmission across the socket.
432 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000433 ei = record.exc_info
434 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000435 dummy = self.format(record) # just to get traceback text into record.exc_text
436 record.exc_info = None # to avoid Unpickleable error
Guido van Rossumba205d62006-08-17 08:57:26 +0000437 s = pickle.dumps(record.__dict__, 1)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000438 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000439 record.exc_info = ei # for next handler
Guido van Rossum57102f82002-11-13 16:15:58 +0000440 slen = struct.pack(">L", len(s))
441 return slen + s
442
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000443 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000444 """
445 Handle an error during logging.
446
447 An error has occurred during logging. Most likely cause -
448 connection lost. Close the socket so that we can retry on the
449 next event.
450 """
451 if self.closeOnError and self.sock:
452 self.sock.close()
453 self.sock = None #try to reconnect next time
454 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000455 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000456
457 def emit(self, record):
458 """
459 Emit a record.
460
461 Pickles the record and writes it to the socket in binary format.
462 If there is an error with the socket, silently drop the packet.
463 If there was a problem with the socket, re-establishes the
464 socket.
465 """
466 try:
467 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000468 self.send(s)
Vinay Sajip85c19092005-10-31 13:14:19 +0000469 except (KeyboardInterrupt, SystemExit):
470 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000471 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000472 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000473
474 def close(self):
475 """
476 Closes the socket.
477 """
478 if self.sock:
479 self.sock.close()
480 self.sock = None
Vinay Sajip48cfe382004-02-20 13:17:27 +0000481 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000482
483class DatagramHandler(SocketHandler):
484 """
485 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000486 a datagram socket. The pickle which is sent is that of the LogRecord's
487 attribute dictionary (__dict__), so that the receiver does not need to
488 have the logging module installed in order to process the logging event.
489
490 To unpickle the record at the receiving end into a LogRecord, use the
491 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000492
493 """
494 def __init__(self, host, port):
495 """
496 Initializes the handler with a specific host address and port.
497 """
498 SocketHandler.__init__(self, host, port)
499 self.closeOnError = 0
500
501 def makeSocket(self):
502 """
503 The factory method of SocketHandler is here overridden to create
504 a UDP socket (SOCK_DGRAM).
505 """
506 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
507 return s
508
509 def send(self, s):
510 """
511 Send a pickled string to a socket.
512
513 This function no longer allows for partial sends which can happen
514 when the network is busy - UDP does not guarantee delivery and
515 can deliver packets out of sequence.
516 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000517 if self.sock is None:
518 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000519 self.sock.sendto(s, (self.host, self.port))
520
521class SysLogHandler(logging.Handler):
522 """
523 A handler class which sends formatted logging records to a syslog
524 server. Based on Sam Rushing's syslog module:
525 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
526 Contributed by Nicolas Untz (after which minor refactoring changes
527 have been made).
528 """
529
530 # from <linux/sys/syslog.h>:
531 # ======================================================================
532 # priorities/facilities are encoded into a single 32-bit quantity, where
533 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
534 # facility (0-big number). Both the priorities and the facilities map
535 # roughly one-to-one to strings in the syslogd(8) source code. This
536 # mapping is included in this file.
537 #
538 # priorities (these are ordered)
539
540 LOG_EMERG = 0 # system is unusable
541 LOG_ALERT = 1 # action must be taken immediately
542 LOG_CRIT = 2 # critical conditions
543 LOG_ERR = 3 # error conditions
544 LOG_WARNING = 4 # warning conditions
545 LOG_NOTICE = 5 # normal but significant condition
546 LOG_INFO = 6 # informational
547 LOG_DEBUG = 7 # debug-level messages
548
549 # facility codes
550 LOG_KERN = 0 # kernel messages
551 LOG_USER = 1 # random user-level messages
552 LOG_MAIL = 2 # mail system
553 LOG_DAEMON = 3 # system daemons
554 LOG_AUTH = 4 # security/authorization messages
555 LOG_SYSLOG = 5 # messages generated internally by syslogd
556 LOG_LPR = 6 # line printer subsystem
557 LOG_NEWS = 7 # network news subsystem
558 LOG_UUCP = 8 # UUCP subsystem
559 LOG_CRON = 9 # clock daemon
560 LOG_AUTHPRIV = 10 # security/authorization messages (private)
561
562 # other codes through 15 reserved for system use
563 LOG_LOCAL0 = 16 # reserved for local use
564 LOG_LOCAL1 = 17 # reserved for local use
565 LOG_LOCAL2 = 18 # reserved for local use
566 LOG_LOCAL3 = 19 # reserved for local use
567 LOG_LOCAL4 = 20 # reserved for local use
568 LOG_LOCAL5 = 21 # reserved for local use
569 LOG_LOCAL6 = 22 # reserved for local use
570 LOG_LOCAL7 = 23 # reserved for local use
571
572 priority_names = {
573 "alert": LOG_ALERT,
574 "crit": LOG_CRIT,
575 "critical": LOG_CRIT,
576 "debug": LOG_DEBUG,
577 "emerg": LOG_EMERG,
578 "err": LOG_ERR,
579 "error": LOG_ERR, # DEPRECATED
580 "info": LOG_INFO,
581 "notice": LOG_NOTICE,
582 "panic": LOG_EMERG, # DEPRECATED
583 "warn": LOG_WARNING, # DEPRECATED
584 "warning": LOG_WARNING,
585 }
586
587 facility_names = {
588 "auth": LOG_AUTH,
589 "authpriv": LOG_AUTHPRIV,
590 "cron": LOG_CRON,
591 "daemon": LOG_DAEMON,
592 "kern": LOG_KERN,
593 "lpr": LOG_LPR,
594 "mail": LOG_MAIL,
595 "news": LOG_NEWS,
596 "security": LOG_AUTH, # DEPRECATED
597 "syslog": LOG_SYSLOG,
598 "user": LOG_USER,
599 "uucp": LOG_UUCP,
600 "local0": LOG_LOCAL0,
601 "local1": LOG_LOCAL1,
602 "local2": LOG_LOCAL2,
603 "local3": LOG_LOCAL3,
604 "local4": LOG_LOCAL4,
605 "local5": LOG_LOCAL5,
606 "local6": LOG_LOCAL6,
607 "local7": LOG_LOCAL7,
608 }
609
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000610 #The map below appears to be trivially lowercasing the key. However,
611 #there's more to it than meets the eye - in some locales, lowercasing
612 #gives unexpected results. See SF #1524081: in the Turkish locale,
613 #"INFO".lower() != "info"
614 priority_map = {
615 "DEBUG" : "debug",
616 "INFO" : "info",
617 "WARNING" : "warning",
618 "ERROR" : "error",
619 "CRITICAL" : "critical"
620 }
621
Guido van Rossum57102f82002-11-13 16:15:58 +0000622 def __init__(self, address=('localhost', SYSLOG_UDP_PORT), facility=LOG_USER):
623 """
624 Initialize a handler.
625
Guido van Rossume7ba4952007-06-06 23:52:48 +0000626 If address is specified as a string, a UNIX socket is used. To log to a
627 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000628 If facility is not specified, LOG_USER is used.
629 """
630 logging.Handler.__init__(self)
631
632 self.address = address
633 self.facility = facility
Guido van Rossum13257902007-06-07 23:15:56 +0000634 if isinstance(address, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000635 self.unixsocket = 1
Thomas Wouters89f507f2006-12-13 04:49:30 +0000636 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000637 else:
Guido van Rossum57102f82002-11-13 16:15:58 +0000638 self.unixsocket = 0
Thomas Wouters89f507f2006-12-13 04:49:30 +0000639 self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000640
641 self.formatter = None
642
Vinay Sajipa1974c12005-01-13 08:23:56 +0000643 def _connect_unixsocket(self, address):
644 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
645 # syslog may require either DGRAM or STREAM sockets
646 try:
647 self.socket.connect(address)
648 except socket.error:
649 self.socket.close()
650 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Vinay Sajip8b6b53f2005-11-09 13:55:13 +0000651 self.socket.connect(address)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000652
Guido van Rossum57102f82002-11-13 16:15:58 +0000653 # curious: when talking to the unix-domain '/dev/log' socket, a
654 # zero-terminator seems to be required. this string is placed
655 # into a class variable so that it can be overridden if
656 # necessary.
657 log_format_string = '<%d>%s\000'
658
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000659 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000660 """
661 Encode the facility and priority. You can pass in strings or
662 integers - if strings are passed, the facility_names and
663 priority_names mapping dictionaries are used to convert them to
664 integers.
665 """
Guido van Rossum13257902007-06-07 23:15:56 +0000666 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000667 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000668 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000669 priority = self.priority_names[priority]
670 return (facility << 3) | priority
671
672 def close (self):
673 """
674 Closes the socket.
675 """
676 if self.unixsocket:
677 self.socket.close()
Vinay Sajip48cfe382004-02-20 13:17:27 +0000678 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000679
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000680 def mapPriority(self, levelName):
681 """
682 Map a logging level name to a key in the priority_names map.
683 This is useful in two scenarios: when custom levels are being
684 used, and in the case where you can't do a straightforward
685 mapping by lowercasing the logging level name because of locale-
686 specific issues (see SF #1524081).
687 """
688 return self.priority_map.get(levelName, "warning")
689
Guido van Rossum57102f82002-11-13 16:15:58 +0000690 def emit(self, record):
691 """
692 Emit a record.
693
694 The record is formatted, and then sent to the syslog server. If
695 exception information is present, it is NOT sent to the server.
696 """
697 msg = self.format(record)
698 """
699 We need to convert record level to lowercase, maybe this will
700 change in the future.
701 """
702 msg = self.log_format_string % (
703 self.encodePriority(self.facility,
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000704 self.mapPriority(record.levelname)),
705 msg)
Guido van Rossum57102f82002-11-13 16:15:58 +0000706 try:
707 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000708 try:
709 self.socket.send(msg)
710 except socket.error:
711 self._connect_unixsocket(self.address)
712 self.socket.send(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +0000713 else:
714 self.socket.sendto(msg, self.address)
Vinay Sajip85c19092005-10-31 13:14:19 +0000715 except (KeyboardInterrupt, SystemExit):
716 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000717 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000718 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000719
720class SMTPHandler(logging.Handler):
721 """
722 A handler class which sends an SMTP email for each logging event.
723 """
Guido van Rossum360e4b82007-05-14 22:51:27 +0000724 def __init__(self, mailhost, fromaddr, toaddrs, subject, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000725 """
726 Initialize the handler.
727
728 Initialize the instance with the from and to addresses and subject
729 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000730 (host, port) tuple format for the mailhost argument. To specify
731 authentication credentials, supply a (username, password) tuple
732 for the credentials argument.
Guido van Rossum57102f82002-11-13 16:15:58 +0000733 """
734 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000735 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000736 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000737 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000738 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000739 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000740 self.username, self.password = credentials
741 else:
742 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000743 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000744 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000745 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000746 self.toaddrs = toaddrs
747 self.subject = subject
748
749 def getSubject(self, record):
750 """
751 Determine the subject for the email.
752
753 If you want to specify a subject line which is record-dependent,
754 override this method.
755 """
756 return self.subject
757
Vinay Sajipe7d40662004-10-03 19:12:07 +0000758 weekdayname = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
759
760 monthname = [None,
761 'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun',
762 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
763
764 def date_time(self):
765 """
766 Return the current date and time formatted for a MIME header.
767 Needed for Python 1.5.2 (no email package available)
768 """
769 year, month, day, hh, mm, ss, wd, y, z = time.gmtime(time.time())
770 s = "%s, %02d %3s %4d %02d:%02d:%02d GMT" % (
771 self.weekdayname[wd],
772 day, self.monthname[month], year,
773 hh, mm, ss)
774 return s
775
Guido van Rossum57102f82002-11-13 16:15:58 +0000776 def emit(self, record):
777 """
778 Emit a record.
779
780 Format the record and send it to the specified addressees.
781 """
782 try:
783 import smtplib
Vinay Sajipe7d40662004-10-03 19:12:07 +0000784 try:
Thomas Woutersb2137042007-02-01 18:02:27 +0000785 from email.utils import formatdate
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000786 except ImportError:
Vinay Sajipe7d40662004-10-03 19:12:07 +0000787 formatdate = self.date_time
Guido van Rossum57102f82002-11-13 16:15:58 +0000788 port = self.mailport
789 if not port:
790 port = smtplib.SMTP_PORT
791 smtp = smtplib.SMTP(self.mailhost, port)
792 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000793 msg = "From: %s\r\nTo: %s\r\nSubject: %s\r\nDate: %s\r\n\r\n%s" % (
Guido van Rossum57102f82002-11-13 16:15:58 +0000794 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000795 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000796 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000797 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000798 if self.username:
799 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000800 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
801 smtp.quit()
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000802 except (KeyboardInterrupt, SystemExit):
803 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000804 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000805 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000806
807class NTEventLogHandler(logging.Handler):
808 """
809 A handler class which sends events to the NT Event Log. Adds a
810 registry entry for the specified application name. If no dllname is
811 provided, win32service.pyd (which contains some basic message
812 placeholders) is used. Note that use of these placeholders will make
813 your event logs big, as the entire message source is held in the log.
814 If you want slimmer logs, you have to pass in the name of your own DLL
815 which contains the message definitions you want to use in the event log.
816 """
817 def __init__(self, appname, dllname=None, logtype="Application"):
818 logging.Handler.__init__(self)
819 try:
820 import win32evtlogutil, win32evtlog
821 self.appname = appname
822 self._welu = win32evtlogutil
823 if not dllname:
824 dllname = os.path.split(self._welu.__file__)
825 dllname = os.path.split(dllname[0])
826 dllname = os.path.join(dllname[0], r'win32service.pyd')
827 self.dllname = dllname
828 self.logtype = logtype
829 self._welu.AddSourceToRegistry(appname, dllname, logtype)
830 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
831 self.typemap = {
832 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
833 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000834 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000835 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
836 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
837 }
838 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000839 print("The Python Win32 extensions for NT (service, event "\
840 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000841 self._welu = None
842
843 def getMessageID(self, record):
844 """
845 Return the message ID for the event record. If you are using your
846 own messages, you could do this by having the msg passed to the
847 logger being an ID rather than a formatting string. Then, in here,
848 you could use a dictionary lookup to get the message ID. This
849 version returns 1, which is the base message ID in win32service.pyd.
850 """
851 return 1
852
853 def getEventCategory(self, record):
854 """
855 Return the event category for the record.
856
857 Override this if you want to specify your own categories. This version
858 returns 0.
859 """
860 return 0
861
862 def getEventType(self, record):
863 """
864 Return the event type for the record.
865
866 Override this if you want to specify your own types. This version does
867 a mapping using the handler's typemap attribute, which is set up in
868 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000869 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000870 either need to override this method or place a suitable dictionary in
871 the handler's typemap attribute.
872 """
873 return self.typemap.get(record.levelno, self.deftype)
874
875 def emit(self, record):
876 """
877 Emit a record.
878
879 Determine the message ID, event category and event type. Then
880 log the message in the NT event log.
881 """
882 if self._welu:
883 try:
884 id = self.getMessageID(record)
885 cat = self.getEventCategory(record)
886 type = self.getEventType(record)
887 msg = self.format(record)
888 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000889 except (KeyboardInterrupt, SystemExit):
890 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000891 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000892 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000893
894 def close(self):
895 """
896 Clean up this handler.
897
898 You can remove the application name from the registry as a
899 source of event log entries. However, if you do this, you will
900 not be able to see the events as you intended in the Event Log
901 Viewer - it needs to be able to access the registry to get the
902 DLL name.
903 """
904 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000905 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000906
907class HTTPHandler(logging.Handler):
908 """
909 A class which sends records to a Web server, using either GET or
910 POST semantics.
911 """
912 def __init__(self, host, url, method="GET"):
913 """
914 Initialize the instance with the host, the request URL, and the method
915 ("GET" or "POST")
916 """
917 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000918 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +0000919 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +0000920 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +0000921 self.host = host
922 self.url = url
923 self.method = method
924
Neal Norwitzf297bd12003-04-23 03:49:43 +0000925 def mapLogRecord(self, record):
926 """
927 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +0000928 that is sent as the CGI data. Overwrite in your class.
Neal Norwitzf297bd12003-04-23 03:49:43 +0000929 Contributed by Franz Glasner.
930 """
931 return record.__dict__
932
Guido van Rossum57102f82002-11-13 16:15:58 +0000933 def emit(self, record):
934 """
935 Emit a record.
936
937 Send the record to the Web server as an URL-encoded dictionary
938 """
939 try:
940 import httplib, urllib
Vinay Sajipb7935062005-10-11 13:15:31 +0000941 host = self.host
942 h = httplib.HTTP(host)
Guido van Rossum57102f82002-11-13 16:15:58 +0000943 url = self.url
Neal Norwitzf297bd12003-04-23 03:49:43 +0000944 data = urllib.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +0000945 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000946 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000947 sep = '&'
948 else:
949 sep = '?'
950 url = url + "%c%s" % (sep, data)
951 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +0000952 # support multiple hosts on one IP address...
953 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000954 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +0000955 if i >= 0:
956 host = host[:i]
957 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +0000958 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +0000959 h.putheader("Content-type",
960 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +0000961 h.putheader("Content-length", str(len(data)))
962 h.endheaders()
963 if self.method == "POST":
964 h.send(data)
965 h.getreply() #can't do anything with the result
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000966 except (KeyboardInterrupt, SystemExit):
967 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000968 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000969 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000970
971class BufferingHandler(logging.Handler):
972 """
973 A handler class which buffers logging records in memory. Whenever each
974 record is added to the buffer, a check is made to see if the buffer should
975 be flushed. If it should, then flush() is expected to do what's needed.
976 """
977 def __init__(self, capacity):
978 """
979 Initialize the handler with the buffer size.
980 """
981 logging.Handler.__init__(self)
982 self.capacity = capacity
983 self.buffer = []
984
985 def shouldFlush(self, record):
986 """
987 Should the handler flush its buffer?
988
989 Returns true if the buffer is up to capacity. This method can be
990 overridden to implement custom flushing strategies.
991 """
992 return (len(self.buffer) >= self.capacity)
993
994 def emit(self, record):
995 """
996 Emit a record.
997
998 Append the record. If shouldFlush() tells us to, call flush() to process
999 the buffer.
1000 """
1001 self.buffer.append(record)
1002 if self.shouldFlush(record):
1003 self.flush()
1004
1005 def flush(self):
1006 """
1007 Override to implement custom flushing behaviour.
1008
1009 This version just zaps the buffer to empty.
1010 """
1011 self.buffer = []
1012
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001013 def close(self):
1014 """
1015 Close the handler.
1016
1017 This version just flushes and chains to the parent class' close().
1018 """
1019 self.flush()
1020 logging.Handler.close(self)
1021
Guido van Rossum57102f82002-11-13 16:15:58 +00001022class MemoryHandler(BufferingHandler):
1023 """
1024 A handler class which buffers logging records in memory, periodically
1025 flushing them to a target handler. Flushing occurs whenever the buffer
1026 is full, or when an event of a certain severity or greater is seen.
1027 """
1028 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1029 """
1030 Initialize the handler with the buffer size, the level at which
1031 flushing should occur and an optional target.
1032
1033 Note that without a target being set either here or via setTarget(),
1034 a MemoryHandler is no use to anyone!
1035 """
1036 BufferingHandler.__init__(self, capacity)
1037 self.flushLevel = flushLevel
1038 self.target = target
1039
1040 def shouldFlush(self, record):
1041 """
1042 Check for buffer full or a record at the flushLevel or higher.
1043 """
1044 return (len(self.buffer) >= self.capacity) or \
1045 (record.levelno >= self.flushLevel)
1046
1047 def setTarget(self, target):
1048 """
1049 Set the target handler for this handler.
1050 """
1051 self.target = target
1052
1053 def flush(self):
1054 """
1055 For a MemoryHandler, flushing means just sending the buffered
1056 records to the target, if there is one. Override if you want
1057 different behaviour.
1058 """
1059 if self.target:
1060 for record in self.buffer:
1061 self.target.handle(record)
1062 self.buffer = []
1063
1064 def close(self):
1065 """
1066 Flush, set the target to None and lose the buffer.
1067 """
1068 self.flush()
1069 self.target = None
Vinay Sajip48cfe382004-02-20 13:17:27 +00001070 BufferingHandler.close(self)