blob: 95e23a8b78bdfee32d28be66a2c83314f37c1835 [file] [log] [blame]
Vinay Sajipe9170522013-04-22 10:07:40 +01001# Copyright 2001-2013 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 Sajipe9170522013-04-22 10:07:40 +010021Copyright (C) 2001-2013 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),
Vinay Sajipe9170522013-04-22 10:07:40 +0100770 facility=LOG_USER, socktype=None):
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 Sajipe9170522013-04-22 10:07:40 +0100789 if socktype is None:
790 socktype = socket.SOCK_DGRAM
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)
Vinay Sajipe9170522013-04-22 10:07:40 +0100794 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000795 self.formatter = None
796
Vinay Sajipa1974c12005-01-13 08:23:56 +0000797 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100798 use_socktype = self.socktype
799 if use_socktype is None:
800 use_socktype = socket.SOCK_DGRAM
801 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000802 try:
803 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100804 # it worked, so set self.socktype to the used type
805 self.socktype = use_socktype
Vinay Sajipa1974c12005-01-13 08:23:56 +0000806 except socket.error:
807 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100808 if self.socktype is not None:
809 # user didn't specify falling back, so fail
810 raise
811 use_socktype = socket.SOCK_STREAM
812 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
813 try:
814 self.socket.connect(address)
815 # it worked, so set self.socktype to the used type
816 self.socktype = use_socktype
817 except socket.error:
818 self.socket.close()
819 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000820
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000821 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000822 """
823 Encode the facility and priority. You can pass in strings or
824 integers - if strings are passed, the facility_names and
825 priority_names mapping dictionaries are used to convert them to
826 integers.
827 """
Guido van Rossum13257902007-06-07 23:15:56 +0000828 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000829 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000830 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000831 priority = self.priority_names[priority]
832 return (facility << 3) | priority
833
834 def close (self):
835 """
836 Closes the socket.
837 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000838 self.acquire()
839 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000840 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000841 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000842 finally:
843 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000844
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000845 def mapPriority(self, levelName):
846 """
847 Map a logging level name to a key in the priority_names map.
848 This is useful in two scenarios: when custom levels are being
849 used, and in the case where you can't do a straightforward
850 mapping by lowercasing the logging level name because of locale-
851 specific issues (see SF #1524081).
852 """
853 return self.priority_map.get(levelName, "warning")
854
Vinay Sajip2353e352011-06-27 15:40:06 +0100855 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100856 append_nul = True # some old syslog daemons expect a NUL terminator
857
Guido van Rossum57102f82002-11-13 16:15:58 +0000858 def emit(self, record):
859 """
860 Emit a record.
861
862 The record is formatted, and then sent to the syslog server. If
863 exception information is present, it is NOT sent to the server.
864 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100865 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100866 if self.ident:
867 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100868 if self.append_nul:
869 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000870 """
871 We need to convert record level to lowercase, maybe this will
872 change in the future.
873 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000874 prio = '<%d>' % self.encodePriority(self.facility,
875 self.mapPriority(record.levelname))
876 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000877 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000878 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000879 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000880 try:
881 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000882 try:
883 self.socket.send(msg)
884 except socket.error:
885 self._connect_unixsocket(self.address)
886 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000887 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000888 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000889 else:
890 self.socket.sendall(msg)
Vinay Sajip985ef872011-04-26 19:34:04 +0100891 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000892 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000893 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000894 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000895
896class SMTPHandler(logging.Handler):
897 """
898 A handler class which sends an SMTP email for each logging event.
899 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000900 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100901 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000902 """
903 Initialize the handler.
904
905 Initialize the instance with the from and to addresses and subject
906 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000907 (host, port) tuple format for the mailhost argument. To specify
908 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000909 for the credentials argument. To specify the use of a secure
910 protocol (TLS), pass in a tuple for the secure argument. This will
911 only be used when authentication credentials are supplied. The tuple
912 will be either an empty tuple, or a single-value tuple with the name
913 of a keyfile, or a 2-value tuple with the names of the keyfile and
914 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000915 A timeout in seconds can be specified for the SMTP connection (the
916 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000917 """
918 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000919 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000920 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000921 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000922 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000923 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000924 self.username, self.password = credentials
925 else:
926 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000927 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000928 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000929 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000930 self.toaddrs = toaddrs
931 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000932 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000933 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000934
935 def getSubject(self, record):
936 """
937 Determine the subject for the email.
938
939 If you want to specify a subject line which is record-dependent,
940 override this method.
941 """
942 return self.subject
943
944 def emit(self, record):
945 """
946 Emit a record.
947
948 Format the record and send it to the specified addressees.
949 """
950 try:
951 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000952 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000953 port = self.mailport
954 if not port:
955 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000956 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000957 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000958 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 +0000959 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000960 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000961 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000962 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000963 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000964 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000965 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000966 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000967 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000968 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000969 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
970 smtp.quit()
Vinay Sajip985ef872011-04-26 19:34:04 +0100971 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000972 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000973 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000974 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000975
976class NTEventLogHandler(logging.Handler):
977 """
978 A handler class which sends events to the NT Event Log. Adds a
979 registry entry for the specified application name. If no dllname is
980 provided, win32service.pyd (which contains some basic message
981 placeholders) is used. Note that use of these placeholders will make
982 your event logs big, as the entire message source is held in the log.
983 If you want slimmer logs, you have to pass in the name of your own DLL
984 which contains the message definitions you want to use in the event log.
985 """
986 def __init__(self, appname, dllname=None, logtype="Application"):
987 logging.Handler.__init__(self)
988 try:
989 import win32evtlogutil, win32evtlog
990 self.appname = appname
991 self._welu = win32evtlogutil
992 if not dllname:
993 dllname = os.path.split(self._welu.__file__)
994 dllname = os.path.split(dllname[0])
995 dllname = os.path.join(dllname[0], r'win32service.pyd')
996 self.dllname = dllname
997 self.logtype = logtype
998 self._welu.AddSourceToRegistry(appname, dllname, logtype)
999 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1000 self.typemap = {
1001 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1002 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001003 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001004 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1005 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1006 }
1007 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001008 print("The Python Win32 extensions for NT (service, event "\
1009 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001010 self._welu = None
1011
1012 def getMessageID(self, record):
1013 """
1014 Return the message ID for the event record. If you are using your
1015 own messages, you could do this by having the msg passed to the
1016 logger being an ID rather than a formatting string. Then, in here,
1017 you could use a dictionary lookup to get the message ID. This
1018 version returns 1, which is the base message ID in win32service.pyd.
1019 """
1020 return 1
1021
1022 def getEventCategory(self, record):
1023 """
1024 Return the event category for the record.
1025
1026 Override this if you want to specify your own categories. This version
1027 returns 0.
1028 """
1029 return 0
1030
1031 def getEventType(self, record):
1032 """
1033 Return the event type for the record.
1034
1035 Override this if you want to specify your own types. This version does
1036 a mapping using the handler's typemap attribute, which is set up in
1037 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001038 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001039 either need to override this method or place a suitable dictionary in
1040 the handler's typemap attribute.
1041 """
1042 return self.typemap.get(record.levelno, self.deftype)
1043
1044 def emit(self, record):
1045 """
1046 Emit a record.
1047
1048 Determine the message ID, event category and event type. Then
1049 log the message in the NT event log.
1050 """
1051 if self._welu:
1052 try:
1053 id = self.getMessageID(record)
1054 cat = self.getEventCategory(record)
1055 type = self.getEventType(record)
1056 msg = self.format(record)
1057 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip985ef872011-04-26 19:34:04 +01001058 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001059 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001060 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001061 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001062
1063 def close(self):
1064 """
1065 Clean up this handler.
1066
1067 You can remove the application name from the registry as a
1068 source of event log entries. However, if you do this, you will
1069 not be able to see the events as you intended in the Event Log
1070 Viewer - it needs to be able to access the registry to get the
1071 DLL name.
1072 """
1073 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001074 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001075
1076class HTTPHandler(logging.Handler):
1077 """
1078 A class which sends records to a Web server, using either GET or
1079 POST semantics.
1080 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001081 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001082 """
1083 Initialize the instance with the host, the request URL, and the method
1084 ("GET" or "POST")
1085 """
1086 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001087 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001088 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001089 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001090 self.host = host
1091 self.url = url
1092 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001093 self.secure = secure
1094 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001095
Neal Norwitzf297bd12003-04-23 03:49:43 +00001096 def mapLogRecord(self, record):
1097 """
1098 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001099 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001100 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001101 """
1102 return record.__dict__
1103
Guido van Rossum57102f82002-11-13 16:15:58 +00001104 def emit(self, record):
1105 """
1106 Emit a record.
1107
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001108 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001109 """
1110 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001111 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001112 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001113 if self.secure:
1114 h = http.client.HTTPSConnection(host)
1115 else:
1116 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001117 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001118 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001119 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001120 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001121 sep = '&'
1122 else:
1123 sep = '?'
1124 url = url + "%c%s" % (sep, data)
1125 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001126 # support multiple hosts on one IP address...
1127 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001128 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001129 if i >= 0:
1130 host = host[:i]
1131 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001132 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001133 h.putheader("Content-type",
1134 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001135 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001136 if self.credentials:
1137 import base64
1138 s = ('u%s:%s' % self.credentials).encode('utf-8')
1139 s = 'Basic ' + base64.b64encode(s).strip()
1140 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001141 h.endheaders()
1142 if self.method == "POST":
1143 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001144 h.getresponse() #can't do anything with the result
Vinay Sajip985ef872011-04-26 19:34:04 +01001145 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001146 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001147 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001148 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001149
1150class BufferingHandler(logging.Handler):
1151 """
1152 A handler class which buffers logging records in memory. Whenever each
1153 record is added to the buffer, a check is made to see if the buffer should
1154 be flushed. If it should, then flush() is expected to do what's needed.
1155 """
1156 def __init__(self, capacity):
1157 """
1158 Initialize the handler with the buffer size.
1159 """
1160 logging.Handler.__init__(self)
1161 self.capacity = capacity
1162 self.buffer = []
1163
1164 def shouldFlush(self, record):
1165 """
1166 Should the handler flush its buffer?
1167
1168 Returns true if the buffer is up to capacity. This method can be
1169 overridden to implement custom flushing strategies.
1170 """
1171 return (len(self.buffer) >= self.capacity)
1172
1173 def emit(self, record):
1174 """
1175 Emit a record.
1176
1177 Append the record. If shouldFlush() tells us to, call flush() to process
1178 the buffer.
1179 """
1180 self.buffer.append(record)
1181 if self.shouldFlush(record):
1182 self.flush()
1183
1184 def flush(self):
1185 """
1186 Override to implement custom flushing behaviour.
1187
1188 This version just zaps the buffer to empty.
1189 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001190 self.acquire()
1191 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001192 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001193 finally:
1194 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001195
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001196 def close(self):
1197 """
1198 Close the handler.
1199
1200 This version just flushes and chains to the parent class' close().
1201 """
1202 self.flush()
1203 logging.Handler.close(self)
1204
Guido van Rossum57102f82002-11-13 16:15:58 +00001205class MemoryHandler(BufferingHandler):
1206 """
1207 A handler class which buffers logging records in memory, periodically
1208 flushing them to a target handler. Flushing occurs whenever the buffer
1209 is full, or when an event of a certain severity or greater is seen.
1210 """
1211 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1212 """
1213 Initialize the handler with the buffer size, the level at which
1214 flushing should occur and an optional target.
1215
1216 Note that without a target being set either here or via setTarget(),
1217 a MemoryHandler is no use to anyone!
1218 """
1219 BufferingHandler.__init__(self, capacity)
1220 self.flushLevel = flushLevel
1221 self.target = target
1222
1223 def shouldFlush(self, record):
1224 """
1225 Check for buffer full or a record at the flushLevel or higher.
1226 """
1227 return (len(self.buffer) >= self.capacity) or \
1228 (record.levelno >= self.flushLevel)
1229
1230 def setTarget(self, target):
1231 """
1232 Set the target handler for this handler.
1233 """
1234 self.target = target
1235
1236 def flush(self):
1237 """
1238 For a MemoryHandler, flushing means just sending the buffered
1239 records to the target, if there is one. Override if you want
1240 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001241
1242 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001243 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001244 self.acquire()
1245 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001246 if self.target:
1247 for record in self.buffer:
1248 self.target.handle(record)
1249 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001250 finally:
1251 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001252
1253 def close(self):
1254 """
1255 Flush, set the target to None and lose the buffer.
1256 """
1257 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001258 self.acquire()
1259 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001260 self.target = None
1261 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001262 finally:
1263 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001264
1265
1266class QueueHandler(logging.Handler):
1267 """
1268 This handler sends events to a queue. Typically, it would be used together
1269 with a multiprocessing Queue to centralise logging to file in one process
1270 (in a multi-process application), so as to avoid file write contention
1271 between processes.
1272
1273 This code is new in Python 3.2, but this class can be copy pasted into
1274 user code for use with earlier Python versions.
1275 """
1276
1277 def __init__(self, queue):
1278 """
1279 Initialise an instance, using the passed queue.
1280 """
1281 logging.Handler.__init__(self)
1282 self.queue = queue
1283
1284 def enqueue(self, record):
1285 """
1286 Enqueue a record.
1287
1288 The base implementation uses put_nowait. You may want to override
1289 this method if you want to use blocking, timeouts or custom queue
1290 implementations.
1291 """
1292 self.queue.put_nowait(record)
1293
Vinay Sajip0258ce82010-09-22 20:34:53 +00001294 def prepare(self, record):
1295 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001296 Prepares a record for queuing. The object returned by this method is
1297 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001298
1299 The base implementation formats the record to merge the message
1300 and arguments, and removes unpickleable items from the record
1301 in-place.
1302
1303 You might want to override this method if you want to convert
1304 the record to a dict or JSON string, or send a modified copy
1305 of the record while leaving the original intact.
1306 """
1307 # The format operation gets traceback text into record.exc_text
1308 # (if there's exception data), and also puts the message into
1309 # record.message. We can then use this to replace the original
1310 # msg + args, as these might be unpickleable. We also zap the
1311 # exc_info attribute, as it's no longer needed and, if not None,
1312 # will typically not be pickleable.
1313 self.format(record)
1314 record.msg = record.message
1315 record.args = None
1316 record.exc_info = None
1317 return record
1318
Vinay Sajip121a1c42010-09-08 10:46:15 +00001319 def emit(self, record):
1320 """
1321 Emit a record.
1322
Vinay Sajip0637d492010-09-23 08:15:54 +00001323 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001324 """
1325 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001326 self.enqueue(self.prepare(record))
Vinay Sajip985ef872011-04-26 19:34:04 +01001327 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip121a1c42010-09-08 10:46:15 +00001328 raise
1329 except:
1330 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001331
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001332if threading:
1333 class QueueListener(object):
1334 """
1335 This class implements an internal threaded listener which watches for
1336 LogRecords being added to a queue, removes them and passes them to a
1337 list of handlers for processing.
1338 """
1339 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001340
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001341 def __init__(self, queue, *handlers):
1342 """
1343 Initialise an instance with the specified queue and
1344 handlers.
1345 """
1346 self.queue = queue
1347 self.handlers = handlers
1348 self._stop = threading.Event()
1349 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001350
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001351 def dequeue(self, block):
1352 """
1353 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001354
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001355 The base implementation uses get. You may want to override this method
1356 if you want to use timeouts or work with custom queue implementations.
1357 """
1358 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001359
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001360 def start(self):
1361 """
1362 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001363
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001364 This starts up a background thread to monitor the queue for
1365 LogRecords to process.
1366 """
1367 self._thread = t = threading.Thread(target=self._monitor)
1368 t.setDaemon(True)
1369 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001370
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001371 def prepare(self , record):
1372 """
1373 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001374
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001375 This method just returns the passed-in record. You may want to
1376 override this method if you need to do any custom marshalling or
1377 manipulation of the record before passing it to the handlers.
1378 """
1379 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001380
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001381 def handle(self, record):
1382 """
1383 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001384
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001385 This just loops through the handlers offering them the record
1386 to handle.
1387 """
1388 record = self.prepare(record)
1389 for handler in self.handlers:
1390 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001391
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001392 def _monitor(self):
1393 """
1394 Monitor the queue for records, and ask the handler
1395 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001396
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001397 This method runs on a separate, internal thread.
1398 The thread will terminate if it sees a sentinel object in the queue.
1399 """
1400 q = self.queue
1401 has_task_done = hasattr(q, 'task_done')
1402 while not self._stop.isSet():
1403 try:
1404 record = self.dequeue(True)
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:
1411 pass
1412 # There might still be records in the queue.
1413 while True:
1414 try:
1415 record = self.dequeue(False)
1416 if record is self._sentinel:
1417 break
1418 self.handle(record)
1419 if has_task_done:
1420 q.task_done()
1421 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001422 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001423
Victor Stinner59bec362011-05-02 16:14:16 +02001424 def enqueue_sentinel(self):
1425 """
1426 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001427
Victor Stinner59bec362011-05-02 16:14:16 +02001428 The base implementation uses put_nowait. You may want to override this
1429 method if you want to use timeouts or work with custom queue
1430 implementations.
1431 """
1432 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001433
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001434 def stop(self):
1435 """
1436 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001437
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001438 This asks the thread to terminate, and then waits for it to do so.
1439 Note that if you don't call this before your application exits, there
1440 may be some records still left on the queue, which won't be processed.
1441 """
1442 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001443 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001444 self._thread.join()
1445 self._thread = None