blob: f286cd61d4496c696ed568a91c75c00b7a19c80a [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
Vinay Sajip66b8b082012-04-24 23:25:30 +010026import errno, logging, socket, os, pickle, struct, time, re
Florent Xicluna5252f9f2011-11-07 19:43:05 +010027from codecs import BOM_UTF8
Benjamin Peterson9451a1c2010-03-13 22:30:34 +000028from stat import ST_DEV, ST_INO, ST_MTIME
Vinay Sajipe723e962011-04-15 22:27:17 +010029import queue
Victor Stinnercafa2ef2011-05-02 16:11:28 +020030try:
31 import threading
Vinay Sajip0372e102011-05-05 12:59:14 +010032except ImportError: #pragma: no cover
Victor Stinnercafa2ef2011-05-02 16:11:28 +020033 threading = None
Guido van Rossum57102f82002-11-13 16:15:58 +000034
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
Vinay Sajipcbabd7e2009-10-10 20:32:36 +000044SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000045
Thomas Wouters477c8d52006-05-27 19:21:47 +000046_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
47
Vinay Sajip17c52d82004-07-03 11:48:34 +000048class BaseRotatingHandler(logging.FileHandler):
49 """
50 Base class for handlers that rotate log files at a certain point.
51 Not meant to be instantiated directly. Instead, use RotatingFileHandler
52 or TimedRotatingFileHandler.
53 """
Vinay Sajip23b94d02012-01-04 12:02:26 +000054 def __init__(self, filename, mode, encoding=None, delay=False):
Vinay Sajip17c52d82004-07-03 11:48:34 +000055 """
56 Use the specified filename for streamed logging
57 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000058 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip4600f112005-03-13 09:56:36 +000059 self.mode = mode
60 self.encoding = encoding
Vinay Sajip23b94d02012-01-04 12:02:26 +000061 self.namer = None
62 self.rotator = None
Guido van Rossum57102f82002-11-13 16:15:58 +000063
Vinay Sajip17c52d82004-07-03 11:48:34 +000064 def emit(self, record):
65 """
66 Emit a record.
67
68 Output the record to the file, catering for rollover as described
69 in doRollover().
70 """
Vinay Sajip3970c112004-07-08 10:24:04 +000071 try:
72 if self.shouldRollover(record):
73 self.doRollover()
74 logging.FileHandler.emit(self, record)
Vinay Sajip985ef872011-04-26 19:34:04 +010075 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +000076 raise
Vinay Sajip3970c112004-07-08 10:24:04 +000077 except:
78 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000079
Vinay Sajip23b94d02012-01-04 12:02:26 +000080 def rotation_filename(self, default_name):
81 """
82 Modify the filename of a log file when rotating.
83
84 This is provided so that a custom filename can be provided.
85
86 The default implementation calls the 'namer' attribute of the
87 handler, if it's callable, passing the default name to
88 it. If the attribute isn't callable (the default is None), the name
89 is returned unchanged.
90
91 :param default_name: The default name for the log file.
92 """
93 if not callable(self.namer):
94 result = default_name
95 else:
96 result = self.namer(default_name)
97 return result
98
99 def rotate(self, source, dest):
100 """
101 When rotating, rotate the current log.
102
103 The default implementation calls the 'rotator' attribute of the
104 handler, if it's callable, passing the source and dest arguments to
105 it. If the attribute isn't callable (the default is None), the source
106 is simply renamed to the destination.
107
108 :param source: The source filename. This is normally the base
109 filename, e.g. 'test.log'
110 :param dest: The destination filename. This is normally
111 what the source is rotated to, e.g. 'test.log.1'.
112 """
113 if not callable(self.rotator):
114 os.rename(source, dest)
115 else:
116 self.rotator(source, dest)
117
Vinay Sajip17c52d82004-07-03 11:48:34 +0000118class RotatingFileHandler(BaseRotatingHandler):
119 """
120 Handler for logging to a set of files, which switches from one file
121 to the next when the current file reaches a certain size.
122 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000123 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
Guido van Rossum57102f82002-11-13 16:15:58 +0000124 """
125 Open the specified file and use it as the stream for logging.
126
127 By default, the file grows indefinitely. You can specify particular
128 values of maxBytes and backupCount to allow the file to rollover at
129 a predetermined size.
130
131 Rollover occurs whenever the current log file is nearly maxBytes in
132 length. If backupCount is >= 1, the system will successively create
133 new files with the same pathname as the base file, but with extensions
134 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
135 and a base file name of "app.log", you would get "app.log",
136 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
137 written to is always "app.log" - when it gets filled up, it is closed
138 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
139 exist, then they are renamed to "app.log.2", "app.log.3" etc.
140 respectively.
141
142 If maxBytes is zero, rollover never occurs.
143 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000144 # If rotation/rollover is wanted, it doesn't make sense to use another
145 # mode. If for example 'w' were specified, then if there were multiple
146 # runs of the calling application, the logs from previous runs would be
147 # lost if the 'w' is respected, because the log file would be truncated
148 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000149 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000150 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000151 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000152 self.maxBytes = maxBytes
153 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000154
155 def doRollover(self):
156 """
157 Do a rollover, as described in __init__().
158 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000159 if self.stream:
160 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000161 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000162 if self.backupCount > 0:
163 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000164 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
165 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
166 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000167 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000168 if os.path.exists(dfn):
169 os.remove(dfn)
170 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000171 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000172 if os.path.exists(dfn):
173 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000174 self.rotate(self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000175 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000176
Vinay Sajip17c52d82004-07-03 11:48:34 +0000177 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000178 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000179 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000180
Vinay Sajip17c52d82004-07-03 11:48:34 +0000181 Basically, see if the supplied record would cause the file to exceed
182 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000183 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000184 if self.stream is None: # delay was set...
185 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000186 if self.maxBytes > 0: # are we rolling over?
187 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000188 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000189 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000190 return 1
191 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000192
Vinay Sajip17c52d82004-07-03 11:48:34 +0000193class TimedRotatingFileHandler(BaseRotatingHandler):
194 """
195 Handler for logging to a file, rotating the log file at certain timed
196 intervals.
197
198 If backupCount is > 0, when rollover is done, no more than backupCount
199 files are kept - the oldest ones are deleted.
200 """
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000201 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000202 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000203 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000204 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000205 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000206 # Calculate the real rollover interval, which is just the number of
207 # seconds between rollovers. Also set the filename suffix used when
208 # a rollover occurs. Current 'when' events supported:
209 # S - Seconds
210 # M - Minutes
211 # H - Hours
212 # D - Days
213 # midnight - roll over at midnight
214 # W{0-6} - roll over on a certain day; 0 - Monday
215 #
216 # Case of the 'when' specifier is not important; lower or upper case
217 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000218 if self.when == 'S':
219 self.interval = 1 # one second
220 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000221 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000222 elif self.when == 'M':
223 self.interval = 60 # one minute
224 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000225 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000226 elif self.when == 'H':
227 self.interval = 60 * 60 # one hour
228 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000229 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000230 elif self.when == 'D' or self.when == 'MIDNIGHT':
231 self.interval = 60 * 60 * 24 # one day
232 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000233 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000234 elif self.when.startswith('W'):
235 self.interval = 60 * 60 * 24 * 7 # one week
236 if len(self.when) != 2:
237 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
238 if self.when[1] < '0' or self.when[1] > '6':
239 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
240 self.dayOfWeek = int(self.when[1])
241 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000242 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000243 else:
244 raise ValueError("Invalid rollover interval specified: %s" % self.when)
245
Antoine Pitroufd036452008-08-19 17:56:33 +0000246 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000247 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000248 if os.path.exists(filename):
249 t = os.stat(filename)[ST_MTIME]
250 else:
251 t = int(time.time())
252 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000253
Vinay Sajipaffbd872009-06-11 10:11:47 +0000254 def computeRollover(self, currentTime):
255 """
256 Work out the rollover time based on the specified time.
257 """
258 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000259 # If we are rolling over at midnight or weekly, then the interval is already known.
260 # What we need to figure out is WHEN the next interval is. In other words,
261 # if you are rolling over at midnight, then your base interval is 1 day,
262 # but you want to start that one day clock at midnight, not now. So, we
263 # have to fudge the rolloverAt value in order to trigger the first rollover
264 # at the right time. After that, the regular interval will take care of
265 # the rest. Note that this code doesn't care about leap seconds. :)
266 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
267 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000268 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000269 t = time.gmtime(currentTime)
270 else:
271 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000272 currentHour = t[3]
273 currentMinute = t[4]
274 currentSecond = t[5]
275 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000276 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
277 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000278 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000279 # If we are rolling over on a certain day, add in the number of days until
280 # the next rollover, but offset by 1 since we just calculated the time
281 # until the next day starts. There are three cases:
282 # Case 1) The day to rollover is today; in this case, do nothing
283 # Case 2) The day to rollover is further in the interval (i.e., today is
284 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
285 # next rollover is simply 6 - 2 - 1, or 3.
286 # Case 3) The day to rollover is behind us in the interval (i.e., today
287 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
288 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
289 # number of days left in the current week (1) plus the number
290 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000291 # The calculations described in 2) and 3) above need to have a day added.
292 # This is because the above time calculation takes us to midnight on this
293 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000294 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000295 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000296 if day != self.dayOfWeek:
297 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000298 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000299 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000300 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000301 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
302 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000303 dstNow = t[-1]
304 dstAtRollover = time.localtime(newRolloverAt)[-1]
305 if dstNow != dstAtRollover:
306 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000307 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000308 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000309 addend = 3600
310 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000311 result = newRolloverAt
312 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000313
314 def shouldRollover(self, record):
315 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000316 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000317
318 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000319 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000320 """
321 t = int(time.time())
322 if t >= self.rolloverAt:
323 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000324 return 0
325
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000326 def getFilesToDelete(self):
327 """
328 Determine the files to delete when rolling over.
329
330 More specific than the earlier method, which just used glob.glob().
331 """
332 dirName, baseName = os.path.split(self.baseFilename)
333 fileNames = os.listdir(dirName)
334 result = []
335 prefix = baseName + "."
336 plen = len(prefix)
337 for fileName in fileNames:
338 if fileName[:plen] == prefix:
339 suffix = fileName[plen:]
340 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000341 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000342 result.sort()
343 if len(result) < self.backupCount:
344 result = []
345 else:
346 result = result[:len(result) - self.backupCount]
347 return result
348
Vinay Sajip17c52d82004-07-03 11:48:34 +0000349 def doRollover(self):
350 """
351 do a rollover; in this case, a date/time stamp is appended to the filename
352 when the rollover happens. However, you want the file to be named for the
353 start of the interval, not the current time. If there is a backup count,
354 then we have to get a list of matching filenames, sort them and remove
355 the one with the oldest suffix.
356 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000357 if self.stream:
358 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000359 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000360 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000361 currentTime = int(time.time())
362 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000363 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000364 if self.utc:
365 timeTuple = time.gmtime(t)
366 else:
367 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000368 dstThen = timeTuple[-1]
369 if dstNow != dstThen:
370 if dstNow:
371 addend = 3600
372 else:
373 addend = -3600
374 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000375 dfn = self.rotation_filename(self.baseFilename + "." +
376 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000377 if os.path.exists(dfn):
378 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000379 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000380 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000381 for s in self.getFilesToDelete():
382 os.remove(s)
Thomas Woutersb2137042007-02-01 18:02:27 +0000383 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000384 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000385 while newRolloverAt <= currentTime:
386 newRolloverAt = newRolloverAt + self.interval
387 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000388 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000389 dstAtRollover = time.localtime(newRolloverAt)[-1]
390 if dstNow != dstAtRollover:
391 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000392 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000393 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000394 addend = 3600
395 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000396 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000397
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000398class WatchedFileHandler(logging.FileHandler):
399 """
400 A handler for logging to a file, which watches the file
401 to see if it has changed while in use. This can happen because of
402 usage of programs such as newsyslog and logrotate which perform
403 log file rotation. This handler, intended for use under Unix,
404 watches the file to see if it has changed since the last emit.
405 (A file has changed if its device or inode have changed.)
406 If it has changed, the old file stream is closed, and the file
407 opened to get a new stream.
408
409 This handler is not appropriate for use under Windows, because
410 under Windows open files cannot be moved or renamed - logging
411 opens the files with exclusive locks - and so there is no need
412 for such a handler. Furthermore, ST_INO is not supported under
413 Windows; stat always returns zero for this value.
414
415 This handler is based on a suggestion and patch by Chad J.
416 Schroeder.
417 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000418 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000419 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100420 self.dev, self.ino = -1, -1
421 self._statstream()
422
423 def _statstream(self):
424 if self.stream:
425 sres = os.fstat(self.stream.fileno())
426 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000427
428 def emit(self, record):
429 """
430 Emit a record.
431
432 First check if the underlying file has changed, and if it
433 has, close the old stream and reopen the file to get the
434 current stream.
435 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100436 # Reduce the chance of race conditions by stat'ing by path only
437 # once and then fstat'ing our new fd if we opened a new log stream.
438 # See issue #14632: Thanks to John Mulligan for the problem report
439 # and patch.
440 try:
441 # stat the file by path, checking for existence
442 sres = os.stat(self.baseFilename)
443 except OSError as err:
444 if err.errno == errno.ENOENT:
445 sres = None
446 else:
447 raise
448 # compare file system stat with that of our stream file handle
449 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
450 if self.stream is not None:
451 # we have an open file handle, clean it up
452 self.stream.flush()
453 self.stream.close()
454 # open a new file handle and get new stat info from that fd
455 self.stream = self._open()
456 self._statstream()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000457 logging.FileHandler.emit(self, record)
458
Vinay Sajip66b8b082012-04-24 23:25:30 +0100459
Guido van Rossum57102f82002-11-13 16:15:58 +0000460class SocketHandler(logging.Handler):
461 """
462 A handler class which writes logging records, in pickle format, to
463 a streaming socket. The socket is kept open across logging calls.
464 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000465 The pickle which is sent is that of the LogRecord's attribute dictionary
466 (__dict__), so that the receiver does not need to have the logging module
467 installed in order to process the logging event.
468
469 To unpickle the record at the receiving end into a LogRecord, use the
470 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000471 """
472
473 def __init__(self, host, port):
474 """
475 Initializes the handler with a specific host address and port.
476
Vinay Sajipde19e082011-04-30 21:52:26 +0100477 When the attribute *closeOnError* is set to True - if a socket error
478 occurs, the socket is silently closed and then reopened on the next
479 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000480 """
481 logging.Handler.__init__(self)
482 self.host = host
483 self.port = port
484 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100485 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000486 self.retryTime = None
487 #
488 # Exponential backoff parameters.
489 #
490 self.retryStart = 1.0
491 self.retryMax = 30.0
492 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000493
Guido van Rossumd8faa362007-04-27 19:54:29 +0000494 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000495 """
496 A factory method which allows subclasses to define the precise
497 type of socket they want.
498 """
499 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000500 if hasattr(s, 'settimeout'):
501 s.settimeout(timeout)
Vinay Sajipb37cd392011-05-07 15:55:47 +0100502 try:
503 s.connect((self.host, self.port))
504 return s
505 except socket.error:
506 s.close()
507 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000508
Vinay Sajip48cfe382004-02-20 13:17:27 +0000509 def createSocket(self):
510 """
511 Try to create a socket, using an exponential backoff with
512 a max retry time. Thanks to Robert Olson for the original patch
513 (SF #815911) which has been slightly refactored.
514 """
515 now = time.time()
516 # Either retryTime is None, in which case this
517 # is the first time back after a disconnect, or
518 # we've waited long enough.
519 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100520 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000521 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000522 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000523 if attempt:
524 try:
525 self.sock = self.makeSocket()
526 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000527 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000528 #Creation failed, so set the retry time and return.
529 if self.retryTime is None:
530 self.retryPeriod = self.retryStart
531 else:
532 self.retryPeriod = self.retryPeriod * self.retryFactor
533 if self.retryPeriod > self.retryMax:
534 self.retryPeriod = self.retryMax
535 self.retryTime = now + self.retryPeriod
536
Guido van Rossum57102f82002-11-13 16:15:58 +0000537 def send(self, s):
538 """
539 Send a pickled string to the socket.
540
541 This function allows for partial sends which can happen when the
542 network is busy.
543 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000544 if self.sock is None:
545 self.createSocket()
546 #self.sock can be None either because we haven't reached the retry
547 #time yet, or because we have reached the retry time and retried,
548 #but are still unable to connect.
549 if self.sock:
550 try:
551 if hasattr(self.sock, "sendall"):
552 self.sock.sendall(s)
Vinay Sajip7fe1d512011-04-28 12:04:58 +0100553 else: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000554 sentsofar = 0
555 left = len(s)
556 while left > 0:
557 sent = self.sock.send(s[sentsofar:])
558 sentsofar = sentsofar + sent
559 left = left - sent
Vinay Sajip7fe1d512011-04-28 12:04:58 +0100560 except socket.error: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000561 self.sock.close()
562 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000563
564 def makePickle(self, record):
565 """
566 Pickles the record in binary format with a length prefix, and
567 returns it ready for transmission across the socket.
568 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000569 ei = record.exc_info
570 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100571 # just to get traceback text into record.exc_text ...
572 dummy = self.format(record)
573 # See issue #14436: If msg or args are objects, they may not be
574 # available on the receiving end. So we convert the msg % args
575 # to a string, save it as msg and zap the args.
576 d = dict(record.__dict__)
577 d['msg'] = record.getMessage()
578 d['args'] = None
579 d['exc_info'] = None
580 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000581 slen = struct.pack(">L", len(s))
582 return slen + s
583
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000584 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000585 """
586 Handle an error during logging.
587
588 An error has occurred during logging. Most likely cause -
589 connection lost. Close the socket so that we can retry on the
590 next event.
591 """
592 if self.closeOnError and self.sock:
593 self.sock.close()
594 self.sock = None #try to reconnect next time
595 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000596 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000597
598 def emit(self, record):
599 """
600 Emit a record.
601
602 Pickles the record and writes it to the socket in binary format.
603 If there is an error with the socket, silently drop the packet.
604 If there was a problem with the socket, re-establishes the
605 socket.
606 """
607 try:
608 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000609 self.send(s)
Vinay Sajip985ef872011-04-26 19:34:04 +0100610 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000611 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000612 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000613 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000614
615 def close(self):
616 """
617 Closes the socket.
618 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000619 self.acquire()
620 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000621 if self.sock:
622 self.sock.close()
623 self.sock = None
624 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000625 finally:
626 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000627
628class DatagramHandler(SocketHandler):
629 """
630 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000631 a datagram socket. The pickle which is sent is that of the LogRecord's
632 attribute dictionary (__dict__), so that the receiver does not need to
633 have the logging module installed in order to process the logging event.
634
635 To unpickle the record at the receiving end into a LogRecord, use the
636 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000637
638 """
639 def __init__(self, host, port):
640 """
641 Initializes the handler with a specific host address and port.
642 """
643 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100644 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000645
646 def makeSocket(self):
647 """
648 The factory method of SocketHandler is here overridden to create
649 a UDP socket (SOCK_DGRAM).
650 """
651 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
652 return s
653
654 def send(self, s):
655 """
656 Send a pickled string to a socket.
657
658 This function no longer allows for partial sends which can happen
659 when the network is busy - UDP does not guarantee delivery and
660 can deliver packets out of sequence.
661 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000662 if self.sock is None:
663 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000664 self.sock.sendto(s, (self.host, self.port))
665
666class SysLogHandler(logging.Handler):
667 """
668 A handler class which sends formatted logging records to a syslog
669 server. Based on Sam Rushing's syslog module:
670 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
671 Contributed by Nicolas Untz (after which minor refactoring changes
672 have been made).
673 """
674
675 # from <linux/sys/syslog.h>:
676 # ======================================================================
677 # priorities/facilities are encoded into a single 32-bit quantity, where
678 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
679 # facility (0-big number). Both the priorities and the facilities map
680 # roughly one-to-one to strings in the syslogd(8) source code. This
681 # mapping is included in this file.
682 #
683 # priorities (these are ordered)
684
685 LOG_EMERG = 0 # system is unusable
686 LOG_ALERT = 1 # action must be taken immediately
687 LOG_CRIT = 2 # critical conditions
688 LOG_ERR = 3 # error conditions
689 LOG_WARNING = 4 # warning conditions
690 LOG_NOTICE = 5 # normal but significant condition
691 LOG_INFO = 6 # informational
692 LOG_DEBUG = 7 # debug-level messages
693
694 # facility codes
695 LOG_KERN = 0 # kernel messages
696 LOG_USER = 1 # random user-level messages
697 LOG_MAIL = 2 # mail system
698 LOG_DAEMON = 3 # system daemons
699 LOG_AUTH = 4 # security/authorization messages
700 LOG_SYSLOG = 5 # messages generated internally by syslogd
701 LOG_LPR = 6 # line printer subsystem
702 LOG_NEWS = 7 # network news subsystem
703 LOG_UUCP = 8 # UUCP subsystem
704 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000705 LOG_AUTHPRIV = 10 # security/authorization messages (private)
706 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000707
708 # other codes through 15 reserved for system use
709 LOG_LOCAL0 = 16 # reserved for local use
710 LOG_LOCAL1 = 17 # reserved for local use
711 LOG_LOCAL2 = 18 # reserved for local use
712 LOG_LOCAL3 = 19 # reserved for local use
713 LOG_LOCAL4 = 20 # reserved for local use
714 LOG_LOCAL5 = 21 # reserved for local use
715 LOG_LOCAL6 = 22 # reserved for local use
716 LOG_LOCAL7 = 23 # reserved for local use
717
718 priority_names = {
719 "alert": LOG_ALERT,
720 "crit": LOG_CRIT,
721 "critical": LOG_CRIT,
722 "debug": LOG_DEBUG,
723 "emerg": LOG_EMERG,
724 "err": LOG_ERR,
725 "error": LOG_ERR, # DEPRECATED
726 "info": LOG_INFO,
727 "notice": LOG_NOTICE,
728 "panic": LOG_EMERG, # DEPRECATED
729 "warn": LOG_WARNING, # DEPRECATED
730 "warning": LOG_WARNING,
731 }
732
733 facility_names = {
734 "auth": LOG_AUTH,
735 "authpriv": LOG_AUTHPRIV,
736 "cron": LOG_CRON,
737 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000738 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000739 "kern": LOG_KERN,
740 "lpr": LOG_LPR,
741 "mail": LOG_MAIL,
742 "news": LOG_NEWS,
743 "security": LOG_AUTH, # DEPRECATED
744 "syslog": LOG_SYSLOG,
745 "user": LOG_USER,
746 "uucp": LOG_UUCP,
747 "local0": LOG_LOCAL0,
748 "local1": LOG_LOCAL1,
749 "local2": LOG_LOCAL2,
750 "local3": LOG_LOCAL3,
751 "local4": LOG_LOCAL4,
752 "local5": LOG_LOCAL5,
753 "local6": LOG_LOCAL6,
754 "local7": LOG_LOCAL7,
755 }
756
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000757 #The map below appears to be trivially lowercasing the key. However,
758 #there's more to it than meets the eye - in some locales, lowercasing
759 #gives unexpected results. See SF #1524081: in the Turkish locale,
760 #"INFO".lower() != "info"
761 priority_map = {
762 "DEBUG" : "debug",
763 "INFO" : "info",
764 "WARNING" : "warning",
765 "ERROR" : "error",
766 "CRITICAL" : "critical"
767 }
768
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000769 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
770 facility=LOG_USER, socktype=socket.SOCK_DGRAM):
Guido van Rossum57102f82002-11-13 16:15:58 +0000771 """
772 Initialize a handler.
773
Guido van Rossume7ba4952007-06-06 23:52:48 +0000774 If address is specified as a string, a UNIX socket is used. To log to a
775 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000776 If facility is not specified, LOG_USER is used.
777 """
778 logging.Handler.__init__(self)
779
780 self.address = address
781 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000782 self.socktype = socktype
783
Guido van Rossum13257902007-06-07 23:15:56 +0000784 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100785 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000786 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000787 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100788 self.unixsocket = False
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000789 self.socket = socket.socket(socket.AF_INET, socktype)
790 if socktype == socket.SOCK_STREAM:
791 self.socket.connect(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000792 self.formatter = None
793
Vinay Sajipa1974c12005-01-13 08:23:56 +0000794 def _connect_unixsocket(self, address):
Vinay Sajipb5ca9322012-10-09 08:19:25 +0100795 self.socket = socket.socket(socket.AF_UNIX, self.socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000796 try:
797 self.socket.connect(address)
798 except socket.error:
799 self.socket.close()
Vinay Sajipb5ca9322012-10-09 08:19:25 +0100800 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000801
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000802 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000803 """
804 Encode the facility and priority. You can pass in strings or
805 integers - if strings are passed, the facility_names and
806 priority_names mapping dictionaries are used to convert them to
807 integers.
808 """
Guido van Rossum13257902007-06-07 23:15:56 +0000809 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000810 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000811 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000812 priority = self.priority_names[priority]
813 return (facility << 3) | priority
814
815 def close (self):
816 """
817 Closes the socket.
818 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000819 self.acquire()
820 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000821 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000822 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000823 finally:
824 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000825
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000826 def mapPriority(self, levelName):
827 """
828 Map a logging level name to a key in the priority_names map.
829 This is useful in two scenarios: when custom levels are being
830 used, and in the case where you can't do a straightforward
831 mapping by lowercasing the logging level name because of locale-
832 specific issues (see SF #1524081).
833 """
834 return self.priority_map.get(levelName, "warning")
835
Vinay Sajip2353e352011-06-27 15:40:06 +0100836 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100837 append_nul = True # some old syslog daemons expect a NUL terminator
838
Guido van Rossum57102f82002-11-13 16:15:58 +0000839 def emit(self, record):
840 """
841 Emit a record.
842
843 The record is formatted, and then sent to the syslog server. If
844 exception information is present, it is NOT sent to the server.
845 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100846 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100847 if self.ident:
848 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100849 if self.append_nul:
850 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000851 """
852 We need to convert record level to lowercase, maybe this will
853 change in the future.
854 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000855 prio = '<%d>' % self.encodePriority(self.facility,
856 self.mapPriority(record.levelname))
857 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000858 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000859 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000860 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000861 try:
862 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000863 try:
864 self.socket.send(msg)
865 except socket.error:
866 self._connect_unixsocket(self.address)
867 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000868 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000869 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000870 else:
871 self.socket.sendall(msg)
Vinay Sajip985ef872011-04-26 19:34:04 +0100872 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000873 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000874 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000875 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000876
877class SMTPHandler(logging.Handler):
878 """
879 A handler class which sends an SMTP email for each logging event.
880 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000881 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100882 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000883 """
884 Initialize the handler.
885
886 Initialize the instance with the from and to addresses and subject
887 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000888 (host, port) tuple format for the mailhost argument. To specify
889 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000890 for the credentials argument. To specify the use of a secure
891 protocol (TLS), pass in a tuple for the secure argument. This will
892 only be used when authentication credentials are supplied. The tuple
893 will be either an empty tuple, or a single-value tuple with the name
894 of a keyfile, or a 2-value tuple with the names of the keyfile and
895 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000896 A timeout in seconds can be specified for the SMTP connection (the
897 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000898 """
899 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000900 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000901 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000902 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000903 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000904 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000905 self.username, self.password = credentials
906 else:
907 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000908 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000909 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000910 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000911 self.toaddrs = toaddrs
912 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000913 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000914 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000915
916 def getSubject(self, record):
917 """
918 Determine the subject for the email.
919
920 If you want to specify a subject line which is record-dependent,
921 override this method.
922 """
923 return self.subject
924
925 def emit(self, record):
926 """
927 Emit a record.
928
929 Format the record and send it to the specified addressees.
930 """
931 try:
932 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000933 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000934 port = self.mailport
935 if not port:
936 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000937 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000938 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000939 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 +0000940 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000941 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000942 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000943 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000944 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000945 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000946 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000947 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000948 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000949 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000950 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
951 smtp.quit()
Vinay Sajip985ef872011-04-26 19:34:04 +0100952 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000953 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000954 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000955 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000956
957class NTEventLogHandler(logging.Handler):
958 """
959 A handler class which sends events to the NT Event Log. Adds a
960 registry entry for the specified application name. If no dllname is
961 provided, win32service.pyd (which contains some basic message
962 placeholders) is used. Note that use of these placeholders will make
963 your event logs big, as the entire message source is held in the log.
964 If you want slimmer logs, you have to pass in the name of your own DLL
965 which contains the message definitions you want to use in the event log.
966 """
967 def __init__(self, appname, dllname=None, logtype="Application"):
968 logging.Handler.__init__(self)
969 try:
970 import win32evtlogutil, win32evtlog
971 self.appname = appname
972 self._welu = win32evtlogutil
973 if not dllname:
974 dllname = os.path.split(self._welu.__file__)
975 dllname = os.path.split(dllname[0])
976 dllname = os.path.join(dllname[0], r'win32service.pyd')
977 self.dllname = dllname
978 self.logtype = logtype
979 self._welu.AddSourceToRegistry(appname, dllname, logtype)
980 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
981 self.typemap = {
982 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
983 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000984 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000985 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
986 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
987 }
988 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000989 print("The Python Win32 extensions for NT (service, event "\
990 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000991 self._welu = None
992
993 def getMessageID(self, record):
994 """
995 Return the message ID for the event record. If you are using your
996 own messages, you could do this by having the msg passed to the
997 logger being an ID rather than a formatting string. Then, in here,
998 you could use a dictionary lookup to get the message ID. This
999 version returns 1, which is the base message ID in win32service.pyd.
1000 """
1001 return 1
1002
1003 def getEventCategory(self, record):
1004 """
1005 Return the event category for the record.
1006
1007 Override this if you want to specify your own categories. This version
1008 returns 0.
1009 """
1010 return 0
1011
1012 def getEventType(self, record):
1013 """
1014 Return the event type for the record.
1015
1016 Override this if you want to specify your own types. This version does
1017 a mapping using the handler's typemap attribute, which is set up in
1018 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001019 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001020 either need to override this method or place a suitable dictionary in
1021 the handler's typemap attribute.
1022 """
1023 return self.typemap.get(record.levelno, self.deftype)
1024
1025 def emit(self, record):
1026 """
1027 Emit a record.
1028
1029 Determine the message ID, event category and event type. Then
1030 log the message in the NT event log.
1031 """
1032 if self._welu:
1033 try:
1034 id = self.getMessageID(record)
1035 cat = self.getEventCategory(record)
1036 type = self.getEventType(record)
1037 msg = self.format(record)
1038 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip985ef872011-04-26 19:34:04 +01001039 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001040 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001041 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001042 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001043
1044 def close(self):
1045 """
1046 Clean up this handler.
1047
1048 You can remove the application name from the registry as a
1049 source of event log entries. However, if you do this, you will
1050 not be able to see the events as you intended in the Event Log
1051 Viewer - it needs to be able to access the registry to get the
1052 DLL name.
1053 """
1054 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001055 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001056
1057class HTTPHandler(logging.Handler):
1058 """
1059 A class which sends records to a Web server, using either GET or
1060 POST semantics.
1061 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001062 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001063 """
1064 Initialize the instance with the host, the request URL, and the method
1065 ("GET" or "POST")
1066 """
1067 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001068 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001069 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001070 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001071 self.host = host
1072 self.url = url
1073 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001074 self.secure = secure
1075 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001076
Neal Norwitzf297bd12003-04-23 03:49:43 +00001077 def mapLogRecord(self, record):
1078 """
1079 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001080 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001081 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001082 """
1083 return record.__dict__
1084
Guido van Rossum57102f82002-11-13 16:15:58 +00001085 def emit(self, record):
1086 """
1087 Emit a record.
1088
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001089 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001090 """
1091 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001092 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001093 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001094 if self.secure:
1095 h = http.client.HTTPSConnection(host)
1096 else:
1097 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001098 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001099 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001100 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001101 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001102 sep = '&'
1103 else:
1104 sep = '?'
1105 url = url + "%c%s" % (sep, data)
1106 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001107 # support multiple hosts on one IP address...
1108 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001109 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001110 if i >= 0:
1111 host = host[:i]
1112 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001113 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001114 h.putheader("Content-type",
1115 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001116 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001117 if self.credentials:
1118 import base64
1119 s = ('u%s:%s' % self.credentials).encode('utf-8')
1120 s = 'Basic ' + base64.b64encode(s).strip()
1121 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001122 h.endheaders()
1123 if self.method == "POST":
1124 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001125 h.getresponse() #can't do anything with the result
Vinay Sajip985ef872011-04-26 19:34:04 +01001126 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001127 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001128 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001129 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001130
1131class BufferingHandler(logging.Handler):
1132 """
1133 A handler class which buffers logging records in memory. Whenever each
1134 record is added to the buffer, a check is made to see if the buffer should
1135 be flushed. If it should, then flush() is expected to do what's needed.
1136 """
1137 def __init__(self, capacity):
1138 """
1139 Initialize the handler with the buffer size.
1140 """
1141 logging.Handler.__init__(self)
1142 self.capacity = capacity
1143 self.buffer = []
1144
1145 def shouldFlush(self, record):
1146 """
1147 Should the handler flush its buffer?
1148
1149 Returns true if the buffer is up to capacity. This method can be
1150 overridden to implement custom flushing strategies.
1151 """
1152 return (len(self.buffer) >= self.capacity)
1153
1154 def emit(self, record):
1155 """
1156 Emit a record.
1157
1158 Append the record. If shouldFlush() tells us to, call flush() to process
1159 the buffer.
1160 """
1161 self.buffer.append(record)
1162 if self.shouldFlush(record):
1163 self.flush()
1164
1165 def flush(self):
1166 """
1167 Override to implement custom flushing behaviour.
1168
1169 This version just zaps the buffer to empty.
1170 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001171 self.acquire()
1172 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001173 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001174 finally:
1175 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001176
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001177 def close(self):
1178 """
1179 Close the handler.
1180
1181 This version just flushes and chains to the parent class' close().
1182 """
1183 self.flush()
1184 logging.Handler.close(self)
1185
Guido van Rossum57102f82002-11-13 16:15:58 +00001186class MemoryHandler(BufferingHandler):
1187 """
1188 A handler class which buffers logging records in memory, periodically
1189 flushing them to a target handler. Flushing occurs whenever the buffer
1190 is full, or when an event of a certain severity or greater is seen.
1191 """
1192 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1193 """
1194 Initialize the handler with the buffer size, the level at which
1195 flushing should occur and an optional target.
1196
1197 Note that without a target being set either here or via setTarget(),
1198 a MemoryHandler is no use to anyone!
1199 """
1200 BufferingHandler.__init__(self, capacity)
1201 self.flushLevel = flushLevel
1202 self.target = target
1203
1204 def shouldFlush(self, record):
1205 """
1206 Check for buffer full or a record at the flushLevel or higher.
1207 """
1208 return (len(self.buffer) >= self.capacity) or \
1209 (record.levelno >= self.flushLevel)
1210
1211 def setTarget(self, target):
1212 """
1213 Set the target handler for this handler.
1214 """
1215 self.target = target
1216
1217 def flush(self):
1218 """
1219 For a MemoryHandler, flushing means just sending the buffered
1220 records to the target, if there is one. Override if you want
1221 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001222
1223 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001224 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001225 self.acquire()
1226 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001227 if self.target:
1228 for record in self.buffer:
1229 self.target.handle(record)
1230 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001231 finally:
1232 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001233
1234 def close(self):
1235 """
1236 Flush, set the target to None and lose the buffer.
1237 """
1238 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001239 self.acquire()
1240 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001241 self.target = None
1242 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001243 finally:
1244 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001245
1246
1247class QueueHandler(logging.Handler):
1248 """
1249 This handler sends events to a queue. Typically, it would be used together
1250 with a multiprocessing Queue to centralise logging to file in one process
1251 (in a multi-process application), so as to avoid file write contention
1252 between processes.
1253
1254 This code is new in Python 3.2, but this class can be copy pasted into
1255 user code for use with earlier Python versions.
1256 """
1257
1258 def __init__(self, queue):
1259 """
1260 Initialise an instance, using the passed queue.
1261 """
1262 logging.Handler.__init__(self)
1263 self.queue = queue
1264
1265 def enqueue(self, record):
1266 """
1267 Enqueue a record.
1268
1269 The base implementation uses put_nowait. You may want to override
1270 this method if you want to use blocking, timeouts or custom queue
1271 implementations.
1272 """
1273 self.queue.put_nowait(record)
1274
Vinay Sajip0258ce82010-09-22 20:34:53 +00001275 def prepare(self, record):
1276 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001277 Prepares a record for queuing. The object returned by this method is
1278 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001279
1280 The base implementation formats the record to merge the message
1281 and arguments, and removes unpickleable items from the record
1282 in-place.
1283
1284 You might want to override this method if you want to convert
1285 the record to a dict or JSON string, or send a modified copy
1286 of the record while leaving the original intact.
1287 """
1288 # The format operation gets traceback text into record.exc_text
1289 # (if there's exception data), and also puts the message into
1290 # record.message. We can then use this to replace the original
1291 # msg + args, as these might be unpickleable. We also zap the
1292 # exc_info attribute, as it's no longer needed and, if not None,
1293 # will typically not be pickleable.
1294 self.format(record)
1295 record.msg = record.message
1296 record.args = None
1297 record.exc_info = None
1298 return record
1299
Vinay Sajip121a1c42010-09-08 10:46:15 +00001300 def emit(self, record):
1301 """
1302 Emit a record.
1303
Vinay Sajip0637d492010-09-23 08:15:54 +00001304 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001305 """
1306 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001307 self.enqueue(self.prepare(record))
Vinay Sajip985ef872011-04-26 19:34:04 +01001308 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip121a1c42010-09-08 10:46:15 +00001309 raise
1310 except:
1311 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001312
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001313if threading:
1314 class QueueListener(object):
1315 """
1316 This class implements an internal threaded listener which watches for
1317 LogRecords being added to a queue, removes them and passes them to a
1318 list of handlers for processing.
1319 """
1320 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001321
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001322 def __init__(self, queue, *handlers):
1323 """
1324 Initialise an instance with the specified queue and
1325 handlers.
1326 """
1327 self.queue = queue
1328 self.handlers = handlers
1329 self._stop = threading.Event()
1330 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001331
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001332 def dequeue(self, block):
1333 """
1334 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001335
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001336 The base implementation uses get. You may want to override this method
1337 if you want to use timeouts or work with custom queue implementations.
1338 """
1339 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001340
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001341 def start(self):
1342 """
1343 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001344
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001345 This starts up a background thread to monitor the queue for
1346 LogRecords to process.
1347 """
1348 self._thread = t = threading.Thread(target=self._monitor)
1349 t.setDaemon(True)
1350 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001351
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001352 def prepare(self , record):
1353 """
1354 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001355
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001356 This method just returns the passed-in record. You may want to
1357 override this method if you need to do any custom marshalling or
1358 manipulation of the record before passing it to the handlers.
1359 """
1360 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001361
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001362 def handle(self, record):
1363 """
1364 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001365
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001366 This just loops through the handlers offering them the record
1367 to handle.
1368 """
1369 record = self.prepare(record)
1370 for handler in self.handlers:
1371 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001372
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001373 def _monitor(self):
1374 """
1375 Monitor the queue for records, and ask the handler
1376 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001377
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001378 This method runs on a separate, internal thread.
1379 The thread will terminate if it sees a sentinel object in the queue.
1380 """
1381 q = self.queue
1382 has_task_done = hasattr(q, 'task_done')
1383 while not self._stop.isSet():
1384 try:
1385 record = self.dequeue(True)
1386 if record is self._sentinel:
1387 break
1388 self.handle(record)
1389 if has_task_done:
1390 q.task_done()
1391 except queue.Empty:
1392 pass
1393 # There might still be records in the queue.
1394 while True:
1395 try:
1396 record = self.dequeue(False)
1397 if record is self._sentinel:
1398 break
1399 self.handle(record)
1400 if has_task_done:
1401 q.task_done()
1402 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001403 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001404
Victor Stinner59bec362011-05-02 16:14:16 +02001405 def enqueue_sentinel(self):
1406 """
1407 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001408
Victor Stinner59bec362011-05-02 16:14:16 +02001409 The base implementation uses put_nowait. You may want to override this
1410 method if you want to use timeouts or work with custom queue
1411 implementations.
1412 """
1413 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001414
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001415 def stop(self):
1416 """
1417 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001418
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001419 This asks the thread to terminate, and then waits for it to do so.
1420 Note that if you don't call this before your application exits, there
1421 may be some records still left on the queue, which won't be processed.
1422 """
1423 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001424 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001425 self._thread.join()
1426 self._thread = None