blob: 9cb0e09664633902eac24df3c14cbf3745f63152 [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
Benjamin Petersonad9d48d2008-04-02 21:49:44 +000022Copyright (C) 2001-2008 Vinay Sajip. All Rights Reserved.
Guido van Rossum57102f82002-11-13 16:15:58 +000023
24To use, simply 'import logging' and log away!
25"""
26
Benjamin Petersonad9d48d2008-04-02 21:49:44 +000027import logging, socket, os, pickle, struct, time, re
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +000028from stat import ST_DEV, ST_INO
Guido van Rossum57102f82002-11-13 16:15:58 +000029
Vinay Sajip4600f112005-03-13 09:56:36 +000030try:
31 import codecs
32except ImportError:
33 codecs = None
34
Guido van Rossum57102f82002-11-13 16:15:58 +000035#
36# Some constants...
37#
38
39DEFAULT_TCP_LOGGING_PORT = 9020
40DEFAULT_UDP_LOGGING_PORT = 9021
41DEFAULT_HTTP_LOGGING_PORT = 9022
42DEFAULT_SOAP_LOGGING_PORT = 9023
43SYSLOG_UDP_PORT = 514
44
Thomas Wouters477c8d52006-05-27 19:21:47 +000045_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
46
Vinay Sajip17c52d82004-07-03 11:48:34 +000047class BaseRotatingHandler(logging.FileHandler):
48 """
49 Base class for handlers that rotate log files at a certain point.
50 Not meant to be instantiated directly. Instead, use RotatingFileHandler
51 or TimedRotatingFileHandler.
52 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000053 def __init__(self, filename, mode, encoding=None, delay=0):
Vinay Sajip17c52d82004-07-03 11:48:34 +000054 """
55 Use the specified filename for streamed logging
56 """
Vinay Sajip4600f112005-03-13 09:56:36 +000057 if codecs is None:
58 encoding = None
Christian Heimese7a15bb2008-01-24 16:21:45 +000059 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip4600f112005-03-13 09:56:36 +000060 self.mode = mode
61 self.encoding = encoding
Guido van Rossum57102f82002-11-13 16:15:58 +000062
Vinay Sajip17c52d82004-07-03 11:48:34 +000063 def emit(self, record):
64 """
65 Emit a record.
66
67 Output the record to the file, catering for rollover as described
68 in doRollover().
69 """
Vinay Sajip3970c112004-07-08 10:24:04 +000070 try:
71 if self.shouldRollover(record):
72 self.doRollover()
73 logging.FileHandler.emit(self, record)
Vinay Sajip85c19092005-10-31 13:14:19 +000074 except (KeyboardInterrupt, SystemExit):
75 raise
Vinay Sajip3970c112004-07-08 10:24:04 +000076 except:
77 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000078
79class RotatingFileHandler(BaseRotatingHandler):
80 """
81 Handler for logging to a set of files, which switches from one file
82 to the next when the current file reaches a certain size.
83 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000084 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=0):
Guido van Rossum57102f82002-11-13 16:15:58 +000085 """
86 Open the specified file and use it as the stream for logging.
87
88 By default, the file grows indefinitely. You can specify particular
89 values of maxBytes and backupCount to allow the file to rollover at
90 a predetermined size.
91
92 Rollover occurs whenever the current log file is nearly maxBytes in
93 length. If backupCount is >= 1, the system will successively create
94 new files with the same pathname as the base file, but with extensions
95 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
96 and a base file name of "app.log", you would get "app.log",
97 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
98 written to is always "app.log" - when it gets filled up, it is closed
99 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
100 exist, then they are renamed to "app.log.2", "app.log.3" etc.
101 respectively.
102
103 If maxBytes is zero, rollover never occurs.
104 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000105 if maxBytes > 0:
Vinay Sajip4600f112005-03-13 09:56:36 +0000106 mode = 'a' # doesn't make sense otherwise!
Christian Heimese7a15bb2008-01-24 16:21:45 +0000107 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000108 self.maxBytes = maxBytes
109 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000110
111 def doRollover(self):
112 """
113 Do a rollover, as described in __init__().
114 """
115
116 self.stream.close()
117 if self.backupCount > 0:
118 for i in range(self.backupCount - 1, 0, -1):
119 sfn = "%s.%d" % (self.baseFilename, i)
120 dfn = "%s.%d" % (self.baseFilename, i + 1)
121 if os.path.exists(sfn):
122 #print "%s -> %s" % (sfn, dfn)
123 if os.path.exists(dfn):
124 os.remove(dfn)
125 os.rename(sfn, dfn)
126 dfn = self.baseFilename + ".1"
127 if os.path.exists(dfn):
128 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000129 os.rename(self.baseFilename, dfn)
Guido van Rossum57102f82002-11-13 16:15:58 +0000130 #print "%s -> %s" % (self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000131 self.mode = 'w'
132 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000133
Vinay Sajip17c52d82004-07-03 11:48:34 +0000134 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000135 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000136 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000137
Vinay Sajip17c52d82004-07-03 11:48:34 +0000138 Basically, see if the supplied record would cause the file to exceed
139 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000140 """
141 if self.maxBytes > 0: # are we rolling over?
142 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000143 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000144 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000145 return 1
146 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000147
Vinay Sajip17c52d82004-07-03 11:48:34 +0000148class TimedRotatingFileHandler(BaseRotatingHandler):
149 """
150 Handler for logging to a file, rotating the log file at certain timed
151 intervals.
152
153 If backupCount is > 0, when rollover is done, no more than backupCount
154 files are kept - the oldest ones are deleted.
155 """
Christian Heimese7a15bb2008-01-24 16:21:45 +0000156 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=0):
157 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000158 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000159 self.backupCount = backupCount
160 # Calculate the real rollover interval, which is just the number of
161 # seconds between rollovers. Also set the filename suffix used when
162 # a rollover occurs. Current 'when' events supported:
163 # S - Seconds
164 # M - Minutes
165 # H - Hours
166 # D - Days
167 # midnight - roll over at midnight
168 # W{0-6} - roll over on a certain day; 0 - Monday
169 #
170 # Case of the 'when' specifier is not important; lower or upper case
171 # will work.
172 currentTime = int(time.time())
173 if self.when == 'S':
174 self.interval = 1 # one second
175 self.suffix = "%Y-%m-%d_%H-%M-%S"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000176 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000177 elif self.when == 'M':
178 self.interval = 60 # one minute
179 self.suffix = "%Y-%m-%d_%H-%M"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000180 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000181 elif self.when == 'H':
182 self.interval = 60 * 60 # one hour
183 self.suffix = "%Y-%m-%d_%H"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000184 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000185 elif self.when == 'D' or self.when == 'MIDNIGHT':
186 self.interval = 60 * 60 * 24 # one day
187 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000188 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000189 elif self.when.startswith('W'):
190 self.interval = 60 * 60 * 24 * 7 # one week
191 if len(self.when) != 2:
192 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
193 if self.when[1] < '0' or self.when[1] > '6':
194 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
195 self.dayOfWeek = int(self.when[1])
196 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000197 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000198 else:
199 raise ValueError("Invalid rollover interval specified: %s" % self.when)
200
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000201 self.extMatch = re.compile(self.extMatch)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000202 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip17c52d82004-07-03 11:48:34 +0000203 self.rolloverAt = currentTime + self.interval
204
205 # If we are rolling over at midnight or weekly, then the interval is already known.
206 # What we need to figure out is WHEN the next interval is. In other words,
207 # if you are rolling over at midnight, then your base interval is 1 day,
208 # but you want to start that one day clock at midnight, not now. So, we
209 # have to fudge the rolloverAt value in order to trigger the first rollover
210 # at the right time. After that, the regular interval will take care of
211 # the rest. Note that this code doesn't care about leap seconds. :)
212 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
213 # This could be done with less code, but I wanted it to be clear
214 t = time.localtime(currentTime)
215 currentHour = t[3]
216 currentMinute = t[4]
217 currentSecond = t[5]
218 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000219 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
220 currentSecond)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000221 self.rolloverAt = currentTime + r
222 # If we are rolling over on a certain day, add in the number of days until
223 # the next rollover, but offset by 1 since we just calculated the time
224 # until the next day starts. There are three cases:
225 # Case 1) The day to rollover is today; in this case, do nothing
226 # Case 2) The day to rollover is further in the interval (i.e., today is
227 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
228 # next rollover is simply 6 - 2 - 1, or 3.
229 # Case 3) The day to rollover is behind us in the interval (i.e., today
230 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
231 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
232 # number of days left in the current week (1) plus the number
233 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000234 # The calculations described in 2) and 3) above need to have a day added.
235 # This is because the above time calculation takes us to midnight on this
236 # day, i.e. the start of the next day.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000237 if when.startswith('W'):
238 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000239 if day != self.dayOfWeek:
240 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000241 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000242 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000243 daysToWait = 6 - day + self.dayOfWeek + 1
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000244 newRolloverAt = self.rolloverAt + (daysToWait * (60 * 60 * 24))
245 dstNow = t[-1]
246 dstAtRollover = time.localtime(newRolloverAt)[-1]
247 if dstNow != dstAtRollover:
248 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
249 newRolloverAt = newRolloverAt - 3600
250 else: # DST bows out before next rollover, so we need to add an hour
251 newRolloverAt = newRolloverAt + 3600
252 self.rolloverAt = newRolloverAt
Vinay Sajip17c52d82004-07-03 11:48:34 +0000253
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000254 #print "Will rollover at %d, %d seconds from now" % (self.rolloverAt, self.rolloverAt - currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000255
256 def shouldRollover(self, record):
257 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000258 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000259
260 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000261 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000262 """
263 t = int(time.time())
264 if t >= self.rolloverAt:
265 return 1
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000266 #print "No need to rollover: %d, %d" % (t, self.rolloverAt)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000267 return 0
268
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000269 def getFilesToDelete(self):
270 """
271 Determine the files to delete when rolling over.
272
273 More specific than the earlier method, which just used glob.glob().
274 """
275 dirName, baseName = os.path.split(self.baseFilename)
276 fileNames = os.listdir(dirName)
277 result = []
278 prefix = baseName + "."
279 plen = len(prefix)
280 for fileName in fileNames:
281 if fileName[:plen] == prefix:
282 suffix = fileName[plen:]
283 if self.extMatch.match(suffix):
284 result.append(fileName)
285 result.sort()
286 if len(result) < self.backupCount:
287 result = []
288 else:
289 result = result[:len(result) - self.backupCount]
290 return result
291
Vinay Sajip17c52d82004-07-03 11:48:34 +0000292 def doRollover(self):
293 """
294 do a rollover; in this case, a date/time stamp is appended to the filename
295 when the rollover happens. However, you want the file to be named for the
296 start of the interval, not the current time. If there is a backup count,
297 then we have to get a list of matching filenames, sort them and remove
298 the one with the oldest suffix.
299 """
300 self.stream.close()
301 # get the time that this sequence started at and make it a TimeTuple
302 t = self.rolloverAt - self.interval
303 timeTuple = time.localtime(t)
304 dfn = self.baseFilename + "." + time.strftime(self.suffix, timeTuple)
305 if os.path.exists(dfn):
306 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000307 os.rename(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000308 if self.backupCount > 0:
309 # find the oldest log file and delete it
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000310 #s = glob.glob(self.baseFilename + ".20*")
311 #if len(s) > self.backupCount:
312 # s.sort()
313 # os.remove(s[0])
314 for s in self.getFilesToDelete():
315 os.remove(s)
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000316 #print "%s -> %s" % (self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000317 self.mode = 'w'
318 self.stream = self._open()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000319 newRolloverAt = self.rolloverAt + self.interval
320 currentTime = int(time.time())
321 while newRolloverAt <= currentTime:
322 newRolloverAt = newRolloverAt + self.interval
323 #If DST changes and midnight or weekly rollover, adjust for this.
324 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
325 dstNow = time.localtime(currentTime)[-1]
326 dstAtRollover = time.localtime(newRolloverAt)[-1]
327 if dstNow != dstAtRollover:
328 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
329 newRolloverAt = newRolloverAt - 3600
330 else: # DST bows out before next rollover, so we need to add an hour
331 newRolloverAt = newRolloverAt + 3600
332 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000333
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000334class WatchedFileHandler(logging.FileHandler):
335 """
336 A handler for logging to a file, which watches the file
337 to see if it has changed while in use. This can happen because of
338 usage of programs such as newsyslog and logrotate which perform
339 log file rotation. This handler, intended for use under Unix,
340 watches the file to see if it has changed since the last emit.
341 (A file has changed if its device or inode have changed.)
342 If it has changed, the old file stream is closed, and the file
343 opened to get a new stream.
344
345 This handler is not appropriate for use under Windows, because
346 under Windows open files cannot be moved or renamed - logging
347 opens the files with exclusive locks - and so there is no need
348 for such a handler. Furthermore, ST_INO is not supported under
349 Windows; stat always returns zero for this value.
350
351 This handler is based on a suggestion and patch by Chad J.
352 Schroeder.
353 """
Christian Heimese7a15bb2008-01-24 16:21:45 +0000354 def __init__(self, filename, mode='a', encoding=None, delay=0):
355 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
356 if not os.path.exists(self.baseFilename):
357 self.dev, self.ino = -1, -1
358 else:
359 stat = os.stat(self.baseFilename)
360 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000361
362 def emit(self, record):
363 """
364 Emit a record.
365
366 First check if the underlying file has changed, and if it
367 has, close the old stream and reopen the file to get the
368 current stream.
369 """
370 if not os.path.exists(self.baseFilename):
371 stat = None
372 changed = 1
373 else:
374 stat = os.stat(self.baseFilename)
375 changed = (stat[ST_DEV] != self.dev) or (stat[ST_INO] != self.ino)
Christian Heimese7a15bb2008-01-24 16:21:45 +0000376 if changed and self.stream is not None:
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000377 self.stream.flush()
378 self.stream.close()
379 self.stream = self._open()
380 if stat is None:
381 stat = os.stat(self.baseFilename)
382 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
383 logging.FileHandler.emit(self, record)
384
Guido van Rossum57102f82002-11-13 16:15:58 +0000385class SocketHandler(logging.Handler):
386 """
387 A handler class which writes logging records, in pickle format, to
388 a streaming socket. The socket is kept open across logging calls.
389 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000390 The pickle which is sent is that of the LogRecord's attribute dictionary
391 (__dict__), so that the receiver does not need to have the logging module
392 installed in order to process the logging event.
393
394 To unpickle the record at the receiving end into a LogRecord, use the
395 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000396 """
397
398 def __init__(self, host, port):
399 """
400 Initializes the handler with a specific host address and port.
401
402 The attribute 'closeOnError' is set to 1 - which means that if
403 a socket error occurs, the socket is silently closed and then
404 reopened on the next logging call.
405 """
406 logging.Handler.__init__(self)
407 self.host = host
408 self.port = port
409 self.sock = None
410 self.closeOnError = 0
Vinay Sajip48cfe382004-02-20 13:17:27 +0000411 self.retryTime = None
412 #
413 # Exponential backoff parameters.
414 #
415 self.retryStart = 1.0
416 self.retryMax = 30.0
417 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000418
Guido van Rossumd8faa362007-04-27 19:54:29 +0000419 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000420 """
421 A factory method which allows subclasses to define the precise
422 type of socket they want.
423 """
424 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000425 if hasattr(s, 'settimeout'):
426 s.settimeout(timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000427 s.connect((self.host, self.port))
428 return s
429
Vinay Sajip48cfe382004-02-20 13:17:27 +0000430 def createSocket(self):
431 """
432 Try to create a socket, using an exponential backoff with
433 a max retry time. Thanks to Robert Olson for the original patch
434 (SF #815911) which has been slightly refactored.
435 """
436 now = time.time()
437 # Either retryTime is None, in which case this
438 # is the first time back after a disconnect, or
439 # we've waited long enough.
440 if self.retryTime is None:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000441 attempt = 1
Vinay Sajip48cfe382004-02-20 13:17:27 +0000442 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000443 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000444 if attempt:
445 try:
446 self.sock = self.makeSocket()
447 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000448 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000449 #Creation failed, so set the retry time and return.
450 if self.retryTime is None:
451 self.retryPeriod = self.retryStart
452 else:
453 self.retryPeriod = self.retryPeriod * self.retryFactor
454 if self.retryPeriod > self.retryMax:
455 self.retryPeriod = self.retryMax
456 self.retryTime = now + self.retryPeriod
457
Guido van Rossum57102f82002-11-13 16:15:58 +0000458 def send(self, s):
459 """
460 Send a pickled string to the socket.
461
462 This function allows for partial sends which can happen when the
463 network is busy.
464 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000465 if self.sock is None:
466 self.createSocket()
467 #self.sock can be None either because we haven't reached the retry
468 #time yet, or because we have reached the retry time and retried,
469 #but are still unable to connect.
470 if self.sock:
471 try:
472 if hasattr(self.sock, "sendall"):
473 self.sock.sendall(s)
474 else:
475 sentsofar = 0
476 left = len(s)
477 while left > 0:
478 sent = self.sock.send(s[sentsofar:])
479 sentsofar = sentsofar + sent
480 left = left - sent
481 except socket.error:
482 self.sock.close()
483 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000484
485 def makePickle(self, record):
486 """
487 Pickles the record in binary format with a length prefix, and
488 returns it ready for transmission across the socket.
489 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000490 ei = record.exc_info
491 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000492 dummy = self.format(record) # just to get traceback text into record.exc_text
493 record.exc_info = None # to avoid Unpickleable error
Guido van Rossumba205d62006-08-17 08:57:26 +0000494 s = pickle.dumps(record.__dict__, 1)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000495 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000496 record.exc_info = ei # for next handler
Guido van Rossum57102f82002-11-13 16:15:58 +0000497 slen = struct.pack(">L", len(s))
498 return slen + s
499
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000500 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000501 """
502 Handle an error during logging.
503
504 An error has occurred during logging. Most likely cause -
505 connection lost. Close the socket so that we can retry on the
506 next event.
507 """
508 if self.closeOnError and self.sock:
509 self.sock.close()
510 self.sock = None #try to reconnect next time
511 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000512 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000513
514 def emit(self, record):
515 """
516 Emit a record.
517
518 Pickles the record and writes it to the socket in binary format.
519 If there is an error with the socket, silently drop the packet.
520 If there was a problem with the socket, re-establishes the
521 socket.
522 """
523 try:
524 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000525 self.send(s)
Vinay Sajip85c19092005-10-31 13:14:19 +0000526 except (KeyboardInterrupt, SystemExit):
527 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000528 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000529 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000530
531 def close(self):
532 """
533 Closes the socket.
534 """
535 if self.sock:
536 self.sock.close()
537 self.sock = None
Vinay Sajip48cfe382004-02-20 13:17:27 +0000538 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000539
540class DatagramHandler(SocketHandler):
541 """
542 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000543 a datagram socket. The pickle which is sent is that of the LogRecord's
544 attribute dictionary (__dict__), so that the receiver does not need to
545 have the logging module installed in order to process the logging event.
546
547 To unpickle the record at the receiving end into a LogRecord, use the
548 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000549
550 """
551 def __init__(self, host, port):
552 """
553 Initializes the handler with a specific host address and port.
554 """
555 SocketHandler.__init__(self, host, port)
556 self.closeOnError = 0
557
558 def makeSocket(self):
559 """
560 The factory method of SocketHandler is here overridden to create
561 a UDP socket (SOCK_DGRAM).
562 """
563 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
564 return s
565
566 def send(self, s):
567 """
568 Send a pickled string to a socket.
569
570 This function no longer allows for partial sends which can happen
571 when the network is busy - UDP does not guarantee delivery and
572 can deliver packets out of sequence.
573 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000574 if self.sock is None:
575 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000576 self.sock.sendto(s, (self.host, self.port))
577
578class SysLogHandler(logging.Handler):
579 """
580 A handler class which sends formatted logging records to a syslog
581 server. Based on Sam Rushing's syslog module:
582 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
583 Contributed by Nicolas Untz (after which minor refactoring changes
584 have been made).
585 """
586
587 # from <linux/sys/syslog.h>:
588 # ======================================================================
589 # priorities/facilities are encoded into a single 32-bit quantity, where
590 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
591 # facility (0-big number). Both the priorities and the facilities map
592 # roughly one-to-one to strings in the syslogd(8) source code. This
593 # mapping is included in this file.
594 #
595 # priorities (these are ordered)
596
597 LOG_EMERG = 0 # system is unusable
598 LOG_ALERT = 1 # action must be taken immediately
599 LOG_CRIT = 2 # critical conditions
600 LOG_ERR = 3 # error conditions
601 LOG_WARNING = 4 # warning conditions
602 LOG_NOTICE = 5 # normal but significant condition
603 LOG_INFO = 6 # informational
604 LOG_DEBUG = 7 # debug-level messages
605
606 # facility codes
607 LOG_KERN = 0 # kernel messages
608 LOG_USER = 1 # random user-level messages
609 LOG_MAIL = 2 # mail system
610 LOG_DAEMON = 3 # system daemons
611 LOG_AUTH = 4 # security/authorization messages
612 LOG_SYSLOG = 5 # messages generated internally by syslogd
613 LOG_LPR = 6 # line printer subsystem
614 LOG_NEWS = 7 # network news subsystem
615 LOG_UUCP = 8 # UUCP subsystem
616 LOG_CRON = 9 # clock daemon
617 LOG_AUTHPRIV = 10 # security/authorization messages (private)
618
619 # other codes through 15 reserved for system use
620 LOG_LOCAL0 = 16 # reserved for local use
621 LOG_LOCAL1 = 17 # reserved for local use
622 LOG_LOCAL2 = 18 # reserved for local use
623 LOG_LOCAL3 = 19 # reserved for local use
624 LOG_LOCAL4 = 20 # reserved for local use
625 LOG_LOCAL5 = 21 # reserved for local use
626 LOG_LOCAL6 = 22 # reserved for local use
627 LOG_LOCAL7 = 23 # reserved for local use
628
629 priority_names = {
630 "alert": LOG_ALERT,
631 "crit": LOG_CRIT,
632 "critical": LOG_CRIT,
633 "debug": LOG_DEBUG,
634 "emerg": LOG_EMERG,
635 "err": LOG_ERR,
636 "error": LOG_ERR, # DEPRECATED
637 "info": LOG_INFO,
638 "notice": LOG_NOTICE,
639 "panic": LOG_EMERG, # DEPRECATED
640 "warn": LOG_WARNING, # DEPRECATED
641 "warning": LOG_WARNING,
642 }
643
644 facility_names = {
645 "auth": LOG_AUTH,
646 "authpriv": LOG_AUTHPRIV,
647 "cron": LOG_CRON,
648 "daemon": LOG_DAEMON,
649 "kern": LOG_KERN,
650 "lpr": LOG_LPR,
651 "mail": LOG_MAIL,
652 "news": LOG_NEWS,
653 "security": LOG_AUTH, # DEPRECATED
654 "syslog": LOG_SYSLOG,
655 "user": LOG_USER,
656 "uucp": LOG_UUCP,
657 "local0": LOG_LOCAL0,
658 "local1": LOG_LOCAL1,
659 "local2": LOG_LOCAL2,
660 "local3": LOG_LOCAL3,
661 "local4": LOG_LOCAL4,
662 "local5": LOG_LOCAL5,
663 "local6": LOG_LOCAL6,
664 "local7": LOG_LOCAL7,
665 }
666
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000667 #The map below appears to be trivially lowercasing the key. However,
668 #there's more to it than meets the eye - in some locales, lowercasing
669 #gives unexpected results. See SF #1524081: in the Turkish locale,
670 #"INFO".lower() != "info"
671 priority_map = {
672 "DEBUG" : "debug",
673 "INFO" : "info",
674 "WARNING" : "warning",
675 "ERROR" : "error",
676 "CRITICAL" : "critical"
677 }
678
Guido van Rossum57102f82002-11-13 16:15:58 +0000679 def __init__(self, address=('localhost', SYSLOG_UDP_PORT), facility=LOG_USER):
680 """
681 Initialize a handler.
682
Guido van Rossume7ba4952007-06-06 23:52:48 +0000683 If address is specified as a string, a UNIX socket is used. To log to a
684 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000685 If facility is not specified, LOG_USER is used.
686 """
687 logging.Handler.__init__(self)
688
689 self.address = address
690 self.facility = facility
Guido van Rossum13257902007-06-07 23:15:56 +0000691 if isinstance(address, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000692 self.unixsocket = 1
Thomas Wouters89f507f2006-12-13 04:49:30 +0000693 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000694 else:
Guido van Rossum57102f82002-11-13 16:15:58 +0000695 self.unixsocket = 0
Thomas Wouters89f507f2006-12-13 04:49:30 +0000696 self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000697
698 self.formatter = None
699
Vinay Sajipa1974c12005-01-13 08:23:56 +0000700 def _connect_unixsocket(self, address):
701 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
702 # syslog may require either DGRAM or STREAM sockets
703 try:
704 self.socket.connect(address)
705 except socket.error:
706 self.socket.close()
707 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Vinay Sajip8b6b53f2005-11-09 13:55:13 +0000708 self.socket.connect(address)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000709
Guido van Rossum57102f82002-11-13 16:15:58 +0000710 # curious: when talking to the unix-domain '/dev/log' socket, a
711 # zero-terminator seems to be required. this string is placed
712 # into a class variable so that it can be overridden if
713 # necessary.
714 log_format_string = '<%d>%s\000'
715
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000716 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000717 """
718 Encode the facility and priority. You can pass in strings or
719 integers - if strings are passed, the facility_names and
720 priority_names mapping dictionaries are used to convert them to
721 integers.
722 """
Guido van Rossum13257902007-06-07 23:15:56 +0000723 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000724 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000725 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000726 priority = self.priority_names[priority]
727 return (facility << 3) | priority
728
729 def close (self):
730 """
731 Closes the socket.
732 """
733 if self.unixsocket:
734 self.socket.close()
Vinay Sajip48cfe382004-02-20 13:17:27 +0000735 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000736
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000737 def mapPriority(self, levelName):
738 """
739 Map a logging level name to a key in the priority_names map.
740 This is useful in two scenarios: when custom levels are being
741 used, and in the case where you can't do a straightforward
742 mapping by lowercasing the logging level name because of locale-
743 specific issues (see SF #1524081).
744 """
745 return self.priority_map.get(levelName, "warning")
746
Guido van Rossum57102f82002-11-13 16:15:58 +0000747 def emit(self, record):
748 """
749 Emit a record.
750
751 The record is formatted, and then sent to the syslog server. If
752 exception information is present, it is NOT sent to the server.
753 """
754 msg = self.format(record)
755 """
756 We need to convert record level to lowercase, maybe this will
757 change in the future.
758 """
759 msg = self.log_format_string % (
760 self.encodePriority(self.facility,
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000761 self.mapPriority(record.levelname)),
762 msg)
Guido van Rossum57102f82002-11-13 16:15:58 +0000763 try:
764 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000765 try:
766 self.socket.send(msg)
767 except socket.error:
768 self._connect_unixsocket(self.address)
769 self.socket.send(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +0000770 else:
771 self.socket.sendto(msg, self.address)
Vinay Sajip85c19092005-10-31 13:14:19 +0000772 except (KeyboardInterrupt, SystemExit):
773 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000774 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000775 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000776
777class SMTPHandler(logging.Handler):
778 """
779 A handler class which sends an SMTP email for each logging event.
780 """
Guido van Rossum360e4b82007-05-14 22:51:27 +0000781 def __init__(self, mailhost, fromaddr, toaddrs, subject, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000782 """
783 Initialize the handler.
784
785 Initialize the instance with the from and to addresses and subject
786 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000787 (host, port) tuple format for the mailhost argument. To specify
788 authentication credentials, supply a (username, password) tuple
789 for the credentials argument.
Guido van Rossum57102f82002-11-13 16:15:58 +0000790 """
791 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000792 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000793 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000794 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000795 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000796 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000797 self.username, self.password = credentials
798 else:
799 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000800 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000801 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000802 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000803 self.toaddrs = toaddrs
804 self.subject = subject
805
806 def getSubject(self, record):
807 """
808 Determine the subject for the email.
809
810 If you want to specify a subject line which is record-dependent,
811 override this method.
812 """
813 return self.subject
814
Vinay Sajipe7d40662004-10-03 19:12:07 +0000815 weekdayname = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
816
817 monthname = [None,
818 'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun',
819 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
820
821 def date_time(self):
822 """
823 Return the current date and time formatted for a MIME header.
824 Needed for Python 1.5.2 (no email package available)
825 """
826 year, month, day, hh, mm, ss, wd, y, z = time.gmtime(time.time())
827 s = "%s, %02d %3s %4d %02d:%02d:%02d GMT" % (
828 self.weekdayname[wd],
829 day, self.monthname[month], year,
830 hh, mm, ss)
831 return s
832
Guido van Rossum57102f82002-11-13 16:15:58 +0000833 def emit(self, record):
834 """
835 Emit a record.
836
837 Format the record and send it to the specified addressees.
838 """
839 try:
840 import smtplib
Vinay Sajipe7d40662004-10-03 19:12:07 +0000841 try:
Thomas Woutersb2137042007-02-01 18:02:27 +0000842 from email.utils import formatdate
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000843 except ImportError:
Vinay Sajipe7d40662004-10-03 19:12:07 +0000844 formatdate = self.date_time
Guido van Rossum57102f82002-11-13 16:15:58 +0000845 port = self.mailport
846 if not port:
847 port = smtplib.SMTP_PORT
848 smtp = smtplib.SMTP(self.mailhost, port)
849 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000850 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 +0000851 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000852 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000853 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000854 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000855 if self.username:
856 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000857 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
858 smtp.quit()
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000859 except (KeyboardInterrupt, SystemExit):
860 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000861 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000862 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000863
864class NTEventLogHandler(logging.Handler):
865 """
866 A handler class which sends events to the NT Event Log. Adds a
867 registry entry for the specified application name. If no dllname is
868 provided, win32service.pyd (which contains some basic message
869 placeholders) is used. Note that use of these placeholders will make
870 your event logs big, as the entire message source is held in the log.
871 If you want slimmer logs, you have to pass in the name of your own DLL
872 which contains the message definitions you want to use in the event log.
873 """
874 def __init__(self, appname, dllname=None, logtype="Application"):
875 logging.Handler.__init__(self)
876 try:
877 import win32evtlogutil, win32evtlog
878 self.appname = appname
879 self._welu = win32evtlogutil
880 if not dllname:
881 dllname = os.path.split(self._welu.__file__)
882 dllname = os.path.split(dllname[0])
883 dllname = os.path.join(dllname[0], r'win32service.pyd')
884 self.dllname = dllname
885 self.logtype = logtype
886 self._welu.AddSourceToRegistry(appname, dllname, logtype)
887 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
888 self.typemap = {
889 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
890 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000891 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000892 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
893 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
894 }
895 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000896 print("The Python Win32 extensions for NT (service, event "\
897 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000898 self._welu = None
899
900 def getMessageID(self, record):
901 """
902 Return the message ID for the event record. If you are using your
903 own messages, you could do this by having the msg passed to the
904 logger being an ID rather than a formatting string. Then, in here,
905 you could use a dictionary lookup to get the message ID. This
906 version returns 1, which is the base message ID in win32service.pyd.
907 """
908 return 1
909
910 def getEventCategory(self, record):
911 """
912 Return the event category for the record.
913
914 Override this if you want to specify your own categories. This version
915 returns 0.
916 """
917 return 0
918
919 def getEventType(self, record):
920 """
921 Return the event type for the record.
922
923 Override this if you want to specify your own types. This version does
924 a mapping using the handler's typemap attribute, which is set up in
925 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000926 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000927 either need to override this method or place a suitable dictionary in
928 the handler's typemap attribute.
929 """
930 return self.typemap.get(record.levelno, self.deftype)
931
932 def emit(self, record):
933 """
934 Emit a record.
935
936 Determine the message ID, event category and event type. Then
937 log the message in the NT event log.
938 """
939 if self._welu:
940 try:
941 id = self.getMessageID(record)
942 cat = self.getEventCategory(record)
943 type = self.getEventType(record)
944 msg = self.format(record)
945 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000946 except (KeyboardInterrupt, SystemExit):
947 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000948 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000949 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000950
951 def close(self):
952 """
953 Clean up this handler.
954
955 You can remove the application name from the registry as a
956 source of event log entries. However, if you do this, you will
957 not be able to see the events as you intended in the Event Log
958 Viewer - it needs to be able to access the registry to get the
959 DLL name.
960 """
961 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000962 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000963
964class HTTPHandler(logging.Handler):
965 """
966 A class which sends records to a Web server, using either GET or
967 POST semantics.
968 """
969 def __init__(self, host, url, method="GET"):
970 """
971 Initialize the instance with the host, the request URL, and the method
972 ("GET" or "POST")
973 """
974 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000975 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +0000976 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +0000977 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +0000978 self.host = host
979 self.url = url
980 self.method = method
981
Neal Norwitzf297bd12003-04-23 03:49:43 +0000982 def mapLogRecord(self, record):
983 """
984 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +0000985 that is sent as the CGI data. Overwrite in your class.
Neal Norwitzf297bd12003-04-23 03:49:43 +0000986 Contributed by Franz Glasner.
987 """
988 return record.__dict__
989
Guido van Rossum57102f82002-11-13 16:15:58 +0000990 def emit(self, record):
991 """
992 Emit a record.
993
994 Send the record to the Web server as an URL-encoded dictionary
995 """
996 try:
997 import httplib, urllib
Vinay Sajipb7935062005-10-11 13:15:31 +0000998 host = self.host
999 h = httplib.HTTP(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001000 url = self.url
Neal Norwitzf297bd12003-04-23 03:49:43 +00001001 data = urllib.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001002 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001003 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001004 sep = '&'
1005 else:
1006 sep = '?'
1007 url = url + "%c%s" % (sep, data)
1008 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001009 # support multiple hosts on one IP address...
1010 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001011 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001012 if i >= 0:
1013 host = host[:i]
1014 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001015 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001016 h.putheader("Content-type",
1017 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001018 h.putheader("Content-length", str(len(data)))
1019 h.endheaders()
1020 if self.method == "POST":
1021 h.send(data)
1022 h.getreply() #can't do anything with the result
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001023 except (KeyboardInterrupt, SystemExit):
1024 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001025 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001026 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001027
1028class BufferingHandler(logging.Handler):
1029 """
1030 A handler class which buffers logging records in memory. Whenever each
1031 record is added to the buffer, a check is made to see if the buffer should
1032 be flushed. If it should, then flush() is expected to do what's needed.
1033 """
1034 def __init__(self, capacity):
1035 """
1036 Initialize the handler with the buffer size.
1037 """
1038 logging.Handler.__init__(self)
1039 self.capacity = capacity
1040 self.buffer = []
1041
1042 def shouldFlush(self, record):
1043 """
1044 Should the handler flush its buffer?
1045
1046 Returns true if the buffer is up to capacity. This method can be
1047 overridden to implement custom flushing strategies.
1048 """
1049 return (len(self.buffer) >= self.capacity)
1050
1051 def emit(self, record):
1052 """
1053 Emit a record.
1054
1055 Append the record. If shouldFlush() tells us to, call flush() to process
1056 the buffer.
1057 """
1058 self.buffer.append(record)
1059 if self.shouldFlush(record):
1060 self.flush()
1061
1062 def flush(self):
1063 """
1064 Override to implement custom flushing behaviour.
1065
1066 This version just zaps the buffer to empty.
1067 """
1068 self.buffer = []
1069
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001070 def close(self):
1071 """
1072 Close the handler.
1073
1074 This version just flushes and chains to the parent class' close().
1075 """
1076 self.flush()
1077 logging.Handler.close(self)
1078
Guido van Rossum57102f82002-11-13 16:15:58 +00001079class MemoryHandler(BufferingHandler):
1080 """
1081 A handler class which buffers logging records in memory, periodically
1082 flushing them to a target handler. Flushing occurs whenever the buffer
1083 is full, or when an event of a certain severity or greater is seen.
1084 """
1085 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1086 """
1087 Initialize the handler with the buffer size, the level at which
1088 flushing should occur and an optional target.
1089
1090 Note that without a target being set either here or via setTarget(),
1091 a MemoryHandler is no use to anyone!
1092 """
1093 BufferingHandler.__init__(self, capacity)
1094 self.flushLevel = flushLevel
1095 self.target = target
1096
1097 def shouldFlush(self, record):
1098 """
1099 Check for buffer full or a record at the flushLevel or higher.
1100 """
1101 return (len(self.buffer) >= self.capacity) or \
1102 (record.levelno >= self.flushLevel)
1103
1104 def setTarget(self, target):
1105 """
1106 Set the target handler for this handler.
1107 """
1108 self.target = target
1109
1110 def flush(self):
1111 """
1112 For a MemoryHandler, flushing means just sending the buffered
1113 records to the target, if there is one. Override if you want
1114 different behaviour.
1115 """
1116 if self.target:
1117 for record in self.buffer:
1118 self.target.handle(record)
1119 self.buffer = []
1120
1121 def close(self):
1122 """
1123 Flush, set the target to None and lose the buffer.
1124 """
1125 self.flush()
1126 self.target = None
Vinay Sajip48cfe382004-02-20 13:17:27 +00001127 BufferingHandler.close(self)