blob: b0b0a1660f08dcc6ab487124f34c203ef729a285 [file] [log] [blame]
Vinay Sajipa7130792013-04-12 17:04:23 +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 Sajipa7130792013-04-12 17:04:23 +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
Brett Cannoncd171c82013-07-04 17:43:24 -040032except 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 Sajip8cf4eb12012-10-09 08:06:13 +010075 except Exception:
Vinay Sajip3970c112004-07-08 10:24:04 +000076 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000077
Vinay Sajip23b94d02012-01-04 12:02:26 +000078 def rotation_filename(self, default_name):
79 """
80 Modify the filename of a log file when rotating.
81
82 This is provided so that a custom filename can be provided.
83
84 The default implementation calls the 'namer' attribute of the
85 handler, if it's callable, passing the default name to
86 it. If the attribute isn't callable (the default is None), the name
87 is returned unchanged.
88
89 :param default_name: The default name for the log file.
90 """
91 if not callable(self.namer):
92 result = default_name
93 else:
94 result = self.namer(default_name)
95 return result
96
97 def rotate(self, source, dest):
98 """
99 When rotating, rotate the current log.
100
101 The default implementation calls the 'rotator' attribute of the
102 handler, if it's callable, passing the source and dest arguments to
103 it. If the attribute isn't callable (the default is None), the source
104 is simply renamed to the destination.
105
106 :param source: The source filename. This is normally the base
107 filename, e.g. 'test.log'
108 :param dest: The destination filename. This is normally
109 what the source is rotated to, e.g. 'test.log.1'.
110 """
111 if not callable(self.rotator):
Vinay Sajipd8599262013-09-06 10:10:22 +0100112 # Issue 18940: A file may not have been created if delay is True.
113 if os.path.exists(source):
114 os.rename(source, dest)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000115 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)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100175 if not self.delay:
176 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000177
Vinay Sajip17c52d82004-07-03 11:48:34 +0000178 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000179 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000180 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000181
Vinay Sajip17c52d82004-07-03 11:48:34 +0000182 Basically, see if the supplied record would cause the file to exceed
183 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000184 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000185 if self.stream is None: # delay was set...
186 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000187 if self.maxBytes > 0: # are we rolling over?
188 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000189 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000190 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000191 return 1
192 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000193
Vinay Sajip17c52d82004-07-03 11:48:34 +0000194class TimedRotatingFileHandler(BaseRotatingHandler):
195 """
196 Handler for logging to a file, rotating the log file at certain timed
197 intervals.
198
199 If backupCount is > 0, when rollover is done, no more than backupCount
200 files are kept - the oldest ones are deleted.
201 """
Vinay Sajipa7130792013-04-12 17:04:23 +0100202 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False, atTime=None):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000203 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000204 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000205 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000206 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100207 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000208 # Calculate the real rollover interval, which is just the number of
209 # seconds between rollovers. Also set the filename suffix used when
210 # a rollover occurs. Current 'when' events supported:
211 # S - Seconds
212 # M - Minutes
213 # H - Hours
214 # D - Days
215 # midnight - roll over at midnight
216 # W{0-6} - roll over on a certain day; 0 - Monday
217 #
218 # Case of the 'when' specifier is not important; lower or upper case
219 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000220 if self.when == 'S':
221 self.interval = 1 # one second
222 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000223 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000224 elif self.when == 'M':
225 self.interval = 60 # one minute
226 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000227 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000228 elif self.when == 'H':
229 self.interval = 60 * 60 # one hour
230 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000231 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000232 elif self.when == 'D' or self.when == 'MIDNIGHT':
233 self.interval = 60 * 60 * 24 # one day
234 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000235 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000236 elif self.when.startswith('W'):
237 self.interval = 60 * 60 * 24 * 7 # one week
238 if len(self.when) != 2:
239 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
240 if self.when[1] < '0' or self.when[1] > '6':
241 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
242 self.dayOfWeek = int(self.when[1])
243 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000244 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000245 else:
246 raise ValueError("Invalid rollover interval specified: %s" % self.when)
247
Antoine Pitroufd036452008-08-19 17:56:33 +0000248 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000249 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000250 if os.path.exists(filename):
251 t = os.stat(filename)[ST_MTIME]
252 else:
253 t = int(time.time())
254 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000255
Vinay Sajipaffbd872009-06-11 10:11:47 +0000256 def computeRollover(self, currentTime):
257 """
258 Work out the rollover time based on the specified time.
259 """
260 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000261 # If we are rolling over at midnight or weekly, then the interval is already known.
262 # What we need to figure out is WHEN the next interval is. In other words,
263 # if you are rolling over at midnight, then your base interval is 1 day,
264 # but you want to start that one day clock at midnight, not now. So, we
265 # have to fudge the rolloverAt value in order to trigger the first rollover
266 # at the right time. After that, the regular interval will take care of
267 # the rest. Note that this code doesn't care about leap seconds. :)
268 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
269 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000270 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000271 t = time.gmtime(currentTime)
272 else:
273 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000274 currentHour = t[3]
275 currentMinute = t[4]
276 currentSecond = t[5]
Vinay Sajipa7130792013-04-12 17:04:23 +0100277 currentDay = t[6]
278 # r is the number of seconds left between now and the next rotation
279 if self.atTime is None:
280 rotate_ts = _MIDNIGHT
281 else:
282 rotate_ts = ((self.atTime.hour * 60 + self.atTime.minute)*60 +
283 self.atTime.second)
284
285 r = rotate_ts - ((currentHour * 60 + currentMinute) * 60 +
286 currentSecond)
287 if r < 0:
288 # Rotate time is before the current time (for example when
289 # self.rotateAt is 13:45 and it now 14:15), rotation is
290 # tomorrow.
291 r += _MIDNIGHT
292 currentDay = (currentDay + 1) % 7
Vinay Sajipaffbd872009-06-11 10:11:47 +0000293 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000294 # If we are rolling over on a certain day, add in the number of days until
295 # the next rollover, but offset by 1 since we just calculated the time
296 # until the next day starts. There are three cases:
297 # Case 1) The day to rollover is today; in this case, do nothing
298 # Case 2) The day to rollover is further in the interval (i.e., today is
299 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
300 # next rollover is simply 6 - 2 - 1, or 3.
301 # Case 3) The day to rollover is behind us in the interval (i.e., today
302 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
303 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
304 # number of days left in the current week (1) plus the number
305 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000306 # The calculations described in 2) and 3) above need to have a day added.
307 # This is because the above time calculation takes us to midnight on this
308 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000309 if self.when.startswith('W'):
Vinay Sajipa7130792013-04-12 17:04:23 +0100310 day = currentDay # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000311 if day != self.dayOfWeek:
312 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000313 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000314 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000315 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000316 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
317 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000318 dstNow = t[-1]
319 dstAtRollover = time.localtime(newRolloverAt)[-1]
320 if dstNow != dstAtRollover:
321 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000322 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000323 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000324 addend = 3600
325 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000326 result = newRolloverAt
327 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000328
329 def shouldRollover(self, record):
330 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000331 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000332
333 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000334 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000335 """
336 t = int(time.time())
337 if t >= self.rolloverAt:
338 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000339 return 0
340
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000341 def getFilesToDelete(self):
342 """
343 Determine the files to delete when rolling over.
344
345 More specific than the earlier method, which just used glob.glob().
346 """
347 dirName, baseName = os.path.split(self.baseFilename)
348 fileNames = os.listdir(dirName)
349 result = []
350 prefix = baseName + "."
351 plen = len(prefix)
352 for fileName in fileNames:
353 if fileName[:plen] == prefix:
354 suffix = fileName[plen:]
355 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000356 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000357 result.sort()
358 if len(result) < self.backupCount:
359 result = []
360 else:
361 result = result[:len(result) - self.backupCount]
362 return result
363
Vinay Sajip17c52d82004-07-03 11:48:34 +0000364 def doRollover(self):
365 """
366 do a rollover; in this case, a date/time stamp is appended to the filename
367 when the rollover happens. However, you want the file to be named for the
368 start of the interval, not the current time. If there is a backup count,
369 then we have to get a list of matching filenames, sort them and remove
370 the one with the oldest suffix.
371 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000372 if self.stream:
373 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000374 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000375 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000376 currentTime = int(time.time())
377 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000378 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000379 if self.utc:
380 timeTuple = time.gmtime(t)
381 else:
382 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000383 dstThen = timeTuple[-1]
384 if dstNow != dstThen:
385 if dstNow:
386 addend = 3600
387 else:
388 addend = -3600
389 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000390 dfn = self.rotation_filename(self.baseFilename + "." +
391 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000392 if os.path.exists(dfn):
393 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000394 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000395 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000396 for s in self.getFilesToDelete():
397 os.remove(s)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100398 if not self.delay:
399 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000400 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000401 while newRolloverAt <= currentTime:
402 newRolloverAt = newRolloverAt + self.interval
403 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000404 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000405 dstAtRollover = time.localtime(newRolloverAt)[-1]
406 if dstNow != dstAtRollover:
407 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000408 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000409 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000410 addend = 3600
411 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000412 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000413
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000414class WatchedFileHandler(logging.FileHandler):
415 """
416 A handler for logging to a file, which watches the file
417 to see if it has changed while in use. This can happen because of
418 usage of programs such as newsyslog and logrotate which perform
419 log file rotation. This handler, intended for use under Unix,
420 watches the file to see if it has changed since the last emit.
421 (A file has changed if its device or inode have changed.)
422 If it has changed, the old file stream is closed, and the file
423 opened to get a new stream.
424
425 This handler is not appropriate for use under Windows, because
426 under Windows open files cannot be moved or renamed - logging
427 opens the files with exclusive locks - and so there is no need
428 for such a handler. Furthermore, ST_INO is not supported under
429 Windows; stat always returns zero for this value.
430
431 This handler is based on a suggestion and patch by Chad J.
432 Schroeder.
433 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000434 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000435 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100436 self.dev, self.ino = -1, -1
437 self._statstream()
438
439 def _statstream(self):
440 if self.stream:
441 sres = os.fstat(self.stream.fileno())
442 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000443
444 def emit(self, record):
445 """
446 Emit a record.
447
448 First check if the underlying file has changed, and if it
449 has, close the old stream and reopen the file to get the
450 current stream.
451 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100452 # Reduce the chance of race conditions by stat'ing by path only
453 # once and then fstat'ing our new fd if we opened a new log stream.
454 # See issue #14632: Thanks to John Mulligan for the problem report
455 # and patch.
456 try:
457 # stat the file by path, checking for existence
458 sres = os.stat(self.baseFilename)
Giampaolo Rodola'0166a282013-02-12 15:14:17 +0100459 except FileNotFoundError:
460 sres = None
Vinay Sajip66b8b082012-04-24 23:25:30 +0100461 # compare file system stat with that of our stream file handle
462 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
463 if self.stream is not None:
464 # we have an open file handle, clean it up
465 self.stream.flush()
466 self.stream.close()
467 # open a new file handle and get new stat info from that fd
468 self.stream = self._open()
469 self._statstream()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000470 logging.FileHandler.emit(self, record)
471
Vinay Sajip66b8b082012-04-24 23:25:30 +0100472
Guido van Rossum57102f82002-11-13 16:15:58 +0000473class SocketHandler(logging.Handler):
474 """
475 A handler class which writes logging records, in pickle format, to
476 a streaming socket. The socket is kept open across logging calls.
477 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000478 The pickle which is sent is that of the LogRecord's attribute dictionary
479 (__dict__), so that the receiver does not need to have the logging module
480 installed in order to process the logging event.
481
482 To unpickle the record at the receiving end into a LogRecord, use the
483 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000484 """
485
486 def __init__(self, host, port):
487 """
488 Initializes the handler with a specific host address and port.
489
Vinay Sajipde19e082011-04-30 21:52:26 +0100490 When the attribute *closeOnError* is set to True - if a socket error
491 occurs, the socket is silently closed and then reopened on the next
492 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000493 """
494 logging.Handler.__init__(self)
495 self.host = host
496 self.port = port
Vinay Sajip5421f352013-09-27 18:18:28 +0100497 if port is None:
498 self.address = host
499 else:
500 self.address = (host, port)
Guido van Rossum57102f82002-11-13 16:15:58 +0000501 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100502 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000503 self.retryTime = None
504 #
505 # Exponential backoff parameters.
506 #
507 self.retryStart = 1.0
508 self.retryMax = 30.0
509 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000510
Guido van Rossumd8faa362007-04-27 19:54:29 +0000511 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000512 """
513 A factory method which allows subclasses to define the precise
514 type of socket they want.
515 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100516 if self.port is not None:
517 result = socket.create_connection(self.address, timeout=timeout)
518 else:
519 result = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
520 result.settimeout(timeout)
521 result.connect(self.address)
522 return result
Guido van Rossum57102f82002-11-13 16:15:58 +0000523
Vinay Sajip48cfe382004-02-20 13:17:27 +0000524 def createSocket(self):
525 """
526 Try to create a socket, using an exponential backoff with
527 a max retry time. Thanks to Robert Olson for the original patch
528 (SF #815911) which has been slightly refactored.
529 """
530 now = time.time()
531 # Either retryTime is None, in which case this
532 # is the first time back after a disconnect, or
533 # we've waited long enough.
534 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100535 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000536 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000537 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000538 if attempt:
539 try:
540 self.sock = self.makeSocket()
541 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200542 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000543 #Creation failed, so set the retry time and return.
544 if self.retryTime is None:
545 self.retryPeriod = self.retryStart
546 else:
547 self.retryPeriod = self.retryPeriod * self.retryFactor
548 if self.retryPeriod > self.retryMax:
549 self.retryPeriod = self.retryMax
550 self.retryTime = now + self.retryPeriod
551
Guido van Rossum57102f82002-11-13 16:15:58 +0000552 def send(self, s):
553 """
554 Send a pickled string to the socket.
555
556 This function allows for partial sends which can happen when the
557 network is busy.
558 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000559 if self.sock is None:
560 self.createSocket()
561 #self.sock can be None either because we haven't reached the retry
562 #time yet, or because we have reached the retry time and retried,
563 #but are still unable to connect.
564 if self.sock:
565 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200566 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200567 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000568 self.sock.close()
569 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000570
571 def makePickle(self, record):
572 """
573 Pickles the record in binary format with a length prefix, and
574 returns it ready for transmission across the socket.
575 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000576 ei = record.exc_info
577 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100578 # just to get traceback text into record.exc_text ...
579 dummy = self.format(record)
580 # See issue #14436: If msg or args are objects, they may not be
581 # available on the receiving end. So we convert the msg % args
582 # to a string, save it as msg and zap the args.
583 d = dict(record.__dict__)
584 d['msg'] = record.getMessage()
585 d['args'] = None
586 d['exc_info'] = None
587 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000588 slen = struct.pack(">L", len(s))
589 return slen + s
590
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000591 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000592 """
593 Handle an error during logging.
594
595 An error has occurred during logging. Most likely cause -
596 connection lost. Close the socket so that we can retry on the
597 next event.
598 """
599 if self.closeOnError and self.sock:
600 self.sock.close()
601 self.sock = None #try to reconnect next time
602 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000603 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000604
605 def emit(self, record):
606 """
607 Emit a record.
608
609 Pickles the record and writes it to the socket in binary format.
610 If there is an error with the socket, silently drop the packet.
611 If there was a problem with the socket, re-establishes the
612 socket.
613 """
614 try:
615 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000616 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100617 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000618 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000619
620 def close(self):
621 """
622 Closes the socket.
623 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000624 self.acquire()
625 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000626 if self.sock:
627 self.sock.close()
628 self.sock = None
629 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000630 finally:
631 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000632
633class DatagramHandler(SocketHandler):
634 """
635 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000636 a datagram socket. The pickle which is sent is that of the LogRecord's
637 attribute dictionary (__dict__), so that the receiver does not need to
638 have the logging module installed in order to process the logging event.
639
640 To unpickle the record at the receiving end into a LogRecord, use the
641 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000642
643 """
644 def __init__(self, host, port):
645 """
646 Initializes the handler with a specific host address and port.
647 """
648 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100649 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000650
651 def makeSocket(self):
652 """
653 The factory method of SocketHandler is here overridden to create
654 a UDP socket (SOCK_DGRAM).
655 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100656 if self.port is None:
657 family = socket.AF_UNIX
658 else:
659 family = socket.AF_INET
660 s = socket.socket(family, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000661 return s
662
663 def send(self, s):
664 """
665 Send a pickled string to a socket.
666
667 This function no longer allows for partial sends which can happen
668 when the network is busy - UDP does not guarantee delivery and
669 can deliver packets out of sequence.
670 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000671 if self.sock is None:
672 self.createSocket()
Vinay Sajip5421f352013-09-27 18:18:28 +0100673 self.sock.sendto(s, self.address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000674
675class SysLogHandler(logging.Handler):
676 """
677 A handler class which sends formatted logging records to a syslog
678 server. Based on Sam Rushing's syslog module:
679 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
680 Contributed by Nicolas Untz (after which minor refactoring changes
681 have been made).
682 """
683
684 # from <linux/sys/syslog.h>:
685 # ======================================================================
686 # priorities/facilities are encoded into a single 32-bit quantity, where
687 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
688 # facility (0-big number). Both the priorities and the facilities map
689 # roughly one-to-one to strings in the syslogd(8) source code. This
690 # mapping is included in this file.
691 #
692 # priorities (these are ordered)
693
694 LOG_EMERG = 0 # system is unusable
695 LOG_ALERT = 1 # action must be taken immediately
696 LOG_CRIT = 2 # critical conditions
697 LOG_ERR = 3 # error conditions
698 LOG_WARNING = 4 # warning conditions
699 LOG_NOTICE = 5 # normal but significant condition
700 LOG_INFO = 6 # informational
701 LOG_DEBUG = 7 # debug-level messages
702
703 # facility codes
704 LOG_KERN = 0 # kernel messages
705 LOG_USER = 1 # random user-level messages
706 LOG_MAIL = 2 # mail system
707 LOG_DAEMON = 3 # system daemons
708 LOG_AUTH = 4 # security/authorization messages
709 LOG_SYSLOG = 5 # messages generated internally by syslogd
710 LOG_LPR = 6 # line printer subsystem
711 LOG_NEWS = 7 # network news subsystem
712 LOG_UUCP = 8 # UUCP subsystem
713 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000714 LOG_AUTHPRIV = 10 # security/authorization messages (private)
715 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000716
717 # other codes through 15 reserved for system use
718 LOG_LOCAL0 = 16 # reserved for local use
719 LOG_LOCAL1 = 17 # reserved for local use
720 LOG_LOCAL2 = 18 # reserved for local use
721 LOG_LOCAL3 = 19 # reserved for local use
722 LOG_LOCAL4 = 20 # reserved for local use
723 LOG_LOCAL5 = 21 # reserved for local use
724 LOG_LOCAL6 = 22 # reserved for local use
725 LOG_LOCAL7 = 23 # reserved for local use
726
727 priority_names = {
728 "alert": LOG_ALERT,
729 "crit": LOG_CRIT,
730 "critical": LOG_CRIT,
731 "debug": LOG_DEBUG,
732 "emerg": LOG_EMERG,
733 "err": LOG_ERR,
734 "error": LOG_ERR, # DEPRECATED
735 "info": LOG_INFO,
736 "notice": LOG_NOTICE,
737 "panic": LOG_EMERG, # DEPRECATED
738 "warn": LOG_WARNING, # DEPRECATED
739 "warning": LOG_WARNING,
740 }
741
742 facility_names = {
743 "auth": LOG_AUTH,
744 "authpriv": LOG_AUTHPRIV,
745 "cron": LOG_CRON,
746 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000747 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000748 "kern": LOG_KERN,
749 "lpr": LOG_LPR,
750 "mail": LOG_MAIL,
751 "news": LOG_NEWS,
752 "security": LOG_AUTH, # DEPRECATED
753 "syslog": LOG_SYSLOG,
754 "user": LOG_USER,
755 "uucp": LOG_UUCP,
756 "local0": LOG_LOCAL0,
757 "local1": LOG_LOCAL1,
758 "local2": LOG_LOCAL2,
759 "local3": LOG_LOCAL3,
760 "local4": LOG_LOCAL4,
761 "local5": LOG_LOCAL5,
762 "local6": LOG_LOCAL6,
763 "local7": LOG_LOCAL7,
764 }
765
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000766 #The map below appears to be trivially lowercasing the key. However,
767 #there's more to it than meets the eye - in some locales, lowercasing
768 #gives unexpected results. See SF #1524081: in the Turkish locale,
769 #"INFO".lower() != "info"
770 priority_map = {
771 "DEBUG" : "debug",
772 "INFO" : "info",
773 "WARNING" : "warning",
774 "ERROR" : "error",
775 "CRITICAL" : "critical"
776 }
777
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000778 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100779 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000780 """
781 Initialize a handler.
782
Guido van Rossume7ba4952007-06-06 23:52:48 +0000783 If address is specified as a string, a UNIX socket is used. To log to a
784 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100785 If facility is not specified, LOG_USER is used. If socktype is
786 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
787 socket type will be used. For Unix sockets, you can also specify a
788 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
789 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000790 """
791 logging.Handler.__init__(self)
792
793 self.address = address
794 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000795 self.socktype = socktype
796
Guido van Rossum13257902007-06-07 23:15:56 +0000797 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100798 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000799 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000800 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100801 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100802 if socktype is None:
803 socktype = socket.SOCK_DGRAM
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000804 self.socket = socket.socket(socket.AF_INET, socktype)
805 if socktype == socket.SOCK_STREAM:
806 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100807 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000808 self.formatter = None
809
Vinay Sajipa1974c12005-01-13 08:23:56 +0000810 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100811 use_socktype = self.socktype
812 if use_socktype is None:
813 use_socktype = socket.SOCK_DGRAM
814 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000815 try:
816 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100817 # it worked, so set self.socktype to the used type
818 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200819 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000820 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100821 if self.socktype is not None:
822 # user didn't specify falling back, so fail
823 raise
824 use_socktype = socket.SOCK_STREAM
825 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
826 try:
827 self.socket.connect(address)
828 # it worked, so set self.socktype to the used type
829 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100830 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100831 self.socket.close()
832 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000833
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000834 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000835 """
836 Encode the facility and priority. You can pass in strings or
837 integers - if strings are passed, the facility_names and
838 priority_names mapping dictionaries are used to convert them to
839 integers.
840 """
Guido van Rossum13257902007-06-07 23:15:56 +0000841 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000842 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000843 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000844 priority = self.priority_names[priority]
845 return (facility << 3) | priority
846
847 def close (self):
848 """
849 Closes the socket.
850 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000851 self.acquire()
852 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000853 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000854 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000855 finally:
856 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000857
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000858 def mapPriority(self, levelName):
859 """
860 Map a logging level name to a key in the priority_names map.
861 This is useful in two scenarios: when custom levels are being
862 used, and in the case where you can't do a straightforward
863 mapping by lowercasing the logging level name because of locale-
864 specific issues (see SF #1524081).
865 """
866 return self.priority_map.get(levelName, "warning")
867
Vinay Sajip2353e352011-06-27 15:40:06 +0100868 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100869 append_nul = True # some old syslog daemons expect a NUL terminator
870
Guido van Rossum57102f82002-11-13 16:15:58 +0000871 def emit(self, record):
872 """
873 Emit a record.
874
875 The record is formatted, and then sent to the syslog server. If
876 exception information is present, it is NOT sent to the server.
877 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100878 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100879 if self.ident:
880 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100881 if self.append_nul:
882 msg += '\000'
Victor Stinner765531d2013-03-26 01:11:54 +0100883
884 # We need to convert record level to lowercase, maybe this will
885 # change in the future.
Vinay Sajip467d12f2010-08-19 22:20:22 +0000886 prio = '<%d>' % self.encodePriority(self.facility,
887 self.mapPriority(record.levelname))
888 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000889 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000890 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000891 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000892 try:
893 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000894 try:
895 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200896 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100897 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000898 self._connect_unixsocket(self.address)
899 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000900 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000901 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000902 else:
903 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100904 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000905 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000906
907class SMTPHandler(logging.Handler):
908 """
909 A handler class which sends an SMTP email for each logging event.
910 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000911 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100912 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000913 """
914 Initialize the handler.
915
916 Initialize the instance with the from and to addresses and subject
917 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000918 (host, port) tuple format for the mailhost argument. To specify
919 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000920 for the credentials argument. To specify the use of a secure
921 protocol (TLS), pass in a tuple for the secure argument. This will
922 only be used when authentication credentials are supplied. The tuple
923 will be either an empty tuple, or a single-value tuple with the name
924 of a keyfile, or a 2-value tuple with the names of the keyfile and
925 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000926 A timeout in seconds can be specified for the SMTP connection (the
927 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000928 """
929 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000930 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000931 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000932 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000933 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000934 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000935 self.username, self.password = credentials
936 else:
937 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000938 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000939 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000940 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000941 self.toaddrs = toaddrs
942 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000943 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000944 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000945
946 def getSubject(self, record):
947 """
948 Determine the subject for the email.
949
950 If you want to specify a subject line which is record-dependent,
951 override this method.
952 """
953 return self.subject
954
955 def emit(self, record):
956 """
957 Emit a record.
958
959 Format the record and send it to the specified addressees.
960 """
961 try:
962 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000963 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000964 port = self.mailport
965 if not port:
966 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000967 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000968 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000969 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 +0000970 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000971 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000972 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000973 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000974 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000975 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000976 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000977 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000978 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000979 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000980 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
981 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100982 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000983 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000984
985class NTEventLogHandler(logging.Handler):
986 """
987 A handler class which sends events to the NT Event Log. Adds a
988 registry entry for the specified application name. If no dllname is
989 provided, win32service.pyd (which contains some basic message
990 placeholders) is used. Note that use of these placeholders will make
991 your event logs big, as the entire message source is held in the log.
992 If you want slimmer logs, you have to pass in the name of your own DLL
993 which contains the message definitions you want to use in the event log.
994 """
995 def __init__(self, appname, dllname=None, logtype="Application"):
996 logging.Handler.__init__(self)
997 try:
998 import win32evtlogutil, win32evtlog
999 self.appname = appname
1000 self._welu = win32evtlogutil
1001 if not dllname:
1002 dllname = os.path.split(self._welu.__file__)
1003 dllname = os.path.split(dllname[0])
1004 dllname = os.path.join(dllname[0], r'win32service.pyd')
1005 self.dllname = dllname
1006 self.logtype = logtype
1007 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1008 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1009 self.typemap = {
1010 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1011 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001012 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001013 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1014 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1015 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001016 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001017 print("The Python Win32 extensions for NT (service, event "\
1018 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001019 self._welu = None
1020
1021 def getMessageID(self, record):
1022 """
1023 Return the message ID for the event record. If you are using your
1024 own messages, you could do this by having the msg passed to the
1025 logger being an ID rather than a formatting string. Then, in here,
1026 you could use a dictionary lookup to get the message ID. This
1027 version returns 1, which is the base message ID in win32service.pyd.
1028 """
1029 return 1
1030
1031 def getEventCategory(self, record):
1032 """
1033 Return the event category for the record.
1034
1035 Override this if you want to specify your own categories. This version
1036 returns 0.
1037 """
1038 return 0
1039
1040 def getEventType(self, record):
1041 """
1042 Return the event type for the record.
1043
1044 Override this if you want to specify your own types. This version does
1045 a mapping using the handler's typemap attribute, which is set up in
1046 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001047 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001048 either need to override this method or place a suitable dictionary in
1049 the handler's typemap attribute.
1050 """
1051 return self.typemap.get(record.levelno, self.deftype)
1052
1053 def emit(self, record):
1054 """
1055 Emit a record.
1056
1057 Determine the message ID, event category and event type. Then
1058 log the message in the NT event log.
1059 """
1060 if self._welu:
1061 try:
1062 id = self.getMessageID(record)
1063 cat = self.getEventCategory(record)
1064 type = self.getEventType(record)
1065 msg = self.format(record)
1066 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001067 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001068 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001069
1070 def close(self):
1071 """
1072 Clean up this handler.
1073
1074 You can remove the application name from the registry as a
1075 source of event log entries. However, if you do this, you will
1076 not be able to see the events as you intended in the Event Log
1077 Viewer - it needs to be able to access the registry to get the
1078 DLL name.
1079 """
1080 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001081 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001082
1083class HTTPHandler(logging.Handler):
1084 """
1085 A class which sends records to a Web server, using either GET or
1086 POST semantics.
1087 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001088 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001089 """
1090 Initialize the instance with the host, the request URL, and the method
1091 ("GET" or "POST")
1092 """
1093 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001094 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001095 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001096 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001097 self.host = host
1098 self.url = url
1099 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001100 self.secure = secure
1101 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001102
Neal Norwitzf297bd12003-04-23 03:49:43 +00001103 def mapLogRecord(self, record):
1104 """
1105 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001106 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001107 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001108 """
1109 return record.__dict__
1110
Guido van Rossum57102f82002-11-13 16:15:58 +00001111 def emit(self, record):
1112 """
1113 Emit a record.
1114
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001115 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001116 """
1117 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001118 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001119 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001120 if self.secure:
1121 h = http.client.HTTPSConnection(host)
1122 else:
1123 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001124 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001125 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001126 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001127 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001128 sep = '&'
1129 else:
1130 sep = '?'
1131 url = url + "%c%s" % (sep, data)
1132 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001133 # support multiple hosts on one IP address...
1134 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001135 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001136 if i >= 0:
1137 host = host[:i]
1138 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001139 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001140 h.putheader("Content-type",
1141 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001142 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001143 if self.credentials:
1144 import base64
1145 s = ('u%s:%s' % self.credentials).encode('utf-8')
1146 s = 'Basic ' + base64.b64encode(s).strip()
1147 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001148 h.endheaders()
1149 if self.method == "POST":
1150 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001151 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001152 except Exception:
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 Sajip8cf4eb12012-10-09 08:06:13 +01001332 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001333 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001334
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001335if threading:
1336 class QueueListener(object):
1337 """
1338 This class implements an internal threaded listener which watches for
1339 LogRecords being added to a queue, removes them and passes them to a
1340 list of handlers for processing.
1341 """
1342 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001343
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001344 def __init__(self, queue, *handlers):
1345 """
1346 Initialise an instance with the specified queue and
1347 handlers.
1348 """
1349 self.queue = queue
1350 self.handlers = handlers
1351 self._stop = threading.Event()
1352 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001353
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001354 def dequeue(self, block):
1355 """
1356 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001357
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001358 The base implementation uses get. You may want to override this method
1359 if you want to use timeouts or work with custom queue implementations.
1360 """
1361 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001362
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001363 def start(self):
1364 """
1365 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001366
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001367 This starts up a background thread to monitor the queue for
1368 LogRecords to process.
1369 """
1370 self._thread = t = threading.Thread(target=self._monitor)
1371 t.setDaemon(True)
1372 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001373
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001374 def prepare(self , record):
1375 """
1376 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001377
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001378 This method just returns the passed-in record. You may want to
1379 override this method if you need to do any custom marshalling or
1380 manipulation of the record before passing it to the handlers.
1381 """
1382 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001383
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001384 def handle(self, record):
1385 """
1386 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001387
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001388 This just loops through the handlers offering them the record
1389 to handle.
1390 """
1391 record = self.prepare(record)
1392 for handler in self.handlers:
1393 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001394
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001395 def _monitor(self):
1396 """
1397 Monitor the queue for records, and ask the handler
1398 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001399
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001400 This method runs on a separate, internal thread.
1401 The thread will terminate if it sees a sentinel object in the queue.
1402 """
1403 q = self.queue
1404 has_task_done = hasattr(q, 'task_done')
1405 while not self._stop.isSet():
1406 try:
1407 record = self.dequeue(True)
1408 if record is self._sentinel:
1409 break
1410 self.handle(record)
1411 if has_task_done:
1412 q.task_done()
1413 except queue.Empty:
1414 pass
1415 # There might still be records in the queue.
1416 while True:
1417 try:
1418 record = self.dequeue(False)
1419 if record is self._sentinel:
1420 break
1421 self.handle(record)
1422 if has_task_done:
1423 q.task_done()
1424 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001425 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001426
Victor Stinner59bec362011-05-02 16:14:16 +02001427 def enqueue_sentinel(self):
1428 """
1429 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001430
Victor Stinner59bec362011-05-02 16:14:16 +02001431 The base implementation uses put_nowait. You may want to override this
1432 method if you want to use timeouts or work with custom queue
1433 implementations.
1434 """
1435 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001436
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001437 def stop(self):
1438 """
1439 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001440
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001441 This asks the thread to terminate, and then waits for it to do so.
1442 Note that if you don't call this before your application exits, there
1443 may be some records still left on the queue, which won't be processed.
1444 """
1445 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001446 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001447 self._thread.join()
1448 self._thread = None