blob: 708edc51ac76184668846ebed8d7151944df6b68 [file] [log] [blame]
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001# Copyright 2001-2012 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
Vinay Sajip0abf61d2012-02-23 19:45:52 +000019based on PEP 282 and comments thereto in comp.lang.python.
Guido van Rossum57102f82002-11-13 16:15:58 +000020
Vinay Sajip0abf61d2012-02-23 19:45:52 +000021Copyright (C) 2001-2012 Vinay Sajip. All Rights Reserved.
Guido van Rossum57102f82002-11-13 16:15:58 +000022
Vinay Sajip6268cbc2009-01-21 00:19:28 +000023To use, simply 'import logging.handlers' and log away!
Guido van Rossum57102f82002-11-13 16:15:58 +000024"""
25
Benjamin Petersonad9d48d2008-04-02 21:49:44 +000026import logging, socket, os, pickle, struct, time, re
Benjamin Peterson9451a1c2010-03-13 22:30:34 +000027from stat import ST_DEV, ST_INO, ST_MTIME
Vinay Sajipe723e962011-04-15 22:27:17 +010028import queue
Victor Stinnercafa2ef2011-05-02 16:11:28 +020029try:
30 import threading
31except ImportError:
32 threading = None
Guido van Rossum57102f82002-11-13 16:15:58 +000033
Vinay Sajip4600f112005-03-13 09:56:36 +000034try:
35 import codecs
36except ImportError:
37 codecs = None
38
Guido van Rossum57102f82002-11-13 16:15:58 +000039#
40# Some constants...
41#
42
43DEFAULT_TCP_LOGGING_PORT = 9020
44DEFAULT_UDP_LOGGING_PORT = 9021
45DEFAULT_HTTP_LOGGING_PORT = 9022
46DEFAULT_SOAP_LOGGING_PORT = 9023
47SYSLOG_UDP_PORT = 514
Vinay Sajipcbabd7e2009-10-10 20:32:36 +000048SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000049
Thomas Wouters477c8d52006-05-27 19:21:47 +000050_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
51
Vinay Sajip17c52d82004-07-03 11:48:34 +000052class BaseRotatingHandler(logging.FileHandler):
53 """
54 Base class for handlers that rotate log files at a certain point.
55 Not meant to be instantiated directly. Instead, use RotatingFileHandler
56 or TimedRotatingFileHandler.
57 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000058 def __init__(self, filename, mode, encoding=None, delay=0):
Vinay Sajip17c52d82004-07-03 11:48:34 +000059 """
60 Use the specified filename for streamed logging
61 """
Vinay Sajip4600f112005-03-13 09:56:36 +000062 if codecs is None:
63 encoding = None
Christian Heimese7a15bb2008-01-24 16:21:45 +000064 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip4600f112005-03-13 09:56:36 +000065 self.mode = mode
66 self.encoding = encoding
Guido van Rossum57102f82002-11-13 16:15:58 +000067
Vinay Sajip17c52d82004-07-03 11:48:34 +000068 def emit(self, record):
69 """
70 Emit a record.
71
72 Output the record to the file, catering for rollover as described
73 in doRollover().
74 """
Vinay Sajip3970c112004-07-08 10:24:04 +000075 try:
76 if self.shouldRollover(record):
77 self.doRollover()
78 logging.FileHandler.emit(self, record)
Vinay Sajip85c19092005-10-31 13:14:19 +000079 except (KeyboardInterrupt, SystemExit):
80 raise
Vinay Sajip3970c112004-07-08 10:24:04 +000081 except:
82 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000083
84class RotatingFileHandler(BaseRotatingHandler):
85 """
86 Handler for logging to a set of files, which switches from one file
87 to the next when the current file reaches a certain size.
88 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000089 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=0):
Guido van Rossum57102f82002-11-13 16:15:58 +000090 """
91 Open the specified file and use it as the stream for logging.
92
93 By default, the file grows indefinitely. You can specify particular
94 values of maxBytes and backupCount to allow the file to rollover at
95 a predetermined size.
96
97 Rollover occurs whenever the current log file is nearly maxBytes in
98 length. If backupCount is >= 1, the system will successively create
99 new files with the same pathname as the base file, but with extensions
100 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
101 and a base file name of "app.log", you would get "app.log",
102 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
103 written to is always "app.log" - when it gets filled up, it is closed
104 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
105 exist, then they are renamed to "app.log.2", "app.log.3" etc.
106 respectively.
107
108 If maxBytes is zero, rollover never occurs.
109 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000110 # If rotation/rollover is wanted, it doesn't make sense to use another
111 # mode. If for example 'w' were specified, then if there were multiple
112 # runs of the calling application, the logs from previous runs would be
113 # lost if the 'w' is respected, because the log file would be truncated
114 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000115 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000116 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000117 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000118 self.maxBytes = maxBytes
119 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000120
121 def doRollover(self):
122 """
123 Do a rollover, as described in __init__().
124 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000125 if self.stream:
126 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000127 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000128 if self.backupCount > 0:
129 for i in range(self.backupCount - 1, 0, -1):
130 sfn = "%s.%d" % (self.baseFilename, i)
131 dfn = "%s.%d" % (self.baseFilename, i + 1)
132 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000133 if os.path.exists(dfn):
134 os.remove(dfn)
135 os.rename(sfn, dfn)
136 dfn = self.baseFilename + ".1"
137 if os.path.exists(dfn):
138 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000139 os.rename(self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000140 self.mode = 'w'
141 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000142
Vinay Sajip17c52d82004-07-03 11:48:34 +0000143 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000144 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000145 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000146
Vinay Sajip17c52d82004-07-03 11:48:34 +0000147 Basically, see if the supplied record would cause the file to exceed
148 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000149 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000150 if self.stream is None: # delay was set...
151 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000152 if self.maxBytes > 0: # are we rolling over?
153 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000154 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000155 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000156 return 1
157 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000158
Vinay Sajip17c52d82004-07-03 11:48:34 +0000159class TimedRotatingFileHandler(BaseRotatingHandler):
160 """
161 Handler for logging to a file, rotating the log file at certain timed
162 intervals.
163
164 If backupCount is > 0, when rollover is done, no more than backupCount
165 files are kept - the oldest ones are deleted.
166 """
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000167 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000168 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000169 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000170 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000171 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000172 # Calculate the real rollover interval, which is just the number of
173 # seconds between rollovers. Also set the filename suffix used when
174 # a rollover occurs. Current 'when' events supported:
175 # S - Seconds
176 # M - Minutes
177 # H - Hours
178 # D - Days
179 # midnight - roll over at midnight
180 # W{0-6} - roll over on a certain day; 0 - Monday
181 #
182 # Case of the 'when' specifier is not important; lower or upper case
183 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000184 if self.when == 'S':
185 self.interval = 1 # one second
186 self.suffix = "%Y-%m-%d_%H-%M-%S"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000187 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000188 elif self.when == 'M':
189 self.interval = 60 # one minute
190 self.suffix = "%Y-%m-%d_%H-%M"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000191 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000192 elif self.when == 'H':
193 self.interval = 60 * 60 # one hour
194 self.suffix = "%Y-%m-%d_%H"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000195 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000196 elif self.when == 'D' or self.when == 'MIDNIGHT':
197 self.interval = 60 * 60 * 24 # one day
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 elif self.when.startswith('W'):
201 self.interval = 60 * 60 * 24 * 7 # one week
202 if len(self.when) != 2:
203 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
204 if self.when[1] < '0' or self.when[1] > '6':
205 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
206 self.dayOfWeek = int(self.when[1])
207 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000208 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000209 else:
210 raise ValueError("Invalid rollover interval specified: %s" % self.when)
211
Antoine Pitroufd036452008-08-19 17:56:33 +0000212 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000213 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000214 if os.path.exists(filename):
215 t = os.stat(filename)[ST_MTIME]
216 else:
217 t = int(time.time())
218 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000219
Vinay Sajipaffbd872009-06-11 10:11:47 +0000220 def computeRollover(self, currentTime):
221 """
222 Work out the rollover time based on the specified time.
223 """
224 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000225 # If we are rolling over at midnight or weekly, then the interval is already known.
226 # What we need to figure out is WHEN the next interval is. In other words,
227 # if you are rolling over at midnight, then your base interval is 1 day,
228 # but you want to start that one day clock at midnight, not now. So, we
229 # have to fudge the rolloverAt value in order to trigger the first rollover
230 # at the right time. After that, the regular interval will take care of
231 # the rest. Note that this code doesn't care about leap seconds. :)
232 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
233 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000234 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000235 t = time.gmtime(currentTime)
236 else:
237 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000238 currentHour = t[3]
239 currentMinute = t[4]
240 currentSecond = t[5]
241 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000242 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
243 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000244 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000245 # If we are rolling over on a certain day, add in the number of days until
246 # the next rollover, but offset by 1 since we just calculated the time
247 # until the next day starts. There are three cases:
248 # Case 1) The day to rollover is today; in this case, do nothing
249 # Case 2) The day to rollover is further in the interval (i.e., today is
250 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
251 # next rollover is simply 6 - 2 - 1, or 3.
252 # Case 3) The day to rollover is behind us in the interval (i.e., today
253 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
254 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
255 # number of days left in the current week (1) plus the number
256 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000257 # The calculations described in 2) and 3) above need to have a day added.
258 # This is because the above time calculation takes us to midnight on this
259 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000260 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000261 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000262 if day != self.dayOfWeek:
263 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000264 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000265 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000266 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000267 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
268 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000269 dstNow = t[-1]
270 dstAtRollover = time.localtime(newRolloverAt)[-1]
271 if dstNow != dstAtRollover:
272 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
273 newRolloverAt = newRolloverAt - 3600
274 else: # DST bows out before next rollover, so we need to add an hour
275 newRolloverAt = newRolloverAt + 3600
Vinay Sajipaffbd872009-06-11 10:11:47 +0000276 result = newRolloverAt
277 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000278
279 def shouldRollover(self, record):
280 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000281 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000282
283 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000284 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000285 """
286 t = int(time.time())
287 if t >= self.rolloverAt:
288 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000289 return 0
290
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000291 def getFilesToDelete(self):
292 """
293 Determine the files to delete when rolling over.
294
295 More specific than the earlier method, which just used glob.glob().
296 """
297 dirName, baseName = os.path.split(self.baseFilename)
298 fileNames = os.listdir(dirName)
299 result = []
300 prefix = baseName + "."
301 plen = len(prefix)
302 for fileName in fileNames:
303 if fileName[:plen] == prefix:
304 suffix = fileName[plen:]
305 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000306 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000307 result.sort()
308 if len(result) < self.backupCount:
309 result = []
310 else:
311 result = result[:len(result) - self.backupCount]
312 return result
313
Vinay Sajip17c52d82004-07-03 11:48:34 +0000314 def doRollover(self):
315 """
316 do a rollover; in this case, a date/time stamp is appended to the filename
317 when the rollover happens. However, you want the file to be named for the
318 start of the interval, not the current time. If there is a backup count,
319 then we have to get a list of matching filenames, sort them and remove
320 the one with the oldest suffix.
321 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000322 if self.stream:
323 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000324 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000325 # get the time that this sequence started at and make it a TimeTuple
326 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000327 if self.utc:
328 timeTuple = time.gmtime(t)
329 else:
330 timeTuple = time.localtime(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000331 dfn = self.baseFilename + "." + time.strftime(self.suffix, timeTuple)
332 if os.path.exists(dfn):
333 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000334 os.rename(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000335 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000336 for s in self.getFilesToDelete():
337 os.remove(s)
Thomas Woutersb2137042007-02-01 18:02:27 +0000338 self.mode = 'w'
339 self.stream = self._open()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000340 currentTime = int(time.time())
Vinay Sajipaffbd872009-06-11 10:11:47 +0000341 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000342 while newRolloverAt <= currentTime:
343 newRolloverAt = newRolloverAt + self.interval
344 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000345 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000346 dstNow = time.localtime(currentTime)[-1]
347 dstAtRollover = time.localtime(newRolloverAt)[-1]
348 if dstNow != dstAtRollover:
349 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
350 newRolloverAt = newRolloverAt - 3600
351 else: # DST bows out before next rollover, so we need to add an hour
352 newRolloverAt = newRolloverAt + 3600
353 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000354
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000355class WatchedFileHandler(logging.FileHandler):
356 """
357 A handler for logging to a file, which watches the file
358 to see if it has changed while in use. This can happen because of
359 usage of programs such as newsyslog and logrotate which perform
360 log file rotation. This handler, intended for use under Unix,
361 watches the file to see if it has changed since the last emit.
362 (A file has changed if its device or inode have changed.)
363 If it has changed, the old file stream is closed, and the file
364 opened to get a new stream.
365
366 This handler is not appropriate for use under Windows, because
367 under Windows open files cannot be moved or renamed - logging
368 opens the files with exclusive locks - and so there is no need
369 for such a handler. Furthermore, ST_INO is not supported under
370 Windows; stat always returns zero for this value.
371
372 This handler is based on a suggestion and patch by Chad J.
373 Schroeder.
374 """
Christian Heimese7a15bb2008-01-24 16:21:45 +0000375 def __init__(self, filename, mode='a', encoding=None, delay=0):
376 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
377 if not os.path.exists(self.baseFilename):
378 self.dev, self.ino = -1, -1
379 else:
380 stat = os.stat(self.baseFilename)
381 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000382
383 def emit(self, record):
384 """
385 Emit a record.
386
387 First check if the underlying file has changed, and if it
388 has, close the old stream and reopen the file to get the
389 current stream.
390 """
391 if not os.path.exists(self.baseFilename):
392 stat = None
393 changed = 1
394 else:
395 stat = os.stat(self.baseFilename)
396 changed = (stat[ST_DEV] != self.dev) or (stat[ST_INO] != self.ino)
Christian Heimese7a15bb2008-01-24 16:21:45 +0000397 if changed and self.stream is not None:
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000398 self.stream.flush()
399 self.stream.close()
400 self.stream = self._open()
401 if stat is None:
402 stat = os.stat(self.baseFilename)
403 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
404 logging.FileHandler.emit(self, record)
405
Guido van Rossum57102f82002-11-13 16:15:58 +0000406class SocketHandler(logging.Handler):
407 """
408 A handler class which writes logging records, in pickle format, to
409 a streaming socket. The socket is kept open across logging calls.
410 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000411 The pickle which is sent is that of the LogRecord's attribute dictionary
412 (__dict__), so that the receiver does not need to have the logging module
413 installed in order to process the logging event.
414
415 To unpickle the record at the receiving end into a LogRecord, use the
416 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000417 """
418
419 def __init__(self, host, port):
420 """
421 Initializes the handler with a specific host address and port.
422
423 The attribute 'closeOnError' is set to 1 - which means that if
424 a socket error occurs, the socket is silently closed and then
425 reopened on the next logging call.
426 """
427 logging.Handler.__init__(self)
428 self.host = host
429 self.port = port
430 self.sock = None
431 self.closeOnError = 0
Vinay Sajip48cfe382004-02-20 13:17:27 +0000432 self.retryTime = None
433 #
434 # Exponential backoff parameters.
435 #
436 self.retryStart = 1.0
437 self.retryMax = 30.0
438 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000439
Guido van Rossumd8faa362007-04-27 19:54:29 +0000440 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000441 """
442 A factory method which allows subclasses to define the precise
443 type of socket they want.
444 """
445 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000446 if hasattr(s, 'settimeout'):
447 s.settimeout(timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000448 s.connect((self.host, self.port))
449 return s
450
Vinay Sajip48cfe382004-02-20 13:17:27 +0000451 def createSocket(self):
452 """
453 Try to create a socket, using an exponential backoff with
454 a max retry time. Thanks to Robert Olson for the original patch
455 (SF #815911) which has been slightly refactored.
456 """
457 now = time.time()
458 # Either retryTime is None, in which case this
459 # is the first time back after a disconnect, or
460 # we've waited long enough.
461 if self.retryTime is None:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000462 attempt = 1
Vinay Sajip48cfe382004-02-20 13:17:27 +0000463 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000464 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000465 if attempt:
466 try:
467 self.sock = self.makeSocket()
468 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000469 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000470 #Creation failed, so set the retry time and return.
471 if self.retryTime is None:
472 self.retryPeriod = self.retryStart
473 else:
474 self.retryPeriod = self.retryPeriod * self.retryFactor
475 if self.retryPeriod > self.retryMax:
476 self.retryPeriod = self.retryMax
477 self.retryTime = now + self.retryPeriod
478
Guido van Rossum57102f82002-11-13 16:15:58 +0000479 def send(self, s):
480 """
481 Send a pickled string to the socket.
482
483 This function allows for partial sends which can happen when the
484 network is busy.
485 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000486 if self.sock is None:
487 self.createSocket()
488 #self.sock can be None either because we haven't reached the retry
489 #time yet, or because we have reached the retry time and retried,
490 #but are still unable to connect.
491 if self.sock:
492 try:
493 if hasattr(self.sock, "sendall"):
494 self.sock.sendall(s)
495 else:
496 sentsofar = 0
497 left = len(s)
498 while left > 0:
499 sent = self.sock.send(s[sentsofar:])
500 sentsofar = sentsofar + sent
501 left = left - sent
502 except socket.error:
503 self.sock.close()
504 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000505
506 def makePickle(self, record):
507 """
508 Pickles the record in binary format with a length prefix, and
509 returns it ready for transmission across the socket.
510 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000511 ei = record.exc_info
512 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000513 dummy = self.format(record) # just to get traceback text into record.exc_text
514 record.exc_info = None # to avoid Unpickleable error
Guido van Rossumba205d62006-08-17 08:57:26 +0000515 s = pickle.dumps(record.__dict__, 1)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000516 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000517 record.exc_info = ei # for next handler
Guido van Rossum57102f82002-11-13 16:15:58 +0000518 slen = struct.pack(">L", len(s))
519 return slen + s
520
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000521 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000522 """
523 Handle an error during logging.
524
525 An error has occurred during logging. Most likely cause -
526 connection lost. Close the socket so that we can retry on the
527 next event.
528 """
529 if self.closeOnError and self.sock:
530 self.sock.close()
531 self.sock = None #try to reconnect next time
532 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000533 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000534
535 def emit(self, record):
536 """
537 Emit a record.
538
539 Pickles the record and writes it to the socket in binary format.
540 If there is an error with the socket, silently drop the packet.
541 If there was a problem with the socket, re-establishes the
542 socket.
543 """
544 try:
545 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000546 self.send(s)
Vinay Sajip85c19092005-10-31 13:14:19 +0000547 except (KeyboardInterrupt, SystemExit):
548 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000549 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000550 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000551
552 def close(self):
553 """
554 Closes the socket.
555 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000556 self.acquire()
557 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000558 if self.sock:
559 self.sock.close()
560 self.sock = None
561 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000562 finally:
563 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000564
565class DatagramHandler(SocketHandler):
566 """
567 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000568 a datagram socket. The pickle which is sent is that of the LogRecord's
569 attribute dictionary (__dict__), so that the receiver does not need to
570 have the logging module installed in order to process the logging event.
571
572 To unpickle the record at the receiving end into a LogRecord, use the
573 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000574
575 """
576 def __init__(self, host, port):
577 """
578 Initializes the handler with a specific host address and port.
579 """
580 SocketHandler.__init__(self, host, port)
581 self.closeOnError = 0
582
583 def makeSocket(self):
584 """
585 The factory method of SocketHandler is here overridden to create
586 a UDP socket (SOCK_DGRAM).
587 """
588 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
589 return s
590
591 def send(self, s):
592 """
593 Send a pickled string to a socket.
594
595 This function no longer allows for partial sends which can happen
596 when the network is busy - UDP does not guarantee delivery and
597 can deliver packets out of sequence.
598 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000599 if self.sock is None:
600 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000601 self.sock.sendto(s, (self.host, self.port))
602
603class SysLogHandler(logging.Handler):
604 """
605 A handler class which sends formatted logging records to a syslog
606 server. Based on Sam Rushing's syslog module:
607 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
608 Contributed by Nicolas Untz (after which minor refactoring changes
609 have been made).
610 """
611
612 # from <linux/sys/syslog.h>:
613 # ======================================================================
614 # priorities/facilities are encoded into a single 32-bit quantity, where
615 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
616 # facility (0-big number). Both the priorities and the facilities map
617 # roughly one-to-one to strings in the syslogd(8) source code. This
618 # mapping is included in this file.
619 #
620 # priorities (these are ordered)
621
622 LOG_EMERG = 0 # system is unusable
623 LOG_ALERT = 1 # action must be taken immediately
624 LOG_CRIT = 2 # critical conditions
625 LOG_ERR = 3 # error conditions
626 LOG_WARNING = 4 # warning conditions
627 LOG_NOTICE = 5 # normal but significant condition
628 LOG_INFO = 6 # informational
629 LOG_DEBUG = 7 # debug-level messages
630
631 # facility codes
632 LOG_KERN = 0 # kernel messages
633 LOG_USER = 1 # random user-level messages
634 LOG_MAIL = 2 # mail system
635 LOG_DAEMON = 3 # system daemons
636 LOG_AUTH = 4 # security/authorization messages
637 LOG_SYSLOG = 5 # messages generated internally by syslogd
638 LOG_LPR = 6 # line printer subsystem
639 LOG_NEWS = 7 # network news subsystem
640 LOG_UUCP = 8 # UUCP subsystem
641 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000642 LOG_AUTHPRIV = 10 # security/authorization messages (private)
643 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000644
645 # other codes through 15 reserved for system use
646 LOG_LOCAL0 = 16 # reserved for local use
647 LOG_LOCAL1 = 17 # reserved for local use
648 LOG_LOCAL2 = 18 # reserved for local use
649 LOG_LOCAL3 = 19 # reserved for local use
650 LOG_LOCAL4 = 20 # reserved for local use
651 LOG_LOCAL5 = 21 # reserved for local use
652 LOG_LOCAL6 = 22 # reserved for local use
653 LOG_LOCAL7 = 23 # reserved for local use
654
655 priority_names = {
656 "alert": LOG_ALERT,
657 "crit": LOG_CRIT,
658 "critical": LOG_CRIT,
659 "debug": LOG_DEBUG,
660 "emerg": LOG_EMERG,
661 "err": LOG_ERR,
662 "error": LOG_ERR, # DEPRECATED
663 "info": LOG_INFO,
664 "notice": LOG_NOTICE,
665 "panic": LOG_EMERG, # DEPRECATED
666 "warn": LOG_WARNING, # DEPRECATED
667 "warning": LOG_WARNING,
668 }
669
670 facility_names = {
671 "auth": LOG_AUTH,
672 "authpriv": LOG_AUTHPRIV,
673 "cron": LOG_CRON,
674 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000675 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000676 "kern": LOG_KERN,
677 "lpr": LOG_LPR,
678 "mail": LOG_MAIL,
679 "news": LOG_NEWS,
680 "security": LOG_AUTH, # DEPRECATED
681 "syslog": LOG_SYSLOG,
682 "user": LOG_USER,
683 "uucp": LOG_UUCP,
684 "local0": LOG_LOCAL0,
685 "local1": LOG_LOCAL1,
686 "local2": LOG_LOCAL2,
687 "local3": LOG_LOCAL3,
688 "local4": LOG_LOCAL4,
689 "local5": LOG_LOCAL5,
690 "local6": LOG_LOCAL6,
691 "local7": LOG_LOCAL7,
692 }
693
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000694 #The map below appears to be trivially lowercasing the key. However,
695 #there's more to it than meets the eye - in some locales, lowercasing
696 #gives unexpected results. See SF #1524081: in the Turkish locale,
697 #"INFO".lower() != "info"
698 priority_map = {
699 "DEBUG" : "debug",
700 "INFO" : "info",
701 "WARNING" : "warning",
702 "ERROR" : "error",
703 "CRITICAL" : "critical"
704 }
705
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000706 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
707 facility=LOG_USER, socktype=socket.SOCK_DGRAM):
Guido van Rossum57102f82002-11-13 16:15:58 +0000708 """
709 Initialize a handler.
710
Guido van Rossume7ba4952007-06-06 23:52:48 +0000711 If address is specified as a string, a UNIX socket is used. To log to a
712 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000713 If facility is not specified, LOG_USER is used.
714 """
715 logging.Handler.__init__(self)
716
717 self.address = address
718 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000719 self.socktype = socktype
720
Guido van Rossum13257902007-06-07 23:15:56 +0000721 if isinstance(address, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000722 self.unixsocket = 1
Thomas Wouters89f507f2006-12-13 04:49:30 +0000723 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000724 else:
Guido van Rossum57102f82002-11-13 16:15:58 +0000725 self.unixsocket = 0
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000726 self.socket = socket.socket(socket.AF_INET, socktype)
727 if socktype == socket.SOCK_STREAM:
728 self.socket.connect(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000729 self.formatter = None
730
Vinay Sajipa1974c12005-01-13 08:23:56 +0000731 def _connect_unixsocket(self, address):
732 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
733 # syslog may require either DGRAM or STREAM sockets
734 try:
735 self.socket.connect(address)
736 except socket.error:
737 self.socket.close()
738 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Vinay Sajip8b6b53f2005-11-09 13:55:13 +0000739 self.socket.connect(address)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000740
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000741 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000742 """
743 Encode the facility and priority. You can pass in strings or
744 integers - if strings are passed, the facility_names and
745 priority_names mapping dictionaries are used to convert them to
746 integers.
747 """
Guido van Rossum13257902007-06-07 23:15:56 +0000748 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000749 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000750 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000751 priority = self.priority_names[priority]
752 return (facility << 3) | priority
753
754 def close (self):
755 """
756 Closes the socket.
757 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000758 self.acquire()
759 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000760 if self.unixsocket:
761 self.socket.close()
762 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000763 finally:
764 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000765
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000766 def mapPriority(self, levelName):
767 """
768 Map a logging level name to a key in the priority_names map.
769 This is useful in two scenarios: when custom levels are being
770 used, and in the case where you can't do a straightforward
771 mapping by lowercasing the logging level name because of locale-
772 specific issues (see SF #1524081).
773 """
774 return self.priority_map.get(levelName, "warning")
775
Vinay Sajip8168d102011-06-09 16:50:49 +0100776 append_nul = True # some old syslog daemons expect a NUL terminator
777
Guido van Rossum57102f82002-11-13 16:15:58 +0000778 def emit(self, record):
779 """
780 Emit a record.
781
782 The record is formatted, and then sent to the syslog server. If
783 exception information is present, it is NOT sent to the server.
784 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100785 msg = self.format(record)
786 if self.append_nul:
787 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000788 """
789 We need to convert record level to lowercase, maybe this will
790 change in the future.
791 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000792 prio = '<%d>' % self.encodePriority(self.facility,
793 self.mapPriority(record.levelname))
794 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000795 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000796 msg = msg.encode('utf-8')
797 if codecs:
798 msg = codecs.BOM_UTF8 + msg
Vinay Sajip467d12f2010-08-19 22:20:22 +0000799 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000800 try:
801 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000802 try:
803 self.socket.send(msg)
804 except socket.error:
805 self._connect_unixsocket(self.address)
806 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000807 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000808 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000809 else:
810 self.socket.sendall(msg)
Vinay Sajip85c19092005-10-31 13:14:19 +0000811 except (KeyboardInterrupt, SystemExit):
812 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000813 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000814 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000815
816class SMTPHandler(logging.Handler):
817 """
818 A handler class which sends an SMTP email for each logging event.
819 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000820 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip25fcd222009-12-06 18:05:04 +0000821 credentials=None, secure=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000822 """
823 Initialize the handler.
824
825 Initialize the instance with the from and to addresses and subject
826 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000827 (host, port) tuple format for the mailhost argument. To specify
828 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000829 for the credentials argument. To specify the use of a secure
830 protocol (TLS), pass in a tuple for the secure argument. This will
831 only be used when authentication credentials are supplied. The tuple
832 will be either an empty tuple, or a single-value tuple with the name
833 of a keyfile, or a 2-value tuple with the names of the keyfile and
834 certificate file. (This tuple is passed to the `starttls` method).
Guido van Rossum57102f82002-11-13 16:15:58 +0000835 """
836 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000837 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000838 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000839 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000840 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000841 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000842 self.username, self.password = credentials
843 else:
844 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000845 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000846 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000847 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000848 self.toaddrs = toaddrs
849 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000850 self.secure = secure
Guido van Rossum57102f82002-11-13 16:15:58 +0000851
852 def getSubject(self, record):
853 """
854 Determine the subject for the email.
855
856 If you want to specify a subject line which is record-dependent,
857 override this method.
858 """
859 return self.subject
860
861 def emit(self, record):
862 """
863 Emit a record.
864
865 Format the record and send it to the specified addressees.
866 """
867 try:
868 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000869 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000870 port = self.mailport
871 if not port:
872 port = smtplib.SMTP_PORT
873 smtp = smtplib.SMTP(self.mailhost, port)
874 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000875 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 +0000876 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000877 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000878 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000879 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000880 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000881 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000882 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000883 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000884 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000885 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000886 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
887 smtp.quit()
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000888 except (KeyboardInterrupt, SystemExit):
889 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000890 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000891 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000892
893class NTEventLogHandler(logging.Handler):
894 """
895 A handler class which sends events to the NT Event Log. Adds a
896 registry entry for the specified application name. If no dllname is
897 provided, win32service.pyd (which contains some basic message
898 placeholders) is used. Note that use of these placeholders will make
899 your event logs big, as the entire message source is held in the log.
900 If you want slimmer logs, you have to pass in the name of your own DLL
901 which contains the message definitions you want to use in the event log.
902 """
903 def __init__(self, appname, dllname=None, logtype="Application"):
904 logging.Handler.__init__(self)
905 try:
906 import win32evtlogutil, win32evtlog
907 self.appname = appname
908 self._welu = win32evtlogutil
909 if not dllname:
910 dllname = os.path.split(self._welu.__file__)
911 dllname = os.path.split(dllname[0])
912 dllname = os.path.join(dllname[0], r'win32service.pyd')
913 self.dllname = dllname
914 self.logtype = logtype
915 self._welu.AddSourceToRegistry(appname, dllname, logtype)
916 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
917 self.typemap = {
918 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
919 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000920 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000921 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
922 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
923 }
924 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000925 print("The Python Win32 extensions for NT (service, event "\
926 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000927 self._welu = None
928
929 def getMessageID(self, record):
930 """
931 Return the message ID for the event record. If you are using your
932 own messages, you could do this by having the msg passed to the
933 logger being an ID rather than a formatting string. Then, in here,
934 you could use a dictionary lookup to get the message ID. This
935 version returns 1, which is the base message ID in win32service.pyd.
936 """
937 return 1
938
939 def getEventCategory(self, record):
940 """
941 Return the event category for the record.
942
943 Override this if you want to specify your own categories. This version
944 returns 0.
945 """
946 return 0
947
948 def getEventType(self, record):
949 """
950 Return the event type for the record.
951
952 Override this if you want to specify your own types. This version does
953 a mapping using the handler's typemap attribute, which is set up in
954 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000955 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000956 either need to override this method or place a suitable dictionary in
957 the handler's typemap attribute.
958 """
959 return self.typemap.get(record.levelno, self.deftype)
960
961 def emit(self, record):
962 """
963 Emit a record.
964
965 Determine the message ID, event category and event type. Then
966 log the message in the NT event log.
967 """
968 if self._welu:
969 try:
970 id = self.getMessageID(record)
971 cat = self.getEventCategory(record)
972 type = self.getEventType(record)
973 msg = self.format(record)
974 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000975 except (KeyboardInterrupt, SystemExit):
976 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000977 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000978 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000979
980 def close(self):
981 """
982 Clean up this handler.
983
984 You can remove the application name from the registry as a
985 source of event log entries. However, if you do this, you will
986 not be able to see the events as you intended in the Event Log
987 Viewer - it needs to be able to access the registry to get the
988 DLL name.
989 """
990 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000991 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000992
993class HTTPHandler(logging.Handler):
994 """
995 A class which sends records to a Web server, using either GET or
996 POST semantics.
997 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +0000998 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000999 """
1000 Initialize the instance with the host, the request URL, and the method
1001 ("GET" or "POST")
1002 """
1003 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001004 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001005 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001006 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001007 self.host = host
1008 self.url = url
1009 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001010 self.secure = secure
1011 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001012
Neal Norwitzf297bd12003-04-23 03:49:43 +00001013 def mapLogRecord(self, record):
1014 """
1015 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001016 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001017 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001018 """
1019 return record.__dict__
1020
Guido van Rossum57102f82002-11-13 16:15:58 +00001021 def emit(self, record):
1022 """
1023 Emit a record.
1024
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001025 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001026 """
1027 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001028 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001029 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001030 if self.secure:
1031 h = http.client.HTTPSConnection(host)
1032 else:
1033 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001034 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001035 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001036 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001037 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001038 sep = '&'
1039 else:
1040 sep = '?'
1041 url = url + "%c%s" % (sep, data)
1042 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001043 # support multiple hosts on one IP address...
1044 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001045 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001046 if i >= 0:
1047 host = host[:i]
1048 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001049 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001050 h.putheader("Content-type",
1051 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001052 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001053 if self.credentials:
1054 import base64
1055 s = ('u%s:%s' % self.credentials).encode('utf-8')
1056 s = 'Basic ' + base64.b64encode(s).strip()
1057 h.putheader('Authorization', s)
Benjamin Peterson458ad472009-01-18 00:08:45 +00001058 h.endheaders(data if self.method == "POST" else None)
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001059 h.getresponse() #can't do anything with the result
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001060 except (KeyboardInterrupt, SystemExit):
1061 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001062 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001063 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001064
1065class BufferingHandler(logging.Handler):
1066 """
1067 A handler class which buffers logging records in memory. Whenever each
1068 record is added to the buffer, a check is made to see if the buffer should
1069 be flushed. If it should, then flush() is expected to do what's needed.
1070 """
1071 def __init__(self, capacity):
1072 """
1073 Initialize the handler with the buffer size.
1074 """
1075 logging.Handler.__init__(self)
1076 self.capacity = capacity
1077 self.buffer = []
1078
1079 def shouldFlush(self, record):
1080 """
1081 Should the handler flush its buffer?
1082
1083 Returns true if the buffer is up to capacity. This method can be
1084 overridden to implement custom flushing strategies.
1085 """
1086 return (len(self.buffer) >= self.capacity)
1087
1088 def emit(self, record):
1089 """
1090 Emit a record.
1091
1092 Append the record. If shouldFlush() tells us to, call flush() to process
1093 the buffer.
1094 """
1095 self.buffer.append(record)
1096 if self.shouldFlush(record):
1097 self.flush()
1098
1099 def flush(self):
1100 """
1101 Override to implement custom flushing behaviour.
1102
1103 This version just zaps the buffer to empty.
1104 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001105 self.acquire()
1106 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001107 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001108 finally:
1109 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001110
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001111 def close(self):
1112 """
1113 Close the handler.
1114
1115 This version just flushes and chains to the parent class' close().
1116 """
1117 self.flush()
1118 logging.Handler.close(self)
1119
Guido van Rossum57102f82002-11-13 16:15:58 +00001120class MemoryHandler(BufferingHandler):
1121 """
1122 A handler class which buffers logging records in memory, periodically
1123 flushing them to a target handler. Flushing occurs whenever the buffer
1124 is full, or when an event of a certain severity or greater is seen.
1125 """
1126 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1127 """
1128 Initialize the handler with the buffer size, the level at which
1129 flushing should occur and an optional target.
1130
1131 Note that without a target being set either here or via setTarget(),
1132 a MemoryHandler is no use to anyone!
1133 """
1134 BufferingHandler.__init__(self, capacity)
1135 self.flushLevel = flushLevel
1136 self.target = target
1137
1138 def shouldFlush(self, record):
1139 """
1140 Check for buffer full or a record at the flushLevel or higher.
1141 """
1142 return (len(self.buffer) >= self.capacity) or \
1143 (record.levelno >= self.flushLevel)
1144
1145 def setTarget(self, target):
1146 """
1147 Set the target handler for this handler.
1148 """
1149 self.target = target
1150
1151 def flush(self):
1152 """
1153 For a MemoryHandler, flushing means just sending the buffered
1154 records to the target, if there is one. Override if you want
1155 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001156
1157 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001158 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001159 self.acquire()
1160 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001161 if self.target:
1162 for record in self.buffer:
1163 self.target.handle(record)
1164 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001165 finally:
1166 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001167
1168 def close(self):
1169 """
1170 Flush, set the target to None and lose the buffer.
1171 """
1172 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001173 self.acquire()
1174 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001175 self.target = None
1176 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001177 finally:
1178 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001179
1180
1181class QueueHandler(logging.Handler):
1182 """
1183 This handler sends events to a queue. Typically, it would be used together
1184 with a multiprocessing Queue to centralise logging to file in one process
1185 (in a multi-process application), so as to avoid file write contention
1186 between processes.
1187
1188 This code is new in Python 3.2, but this class can be copy pasted into
1189 user code for use with earlier Python versions.
1190 """
1191
1192 def __init__(self, queue):
1193 """
1194 Initialise an instance, using the passed queue.
1195 """
1196 logging.Handler.__init__(self)
1197 self.queue = queue
1198
1199 def enqueue(self, record):
1200 """
1201 Enqueue a record.
1202
1203 The base implementation uses put_nowait. You may want to override
1204 this method if you want to use blocking, timeouts or custom queue
1205 implementations.
1206 """
1207 self.queue.put_nowait(record)
1208
Vinay Sajip0258ce82010-09-22 20:34:53 +00001209 def prepare(self, record):
1210 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001211 Prepares a record for queuing. The object returned by this method is
1212 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001213
1214 The base implementation formats the record to merge the message
1215 and arguments, and removes unpickleable items from the record
1216 in-place.
1217
1218 You might want to override this method if you want to convert
1219 the record to a dict or JSON string, or send a modified copy
1220 of the record while leaving the original intact.
1221 """
1222 # The format operation gets traceback text into record.exc_text
1223 # (if there's exception data), and also puts the message into
1224 # record.message. We can then use this to replace the original
1225 # msg + args, as these might be unpickleable. We also zap the
1226 # exc_info attribute, as it's no longer needed and, if not None,
1227 # will typically not be pickleable.
1228 self.format(record)
1229 record.msg = record.message
1230 record.args = None
1231 record.exc_info = None
1232 return record
1233
Vinay Sajip121a1c42010-09-08 10:46:15 +00001234 def emit(self, record):
1235 """
1236 Emit a record.
1237
Vinay Sajip0637d492010-09-23 08:15:54 +00001238 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001239 """
1240 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001241 self.enqueue(self.prepare(record))
Vinay Sajip121a1c42010-09-08 10:46:15 +00001242 except (KeyboardInterrupt, SystemExit):
1243 raise
1244 except:
1245 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001246
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001247if threading:
1248 class QueueListener(object):
1249 """
1250 This class implements an internal threaded listener which watches for
1251 LogRecords being added to a queue, removes them and passes them to a
1252 list of handlers for processing.
1253 """
1254 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001255
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001256 def __init__(self, queue, *handlers):
1257 """
1258 Initialise an instance with the specified queue and
1259 handlers.
1260 """
1261 self.queue = queue
1262 self.handlers = handlers
1263 self._stop = threading.Event()
1264 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001265
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001266 def dequeue(self, block):
1267 """
1268 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001269
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001270 The base implementation uses get. You may want to override this method
1271 if you want to use timeouts or work with custom queue implementations.
1272 """
1273 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001274
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001275 def start(self):
1276 """
1277 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001278
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001279 This starts up a background thread to monitor the queue for
1280 LogRecords to process.
1281 """
1282 self._thread = t = threading.Thread(target=self._monitor)
1283 t.setDaemon(True)
1284 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001285
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001286 def prepare(self , record):
1287 """
1288 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001289
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001290 This method just returns the passed-in record. You may want to
1291 override this method if you need to do any custom marshalling or
1292 manipulation of the record before passing it to the handlers.
1293 """
1294 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001295
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001296 def handle(self, record):
1297 """
1298 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001299
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001300 This just loops through the handlers offering them the record
1301 to handle.
1302 """
1303 record = self.prepare(record)
1304 for handler in self.handlers:
1305 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001306
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001307 def _monitor(self):
1308 """
1309 Monitor the queue for records, and ask the handler
1310 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001311
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001312 This method runs on a separate, internal thread.
1313 The thread will terminate if it sees a sentinel object in the queue.
1314 """
1315 q = self.queue
1316 has_task_done = hasattr(q, 'task_done')
1317 while not self._stop.isSet():
1318 try:
1319 record = self.dequeue(True)
1320 if record is self._sentinel:
1321 break
1322 self.handle(record)
1323 if has_task_done:
1324 q.task_done()
1325 except queue.Empty:
1326 pass
1327 # There might still be records in the queue.
1328 while True:
1329 try:
1330 record = self.dequeue(False)
1331 if record is self._sentinel:
1332 break
1333 self.handle(record)
1334 if has_task_done:
1335 q.task_done()
1336 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001337 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001338
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001339 def stop(self):
1340 """
1341 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001342
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001343 This asks the thread to terminate, and then waits for it to do so.
1344 Note that if you don't call this before your application exits, there
1345 may be some records still left on the queue, which won't be processed.
1346 """
1347 self._stop.set()
1348 self.queue.put_nowait(self._sentinel)
1349 self._thread.join()
1350 self._thread = None