blob: ddec7dd9d01c63395fabe97e9da730c9584976af [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):
Vinay Sajipd8599262013-09-06 10:10:22 +0100114 # Issue 18940: A file may not have been created if delay is True.
115 if os.path.exists(source):
116 os.rename(source, dest)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000117 else:
118 self.rotator(source, dest)
119
Vinay Sajip17c52d82004-07-03 11:48:34 +0000120class RotatingFileHandler(BaseRotatingHandler):
121 """
122 Handler for logging to a set of files, which switches from one file
123 to the next when the current file reaches a certain size.
124 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000125 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
Guido van Rossum57102f82002-11-13 16:15:58 +0000126 """
127 Open the specified file and use it as the stream for logging.
128
129 By default, the file grows indefinitely. You can specify particular
130 values of maxBytes and backupCount to allow the file to rollover at
131 a predetermined size.
132
133 Rollover occurs whenever the current log file is nearly maxBytes in
134 length. If backupCount is >= 1, the system will successively create
135 new files with the same pathname as the base file, but with extensions
136 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
137 and a base file name of "app.log", you would get "app.log",
138 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
139 written to is always "app.log" - when it gets filled up, it is closed
140 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
141 exist, then they are renamed to "app.log.2", "app.log.3" etc.
142 respectively.
143
144 If maxBytes is zero, rollover never occurs.
145 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000146 # If rotation/rollover is wanted, it doesn't make sense to use another
147 # mode. If for example 'w' were specified, then if there were multiple
148 # runs of the calling application, the logs from previous runs would be
149 # lost if the 'w' is respected, because the log file would be truncated
150 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000151 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000152 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000153 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000154 self.maxBytes = maxBytes
155 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000156
157 def doRollover(self):
158 """
159 Do a rollover, as described in __init__().
160 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000161 if self.stream:
162 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000163 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000164 if self.backupCount > 0:
165 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000166 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
167 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
168 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000169 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000170 if os.path.exists(dfn):
171 os.remove(dfn)
172 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000173 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000174 if os.path.exists(dfn):
175 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000176 self.rotate(self.baseFilename, dfn)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100177 if not self.delay:
178 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000179
Vinay Sajip17c52d82004-07-03 11:48:34 +0000180 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000181 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000182 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000183
Vinay Sajip17c52d82004-07-03 11:48:34 +0000184 Basically, see if the supplied record would cause the file to exceed
185 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000186 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000187 if self.stream is None: # delay was set...
188 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000189 if self.maxBytes > 0: # are we rolling over?
190 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000191 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000192 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000193 return 1
194 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000195
Vinay Sajip17c52d82004-07-03 11:48:34 +0000196class TimedRotatingFileHandler(BaseRotatingHandler):
197 """
198 Handler for logging to a file, rotating the log file at certain timed
199 intervals.
200
201 If backupCount is > 0, when rollover is done, no more than backupCount
202 files are kept - the oldest ones are deleted.
203 """
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000204 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000205 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000206 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000207 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000208 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000209 # Calculate the real rollover interval, which is just the number of
210 # seconds between rollovers. Also set the filename suffix used when
211 # a rollover occurs. Current 'when' events supported:
212 # S - Seconds
213 # M - Minutes
214 # H - Hours
215 # D - Days
216 # midnight - roll over at midnight
217 # W{0-6} - roll over on a certain day; 0 - Monday
218 #
219 # Case of the 'when' specifier is not important; lower or upper case
220 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000221 if self.when == 'S':
222 self.interval = 1 # one second
223 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000224 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000225 elif self.when == 'M':
226 self.interval = 60 # one minute
227 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000228 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000229 elif self.when == 'H':
230 self.interval = 60 * 60 # one hour
231 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000232 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000233 elif self.when == 'D' or self.when == 'MIDNIGHT':
234 self.interval = 60 * 60 * 24 # one day
235 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000236 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000237 elif self.when.startswith('W'):
238 self.interval = 60 * 60 * 24 * 7 # one week
239 if len(self.when) != 2:
240 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
241 if self.when[1] < '0' or self.when[1] > '6':
242 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
243 self.dayOfWeek = int(self.when[1])
244 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000245 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000246 else:
247 raise ValueError("Invalid rollover interval specified: %s" % self.when)
248
Antoine Pitroufd036452008-08-19 17:56:33 +0000249 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000250 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000251 if os.path.exists(filename):
252 t = os.stat(filename)[ST_MTIME]
253 else:
254 t = int(time.time())
255 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000256
Vinay Sajipaffbd872009-06-11 10:11:47 +0000257 def computeRollover(self, currentTime):
258 """
259 Work out the rollover time based on the specified time.
260 """
261 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000262 # If we are rolling over at midnight or weekly, then the interval is already known.
263 # What we need to figure out is WHEN the next interval is. In other words,
264 # if you are rolling over at midnight, then your base interval is 1 day,
265 # but you want to start that one day clock at midnight, not now. So, we
266 # have to fudge the rolloverAt value in order to trigger the first rollover
267 # at the right time. After that, the regular interval will take care of
268 # the rest. Note that this code doesn't care about leap seconds. :)
269 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
270 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000271 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000272 t = time.gmtime(currentTime)
273 else:
274 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000275 currentHour = t[3]
276 currentMinute = t[4]
277 currentSecond = t[5]
278 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000279 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
280 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000281 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000282 # If we are rolling over on a certain day, add in the number of days until
283 # the next rollover, but offset by 1 since we just calculated the time
284 # until the next day starts. There are three cases:
285 # Case 1) The day to rollover is today; in this case, do nothing
286 # Case 2) The day to rollover is further in the interval (i.e., today is
287 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
288 # next rollover is simply 6 - 2 - 1, or 3.
289 # Case 3) The day to rollover is behind us in the interval (i.e., today
290 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
291 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
292 # number of days left in the current week (1) plus the number
293 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000294 # The calculations described in 2) and 3) above need to have a day added.
295 # This is because the above time calculation takes us to midnight on this
296 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000297 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000298 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000299 if day != self.dayOfWeek:
300 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000301 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000302 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000303 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000304 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
305 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000306 dstNow = t[-1]
307 dstAtRollover = time.localtime(newRolloverAt)[-1]
308 if dstNow != dstAtRollover:
309 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000310 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000311 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000312 addend = 3600
313 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000314 result = newRolloverAt
315 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000316
317 def shouldRollover(self, record):
318 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000319 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000320
321 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000322 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000323 """
324 t = int(time.time())
325 if t >= self.rolloverAt:
326 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000327 return 0
328
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000329 def getFilesToDelete(self):
330 """
331 Determine the files to delete when rolling over.
332
333 More specific than the earlier method, which just used glob.glob().
334 """
335 dirName, baseName = os.path.split(self.baseFilename)
336 fileNames = os.listdir(dirName)
337 result = []
338 prefix = baseName + "."
339 plen = len(prefix)
340 for fileName in fileNames:
341 if fileName[:plen] == prefix:
342 suffix = fileName[plen:]
343 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000344 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000345 result.sort()
346 if len(result) < self.backupCount:
347 result = []
348 else:
349 result = result[:len(result) - self.backupCount]
350 return result
351
Vinay Sajip17c52d82004-07-03 11:48:34 +0000352 def doRollover(self):
353 """
354 do a rollover; in this case, a date/time stamp is appended to the filename
355 when the rollover happens. However, you want the file to be named for the
356 start of the interval, not the current time. If there is a backup count,
357 then we have to get a list of matching filenames, sort them and remove
358 the one with the oldest suffix.
359 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000360 if self.stream:
361 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000362 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000363 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000364 currentTime = int(time.time())
365 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000366 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000367 if self.utc:
368 timeTuple = time.gmtime(t)
369 else:
370 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000371 dstThen = timeTuple[-1]
372 if dstNow != dstThen:
373 if dstNow:
374 addend = 3600
375 else:
376 addend = -3600
377 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000378 dfn = self.rotation_filename(self.baseFilename + "." +
379 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000380 if os.path.exists(dfn):
381 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000382 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000383 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000384 for s in self.getFilesToDelete():
385 os.remove(s)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100386 if not self.delay:
387 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000388 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000389 while newRolloverAt <= currentTime:
390 newRolloverAt = newRolloverAt + self.interval
391 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000392 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000393 dstAtRollover = time.localtime(newRolloverAt)[-1]
394 if dstNow != dstAtRollover:
395 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000396 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000397 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000398 addend = 3600
399 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000400 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000401
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000402class WatchedFileHandler(logging.FileHandler):
403 """
404 A handler for logging to a file, which watches the file
405 to see if it has changed while in use. This can happen because of
406 usage of programs such as newsyslog and logrotate which perform
407 log file rotation. This handler, intended for use under Unix,
408 watches the file to see if it has changed since the last emit.
409 (A file has changed if its device or inode have changed.)
410 If it has changed, the old file stream is closed, and the file
411 opened to get a new stream.
412
413 This handler is not appropriate for use under Windows, because
414 under Windows open files cannot be moved or renamed - logging
415 opens the files with exclusive locks - and so there is no need
416 for such a handler. Furthermore, ST_INO is not supported under
417 Windows; stat always returns zero for this value.
418
419 This handler is based on a suggestion and patch by Chad J.
420 Schroeder.
421 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000422 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000423 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100424 self.dev, self.ino = -1, -1
425 self._statstream()
426
427 def _statstream(self):
428 if self.stream:
429 sres = os.fstat(self.stream.fileno())
430 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000431
432 def emit(self, record):
433 """
434 Emit a record.
435
436 First check if the underlying file has changed, and if it
437 has, close the old stream and reopen the file to get the
438 current stream.
439 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100440 # Reduce the chance of race conditions by stat'ing by path only
441 # once and then fstat'ing our new fd if we opened a new log stream.
442 # See issue #14632: Thanks to John Mulligan for the problem report
443 # and patch.
444 try:
445 # stat the file by path, checking for existence
446 sres = os.stat(self.baseFilename)
447 except OSError as err:
448 if err.errno == errno.ENOENT:
449 sres = None
450 else:
451 raise
452 # compare file system stat with that of our stream file handle
453 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
454 if self.stream is not None:
455 # we have an open file handle, clean it up
456 self.stream.flush()
457 self.stream.close()
458 # open a new file handle and get new stat info from that fd
459 self.stream = self._open()
460 self._statstream()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000461 logging.FileHandler.emit(self, record)
462
Vinay Sajip66b8b082012-04-24 23:25:30 +0100463
Guido van Rossum57102f82002-11-13 16:15:58 +0000464class SocketHandler(logging.Handler):
465 """
466 A handler class which writes logging records, in pickle format, to
467 a streaming socket. The socket is kept open across logging calls.
468 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000469 The pickle which is sent is that of the LogRecord's attribute dictionary
470 (__dict__), so that the receiver does not need to have the logging module
471 installed in order to process the logging event.
472
473 To unpickle the record at the receiving end into a LogRecord, use the
474 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000475 """
476
477 def __init__(self, host, port):
478 """
479 Initializes the handler with a specific host address and port.
480
Vinay Sajipde19e082011-04-30 21:52:26 +0100481 When the attribute *closeOnError* is set to True - if a socket error
482 occurs, the socket is silently closed and then reopened on the next
483 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000484 """
485 logging.Handler.__init__(self)
486 self.host = host
487 self.port = port
488 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100489 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000490 self.retryTime = None
491 #
492 # Exponential backoff parameters.
493 #
494 self.retryStart = 1.0
495 self.retryMax = 30.0
496 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000497
Guido van Rossumd8faa362007-04-27 19:54:29 +0000498 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000499 """
500 A factory method which allows subclasses to define the precise
501 type of socket they want.
502 """
503 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000504 if hasattr(s, 'settimeout'):
505 s.settimeout(timeout)
Vinay Sajipb37cd392011-05-07 15:55:47 +0100506 try:
507 s.connect((self.host, self.port))
508 return s
509 except socket.error:
510 s.close()
511 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000512
Vinay Sajip48cfe382004-02-20 13:17:27 +0000513 def createSocket(self):
514 """
515 Try to create a socket, using an exponential backoff with
516 a max retry time. Thanks to Robert Olson for the original patch
517 (SF #815911) which has been slightly refactored.
518 """
519 now = time.time()
520 # Either retryTime is None, in which case this
521 # is the first time back after a disconnect, or
522 # we've waited long enough.
523 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100524 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000525 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000526 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000527 if attempt:
528 try:
529 self.sock = self.makeSocket()
530 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000531 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000532 #Creation failed, so set the retry time and return.
533 if self.retryTime is None:
534 self.retryPeriod = self.retryStart
535 else:
536 self.retryPeriod = self.retryPeriod * self.retryFactor
537 if self.retryPeriod > self.retryMax:
538 self.retryPeriod = self.retryMax
539 self.retryTime = now + self.retryPeriod
540
Guido van Rossum57102f82002-11-13 16:15:58 +0000541 def send(self, s):
542 """
543 Send a pickled string to the socket.
544
545 This function allows for partial sends which can happen when the
546 network is busy.
547 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000548 if self.sock is None:
549 self.createSocket()
550 #self.sock can be None either because we haven't reached the retry
551 #time yet, or because we have reached the retry time and retried,
552 #but are still unable to connect.
553 if self.sock:
554 try:
555 if hasattr(self.sock, "sendall"):
556 self.sock.sendall(s)
Vinay Sajip7fe1d512011-04-28 12:04:58 +0100557 else: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000558 sentsofar = 0
559 left = len(s)
560 while left > 0:
561 sent = self.sock.send(s[sentsofar:])
562 sentsofar = sentsofar + sent
563 left = left - sent
Vinay Sajip7fe1d512011-04-28 12:04:58 +0100564 except socket.error: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000565 self.sock.close()
566 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000567
568 def makePickle(self, record):
569 """
570 Pickles the record in binary format with a length prefix, and
571 returns it ready for transmission across the socket.
572 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000573 ei = record.exc_info
574 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100575 # just to get traceback text into record.exc_text ...
576 dummy = self.format(record)
577 # See issue #14436: If msg or args are objects, they may not be
578 # available on the receiving end. So we convert the msg % args
579 # to a string, save it as msg and zap the args.
580 d = dict(record.__dict__)
581 d['msg'] = record.getMessage()
582 d['args'] = None
583 d['exc_info'] = None
584 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000585 slen = struct.pack(">L", len(s))
586 return slen + s
587
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000588 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000589 """
590 Handle an error during logging.
591
592 An error has occurred during logging. Most likely cause -
593 connection lost. Close the socket so that we can retry on the
594 next event.
595 """
596 if self.closeOnError and self.sock:
597 self.sock.close()
598 self.sock = None #try to reconnect next time
599 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000600 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000601
602 def emit(self, record):
603 """
604 Emit a record.
605
606 Pickles the record and writes it to the socket in binary format.
607 If there is an error with the socket, silently drop the packet.
608 If there was a problem with the socket, re-establishes the
609 socket.
610 """
611 try:
612 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000613 self.send(s)
Vinay Sajip985ef872011-04-26 19:34:04 +0100614 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000615 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000616 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000617 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000618
619 def close(self):
620 """
621 Closes the socket.
622 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000623 self.acquire()
624 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000625 if self.sock:
626 self.sock.close()
627 self.sock = None
628 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000629 finally:
630 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000631
632class DatagramHandler(SocketHandler):
633 """
634 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000635 a datagram socket. The pickle which is sent is that of the LogRecord's
636 attribute dictionary (__dict__), so that the receiver does not need to
637 have the logging module installed in order to process the logging event.
638
639 To unpickle the record at the receiving end into a LogRecord, use the
640 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000641
642 """
643 def __init__(self, host, port):
644 """
645 Initializes the handler with a specific host address and port.
646 """
647 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100648 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000649
650 def makeSocket(self):
651 """
652 The factory method of SocketHandler is here overridden to create
653 a UDP socket (SOCK_DGRAM).
654 """
655 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
656 return s
657
658 def send(self, s):
659 """
660 Send a pickled string to a socket.
661
662 This function no longer allows for partial sends which can happen
663 when the network is busy - UDP does not guarantee delivery and
664 can deliver packets out of sequence.
665 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000666 if self.sock is None:
667 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000668 self.sock.sendto(s, (self.host, self.port))
669
670class SysLogHandler(logging.Handler):
671 """
672 A handler class which sends formatted logging records to a syslog
673 server. Based on Sam Rushing's syslog module:
674 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
675 Contributed by Nicolas Untz (after which minor refactoring changes
676 have been made).
677 """
678
679 # from <linux/sys/syslog.h>:
680 # ======================================================================
681 # priorities/facilities are encoded into a single 32-bit quantity, where
682 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
683 # facility (0-big number). Both the priorities and the facilities map
684 # roughly one-to-one to strings in the syslogd(8) source code. This
685 # mapping is included in this file.
686 #
687 # priorities (these are ordered)
688
689 LOG_EMERG = 0 # system is unusable
690 LOG_ALERT = 1 # action must be taken immediately
691 LOG_CRIT = 2 # critical conditions
692 LOG_ERR = 3 # error conditions
693 LOG_WARNING = 4 # warning conditions
694 LOG_NOTICE = 5 # normal but significant condition
695 LOG_INFO = 6 # informational
696 LOG_DEBUG = 7 # debug-level messages
697
698 # facility codes
699 LOG_KERN = 0 # kernel messages
700 LOG_USER = 1 # random user-level messages
701 LOG_MAIL = 2 # mail system
702 LOG_DAEMON = 3 # system daemons
703 LOG_AUTH = 4 # security/authorization messages
704 LOG_SYSLOG = 5 # messages generated internally by syslogd
705 LOG_LPR = 6 # line printer subsystem
706 LOG_NEWS = 7 # network news subsystem
707 LOG_UUCP = 8 # UUCP subsystem
708 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000709 LOG_AUTHPRIV = 10 # security/authorization messages (private)
710 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000711
712 # other codes through 15 reserved for system use
713 LOG_LOCAL0 = 16 # reserved for local use
714 LOG_LOCAL1 = 17 # reserved for local use
715 LOG_LOCAL2 = 18 # reserved for local use
716 LOG_LOCAL3 = 19 # reserved for local use
717 LOG_LOCAL4 = 20 # reserved for local use
718 LOG_LOCAL5 = 21 # reserved for local use
719 LOG_LOCAL6 = 22 # reserved for local use
720 LOG_LOCAL7 = 23 # reserved for local use
721
722 priority_names = {
723 "alert": LOG_ALERT,
724 "crit": LOG_CRIT,
725 "critical": LOG_CRIT,
726 "debug": LOG_DEBUG,
727 "emerg": LOG_EMERG,
728 "err": LOG_ERR,
729 "error": LOG_ERR, # DEPRECATED
730 "info": LOG_INFO,
731 "notice": LOG_NOTICE,
732 "panic": LOG_EMERG, # DEPRECATED
733 "warn": LOG_WARNING, # DEPRECATED
734 "warning": LOG_WARNING,
735 }
736
737 facility_names = {
738 "auth": LOG_AUTH,
739 "authpriv": LOG_AUTHPRIV,
740 "cron": LOG_CRON,
741 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000742 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000743 "kern": LOG_KERN,
744 "lpr": LOG_LPR,
745 "mail": LOG_MAIL,
746 "news": LOG_NEWS,
747 "security": LOG_AUTH, # DEPRECATED
748 "syslog": LOG_SYSLOG,
749 "user": LOG_USER,
750 "uucp": LOG_UUCP,
751 "local0": LOG_LOCAL0,
752 "local1": LOG_LOCAL1,
753 "local2": LOG_LOCAL2,
754 "local3": LOG_LOCAL3,
755 "local4": LOG_LOCAL4,
756 "local5": LOG_LOCAL5,
757 "local6": LOG_LOCAL6,
758 "local7": LOG_LOCAL7,
759 }
760
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000761 #The map below appears to be trivially lowercasing the key. However,
762 #there's more to it than meets the eye - in some locales, lowercasing
763 #gives unexpected results. See SF #1524081: in the Turkish locale,
764 #"INFO".lower() != "info"
765 priority_map = {
766 "DEBUG" : "debug",
767 "INFO" : "info",
768 "WARNING" : "warning",
769 "ERROR" : "error",
770 "CRITICAL" : "critical"
771 }
772
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000773 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100774 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000775 """
776 Initialize a handler.
777
Guido van Rossume7ba4952007-06-06 23:52:48 +0000778 If address is specified as a string, a UNIX socket is used. To log to a
779 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000780 If facility is not specified, LOG_USER is used.
781 """
782 logging.Handler.__init__(self)
783
784 self.address = address
785 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000786 self.socktype = socktype
787
Guido van Rossum13257902007-06-07 23:15:56 +0000788 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100789 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000790 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000791 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100792 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100793 if socktype is None:
794 socktype = socket.SOCK_DGRAM
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000795 self.socket = socket.socket(socket.AF_INET, socktype)
796 if socktype == socket.SOCK_STREAM:
797 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100798 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000799 self.formatter = None
800
Vinay Sajipa1974c12005-01-13 08:23:56 +0000801 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100802 use_socktype = self.socktype
803 if use_socktype is None:
804 use_socktype = socket.SOCK_DGRAM
805 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000806 try:
807 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100808 # it worked, so set self.socktype to the used type
809 self.socktype = use_socktype
Vinay Sajipa1974c12005-01-13 08:23:56 +0000810 except socket.error:
811 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100812 if self.socktype is not None:
813 # user didn't specify falling back, so fail
814 raise
815 use_socktype = socket.SOCK_STREAM
816 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
817 try:
818 self.socket.connect(address)
819 # it worked, so set self.socktype to the used type
820 self.socktype = use_socktype
821 except socket.error:
822 self.socket.close()
823 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000824
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000825 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000826 """
827 Encode the facility and priority. You can pass in strings or
828 integers - if strings are passed, the facility_names and
829 priority_names mapping dictionaries are used to convert them to
830 integers.
831 """
Guido van Rossum13257902007-06-07 23:15:56 +0000832 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000833 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000834 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000835 priority = self.priority_names[priority]
836 return (facility << 3) | priority
837
838 def close (self):
839 """
840 Closes the socket.
841 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000842 self.acquire()
843 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000844 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000845 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000846 finally:
847 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000848
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000849 def mapPriority(self, levelName):
850 """
851 Map a logging level name to a key in the priority_names map.
852 This is useful in two scenarios: when custom levels are being
853 used, and in the case where you can't do a straightforward
854 mapping by lowercasing the logging level name because of locale-
855 specific issues (see SF #1524081).
856 """
857 return self.priority_map.get(levelName, "warning")
858
Vinay Sajip2353e352011-06-27 15:40:06 +0100859 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100860 append_nul = True # some old syslog daemons expect a NUL terminator
861
Guido van Rossum57102f82002-11-13 16:15:58 +0000862 def emit(self, record):
863 """
864 Emit a record.
865
866 The record is formatted, and then sent to the syslog server. If
867 exception information is present, it is NOT sent to the server.
868 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100869 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100870 if self.ident:
871 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100872 if self.append_nul:
873 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000874 """
875 We need to convert record level to lowercase, maybe this will
876 change in the future.
877 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000878 prio = '<%d>' % self.encodePriority(self.facility,
879 self.mapPriority(record.levelname))
880 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000881 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000882 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000883 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000884 try:
885 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000886 try:
887 self.socket.send(msg)
888 except socket.error:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100889 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000890 self._connect_unixsocket(self.address)
891 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000892 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000893 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000894 else:
895 self.socket.sendall(msg)
Vinay Sajip985ef872011-04-26 19:34:04 +0100896 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000897 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000898 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000899 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000900
901class SMTPHandler(logging.Handler):
902 """
903 A handler class which sends an SMTP email for each logging event.
904 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000905 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100906 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000907 """
908 Initialize the handler.
909
910 Initialize the instance with the from and to addresses and subject
911 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000912 (host, port) tuple format for the mailhost argument. To specify
913 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000914 for the credentials argument. To specify the use of a secure
915 protocol (TLS), pass in a tuple for the secure argument. This will
916 only be used when authentication credentials are supplied. The tuple
917 will be either an empty tuple, or a single-value tuple with the name
918 of a keyfile, or a 2-value tuple with the names of the keyfile and
919 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000920 A timeout in seconds can be specified for the SMTP connection (the
921 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000922 """
923 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000924 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000925 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000926 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000927 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000928 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000929 self.username, self.password = credentials
930 else:
931 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000932 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000933 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000934 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000935 self.toaddrs = toaddrs
936 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000937 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000938 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000939
940 def getSubject(self, record):
941 """
942 Determine the subject for the email.
943
944 If you want to specify a subject line which is record-dependent,
945 override this method.
946 """
947 return self.subject
948
949 def emit(self, record):
950 """
951 Emit a record.
952
953 Format the record and send it to the specified addressees.
954 """
955 try:
956 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000957 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000958 port = self.mailport
959 if not port:
960 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000961 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000962 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000963 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 +0000964 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000965 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000966 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000967 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000968 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000969 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000970 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000971 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000972 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000973 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000974 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
975 smtp.quit()
Vinay Sajip985ef872011-04-26 19:34:04 +0100976 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000977 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000978 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000979 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000980
981class NTEventLogHandler(logging.Handler):
982 """
983 A handler class which sends events to the NT Event Log. Adds a
984 registry entry for the specified application name. If no dllname is
985 provided, win32service.pyd (which contains some basic message
986 placeholders) is used. Note that use of these placeholders will make
987 your event logs big, as the entire message source is held in the log.
988 If you want slimmer logs, you have to pass in the name of your own DLL
989 which contains the message definitions you want to use in the event log.
990 """
991 def __init__(self, appname, dllname=None, logtype="Application"):
992 logging.Handler.__init__(self)
993 try:
994 import win32evtlogutil, win32evtlog
995 self.appname = appname
996 self._welu = win32evtlogutil
997 if not dllname:
998 dllname = os.path.split(self._welu.__file__)
999 dllname = os.path.split(dllname[0])
1000 dllname = os.path.join(dllname[0], r'win32service.pyd')
1001 self.dllname = dllname
1002 self.logtype = logtype
1003 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1004 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1005 self.typemap = {
1006 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1007 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001008 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001009 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1010 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1011 }
1012 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001013 print("The Python Win32 extensions for NT (service, event "\
1014 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001015 self._welu = None
1016
1017 def getMessageID(self, record):
1018 """
1019 Return the message ID for the event record. If you are using your
1020 own messages, you could do this by having the msg passed to the
1021 logger being an ID rather than a formatting string. Then, in here,
1022 you could use a dictionary lookup to get the message ID. This
1023 version returns 1, which is the base message ID in win32service.pyd.
1024 """
1025 return 1
1026
1027 def getEventCategory(self, record):
1028 """
1029 Return the event category for the record.
1030
1031 Override this if you want to specify your own categories. This version
1032 returns 0.
1033 """
1034 return 0
1035
1036 def getEventType(self, record):
1037 """
1038 Return the event type for the record.
1039
1040 Override this if you want to specify your own types. This version does
1041 a mapping using the handler's typemap attribute, which is set up in
1042 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001043 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001044 either need to override this method or place a suitable dictionary in
1045 the handler's typemap attribute.
1046 """
1047 return self.typemap.get(record.levelno, self.deftype)
1048
1049 def emit(self, record):
1050 """
1051 Emit a record.
1052
1053 Determine the message ID, event category and event type. Then
1054 log the message in the NT event log.
1055 """
1056 if self._welu:
1057 try:
1058 id = self.getMessageID(record)
1059 cat = self.getEventCategory(record)
1060 type = self.getEventType(record)
1061 msg = self.format(record)
1062 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip985ef872011-04-26 19:34:04 +01001063 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001064 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001065 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001066 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001067
1068 def close(self):
1069 """
1070 Clean up this handler.
1071
1072 You can remove the application name from the registry as a
1073 source of event log entries. However, if you do this, you will
1074 not be able to see the events as you intended in the Event Log
1075 Viewer - it needs to be able to access the registry to get the
1076 DLL name.
1077 """
1078 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001079 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001080
1081class HTTPHandler(logging.Handler):
1082 """
1083 A class which sends records to a Web server, using either GET or
1084 POST semantics.
1085 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001086 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001087 """
1088 Initialize the instance with the host, the request URL, and the method
1089 ("GET" or "POST")
1090 """
1091 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001092 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001093 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001094 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001095 self.host = host
1096 self.url = url
1097 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001098 self.secure = secure
1099 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001100
Neal Norwitzf297bd12003-04-23 03:49:43 +00001101 def mapLogRecord(self, record):
1102 """
1103 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001104 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001105 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001106 """
1107 return record.__dict__
1108
Guido van Rossum57102f82002-11-13 16:15:58 +00001109 def emit(self, record):
1110 """
1111 Emit a record.
1112
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001113 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001114 """
1115 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001116 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001117 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001118 if self.secure:
1119 h = http.client.HTTPSConnection(host)
1120 else:
1121 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001122 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001123 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001124 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001125 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001126 sep = '&'
1127 else:
1128 sep = '?'
1129 url = url + "%c%s" % (sep, data)
1130 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001131 # support multiple hosts on one IP address...
1132 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001133 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001134 if i >= 0:
1135 host = host[:i]
1136 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001137 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001138 h.putheader("Content-type",
1139 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001140 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001141 if self.credentials:
1142 import base64
1143 s = ('u%s:%s' % self.credentials).encode('utf-8')
1144 s = 'Basic ' + base64.b64encode(s).strip()
1145 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001146 h.endheaders()
1147 if self.method == "POST":
1148 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001149 h.getresponse() #can't do anything with the result
Vinay Sajip985ef872011-04-26 19:34:04 +01001150 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001151 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001152 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001153 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001154
1155class BufferingHandler(logging.Handler):
1156 """
1157 A handler class which buffers logging records in memory. Whenever each
1158 record is added to the buffer, a check is made to see if the buffer should
1159 be flushed. If it should, then flush() is expected to do what's needed.
1160 """
1161 def __init__(self, capacity):
1162 """
1163 Initialize the handler with the buffer size.
1164 """
1165 logging.Handler.__init__(self)
1166 self.capacity = capacity
1167 self.buffer = []
1168
1169 def shouldFlush(self, record):
1170 """
1171 Should the handler flush its buffer?
1172
1173 Returns true if the buffer is up to capacity. This method can be
1174 overridden to implement custom flushing strategies.
1175 """
1176 return (len(self.buffer) >= self.capacity)
1177
1178 def emit(self, record):
1179 """
1180 Emit a record.
1181
1182 Append the record. If shouldFlush() tells us to, call flush() to process
1183 the buffer.
1184 """
1185 self.buffer.append(record)
1186 if self.shouldFlush(record):
1187 self.flush()
1188
1189 def flush(self):
1190 """
1191 Override to implement custom flushing behaviour.
1192
1193 This version just zaps the buffer to empty.
1194 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001195 self.acquire()
1196 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001197 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001198 finally:
1199 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001200
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001201 def close(self):
1202 """
1203 Close the handler.
1204
1205 This version just flushes and chains to the parent class' close().
1206 """
1207 self.flush()
1208 logging.Handler.close(self)
1209
Guido van Rossum57102f82002-11-13 16:15:58 +00001210class MemoryHandler(BufferingHandler):
1211 """
1212 A handler class which buffers logging records in memory, periodically
1213 flushing them to a target handler. Flushing occurs whenever the buffer
1214 is full, or when an event of a certain severity or greater is seen.
1215 """
1216 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1217 """
1218 Initialize the handler with the buffer size, the level at which
1219 flushing should occur and an optional target.
1220
1221 Note that without a target being set either here or via setTarget(),
1222 a MemoryHandler is no use to anyone!
1223 """
1224 BufferingHandler.__init__(self, capacity)
1225 self.flushLevel = flushLevel
1226 self.target = target
1227
1228 def shouldFlush(self, record):
1229 """
1230 Check for buffer full or a record at the flushLevel or higher.
1231 """
1232 return (len(self.buffer) >= self.capacity) or \
1233 (record.levelno >= self.flushLevel)
1234
1235 def setTarget(self, target):
1236 """
1237 Set the target handler for this handler.
1238 """
1239 self.target = target
1240
1241 def flush(self):
1242 """
1243 For a MemoryHandler, flushing means just sending the buffered
1244 records to the target, if there is one. Override if you want
1245 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001246
1247 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001248 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001249 self.acquire()
1250 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001251 if self.target:
1252 for record in self.buffer:
1253 self.target.handle(record)
1254 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001255 finally:
1256 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001257
1258 def close(self):
1259 """
1260 Flush, set the target to None and lose the buffer.
1261 """
1262 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001263 self.acquire()
1264 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001265 self.target = None
1266 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001267 finally:
1268 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001269
1270
1271class QueueHandler(logging.Handler):
1272 """
1273 This handler sends events to a queue. Typically, it would be used together
1274 with a multiprocessing Queue to centralise logging to file in one process
1275 (in a multi-process application), so as to avoid file write contention
1276 between processes.
1277
1278 This code is new in Python 3.2, but this class can be copy pasted into
1279 user code for use with earlier Python versions.
1280 """
1281
1282 def __init__(self, queue):
1283 """
1284 Initialise an instance, using the passed queue.
1285 """
1286 logging.Handler.__init__(self)
1287 self.queue = queue
1288
1289 def enqueue(self, record):
1290 """
1291 Enqueue a record.
1292
1293 The base implementation uses put_nowait. You may want to override
1294 this method if you want to use blocking, timeouts or custom queue
1295 implementations.
1296 """
1297 self.queue.put_nowait(record)
1298
Vinay Sajip0258ce82010-09-22 20:34:53 +00001299 def prepare(self, record):
1300 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001301 Prepares a record for queuing. The object returned by this method is
1302 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001303
1304 The base implementation formats the record to merge the message
1305 and arguments, and removes unpickleable items from the record
1306 in-place.
1307
1308 You might want to override this method if you want to convert
1309 the record to a dict or JSON string, or send a modified copy
1310 of the record while leaving the original intact.
1311 """
1312 # The format operation gets traceback text into record.exc_text
1313 # (if there's exception data), and also puts the message into
1314 # record.message. We can then use this to replace the original
1315 # msg + args, as these might be unpickleable. We also zap the
1316 # exc_info attribute, as it's no longer needed and, if not None,
1317 # will typically not be pickleable.
1318 self.format(record)
1319 record.msg = record.message
1320 record.args = None
1321 record.exc_info = None
1322 return record
1323
Vinay Sajip121a1c42010-09-08 10:46:15 +00001324 def emit(self, record):
1325 """
1326 Emit a record.
1327
Vinay Sajip0637d492010-09-23 08:15:54 +00001328 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001329 """
1330 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001331 self.enqueue(self.prepare(record))
Vinay Sajip985ef872011-04-26 19:34:04 +01001332 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip121a1c42010-09-08 10:46:15 +00001333 raise
1334 except:
1335 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001336
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001337if threading:
1338 class QueueListener(object):
1339 """
1340 This class implements an internal threaded listener which watches for
1341 LogRecords being added to a queue, removes them and passes them to a
1342 list of handlers for processing.
1343 """
1344 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001345
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001346 def __init__(self, queue, *handlers):
1347 """
1348 Initialise an instance with the specified queue and
1349 handlers.
1350 """
1351 self.queue = queue
1352 self.handlers = handlers
1353 self._stop = threading.Event()
1354 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001355
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001356 def dequeue(self, block):
1357 """
1358 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001359
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001360 The base implementation uses get. You may want to override this method
1361 if you want to use timeouts or work with custom queue implementations.
1362 """
1363 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001364
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001365 def start(self):
1366 """
1367 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001368
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001369 This starts up a background thread to monitor the queue for
1370 LogRecords to process.
1371 """
1372 self._thread = t = threading.Thread(target=self._monitor)
1373 t.setDaemon(True)
1374 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001375
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001376 def prepare(self , record):
1377 """
1378 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001379
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001380 This method just returns the passed-in record. You may want to
1381 override this method if you need to do any custom marshalling or
1382 manipulation of the record before passing it to the handlers.
1383 """
1384 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001385
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001386 def handle(self, record):
1387 """
1388 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001389
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001390 This just loops through the handlers offering them the record
1391 to handle.
1392 """
1393 record = self.prepare(record)
1394 for handler in self.handlers:
1395 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001396
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001397 def _monitor(self):
1398 """
1399 Monitor the queue for records, and ask the handler
1400 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001401
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001402 This method runs on a separate, internal thread.
1403 The thread will terminate if it sees a sentinel object in the queue.
1404 """
1405 q = self.queue
1406 has_task_done = hasattr(q, 'task_done')
1407 while not self._stop.isSet():
1408 try:
1409 record = self.dequeue(True)
1410 if record is self._sentinel:
1411 break
1412 self.handle(record)
1413 if has_task_done:
1414 q.task_done()
1415 except queue.Empty:
1416 pass
1417 # There might still be records in the queue.
1418 while True:
1419 try:
1420 record = self.dequeue(False)
1421 if record is self._sentinel:
1422 break
1423 self.handle(record)
1424 if has_task_done:
1425 q.task_done()
1426 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001427 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001428
Victor Stinner59bec362011-05-02 16:14:16 +02001429 def enqueue_sentinel(self):
1430 """
1431 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001432
Victor Stinner59bec362011-05-02 16:14:16 +02001433 The base implementation uses put_nowait. You may want to override this
1434 method if you want to use timeouts or work with custom queue
1435 implementations.
1436 """
1437 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001438
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001439 def stop(self):
1440 """
1441 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001442
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001443 This asks the thread to terminate, and then waits for it to do so.
1444 Note that if you don't call this before your application exits, there
1445 may be some records still left on the queue, which won't be processed.
1446 """
1447 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001448 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001449 self._thread.join()
1450 self._thread = None