blob: 31352ad1c7d476faea706212bcc7b46175eca1bc [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
Vinay Sajip6268cbc2009-01-21 00:19:28 +000022Copyright (C) 2001-2009 Vinay Sajip. All Rights Reserved.
Guido van Rossum57102f82002-11-13 16:15:58 +000023
Vinay Sajip6268cbc2009-01-21 00:19:28 +000024To use, simply 'import logging.handlers' and log away!
Guido van Rossum57102f82002-11-13 16:15:58 +000025"""
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 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000115 if self.stream:
116 self.stream.close()
Guido van Rossum57102f82002-11-13 16:15:58 +0000117 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 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000141 if self.stream is None: # delay was set...
142 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000143 if self.maxBytes > 0: # are we rolling over?
144 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000145 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000146 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000147 return 1
148 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000149
Vinay Sajip17c52d82004-07-03 11:48:34 +0000150class TimedRotatingFileHandler(BaseRotatingHandler):
151 """
152 Handler for logging to a file, rotating the log file at certain timed
153 intervals.
154
155 If backupCount is > 0, when rollover is done, no more than backupCount
156 files are kept - the oldest ones are deleted.
157 """
Georg Brandl96f05e52009-06-08 08:53:39 +0000158 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=0, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000159 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000160 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000161 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000162 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000163 # Calculate the real rollover interval, which is just the number of
164 # seconds between rollovers. Also set the filename suffix used when
165 # a rollover occurs. Current 'when' events supported:
166 # S - Seconds
167 # M - Minutes
168 # H - Hours
169 # D - Days
170 # midnight - roll over at midnight
171 # W{0-6} - roll over on a certain day; 0 - Monday
172 #
173 # Case of the 'when' specifier is not important; lower or upper case
174 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000175 if self.when == 'S':
176 self.interval = 1 # one second
177 self.suffix = "%Y-%m-%d_%H-%M-%S"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000178 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000179 elif self.when == 'M':
180 self.interval = 60 # one minute
181 self.suffix = "%Y-%m-%d_%H-%M"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000182 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000183 elif self.when == 'H':
184 self.interval = 60 * 60 # one hour
185 self.suffix = "%Y-%m-%d_%H"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000186 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000187 elif self.when == 'D' or self.when == 'MIDNIGHT':
188 self.interval = 60 * 60 * 24 # one day
189 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000190 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000191 elif self.when.startswith('W'):
192 self.interval = 60 * 60 * 24 * 7 # one week
193 if len(self.when) != 2:
194 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
195 if self.when[1] < '0' or self.when[1] > '6':
196 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
197 self.dayOfWeek = int(self.when[1])
198 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000199 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000200 else:
201 raise ValueError("Invalid rollover interval specified: %s" % self.when)
202
Antoine Pitroufd036452008-08-19 17:56:33 +0000203 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000204 self.interval = self.interval * interval # multiply by units requested
Vinay Sajipaffbd872009-06-11 10:11:47 +0000205 self.rolloverAt = self.computeRollover(int(time.time()))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000206
Vinay Sajipaffbd872009-06-11 10:11:47 +0000207 def computeRollover(self, currentTime):
208 """
209 Work out the rollover time based on the specified time.
210 """
211 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000212 # If we are rolling over at midnight or weekly, then the interval is already known.
213 # What we need to figure out is WHEN the next interval is. In other words,
214 # if you are rolling over at midnight, then your base interval is 1 day,
215 # but you want to start that one day clock at midnight, not now. So, we
216 # have to fudge the rolloverAt value in order to trigger the first rollover
217 # at the right time. After that, the regular interval will take care of
218 # the rest. Note that this code doesn't care about leap seconds. :)
219 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
220 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000221 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000222 t = time.gmtime(currentTime)
223 else:
224 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000225 currentHour = t[3]
226 currentMinute = t[4]
227 currentSecond = t[5]
228 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000229 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
230 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000231 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000232 # If we are rolling over on a certain day, add in the number of days until
233 # the next rollover, but offset by 1 since we just calculated the time
234 # until the next day starts. There are three cases:
235 # Case 1) The day to rollover is today; in this case, do nothing
236 # Case 2) The day to rollover is further in the interval (i.e., today is
237 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
238 # next rollover is simply 6 - 2 - 1, or 3.
239 # Case 3) The day to rollover is behind us in the interval (i.e., today
240 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
241 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
242 # number of days left in the current week (1) plus the number
243 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000244 # The calculations described in 2) and 3) above need to have a day added.
245 # This is because the above time calculation takes us to midnight on this
246 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000247 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000248 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000249 if day != self.dayOfWeek:
250 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000251 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000252 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000253 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000254 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
255 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000256 dstNow = t[-1]
257 dstAtRollover = time.localtime(newRolloverAt)[-1]
258 if dstNow != dstAtRollover:
259 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
260 newRolloverAt = newRolloverAt - 3600
261 else: # DST bows out before next rollover, so we need to add an hour
262 newRolloverAt = newRolloverAt + 3600
Vinay Sajipaffbd872009-06-11 10:11:47 +0000263 result = newRolloverAt
264 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000265
266 def shouldRollover(self, record):
267 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000268 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000269
270 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000271 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000272 """
273 t = int(time.time())
274 if t >= self.rolloverAt:
275 return 1
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000276 #print "No need to rollover: %d, %d" % (t, self.rolloverAt)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000277 return 0
278
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000279 def getFilesToDelete(self):
280 """
281 Determine the files to delete when rolling over.
282
283 More specific than the earlier method, which just used glob.glob().
284 """
285 dirName, baseName = os.path.split(self.baseFilename)
286 fileNames = os.listdir(dirName)
287 result = []
288 prefix = baseName + "."
289 plen = len(prefix)
290 for fileName in fileNames:
291 if fileName[:plen] == prefix:
292 suffix = fileName[plen:]
293 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000294 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000295 result.sort()
296 if len(result) < self.backupCount:
297 result = []
298 else:
299 result = result[:len(result) - self.backupCount]
300 return result
301
Vinay Sajip17c52d82004-07-03 11:48:34 +0000302 def doRollover(self):
303 """
304 do a rollover; in this case, a date/time stamp is appended to the filename
305 when the rollover happens. However, you want the file to be named for the
306 start of the interval, not the current time. If there is a backup count,
307 then we have to get a list of matching filenames, sort them and remove
308 the one with the oldest suffix.
309 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000310 if self.stream:
311 self.stream.close()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000312 # get the time that this sequence started at and make it a TimeTuple
313 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000314 if self.utc:
315 timeTuple = time.gmtime(t)
316 else:
317 timeTuple = time.localtime(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000318 dfn = self.baseFilename + "." + time.strftime(self.suffix, timeTuple)
319 if os.path.exists(dfn):
320 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000321 os.rename(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000322 if self.backupCount > 0:
323 # find the oldest log file and delete it
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000324 #s = glob.glob(self.baseFilename + ".20*")
325 #if len(s) > self.backupCount:
326 # s.sort()
327 # os.remove(s[0])
328 for s in self.getFilesToDelete():
329 os.remove(s)
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000330 #print "%s -> %s" % (self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000331 self.mode = 'w'
332 self.stream = self._open()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000333 currentTime = int(time.time())
Vinay Sajipaffbd872009-06-11 10:11:47 +0000334 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000335 while newRolloverAt <= currentTime:
336 newRolloverAt = newRolloverAt + self.interval
337 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000338 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000339 dstNow = time.localtime(currentTime)[-1]
340 dstAtRollover = time.localtime(newRolloverAt)[-1]
341 if dstNow != dstAtRollover:
342 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
343 newRolloverAt = newRolloverAt - 3600
344 else: # DST bows out before next rollover, so we need to add an hour
345 newRolloverAt = newRolloverAt + 3600
346 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000347
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000348class WatchedFileHandler(logging.FileHandler):
349 """
350 A handler for logging to a file, which watches the file
351 to see if it has changed while in use. This can happen because of
352 usage of programs such as newsyslog and logrotate which perform
353 log file rotation. This handler, intended for use under Unix,
354 watches the file to see if it has changed since the last emit.
355 (A file has changed if its device or inode have changed.)
356 If it has changed, the old file stream is closed, and the file
357 opened to get a new stream.
358
359 This handler is not appropriate for use under Windows, because
360 under Windows open files cannot be moved or renamed - logging
361 opens the files with exclusive locks - and so there is no need
362 for such a handler. Furthermore, ST_INO is not supported under
363 Windows; stat always returns zero for this value.
364
365 This handler is based on a suggestion and patch by Chad J.
366 Schroeder.
367 """
Christian Heimese7a15bb2008-01-24 16:21:45 +0000368 def __init__(self, filename, mode='a', encoding=None, delay=0):
369 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
370 if not os.path.exists(self.baseFilename):
371 self.dev, self.ino = -1, -1
372 else:
373 stat = os.stat(self.baseFilename)
374 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000375
376 def emit(self, record):
377 """
378 Emit a record.
379
380 First check if the underlying file has changed, and if it
381 has, close the old stream and reopen the file to get the
382 current stream.
383 """
384 if not os.path.exists(self.baseFilename):
385 stat = None
386 changed = 1
387 else:
388 stat = os.stat(self.baseFilename)
389 changed = (stat[ST_DEV] != self.dev) or (stat[ST_INO] != self.ino)
Christian Heimese7a15bb2008-01-24 16:21:45 +0000390 if changed and self.stream is not None:
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000391 self.stream.flush()
392 self.stream.close()
393 self.stream = self._open()
394 if stat is None:
395 stat = os.stat(self.baseFilename)
396 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
397 logging.FileHandler.emit(self, record)
398
Guido van Rossum57102f82002-11-13 16:15:58 +0000399class SocketHandler(logging.Handler):
400 """
401 A handler class which writes logging records, in pickle format, to
402 a streaming socket. The socket is kept open across logging calls.
403 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000404 The pickle which is sent is that of the LogRecord's attribute dictionary
405 (__dict__), so that the receiver does not need to have the logging module
406 installed in order to process the logging event.
407
408 To unpickle the record at the receiving end into a LogRecord, use the
409 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000410 """
411
412 def __init__(self, host, port):
413 """
414 Initializes the handler with a specific host address and port.
415
416 The attribute 'closeOnError' is set to 1 - which means that if
417 a socket error occurs, the socket is silently closed and then
418 reopened on the next logging call.
419 """
420 logging.Handler.__init__(self)
421 self.host = host
422 self.port = port
423 self.sock = None
424 self.closeOnError = 0
Vinay Sajip48cfe382004-02-20 13:17:27 +0000425 self.retryTime = None
426 #
427 # Exponential backoff parameters.
428 #
429 self.retryStart = 1.0
430 self.retryMax = 30.0
431 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000432
Guido van Rossumd8faa362007-04-27 19:54:29 +0000433 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000434 """
435 A factory method which allows subclasses to define the precise
436 type of socket they want.
437 """
438 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000439 if hasattr(s, 'settimeout'):
440 s.settimeout(timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000441 s.connect((self.host, self.port))
442 return s
443
Vinay Sajip48cfe382004-02-20 13:17:27 +0000444 def createSocket(self):
445 """
446 Try to create a socket, using an exponential backoff with
447 a max retry time. Thanks to Robert Olson for the original patch
448 (SF #815911) which has been slightly refactored.
449 """
450 now = time.time()
451 # Either retryTime is None, in which case this
452 # is the first time back after a disconnect, or
453 # we've waited long enough.
454 if self.retryTime is None:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000455 attempt = 1
Vinay Sajip48cfe382004-02-20 13:17:27 +0000456 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000457 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000458 if attempt:
459 try:
460 self.sock = self.makeSocket()
461 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000462 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000463 #Creation failed, so set the retry time and return.
464 if self.retryTime is None:
465 self.retryPeriod = self.retryStart
466 else:
467 self.retryPeriod = self.retryPeriod * self.retryFactor
468 if self.retryPeriod > self.retryMax:
469 self.retryPeriod = self.retryMax
470 self.retryTime = now + self.retryPeriod
471
Guido van Rossum57102f82002-11-13 16:15:58 +0000472 def send(self, s):
473 """
474 Send a pickled string to the socket.
475
476 This function allows for partial sends which can happen when the
477 network is busy.
478 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000479 if self.sock is None:
480 self.createSocket()
481 #self.sock can be None either because we haven't reached the retry
482 #time yet, or because we have reached the retry time and retried,
483 #but are still unable to connect.
484 if self.sock:
485 try:
486 if hasattr(self.sock, "sendall"):
487 self.sock.sendall(s)
488 else:
489 sentsofar = 0
490 left = len(s)
491 while left > 0:
492 sent = self.sock.send(s[sentsofar:])
493 sentsofar = sentsofar + sent
494 left = left - sent
495 except socket.error:
496 self.sock.close()
497 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000498
499 def makePickle(self, record):
500 """
501 Pickles the record in binary format with a length prefix, and
502 returns it ready for transmission across the socket.
503 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000504 ei = record.exc_info
505 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000506 dummy = self.format(record) # just to get traceback text into record.exc_text
507 record.exc_info = None # to avoid Unpickleable error
Guido van Rossumba205d62006-08-17 08:57:26 +0000508 s = pickle.dumps(record.__dict__, 1)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000509 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000510 record.exc_info = ei # for next handler
Guido van Rossum57102f82002-11-13 16:15:58 +0000511 slen = struct.pack(">L", len(s))
512 return slen + s
513
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000514 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000515 """
516 Handle an error during logging.
517
518 An error has occurred during logging. Most likely cause -
519 connection lost. Close the socket so that we can retry on the
520 next event.
521 """
522 if self.closeOnError and self.sock:
523 self.sock.close()
524 self.sock = None #try to reconnect next time
525 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000526 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000527
528 def emit(self, record):
529 """
530 Emit a record.
531
532 Pickles the record and writes it to the socket in binary format.
533 If there is an error with the socket, silently drop the packet.
534 If there was a problem with the socket, re-establishes the
535 socket.
536 """
537 try:
538 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000539 self.send(s)
Vinay Sajip85c19092005-10-31 13:14:19 +0000540 except (KeyboardInterrupt, SystemExit):
541 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000542 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000543 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000544
545 def close(self):
546 """
547 Closes the socket.
548 """
549 if self.sock:
550 self.sock.close()
551 self.sock = None
Vinay Sajip48cfe382004-02-20 13:17:27 +0000552 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000553
554class DatagramHandler(SocketHandler):
555 """
556 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000557 a datagram socket. The pickle which is sent is that of the LogRecord's
558 attribute dictionary (__dict__), so that the receiver does not need to
559 have the logging module installed in order to process the logging event.
560
561 To unpickle the record at the receiving end into a LogRecord, use the
562 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000563
564 """
565 def __init__(self, host, port):
566 """
567 Initializes the handler with a specific host address and port.
568 """
569 SocketHandler.__init__(self, host, port)
570 self.closeOnError = 0
571
572 def makeSocket(self):
573 """
574 The factory method of SocketHandler is here overridden to create
575 a UDP socket (SOCK_DGRAM).
576 """
577 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
578 return s
579
580 def send(self, s):
581 """
582 Send a pickled string to a socket.
583
584 This function no longer allows for partial sends which can happen
585 when the network is busy - UDP does not guarantee delivery and
586 can deliver packets out of sequence.
587 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000588 if self.sock is None:
589 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000590 self.sock.sendto(s, (self.host, self.port))
591
592class SysLogHandler(logging.Handler):
593 """
594 A handler class which sends formatted logging records to a syslog
595 server. Based on Sam Rushing's syslog module:
596 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
597 Contributed by Nicolas Untz (after which minor refactoring changes
598 have been made).
599 """
600
601 # from <linux/sys/syslog.h>:
602 # ======================================================================
603 # priorities/facilities are encoded into a single 32-bit quantity, where
604 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
605 # facility (0-big number). Both the priorities and the facilities map
606 # roughly one-to-one to strings in the syslogd(8) source code. This
607 # mapping is included in this file.
608 #
609 # priorities (these are ordered)
610
611 LOG_EMERG = 0 # system is unusable
612 LOG_ALERT = 1 # action must be taken immediately
613 LOG_CRIT = 2 # critical conditions
614 LOG_ERR = 3 # error conditions
615 LOG_WARNING = 4 # warning conditions
616 LOG_NOTICE = 5 # normal but significant condition
617 LOG_INFO = 6 # informational
618 LOG_DEBUG = 7 # debug-level messages
619
620 # facility codes
621 LOG_KERN = 0 # kernel messages
622 LOG_USER = 1 # random user-level messages
623 LOG_MAIL = 2 # mail system
624 LOG_DAEMON = 3 # system daemons
625 LOG_AUTH = 4 # security/authorization messages
626 LOG_SYSLOG = 5 # messages generated internally by syslogd
627 LOG_LPR = 6 # line printer subsystem
628 LOG_NEWS = 7 # network news subsystem
629 LOG_UUCP = 8 # UUCP subsystem
630 LOG_CRON = 9 # clock daemon
631 LOG_AUTHPRIV = 10 # security/authorization messages (private)
632
633 # other codes through 15 reserved for system use
634 LOG_LOCAL0 = 16 # reserved for local use
635 LOG_LOCAL1 = 17 # reserved for local use
636 LOG_LOCAL2 = 18 # reserved for local use
637 LOG_LOCAL3 = 19 # reserved for local use
638 LOG_LOCAL4 = 20 # reserved for local use
639 LOG_LOCAL5 = 21 # reserved for local use
640 LOG_LOCAL6 = 22 # reserved for local use
641 LOG_LOCAL7 = 23 # reserved for local use
642
643 priority_names = {
644 "alert": LOG_ALERT,
645 "crit": LOG_CRIT,
646 "critical": LOG_CRIT,
647 "debug": LOG_DEBUG,
648 "emerg": LOG_EMERG,
649 "err": LOG_ERR,
650 "error": LOG_ERR, # DEPRECATED
651 "info": LOG_INFO,
652 "notice": LOG_NOTICE,
653 "panic": LOG_EMERG, # DEPRECATED
654 "warn": LOG_WARNING, # DEPRECATED
655 "warning": LOG_WARNING,
656 }
657
658 facility_names = {
659 "auth": LOG_AUTH,
660 "authpriv": LOG_AUTHPRIV,
661 "cron": LOG_CRON,
662 "daemon": LOG_DAEMON,
663 "kern": LOG_KERN,
664 "lpr": LOG_LPR,
665 "mail": LOG_MAIL,
666 "news": LOG_NEWS,
667 "security": LOG_AUTH, # DEPRECATED
668 "syslog": LOG_SYSLOG,
669 "user": LOG_USER,
670 "uucp": LOG_UUCP,
671 "local0": LOG_LOCAL0,
672 "local1": LOG_LOCAL1,
673 "local2": LOG_LOCAL2,
674 "local3": LOG_LOCAL3,
675 "local4": LOG_LOCAL4,
676 "local5": LOG_LOCAL5,
677 "local6": LOG_LOCAL6,
678 "local7": LOG_LOCAL7,
679 }
680
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000681 #The map below appears to be trivially lowercasing the key. However,
682 #there's more to it than meets the eye - in some locales, lowercasing
683 #gives unexpected results. See SF #1524081: in the Turkish locale,
684 #"INFO".lower() != "info"
685 priority_map = {
686 "DEBUG" : "debug",
687 "INFO" : "info",
688 "WARNING" : "warning",
689 "ERROR" : "error",
690 "CRITICAL" : "critical"
691 }
692
Guido van Rossum57102f82002-11-13 16:15:58 +0000693 def __init__(self, address=('localhost', SYSLOG_UDP_PORT), facility=LOG_USER):
694 """
695 Initialize a handler.
696
Guido van Rossume7ba4952007-06-06 23:52:48 +0000697 If address is specified as a string, a UNIX socket is used. To log to a
698 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000699 If facility is not specified, LOG_USER is used.
700 """
701 logging.Handler.__init__(self)
702
703 self.address = address
704 self.facility = facility
Guido van Rossum13257902007-06-07 23:15:56 +0000705 if isinstance(address, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000706 self.unixsocket = 1
Thomas Wouters89f507f2006-12-13 04:49:30 +0000707 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000708 else:
Guido van Rossum57102f82002-11-13 16:15:58 +0000709 self.unixsocket = 0
Thomas Wouters89f507f2006-12-13 04:49:30 +0000710 self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000711
712 self.formatter = None
713
Vinay Sajipa1974c12005-01-13 08:23:56 +0000714 def _connect_unixsocket(self, address):
715 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
716 # syslog may require either DGRAM or STREAM sockets
717 try:
718 self.socket.connect(address)
719 except socket.error:
720 self.socket.close()
721 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Vinay Sajip8b6b53f2005-11-09 13:55:13 +0000722 self.socket.connect(address)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000723
Guido van Rossum57102f82002-11-13 16:15:58 +0000724 # curious: when talking to the unix-domain '/dev/log' socket, a
725 # zero-terminator seems to be required. this string is placed
726 # into a class variable so that it can be overridden if
727 # necessary.
728 log_format_string = '<%d>%s\000'
729
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000730 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000731 """
732 Encode the facility and priority. You can pass in strings or
733 integers - if strings are passed, the facility_names and
734 priority_names mapping dictionaries are used to convert them to
735 integers.
736 """
Guido van Rossum13257902007-06-07 23:15:56 +0000737 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000738 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000739 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000740 priority = self.priority_names[priority]
741 return (facility << 3) | priority
742
743 def close (self):
744 """
745 Closes the socket.
746 """
747 if self.unixsocket:
748 self.socket.close()
Vinay Sajip48cfe382004-02-20 13:17:27 +0000749 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000750
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000751 def mapPriority(self, levelName):
752 """
753 Map a logging level name to a key in the priority_names map.
754 This is useful in two scenarios: when custom levels are being
755 used, and in the case where you can't do a straightforward
756 mapping by lowercasing the logging level name because of locale-
757 specific issues (see SF #1524081).
758 """
759 return self.priority_map.get(levelName, "warning")
760
Guido van Rossum57102f82002-11-13 16:15:58 +0000761 def emit(self, record):
762 """
763 Emit a record.
764
765 The record is formatted, and then sent to the syslog server. If
766 exception information is present, it is NOT sent to the server.
767 """
768 msg = self.format(record)
769 """
770 We need to convert record level to lowercase, maybe this will
771 change in the future.
772 """
773 msg = self.log_format_string % (
774 self.encodePriority(self.facility,
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000775 self.mapPriority(record.levelname)),
776 msg)
Guido van Rossum57102f82002-11-13 16:15:58 +0000777 try:
778 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000779 try:
780 self.socket.send(msg)
781 except socket.error:
782 self._connect_unixsocket(self.address)
783 self.socket.send(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +0000784 else:
785 self.socket.sendto(msg, self.address)
Vinay Sajip85c19092005-10-31 13:14:19 +0000786 except (KeyboardInterrupt, SystemExit):
787 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000788 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000789 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000790
791class SMTPHandler(logging.Handler):
792 """
793 A handler class which sends an SMTP email for each logging event.
794 """
Guido van Rossum360e4b82007-05-14 22:51:27 +0000795 def __init__(self, mailhost, fromaddr, toaddrs, subject, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000796 """
797 Initialize the handler.
798
799 Initialize the instance with the from and to addresses and subject
800 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000801 (host, port) tuple format for the mailhost argument. To specify
802 authentication credentials, supply a (username, password) tuple
803 for the credentials argument.
Guido van Rossum57102f82002-11-13 16:15:58 +0000804 """
805 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000806 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000807 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000808 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000809 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000810 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000811 self.username, self.password = credentials
812 else:
813 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000814 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000815 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000816 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000817 self.toaddrs = toaddrs
818 self.subject = subject
819
820 def getSubject(self, record):
821 """
822 Determine the subject for the email.
823
824 If you want to specify a subject line which is record-dependent,
825 override this method.
826 """
827 return self.subject
828
Vinay Sajipe7d40662004-10-03 19:12:07 +0000829 weekdayname = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
830
831 monthname = [None,
832 'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun',
833 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
834
835 def date_time(self):
836 """
837 Return the current date and time formatted for a MIME header.
838 Needed for Python 1.5.2 (no email package available)
839 """
840 year, month, day, hh, mm, ss, wd, y, z = time.gmtime(time.time())
841 s = "%s, %02d %3s %4d %02d:%02d:%02d GMT" % (
842 self.weekdayname[wd],
843 day, self.monthname[month], year,
844 hh, mm, ss)
845 return s
846
Guido van Rossum57102f82002-11-13 16:15:58 +0000847 def emit(self, record):
848 """
849 Emit a record.
850
851 Format the record and send it to the specified addressees.
852 """
853 try:
854 import smtplib
Vinay Sajipe7d40662004-10-03 19:12:07 +0000855 try:
Thomas Woutersb2137042007-02-01 18:02:27 +0000856 from email.utils import formatdate
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000857 except ImportError:
Vinay Sajipe7d40662004-10-03 19:12:07 +0000858 formatdate = self.date_time
Guido van Rossum57102f82002-11-13 16:15:58 +0000859 port = self.mailport
860 if not port:
861 port = smtplib.SMTP_PORT
862 smtp = smtplib.SMTP(self.mailhost, port)
863 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000864 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 +0000865 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000866 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000867 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000868 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000869 if self.username:
870 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000871 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
872 smtp.quit()
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000873 except (KeyboardInterrupt, SystemExit):
874 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000875 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000876 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000877
878class NTEventLogHandler(logging.Handler):
879 """
880 A handler class which sends events to the NT Event Log. Adds a
881 registry entry for the specified application name. If no dllname is
882 provided, win32service.pyd (which contains some basic message
883 placeholders) is used. Note that use of these placeholders will make
884 your event logs big, as the entire message source is held in the log.
885 If you want slimmer logs, you have to pass in the name of your own DLL
886 which contains the message definitions you want to use in the event log.
887 """
888 def __init__(self, appname, dllname=None, logtype="Application"):
889 logging.Handler.__init__(self)
890 try:
891 import win32evtlogutil, win32evtlog
892 self.appname = appname
893 self._welu = win32evtlogutil
894 if not dllname:
895 dllname = os.path.split(self._welu.__file__)
896 dllname = os.path.split(dllname[0])
897 dllname = os.path.join(dllname[0], r'win32service.pyd')
898 self.dllname = dllname
899 self.logtype = logtype
900 self._welu.AddSourceToRegistry(appname, dllname, logtype)
901 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
902 self.typemap = {
903 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
904 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000905 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000906 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
907 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
908 }
909 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000910 print("The Python Win32 extensions for NT (service, event "\
911 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000912 self._welu = None
913
914 def getMessageID(self, record):
915 """
916 Return the message ID for the event record. If you are using your
917 own messages, you could do this by having the msg passed to the
918 logger being an ID rather than a formatting string. Then, in here,
919 you could use a dictionary lookup to get the message ID. This
920 version returns 1, which is the base message ID in win32service.pyd.
921 """
922 return 1
923
924 def getEventCategory(self, record):
925 """
926 Return the event category for the record.
927
928 Override this if you want to specify your own categories. This version
929 returns 0.
930 """
931 return 0
932
933 def getEventType(self, record):
934 """
935 Return the event type for the record.
936
937 Override this if you want to specify your own types. This version does
938 a mapping using the handler's typemap attribute, which is set up in
939 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000940 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000941 either need to override this method or place a suitable dictionary in
942 the handler's typemap attribute.
943 """
944 return self.typemap.get(record.levelno, self.deftype)
945
946 def emit(self, record):
947 """
948 Emit a record.
949
950 Determine the message ID, event category and event type. Then
951 log the message in the NT event log.
952 """
953 if self._welu:
954 try:
955 id = self.getMessageID(record)
956 cat = self.getEventCategory(record)
957 type = self.getEventType(record)
958 msg = self.format(record)
959 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000960 except (KeyboardInterrupt, SystemExit):
961 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000962 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000963 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000964
965 def close(self):
966 """
967 Clean up this handler.
968
969 You can remove the application name from the registry as a
970 source of event log entries. However, if you do this, you will
971 not be able to see the events as you intended in the Event Log
972 Viewer - it needs to be able to access the registry to get the
973 DLL name.
974 """
975 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000976 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000977
978class HTTPHandler(logging.Handler):
979 """
980 A class which sends records to a Web server, using either GET or
981 POST semantics.
982 """
983 def __init__(self, host, url, method="GET"):
984 """
985 Initialize the instance with the host, the request URL, and the method
986 ("GET" or "POST")
987 """
988 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000989 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +0000990 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +0000991 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +0000992 self.host = host
993 self.url = url
994 self.method = method
995
Neal Norwitzf297bd12003-04-23 03:49:43 +0000996 def mapLogRecord(self, record):
997 """
998 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +0000999 that is sent as the CGI data. Overwrite in your class.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001000 Contributed by Franz Glasner.
1001 """
1002 return record.__dict__
1003
Guido van Rossum57102f82002-11-13 16:15:58 +00001004 def emit(self, record):
1005 """
1006 Emit a record.
1007
1008 Send the record to the Web server as an URL-encoded dictionary
1009 """
1010 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001011 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001012 host = self.host
Georg Brandl24420152008-05-26 16:32:26 +00001013 h = http.client.HTTP(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001014 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001015 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001016 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001017 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001018 sep = '&'
1019 else:
1020 sep = '?'
1021 url = url + "%c%s" % (sep, data)
1022 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001023 # support multiple hosts on one IP address...
1024 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001025 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001026 if i >= 0:
1027 host = host[:i]
1028 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001029 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001030 h.putheader("Content-type",
1031 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001032 h.putheader("Content-length", str(len(data)))
Benjamin Peterson458ad472009-01-18 00:08:45 +00001033 h.endheaders(data if self.method == "POST" else None)
Guido van Rossum57102f82002-11-13 16:15:58 +00001034 h.getreply() #can't do anything with the result
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001035 except (KeyboardInterrupt, SystemExit):
1036 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001037 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001038 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001039
1040class BufferingHandler(logging.Handler):
1041 """
1042 A handler class which buffers logging records in memory. Whenever each
1043 record is added to the buffer, a check is made to see if the buffer should
1044 be flushed. If it should, then flush() is expected to do what's needed.
1045 """
1046 def __init__(self, capacity):
1047 """
1048 Initialize the handler with the buffer size.
1049 """
1050 logging.Handler.__init__(self)
1051 self.capacity = capacity
1052 self.buffer = []
1053
1054 def shouldFlush(self, record):
1055 """
1056 Should the handler flush its buffer?
1057
1058 Returns true if the buffer is up to capacity. This method can be
1059 overridden to implement custom flushing strategies.
1060 """
1061 return (len(self.buffer) >= self.capacity)
1062
1063 def emit(self, record):
1064 """
1065 Emit a record.
1066
1067 Append the record. If shouldFlush() tells us to, call flush() to process
1068 the buffer.
1069 """
1070 self.buffer.append(record)
1071 if self.shouldFlush(record):
1072 self.flush()
1073
1074 def flush(self):
1075 """
1076 Override to implement custom flushing behaviour.
1077
1078 This version just zaps the buffer to empty.
1079 """
1080 self.buffer = []
1081
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001082 def close(self):
1083 """
1084 Close the handler.
1085
1086 This version just flushes and chains to the parent class' close().
1087 """
1088 self.flush()
1089 logging.Handler.close(self)
1090
Guido van Rossum57102f82002-11-13 16:15:58 +00001091class MemoryHandler(BufferingHandler):
1092 """
1093 A handler class which buffers logging records in memory, periodically
1094 flushing them to a target handler. Flushing occurs whenever the buffer
1095 is full, or when an event of a certain severity or greater is seen.
1096 """
1097 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1098 """
1099 Initialize the handler with the buffer size, the level at which
1100 flushing should occur and an optional target.
1101
1102 Note that without a target being set either here or via setTarget(),
1103 a MemoryHandler is no use to anyone!
1104 """
1105 BufferingHandler.__init__(self, capacity)
1106 self.flushLevel = flushLevel
1107 self.target = target
1108
1109 def shouldFlush(self, record):
1110 """
1111 Check for buffer full or a record at the flushLevel or higher.
1112 """
1113 return (len(self.buffer) >= self.capacity) or \
1114 (record.levelno >= self.flushLevel)
1115
1116 def setTarget(self, target):
1117 """
1118 Set the target handler for this handler.
1119 """
1120 self.target = target
1121
1122 def flush(self):
1123 """
1124 For a MemoryHandler, flushing means just sending the buffered
1125 records to the target, if there is one. Override if you want
1126 different behaviour.
1127 """
1128 if self.target:
1129 for record in self.buffer:
1130 self.target.handle(record)
1131 self.buffer = []
1132
1133 def close(self):
1134 """
1135 Flush, set the target to None and lose the buffer.
1136 """
1137 self.flush()
1138 self.target = None
Vinay Sajip48cfe382004-02-20 13:17:27 +00001139 BufferingHandler.close(self)