blob: 1bad8560f1ed612f8556f5447c23c916b988b09e [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.mode = 'w'
176 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000177
Vinay Sajip17c52d82004-07-03 11:48:34 +0000178 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000179 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000180 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000181
Vinay Sajip17c52d82004-07-03 11:48:34 +0000182 Basically, see if the supplied record would cause the file to exceed
183 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000184 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000185 if self.stream is None: # delay was set...
186 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000187 if self.maxBytes > 0: # are we rolling over?
188 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000189 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000190 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000191 return 1
192 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000193
Vinay Sajip17c52d82004-07-03 11:48:34 +0000194class TimedRotatingFileHandler(BaseRotatingHandler):
195 """
196 Handler for logging to a file, rotating the log file at certain timed
197 intervals.
198
199 If backupCount is > 0, when rollover is done, no more than backupCount
200 files are kept - the oldest ones are deleted.
201 """
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000202 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000203 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000204 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000205 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000206 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000207 # Calculate the real rollover interval, which is just the number of
208 # seconds between rollovers. Also set the filename suffix used when
209 # a rollover occurs. Current 'when' events supported:
210 # S - Seconds
211 # M - Minutes
212 # H - Hours
213 # D - Days
214 # midnight - roll over at midnight
215 # W{0-6} - roll over on a certain day; 0 - Monday
216 #
217 # Case of the 'when' specifier is not important; lower or upper case
218 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000219 if self.when == 'S':
220 self.interval = 1 # one second
221 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000222 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000223 elif self.when == 'M':
224 self.interval = 60 # one minute
225 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000226 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000227 elif self.when == 'H':
228 self.interval = 60 * 60 # one hour
229 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000230 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000231 elif self.when == 'D' or self.when == 'MIDNIGHT':
232 self.interval = 60 * 60 * 24 # one day
233 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000234 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000235 elif self.when.startswith('W'):
236 self.interval = 60 * 60 * 24 * 7 # one week
237 if len(self.when) != 2:
238 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
239 if self.when[1] < '0' or self.when[1] > '6':
240 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
241 self.dayOfWeek = int(self.when[1])
242 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000243 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000244 else:
245 raise ValueError("Invalid rollover interval specified: %s" % self.when)
246
Antoine Pitroufd036452008-08-19 17:56:33 +0000247 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000248 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000249 if os.path.exists(filename):
250 t = os.stat(filename)[ST_MTIME]
251 else:
252 t = int(time.time())
253 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000254
Vinay Sajipaffbd872009-06-11 10:11:47 +0000255 def computeRollover(self, currentTime):
256 """
257 Work out the rollover time based on the specified time.
258 """
259 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000260 # If we are rolling over at midnight or weekly, then the interval is already known.
261 # What we need to figure out is WHEN the next interval is. In other words,
262 # if you are rolling over at midnight, then your base interval is 1 day,
263 # but you want to start that one day clock at midnight, not now. So, we
264 # have to fudge the rolloverAt value in order to trigger the first rollover
265 # at the right time. After that, the regular interval will take care of
266 # the rest. Note that this code doesn't care about leap seconds. :)
267 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
268 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000269 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000270 t = time.gmtime(currentTime)
271 else:
272 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000273 currentHour = t[3]
274 currentMinute = t[4]
275 currentSecond = t[5]
276 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000277 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
278 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000279 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000280 # If we are rolling over on a certain day, add in the number of days until
281 # the next rollover, but offset by 1 since we just calculated the time
282 # until the next day starts. There are three cases:
283 # Case 1) The day to rollover is today; in this case, do nothing
284 # Case 2) The day to rollover is further in the interval (i.e., today is
285 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
286 # next rollover is simply 6 - 2 - 1, or 3.
287 # Case 3) The day to rollover is behind us in the interval (i.e., today
288 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
289 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
290 # number of days left in the current week (1) plus the number
291 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000292 # The calculations described in 2) and 3) above need to have a day added.
293 # This is because the above time calculation takes us to midnight on this
294 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000295 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000296 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000297 if day != self.dayOfWeek:
298 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000299 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000300 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000301 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000302 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
303 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000304 dstNow = t[-1]
305 dstAtRollover = time.localtime(newRolloverAt)[-1]
306 if dstNow != dstAtRollover:
307 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000308 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000309 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000310 addend = 3600
311 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000312 result = newRolloverAt
313 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000314
315 def shouldRollover(self, record):
316 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000317 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000318
319 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000320 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000321 """
322 t = int(time.time())
323 if t >= self.rolloverAt:
324 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000325 return 0
326
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000327 def getFilesToDelete(self):
328 """
329 Determine the files to delete when rolling over.
330
331 More specific than the earlier method, which just used glob.glob().
332 """
333 dirName, baseName = os.path.split(self.baseFilename)
334 fileNames = os.listdir(dirName)
335 result = []
336 prefix = baseName + "."
337 plen = len(prefix)
338 for fileName in fileNames:
339 if fileName[:plen] == prefix:
340 suffix = fileName[plen:]
341 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000342 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000343 result.sort()
344 if len(result) < self.backupCount:
345 result = []
346 else:
347 result = result[:len(result) - self.backupCount]
348 return result
349
Vinay Sajip17c52d82004-07-03 11:48:34 +0000350 def doRollover(self):
351 """
352 do a rollover; in this case, a date/time stamp is appended to the filename
353 when the rollover happens. However, you want the file to be named for the
354 start of the interval, not the current time. If there is a backup count,
355 then we have to get a list of matching filenames, sort them and remove
356 the one with the oldest suffix.
357 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000358 if self.stream:
359 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000360 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000361 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000362 currentTime = int(time.time())
363 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000364 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000365 if self.utc:
366 timeTuple = time.gmtime(t)
367 else:
368 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000369 dstThen = timeTuple[-1]
370 if dstNow != dstThen:
371 if dstNow:
372 addend = 3600
373 else:
374 addend = -3600
375 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000376 dfn = self.rotation_filename(self.baseFilename + "." +
377 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000378 if os.path.exists(dfn):
379 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000380 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000381 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000382 for s in self.getFilesToDelete():
383 os.remove(s)
Thomas Woutersb2137042007-02-01 18:02:27 +0000384 self.mode = 'w'
385 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000386 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000387 while newRolloverAt <= currentTime:
388 newRolloverAt = newRolloverAt + self.interval
389 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000390 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000391 dstAtRollover = time.localtime(newRolloverAt)[-1]
392 if dstNow != dstAtRollover:
393 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000394 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000395 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000396 addend = 3600
397 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000398 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000399
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000400class WatchedFileHandler(logging.FileHandler):
401 """
402 A handler for logging to a file, which watches the file
403 to see if it has changed while in use. This can happen because of
404 usage of programs such as newsyslog and logrotate which perform
405 log file rotation. This handler, intended for use under Unix,
406 watches the file to see if it has changed since the last emit.
407 (A file has changed if its device or inode have changed.)
408 If it has changed, the old file stream is closed, and the file
409 opened to get a new stream.
410
411 This handler is not appropriate for use under Windows, because
412 under Windows open files cannot be moved or renamed - logging
413 opens the files with exclusive locks - and so there is no need
414 for such a handler. Furthermore, ST_INO is not supported under
415 Windows; stat always returns zero for this value.
416
417 This handler is based on a suggestion and patch by Chad J.
418 Schroeder.
419 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000420 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000421 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100422 self.dev, self.ino = -1, -1
423 self._statstream()
424
425 def _statstream(self):
426 if self.stream:
427 sres = os.fstat(self.stream.fileno())
428 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000429
430 def emit(self, record):
431 """
432 Emit a record.
433
434 First check if the underlying file has changed, and if it
435 has, close the old stream and reopen the file to get the
436 current stream.
437 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100438 # Reduce the chance of race conditions by stat'ing by path only
439 # once and then fstat'ing our new fd if we opened a new log stream.
440 # See issue #14632: Thanks to John Mulligan for the problem report
441 # and patch.
442 try:
443 # stat the file by path, checking for existence
444 sres = os.stat(self.baseFilename)
445 except OSError as err:
446 if err.errno == errno.ENOENT:
447 sres = None
448 else:
449 raise
450 # compare file system stat with that of our stream file handle
451 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
452 if self.stream is not None:
453 # we have an open file handle, clean it up
454 self.stream.flush()
455 self.stream.close()
456 # open a new file handle and get new stat info from that fd
457 self.stream = self._open()
458 self._statstream()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000459 logging.FileHandler.emit(self, record)
460
Vinay Sajip66b8b082012-04-24 23:25:30 +0100461
Guido van Rossum57102f82002-11-13 16:15:58 +0000462class SocketHandler(logging.Handler):
463 """
464 A handler class which writes logging records, in pickle format, to
465 a streaming socket. The socket is kept open across logging calls.
466 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000467 The pickle which is sent is that of the LogRecord's attribute dictionary
468 (__dict__), so that the receiver does not need to have the logging module
469 installed in order to process the logging event.
470
471 To unpickle the record at the receiving end into a LogRecord, use the
472 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000473 """
474
475 def __init__(self, host, port):
476 """
477 Initializes the handler with a specific host address and port.
478
Vinay Sajipde19e082011-04-30 21:52:26 +0100479 When the attribute *closeOnError* is set to True - if a socket error
480 occurs, the socket is silently closed and then reopened on the next
481 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000482 """
483 logging.Handler.__init__(self)
484 self.host = host
485 self.port = port
486 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100487 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000488 self.retryTime = None
489 #
490 # Exponential backoff parameters.
491 #
492 self.retryStart = 1.0
493 self.retryMax = 30.0
494 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000495
Guido van Rossumd8faa362007-04-27 19:54:29 +0000496 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000497 """
498 A factory method which allows subclasses to define the precise
499 type of socket they want.
500 """
501 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000502 if hasattr(s, 'settimeout'):
503 s.settimeout(timeout)
Vinay Sajipb37cd392011-05-07 15:55:47 +0100504 try:
505 s.connect((self.host, self.port))
506 return s
507 except socket.error:
508 s.close()
509 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000510
Vinay Sajip48cfe382004-02-20 13:17:27 +0000511 def createSocket(self):
512 """
513 Try to create a socket, using an exponential backoff with
514 a max retry time. Thanks to Robert Olson for the original patch
515 (SF #815911) which has been slightly refactored.
516 """
517 now = time.time()
518 # Either retryTime is None, in which case this
519 # is the first time back after a disconnect, or
520 # we've waited long enough.
521 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100522 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000523 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000524 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000525 if attempt:
526 try:
527 self.sock = self.makeSocket()
528 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000529 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000530 #Creation failed, so set the retry time and return.
531 if self.retryTime is None:
532 self.retryPeriod = self.retryStart
533 else:
534 self.retryPeriod = self.retryPeriod * self.retryFactor
535 if self.retryPeriod > self.retryMax:
536 self.retryPeriod = self.retryMax
537 self.retryTime = now + self.retryPeriod
538
Guido van Rossum57102f82002-11-13 16:15:58 +0000539 def send(self, s):
540 """
541 Send a pickled string to the socket.
542
543 This function allows for partial sends which can happen when the
544 network is busy.
545 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000546 if self.sock is None:
547 self.createSocket()
548 #self.sock can be None either because we haven't reached the retry
549 #time yet, or because we have reached the retry time and retried,
550 #but are still unable to connect.
551 if self.sock:
552 try:
553 if hasattr(self.sock, "sendall"):
554 self.sock.sendall(s)
Vinay Sajip7fe1d512011-04-28 12:04:58 +0100555 else: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000556 sentsofar = 0
557 left = len(s)
558 while left > 0:
559 sent = self.sock.send(s[sentsofar:])
560 sentsofar = sentsofar + sent
561 left = left - sent
Vinay Sajip7fe1d512011-04-28 12:04:58 +0100562 except socket.error: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000563 self.sock.close()
564 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000565
566 def makePickle(self, record):
567 """
568 Pickles the record in binary format with a length prefix, and
569 returns it ready for transmission across the socket.
570 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000571 ei = record.exc_info
572 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100573 # just to get traceback text into record.exc_text ...
574 dummy = self.format(record)
575 # See issue #14436: If msg or args are objects, they may not be
576 # available on the receiving end. So we convert the msg % args
577 # to a string, save it as msg and zap the args.
578 d = dict(record.__dict__)
579 d['msg'] = record.getMessage()
580 d['args'] = None
581 d['exc_info'] = None
582 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000583 slen = struct.pack(">L", len(s))
584 return slen + s
585
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000586 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000587 """
588 Handle an error during logging.
589
590 An error has occurred during logging. Most likely cause -
591 connection lost. Close the socket so that we can retry on the
592 next event.
593 """
594 if self.closeOnError and self.sock:
595 self.sock.close()
596 self.sock = None #try to reconnect next time
597 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000598 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000599
600 def emit(self, record):
601 """
602 Emit a record.
603
604 Pickles the record and writes it to the socket in binary format.
605 If there is an error with the socket, silently drop the packet.
606 If there was a problem with the socket, re-establishes the
607 socket.
608 """
609 try:
610 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000611 self.send(s)
Vinay Sajip985ef872011-04-26 19:34:04 +0100612 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000613 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000614 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000615 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000616
617 def close(self):
618 """
619 Closes the socket.
620 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000621 self.acquire()
622 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000623 if self.sock:
624 self.sock.close()
625 self.sock = None
626 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000627 finally:
628 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000629
630class DatagramHandler(SocketHandler):
631 """
632 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000633 a datagram socket. The pickle which is sent is that of the LogRecord's
634 attribute dictionary (__dict__), so that the receiver does not need to
635 have the logging module installed in order to process the logging event.
636
637 To unpickle the record at the receiving end into a LogRecord, use the
638 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000639
640 """
641 def __init__(self, host, port):
642 """
643 Initializes the handler with a specific host address and port.
644 """
645 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100646 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000647
648 def makeSocket(self):
649 """
650 The factory method of SocketHandler is here overridden to create
651 a UDP socket (SOCK_DGRAM).
652 """
653 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
654 return s
655
656 def send(self, s):
657 """
658 Send a pickled string to a socket.
659
660 This function no longer allows for partial sends which can happen
661 when the network is busy - UDP does not guarantee delivery and
662 can deliver packets out of sequence.
663 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000664 if self.sock is None:
665 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000666 self.sock.sendto(s, (self.host, self.port))
667
668class SysLogHandler(logging.Handler):
669 """
670 A handler class which sends formatted logging records to a syslog
671 server. Based on Sam Rushing's syslog module:
672 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
673 Contributed by Nicolas Untz (after which minor refactoring changes
674 have been made).
675 """
676
677 # from <linux/sys/syslog.h>:
678 # ======================================================================
679 # priorities/facilities are encoded into a single 32-bit quantity, where
680 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
681 # facility (0-big number). Both the priorities and the facilities map
682 # roughly one-to-one to strings in the syslogd(8) source code. This
683 # mapping is included in this file.
684 #
685 # priorities (these are ordered)
686
687 LOG_EMERG = 0 # system is unusable
688 LOG_ALERT = 1 # action must be taken immediately
689 LOG_CRIT = 2 # critical conditions
690 LOG_ERR = 3 # error conditions
691 LOG_WARNING = 4 # warning conditions
692 LOG_NOTICE = 5 # normal but significant condition
693 LOG_INFO = 6 # informational
694 LOG_DEBUG = 7 # debug-level messages
695
696 # facility codes
697 LOG_KERN = 0 # kernel messages
698 LOG_USER = 1 # random user-level messages
699 LOG_MAIL = 2 # mail system
700 LOG_DAEMON = 3 # system daemons
701 LOG_AUTH = 4 # security/authorization messages
702 LOG_SYSLOG = 5 # messages generated internally by syslogd
703 LOG_LPR = 6 # line printer subsystem
704 LOG_NEWS = 7 # network news subsystem
705 LOG_UUCP = 8 # UUCP subsystem
706 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000707 LOG_AUTHPRIV = 10 # security/authorization messages (private)
708 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000709
710 # other codes through 15 reserved for system use
711 LOG_LOCAL0 = 16 # reserved for local use
712 LOG_LOCAL1 = 17 # reserved for local use
713 LOG_LOCAL2 = 18 # reserved for local use
714 LOG_LOCAL3 = 19 # reserved for local use
715 LOG_LOCAL4 = 20 # reserved for local use
716 LOG_LOCAL5 = 21 # reserved for local use
717 LOG_LOCAL6 = 22 # reserved for local use
718 LOG_LOCAL7 = 23 # reserved for local use
719
720 priority_names = {
721 "alert": LOG_ALERT,
722 "crit": LOG_CRIT,
723 "critical": LOG_CRIT,
724 "debug": LOG_DEBUG,
725 "emerg": LOG_EMERG,
726 "err": LOG_ERR,
727 "error": LOG_ERR, # DEPRECATED
728 "info": LOG_INFO,
729 "notice": LOG_NOTICE,
730 "panic": LOG_EMERG, # DEPRECATED
731 "warn": LOG_WARNING, # DEPRECATED
732 "warning": LOG_WARNING,
733 }
734
735 facility_names = {
736 "auth": LOG_AUTH,
737 "authpriv": LOG_AUTHPRIV,
738 "cron": LOG_CRON,
739 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000740 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000741 "kern": LOG_KERN,
742 "lpr": LOG_LPR,
743 "mail": LOG_MAIL,
744 "news": LOG_NEWS,
745 "security": LOG_AUTH, # DEPRECATED
746 "syslog": LOG_SYSLOG,
747 "user": LOG_USER,
748 "uucp": LOG_UUCP,
749 "local0": LOG_LOCAL0,
750 "local1": LOG_LOCAL1,
751 "local2": LOG_LOCAL2,
752 "local3": LOG_LOCAL3,
753 "local4": LOG_LOCAL4,
754 "local5": LOG_LOCAL5,
755 "local6": LOG_LOCAL6,
756 "local7": LOG_LOCAL7,
757 }
758
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000759 #The map below appears to be trivially lowercasing the key. However,
760 #there's more to it than meets the eye - in some locales, lowercasing
761 #gives unexpected results. See SF #1524081: in the Turkish locale,
762 #"INFO".lower() != "info"
763 priority_map = {
764 "DEBUG" : "debug",
765 "INFO" : "info",
766 "WARNING" : "warning",
767 "ERROR" : "error",
768 "CRITICAL" : "critical"
769 }
770
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000771 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
772 facility=LOG_USER, socktype=socket.SOCK_DGRAM):
Guido van Rossum57102f82002-11-13 16:15:58 +0000773 """
774 Initialize a handler.
775
Guido van Rossume7ba4952007-06-06 23:52:48 +0000776 If address is specified as a string, a UNIX socket is used. To log to a
777 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000778 If facility is not specified, LOG_USER is used.
779 """
780 logging.Handler.__init__(self)
781
782 self.address = address
783 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000784 self.socktype = socktype
785
Guido van Rossum13257902007-06-07 23:15:56 +0000786 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100787 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000788 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000789 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100790 self.unixsocket = False
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000791 self.socket = socket.socket(socket.AF_INET, socktype)
792 if socktype == socket.SOCK_STREAM:
793 self.socket.connect(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000794 self.formatter = None
795
Vinay Sajipa1974c12005-01-13 08:23:56 +0000796 def _connect_unixsocket(self, address):
797 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
798 # syslog may require either DGRAM or STREAM sockets
799 try:
800 self.socket.connect(address)
801 except socket.error:
802 self.socket.close()
803 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Vinay Sajip3e6d71d2012-06-25 23:18:45 +0100804 try:
805 self.socket.connect(address)
806 except socket.error:
807 self.socket.close()
808 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000809
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000810 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000811 """
812 Encode the facility and priority. You can pass in strings or
813 integers - if strings are passed, the facility_names and
814 priority_names mapping dictionaries are used to convert them to
815 integers.
816 """
Guido van Rossum13257902007-06-07 23:15:56 +0000817 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000818 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000819 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000820 priority = self.priority_names[priority]
821 return (facility << 3) | priority
822
823 def close (self):
824 """
825 Closes the socket.
826 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000827 self.acquire()
828 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000829 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000830 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000831 finally:
832 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000833
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000834 def mapPriority(self, levelName):
835 """
836 Map a logging level name to a key in the priority_names map.
837 This is useful in two scenarios: when custom levels are being
838 used, and in the case where you can't do a straightforward
839 mapping by lowercasing the logging level name because of locale-
840 specific issues (see SF #1524081).
841 """
842 return self.priority_map.get(levelName, "warning")
843
Vinay Sajip2353e352011-06-27 15:40:06 +0100844 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100845 append_nul = True # some old syslog daemons expect a NUL terminator
846
Guido van Rossum57102f82002-11-13 16:15:58 +0000847 def emit(self, record):
848 """
849 Emit a record.
850
851 The record is formatted, and then sent to the syslog server. If
852 exception information is present, it is NOT sent to the server.
853 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100854 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100855 if self.ident:
856 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100857 if self.append_nul:
858 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000859 """
860 We need to convert record level to lowercase, maybe this will
861 change in the future.
862 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000863 prio = '<%d>' % self.encodePriority(self.facility,
864 self.mapPriority(record.levelname))
865 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000866 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000867 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000868 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000869 try:
870 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000871 try:
872 self.socket.send(msg)
873 except socket.error:
874 self._connect_unixsocket(self.address)
875 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000876 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000877 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000878 else:
879 self.socket.sendall(msg)
Vinay Sajip985ef872011-04-26 19:34:04 +0100880 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000881 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000882 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000883 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000884
885class SMTPHandler(logging.Handler):
886 """
887 A handler class which sends an SMTP email for each logging event.
888 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000889 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100890 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000891 """
892 Initialize the handler.
893
894 Initialize the instance with the from and to addresses and subject
895 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000896 (host, port) tuple format for the mailhost argument. To specify
897 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000898 for the credentials argument. To specify the use of a secure
899 protocol (TLS), pass in a tuple for the secure argument. This will
900 only be used when authentication credentials are supplied. The tuple
901 will be either an empty tuple, or a single-value tuple with the name
902 of a keyfile, or a 2-value tuple with the names of the keyfile and
903 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000904 A timeout in seconds can be specified for the SMTP connection (the
905 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000906 """
907 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000908 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000909 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000910 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000911 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000912 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000913 self.username, self.password = credentials
914 else:
915 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000916 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000917 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000918 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000919 self.toaddrs = toaddrs
920 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000921 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000922 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000923
924 def getSubject(self, record):
925 """
926 Determine the subject for the email.
927
928 If you want to specify a subject line which is record-dependent,
929 override this method.
930 """
931 return self.subject
932
933 def emit(self, record):
934 """
935 Emit a record.
936
937 Format the record and send it to the specified addressees.
938 """
939 try:
940 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000941 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000942 port = self.mailport
943 if not port:
944 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000945 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000946 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000947 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 +0000948 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000949 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000950 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000951 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000952 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000953 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000954 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000955 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000956 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000957 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000958 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
959 smtp.quit()
Vinay Sajip985ef872011-04-26 19:34:04 +0100960 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000961 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000962 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000963 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000964
965class NTEventLogHandler(logging.Handler):
966 """
967 A handler class which sends events to the NT Event Log. Adds a
968 registry entry for the specified application name. If no dllname is
969 provided, win32service.pyd (which contains some basic message
970 placeholders) is used. Note that use of these placeholders will make
971 your event logs big, as the entire message source is held in the log.
972 If you want slimmer logs, you have to pass in the name of your own DLL
973 which contains the message definitions you want to use in the event log.
974 """
975 def __init__(self, appname, dllname=None, logtype="Application"):
976 logging.Handler.__init__(self)
977 try:
978 import win32evtlogutil, win32evtlog
979 self.appname = appname
980 self._welu = win32evtlogutil
981 if not dllname:
982 dllname = os.path.split(self._welu.__file__)
983 dllname = os.path.split(dllname[0])
984 dllname = os.path.join(dllname[0], r'win32service.pyd')
985 self.dllname = dllname
986 self.logtype = logtype
987 self._welu.AddSourceToRegistry(appname, dllname, logtype)
988 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
989 self.typemap = {
990 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
991 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000992 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000993 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
994 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
995 }
996 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000997 print("The Python Win32 extensions for NT (service, event "\
998 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000999 self._welu = None
1000
1001 def getMessageID(self, record):
1002 """
1003 Return the message ID for the event record. If you are using your
1004 own messages, you could do this by having the msg passed to the
1005 logger being an ID rather than a formatting string. Then, in here,
1006 you could use a dictionary lookup to get the message ID. This
1007 version returns 1, which is the base message ID in win32service.pyd.
1008 """
1009 return 1
1010
1011 def getEventCategory(self, record):
1012 """
1013 Return the event category for the record.
1014
1015 Override this if you want to specify your own categories. This version
1016 returns 0.
1017 """
1018 return 0
1019
1020 def getEventType(self, record):
1021 """
1022 Return the event type for the record.
1023
1024 Override this if you want to specify your own types. This version does
1025 a mapping using the handler's typemap attribute, which is set up in
1026 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001027 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001028 either need to override this method or place a suitable dictionary in
1029 the handler's typemap attribute.
1030 """
1031 return self.typemap.get(record.levelno, self.deftype)
1032
1033 def emit(self, record):
1034 """
1035 Emit a record.
1036
1037 Determine the message ID, event category and event type. Then
1038 log the message in the NT event log.
1039 """
1040 if self._welu:
1041 try:
1042 id = self.getMessageID(record)
1043 cat = self.getEventCategory(record)
1044 type = self.getEventType(record)
1045 msg = self.format(record)
1046 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip985ef872011-04-26 19:34:04 +01001047 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001048 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001049 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001050 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001051
1052 def close(self):
1053 """
1054 Clean up this handler.
1055
1056 You can remove the application name from the registry as a
1057 source of event log entries. However, if you do this, you will
1058 not be able to see the events as you intended in the Event Log
1059 Viewer - it needs to be able to access the registry to get the
1060 DLL name.
1061 """
1062 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001063 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001064
1065class HTTPHandler(logging.Handler):
1066 """
1067 A class which sends records to a Web server, using either GET or
1068 POST semantics.
1069 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001070 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001071 """
1072 Initialize the instance with the host, the request URL, and the method
1073 ("GET" or "POST")
1074 """
1075 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001076 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001077 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001078 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001079 self.host = host
1080 self.url = url
1081 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001082 self.secure = secure
1083 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001084
Neal Norwitzf297bd12003-04-23 03:49:43 +00001085 def mapLogRecord(self, record):
1086 """
1087 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001088 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001089 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001090 """
1091 return record.__dict__
1092
Guido van Rossum57102f82002-11-13 16:15:58 +00001093 def emit(self, record):
1094 """
1095 Emit a record.
1096
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001097 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001098 """
1099 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001100 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001101 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001102 if self.secure:
1103 h = http.client.HTTPSConnection(host)
1104 else:
1105 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001106 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001107 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001108 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001109 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001110 sep = '&'
1111 else:
1112 sep = '?'
1113 url = url + "%c%s" % (sep, data)
1114 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001115 # support multiple hosts on one IP address...
1116 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001117 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001118 if i >= 0:
1119 host = host[:i]
1120 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001121 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001122 h.putheader("Content-type",
1123 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001124 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001125 if self.credentials:
1126 import base64
1127 s = ('u%s:%s' % self.credentials).encode('utf-8')
1128 s = 'Basic ' + base64.b64encode(s).strip()
1129 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001130 h.endheaders()
1131 if self.method == "POST":
1132 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001133 h.getresponse() #can't do anything with the result
Vinay Sajip985ef872011-04-26 19:34:04 +01001134 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001135 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001136 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001137 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001138
1139class BufferingHandler(logging.Handler):
1140 """
1141 A handler class which buffers logging records in memory. Whenever each
1142 record is added to the buffer, a check is made to see if the buffer should
1143 be flushed. If it should, then flush() is expected to do what's needed.
1144 """
1145 def __init__(self, capacity):
1146 """
1147 Initialize the handler with the buffer size.
1148 """
1149 logging.Handler.__init__(self)
1150 self.capacity = capacity
1151 self.buffer = []
1152
1153 def shouldFlush(self, record):
1154 """
1155 Should the handler flush its buffer?
1156
1157 Returns true if the buffer is up to capacity. This method can be
1158 overridden to implement custom flushing strategies.
1159 """
1160 return (len(self.buffer) >= self.capacity)
1161
1162 def emit(self, record):
1163 """
1164 Emit a record.
1165
1166 Append the record. If shouldFlush() tells us to, call flush() to process
1167 the buffer.
1168 """
1169 self.buffer.append(record)
1170 if self.shouldFlush(record):
1171 self.flush()
1172
1173 def flush(self):
1174 """
1175 Override to implement custom flushing behaviour.
1176
1177 This version just zaps the buffer to empty.
1178 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001179 self.acquire()
1180 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001181 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001182 finally:
1183 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001184
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001185 def close(self):
1186 """
1187 Close the handler.
1188
1189 This version just flushes and chains to the parent class' close().
1190 """
1191 self.flush()
1192 logging.Handler.close(self)
1193
Guido van Rossum57102f82002-11-13 16:15:58 +00001194class MemoryHandler(BufferingHandler):
1195 """
1196 A handler class which buffers logging records in memory, periodically
1197 flushing them to a target handler. Flushing occurs whenever the buffer
1198 is full, or when an event of a certain severity or greater is seen.
1199 """
1200 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1201 """
1202 Initialize the handler with the buffer size, the level at which
1203 flushing should occur and an optional target.
1204
1205 Note that without a target being set either here or via setTarget(),
1206 a MemoryHandler is no use to anyone!
1207 """
1208 BufferingHandler.__init__(self, capacity)
1209 self.flushLevel = flushLevel
1210 self.target = target
1211
1212 def shouldFlush(self, record):
1213 """
1214 Check for buffer full or a record at the flushLevel or higher.
1215 """
1216 return (len(self.buffer) >= self.capacity) or \
1217 (record.levelno >= self.flushLevel)
1218
1219 def setTarget(self, target):
1220 """
1221 Set the target handler for this handler.
1222 """
1223 self.target = target
1224
1225 def flush(self):
1226 """
1227 For a MemoryHandler, flushing means just sending the buffered
1228 records to the target, if there is one. Override if you want
1229 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001230
1231 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001232 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001233 self.acquire()
1234 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001235 if self.target:
1236 for record in self.buffer:
1237 self.target.handle(record)
1238 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001239 finally:
1240 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001241
1242 def close(self):
1243 """
1244 Flush, set the target to None and lose the buffer.
1245 """
1246 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001247 self.acquire()
1248 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001249 self.target = None
1250 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001251 finally:
1252 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001253
1254
1255class QueueHandler(logging.Handler):
1256 """
1257 This handler sends events to a queue. Typically, it would be used together
1258 with a multiprocessing Queue to centralise logging to file in one process
1259 (in a multi-process application), so as to avoid file write contention
1260 between processes.
1261
1262 This code is new in Python 3.2, but this class can be copy pasted into
1263 user code for use with earlier Python versions.
1264 """
1265
1266 def __init__(self, queue):
1267 """
1268 Initialise an instance, using the passed queue.
1269 """
1270 logging.Handler.__init__(self)
1271 self.queue = queue
1272
1273 def enqueue(self, record):
1274 """
1275 Enqueue a record.
1276
1277 The base implementation uses put_nowait. You may want to override
1278 this method if you want to use blocking, timeouts or custom queue
1279 implementations.
1280 """
1281 self.queue.put_nowait(record)
1282
Vinay Sajip0258ce82010-09-22 20:34:53 +00001283 def prepare(self, record):
1284 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001285 Prepares a record for queuing. The object returned by this method is
1286 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001287
1288 The base implementation formats the record to merge the message
1289 and arguments, and removes unpickleable items from the record
1290 in-place.
1291
1292 You might want to override this method if you want to convert
1293 the record to a dict or JSON string, or send a modified copy
1294 of the record while leaving the original intact.
1295 """
1296 # The format operation gets traceback text into record.exc_text
1297 # (if there's exception data), and also puts the message into
1298 # record.message. We can then use this to replace the original
1299 # msg + args, as these might be unpickleable. We also zap the
1300 # exc_info attribute, as it's no longer needed and, if not None,
1301 # will typically not be pickleable.
1302 self.format(record)
1303 record.msg = record.message
1304 record.args = None
1305 record.exc_info = None
1306 return record
1307
Vinay Sajip121a1c42010-09-08 10:46:15 +00001308 def emit(self, record):
1309 """
1310 Emit a record.
1311
Vinay Sajip0637d492010-09-23 08:15:54 +00001312 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001313 """
1314 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001315 self.enqueue(self.prepare(record))
Vinay Sajip985ef872011-04-26 19:34:04 +01001316 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip121a1c42010-09-08 10:46:15 +00001317 raise
1318 except:
1319 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001320
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001321if threading:
1322 class QueueListener(object):
1323 """
1324 This class implements an internal threaded listener which watches for
1325 LogRecords being added to a queue, removes them and passes them to a
1326 list of handlers for processing.
1327 """
1328 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001329
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001330 def __init__(self, queue, *handlers):
1331 """
1332 Initialise an instance with the specified queue and
1333 handlers.
1334 """
1335 self.queue = queue
1336 self.handlers = handlers
1337 self._stop = threading.Event()
1338 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001339
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001340 def dequeue(self, block):
1341 """
1342 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001343
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001344 The base implementation uses get. You may want to override this method
1345 if you want to use timeouts or work with custom queue implementations.
1346 """
1347 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001348
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001349 def start(self):
1350 """
1351 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001352
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001353 This starts up a background thread to monitor the queue for
1354 LogRecords to process.
1355 """
1356 self._thread = t = threading.Thread(target=self._monitor)
1357 t.setDaemon(True)
1358 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001359
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001360 def prepare(self , record):
1361 """
1362 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001363
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001364 This method just returns the passed-in record. You may want to
1365 override this method if you need to do any custom marshalling or
1366 manipulation of the record before passing it to the handlers.
1367 """
1368 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001369
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001370 def handle(self, record):
1371 """
1372 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001373
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001374 This just loops through the handlers offering them the record
1375 to handle.
1376 """
1377 record = self.prepare(record)
1378 for handler in self.handlers:
1379 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001380
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001381 def _monitor(self):
1382 """
1383 Monitor the queue for records, and ask the handler
1384 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001385
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001386 This method runs on a separate, internal thread.
1387 The thread will terminate if it sees a sentinel object in the queue.
1388 """
1389 q = self.queue
1390 has_task_done = hasattr(q, 'task_done')
1391 while not self._stop.isSet():
1392 try:
1393 record = self.dequeue(True)
1394 if record is self._sentinel:
1395 break
1396 self.handle(record)
1397 if has_task_done:
1398 q.task_done()
1399 except queue.Empty:
1400 pass
1401 # There might still be records in the queue.
1402 while True:
1403 try:
1404 record = self.dequeue(False)
1405 if record is self._sentinel:
1406 break
1407 self.handle(record)
1408 if has_task_done:
1409 q.task_done()
1410 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001411 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001412
Victor Stinner59bec362011-05-02 16:14:16 +02001413 def enqueue_sentinel(self):
1414 """
1415 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001416
Victor Stinner59bec362011-05-02 16:14:16 +02001417 The base implementation uses put_nowait. You may want to override this
1418 method if you want to use timeouts or work with custom queue
1419 implementations.
1420 """
1421 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001422
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001423 def stop(self):
1424 """
1425 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001426
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001427 This asks the thread to terminate, and then waits for it to do so.
1428 Note that if you don't call this before your application exits, there
1429 may be some records still left on the queue, which won't be processed.
1430 """
1431 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001432 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001433 self._thread.join()
1434 self._thread = None