blob: 0ae6e45abf290b87582c45c5a5a2ff25c43c3dd9 [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
497 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100498 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000499 self.retryTime = None
500 #
501 # Exponential backoff parameters.
502 #
503 self.retryStart = 1.0
504 self.retryMax = 30.0
505 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000506
Guido van Rossumd8faa362007-04-27 19:54:29 +0000507 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000508 """
509 A factory method which allows subclasses to define the precise
510 type of socket they want.
511 """
Giampaolo Rodola'35eeb302012-10-15 16:57:37 +0200512 return socket.create_connection((self.host, self.port), timeout=timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000513
Vinay Sajip48cfe382004-02-20 13:17:27 +0000514 def createSocket(self):
515 """
516 Try to create a socket, using an exponential backoff with
517 a max retry time. Thanks to Robert Olson for the original patch
518 (SF #815911) which has been slightly refactored.
519 """
520 now = time.time()
521 # Either retryTime is None, in which case this
522 # is the first time back after a disconnect, or
523 # we've waited long enough.
524 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100525 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000526 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000527 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000528 if attempt:
529 try:
530 self.sock = self.makeSocket()
531 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200532 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000533 #Creation failed, so set the retry time and return.
534 if self.retryTime is None:
535 self.retryPeriod = self.retryStart
536 else:
537 self.retryPeriod = self.retryPeriod * self.retryFactor
538 if self.retryPeriod > self.retryMax:
539 self.retryPeriod = self.retryMax
540 self.retryTime = now + self.retryPeriod
541
Guido van Rossum57102f82002-11-13 16:15:58 +0000542 def send(self, s):
543 """
544 Send a pickled string to the socket.
545
546 This function allows for partial sends which can happen when the
547 network is busy.
548 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000549 if self.sock is None:
550 self.createSocket()
551 #self.sock can be None either because we haven't reached the retry
552 #time yet, or because we have reached the retry time and retried,
553 #but are still unable to connect.
554 if self.sock:
555 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200556 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200557 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000558 self.sock.close()
559 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000560
561 def makePickle(self, record):
562 """
563 Pickles the record in binary format with a length prefix, and
564 returns it ready for transmission across the socket.
565 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000566 ei = record.exc_info
567 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100568 # just to get traceback text into record.exc_text ...
569 dummy = self.format(record)
570 # See issue #14436: If msg or args are objects, they may not be
571 # available on the receiving end. So we convert the msg % args
572 # to a string, save it as msg and zap the args.
573 d = dict(record.__dict__)
574 d['msg'] = record.getMessage()
575 d['args'] = None
576 d['exc_info'] = None
577 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000578 slen = struct.pack(">L", len(s))
579 return slen + s
580
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000581 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000582 """
583 Handle an error during logging.
584
585 An error has occurred during logging. Most likely cause -
586 connection lost. Close the socket so that we can retry on the
587 next event.
588 """
589 if self.closeOnError and self.sock:
590 self.sock.close()
591 self.sock = None #try to reconnect next time
592 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000593 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000594
595 def emit(self, record):
596 """
597 Emit a record.
598
599 Pickles the record and writes it to the socket in binary format.
600 If there is an error with the socket, silently drop the packet.
601 If there was a problem with the socket, re-establishes the
602 socket.
603 """
604 try:
605 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000606 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100607 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000608 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000609
610 def close(self):
611 """
612 Closes the socket.
613 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000614 self.acquire()
615 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000616 if self.sock:
617 self.sock.close()
618 self.sock = None
619 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000620 finally:
621 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000622
623class DatagramHandler(SocketHandler):
624 """
625 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000626 a datagram socket. The pickle which is sent is that of the LogRecord's
627 attribute dictionary (__dict__), so that the receiver does not need to
628 have the logging module installed in order to process the logging event.
629
630 To unpickle the record at the receiving end into a LogRecord, use the
631 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000632
633 """
634 def __init__(self, host, port):
635 """
636 Initializes the handler with a specific host address and port.
637 """
638 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100639 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000640
641 def makeSocket(self):
642 """
643 The factory method of SocketHandler is here overridden to create
644 a UDP socket (SOCK_DGRAM).
645 """
646 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
647 return s
648
649 def send(self, s):
650 """
651 Send a pickled string to a socket.
652
653 This function no longer allows for partial sends which can happen
654 when the network is busy - UDP does not guarantee delivery and
655 can deliver packets out of sequence.
656 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000657 if self.sock is None:
658 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000659 self.sock.sendto(s, (self.host, self.port))
660
661class SysLogHandler(logging.Handler):
662 """
663 A handler class which sends formatted logging records to a syslog
664 server. Based on Sam Rushing's syslog module:
665 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
666 Contributed by Nicolas Untz (after which minor refactoring changes
667 have been made).
668 """
669
670 # from <linux/sys/syslog.h>:
671 # ======================================================================
672 # priorities/facilities are encoded into a single 32-bit quantity, where
673 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
674 # facility (0-big number). Both the priorities and the facilities map
675 # roughly one-to-one to strings in the syslogd(8) source code. This
676 # mapping is included in this file.
677 #
678 # priorities (these are ordered)
679
680 LOG_EMERG = 0 # system is unusable
681 LOG_ALERT = 1 # action must be taken immediately
682 LOG_CRIT = 2 # critical conditions
683 LOG_ERR = 3 # error conditions
684 LOG_WARNING = 4 # warning conditions
685 LOG_NOTICE = 5 # normal but significant condition
686 LOG_INFO = 6 # informational
687 LOG_DEBUG = 7 # debug-level messages
688
689 # facility codes
690 LOG_KERN = 0 # kernel messages
691 LOG_USER = 1 # random user-level messages
692 LOG_MAIL = 2 # mail system
693 LOG_DAEMON = 3 # system daemons
694 LOG_AUTH = 4 # security/authorization messages
695 LOG_SYSLOG = 5 # messages generated internally by syslogd
696 LOG_LPR = 6 # line printer subsystem
697 LOG_NEWS = 7 # network news subsystem
698 LOG_UUCP = 8 # UUCP subsystem
699 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000700 LOG_AUTHPRIV = 10 # security/authorization messages (private)
701 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000702
703 # other codes through 15 reserved for system use
704 LOG_LOCAL0 = 16 # reserved for local use
705 LOG_LOCAL1 = 17 # reserved for local use
706 LOG_LOCAL2 = 18 # reserved for local use
707 LOG_LOCAL3 = 19 # reserved for local use
708 LOG_LOCAL4 = 20 # reserved for local use
709 LOG_LOCAL5 = 21 # reserved for local use
710 LOG_LOCAL6 = 22 # reserved for local use
711 LOG_LOCAL7 = 23 # reserved for local use
712
713 priority_names = {
714 "alert": LOG_ALERT,
715 "crit": LOG_CRIT,
716 "critical": LOG_CRIT,
717 "debug": LOG_DEBUG,
718 "emerg": LOG_EMERG,
719 "err": LOG_ERR,
720 "error": LOG_ERR, # DEPRECATED
721 "info": LOG_INFO,
722 "notice": LOG_NOTICE,
723 "panic": LOG_EMERG, # DEPRECATED
724 "warn": LOG_WARNING, # DEPRECATED
725 "warning": LOG_WARNING,
726 }
727
728 facility_names = {
729 "auth": LOG_AUTH,
730 "authpriv": LOG_AUTHPRIV,
731 "cron": LOG_CRON,
732 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000733 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000734 "kern": LOG_KERN,
735 "lpr": LOG_LPR,
736 "mail": LOG_MAIL,
737 "news": LOG_NEWS,
738 "security": LOG_AUTH, # DEPRECATED
739 "syslog": LOG_SYSLOG,
740 "user": LOG_USER,
741 "uucp": LOG_UUCP,
742 "local0": LOG_LOCAL0,
743 "local1": LOG_LOCAL1,
744 "local2": LOG_LOCAL2,
745 "local3": LOG_LOCAL3,
746 "local4": LOG_LOCAL4,
747 "local5": LOG_LOCAL5,
748 "local6": LOG_LOCAL6,
749 "local7": LOG_LOCAL7,
750 }
751
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000752 #The map below appears to be trivially lowercasing the key. However,
753 #there's more to it than meets the eye - in some locales, lowercasing
754 #gives unexpected results. See SF #1524081: in the Turkish locale,
755 #"INFO".lower() != "info"
756 priority_map = {
757 "DEBUG" : "debug",
758 "INFO" : "info",
759 "WARNING" : "warning",
760 "ERROR" : "error",
761 "CRITICAL" : "critical"
762 }
763
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000764 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100765 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000766 """
767 Initialize a handler.
768
Guido van Rossume7ba4952007-06-06 23:52:48 +0000769 If address is specified as a string, a UNIX socket is used. To log to a
770 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100771 If facility is not specified, LOG_USER is used. If socktype is
772 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
773 socket type will be used. For Unix sockets, you can also specify a
774 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
775 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000776 """
777 logging.Handler.__init__(self)
778
779 self.address = address
780 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000781 self.socktype = socktype
782
Guido van Rossum13257902007-06-07 23:15:56 +0000783 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100784 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000785 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000786 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100787 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100788 if socktype is None:
789 socktype = socket.SOCK_DGRAM
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000790 self.socket = socket.socket(socket.AF_INET, socktype)
791 if socktype == socket.SOCK_STREAM:
792 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100793 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000794 self.formatter = None
795
Vinay Sajipa1974c12005-01-13 08:23:56 +0000796 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100797 use_socktype = self.socktype
798 if use_socktype is None:
799 use_socktype = socket.SOCK_DGRAM
800 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000801 try:
802 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100803 # it worked, so set self.socktype to the used type
804 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200805 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000806 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100807 if self.socktype is not None:
808 # user didn't specify falling back, so fail
809 raise
810 use_socktype = socket.SOCK_STREAM
811 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
812 try:
813 self.socket.connect(address)
814 # it worked, so set self.socktype to the used type
815 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100816 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100817 self.socket.close()
818 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000819
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000820 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000821 """
822 Encode the facility and priority. You can pass in strings or
823 integers - if strings are passed, the facility_names and
824 priority_names mapping dictionaries are used to convert them to
825 integers.
826 """
Guido van Rossum13257902007-06-07 23:15:56 +0000827 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000828 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000829 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000830 priority = self.priority_names[priority]
831 return (facility << 3) | priority
832
833 def close (self):
834 """
835 Closes the socket.
836 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000837 self.acquire()
838 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000839 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000840 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000841 finally:
842 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000843
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000844 def mapPriority(self, levelName):
845 """
846 Map a logging level name to a key in the priority_names map.
847 This is useful in two scenarios: when custom levels are being
848 used, and in the case where you can't do a straightforward
849 mapping by lowercasing the logging level name because of locale-
850 specific issues (see SF #1524081).
851 """
852 return self.priority_map.get(levelName, "warning")
853
Vinay Sajip2353e352011-06-27 15:40:06 +0100854 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100855 append_nul = True # some old syslog daemons expect a NUL terminator
856
Guido van Rossum57102f82002-11-13 16:15:58 +0000857 def emit(self, record):
858 """
859 Emit a record.
860
861 The record is formatted, and then sent to the syslog server. If
862 exception information is present, it is NOT sent to the server.
863 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100864 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100865 if self.ident:
866 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100867 if self.append_nul:
868 msg += '\000'
Victor Stinner765531d2013-03-26 01:11:54 +0100869
870 # We need to convert record level to lowercase, maybe this will
871 # change in the future.
Vinay Sajip467d12f2010-08-19 22:20:22 +0000872 prio = '<%d>' % self.encodePriority(self.facility,
873 self.mapPriority(record.levelname))
874 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000875 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000876 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000877 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000878 try:
879 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000880 try:
881 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200882 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100883 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000884 self._connect_unixsocket(self.address)
885 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000886 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000887 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000888 else:
889 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100890 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000891 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000892
893class SMTPHandler(logging.Handler):
894 """
895 A handler class which sends an SMTP email for each logging event.
896 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000897 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100898 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000899 """
900 Initialize the handler.
901
902 Initialize the instance with the from and to addresses and subject
903 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000904 (host, port) tuple format for the mailhost argument. To specify
905 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000906 for the credentials argument. To specify the use of a secure
907 protocol (TLS), pass in a tuple for the secure argument. This will
908 only be used when authentication credentials are supplied. The tuple
909 will be either an empty tuple, or a single-value tuple with the name
910 of a keyfile, or a 2-value tuple with the names of the keyfile and
911 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000912 A timeout in seconds can be specified for the SMTP connection (the
913 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000914 """
915 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000916 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000917 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000918 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000919 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000920 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000921 self.username, self.password = credentials
922 else:
923 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000924 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000925 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000926 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000927 self.toaddrs = toaddrs
928 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000929 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000930 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000931
932 def getSubject(self, record):
933 """
934 Determine the subject for the email.
935
936 If you want to specify a subject line which is record-dependent,
937 override this method.
938 """
939 return self.subject
940
941 def emit(self, record):
942 """
943 Emit a record.
944
945 Format the record and send it to the specified addressees.
946 """
947 try:
948 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000949 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000950 port = self.mailport
951 if not port:
952 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000953 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000954 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000955 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 +0000956 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000957 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000958 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000959 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000960 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000961 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000962 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000963 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000964 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000965 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000966 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
967 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100968 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000969 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000970
971class NTEventLogHandler(logging.Handler):
972 """
973 A handler class which sends events to the NT Event Log. Adds a
974 registry entry for the specified application name. If no dllname is
975 provided, win32service.pyd (which contains some basic message
976 placeholders) is used. Note that use of these placeholders will make
977 your event logs big, as the entire message source is held in the log.
978 If you want slimmer logs, you have to pass in the name of your own DLL
979 which contains the message definitions you want to use in the event log.
980 """
981 def __init__(self, appname, dllname=None, logtype="Application"):
982 logging.Handler.__init__(self)
983 try:
984 import win32evtlogutil, win32evtlog
985 self.appname = appname
986 self._welu = win32evtlogutil
987 if not dllname:
988 dllname = os.path.split(self._welu.__file__)
989 dllname = os.path.split(dllname[0])
990 dllname = os.path.join(dllname[0], r'win32service.pyd')
991 self.dllname = dllname
992 self.logtype = logtype
993 self._welu.AddSourceToRegistry(appname, dllname, logtype)
994 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
995 self.typemap = {
996 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
997 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000998 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000999 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1000 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1001 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001002 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001003 print("The Python Win32 extensions for NT (service, event "\
1004 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001005 self._welu = None
1006
1007 def getMessageID(self, record):
1008 """
1009 Return the message ID for the event record. If you are using your
1010 own messages, you could do this by having the msg passed to the
1011 logger being an ID rather than a formatting string. Then, in here,
1012 you could use a dictionary lookup to get the message ID. This
1013 version returns 1, which is the base message ID in win32service.pyd.
1014 """
1015 return 1
1016
1017 def getEventCategory(self, record):
1018 """
1019 Return the event category for the record.
1020
1021 Override this if you want to specify your own categories. This version
1022 returns 0.
1023 """
1024 return 0
1025
1026 def getEventType(self, record):
1027 """
1028 Return the event type for the record.
1029
1030 Override this if you want to specify your own types. This version does
1031 a mapping using the handler's typemap attribute, which is set up in
1032 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001033 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001034 either need to override this method or place a suitable dictionary in
1035 the handler's typemap attribute.
1036 """
1037 return self.typemap.get(record.levelno, self.deftype)
1038
1039 def emit(self, record):
1040 """
1041 Emit a record.
1042
1043 Determine the message ID, event category and event type. Then
1044 log the message in the NT event log.
1045 """
1046 if self._welu:
1047 try:
1048 id = self.getMessageID(record)
1049 cat = self.getEventCategory(record)
1050 type = self.getEventType(record)
1051 msg = self.format(record)
1052 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001053 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001054 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001055
1056 def close(self):
1057 """
1058 Clean up this handler.
1059
1060 You can remove the application name from the registry as a
1061 source of event log entries. However, if you do this, you will
1062 not be able to see the events as you intended in the Event Log
1063 Viewer - it needs to be able to access the registry to get the
1064 DLL name.
1065 """
1066 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001067 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001068
1069class HTTPHandler(logging.Handler):
1070 """
1071 A class which sends records to a Web server, using either GET or
1072 POST semantics.
1073 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001074 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001075 """
1076 Initialize the instance with the host, the request URL, and the method
1077 ("GET" or "POST")
1078 """
1079 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001080 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001081 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001082 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001083 self.host = host
1084 self.url = url
1085 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001086 self.secure = secure
1087 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001088
Neal Norwitzf297bd12003-04-23 03:49:43 +00001089 def mapLogRecord(self, record):
1090 """
1091 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001092 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001093 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001094 """
1095 return record.__dict__
1096
Guido van Rossum57102f82002-11-13 16:15:58 +00001097 def emit(self, record):
1098 """
1099 Emit a record.
1100
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001101 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001102 """
1103 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001104 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001105 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001106 if self.secure:
1107 h = http.client.HTTPSConnection(host)
1108 else:
1109 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001110 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001111 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001112 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001113 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001114 sep = '&'
1115 else:
1116 sep = '?'
1117 url = url + "%c%s" % (sep, data)
1118 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001119 # support multiple hosts on one IP address...
1120 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001121 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001122 if i >= 0:
1123 host = host[:i]
1124 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001125 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001126 h.putheader("Content-type",
1127 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001128 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001129 if self.credentials:
1130 import base64
1131 s = ('u%s:%s' % self.credentials).encode('utf-8')
1132 s = 'Basic ' + base64.b64encode(s).strip()
1133 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001134 h.endheaders()
1135 if self.method == "POST":
1136 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001137 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001138 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001139 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001140
1141class BufferingHandler(logging.Handler):
1142 """
1143 A handler class which buffers logging records in memory. Whenever each
1144 record is added to the buffer, a check is made to see if the buffer should
1145 be flushed. If it should, then flush() is expected to do what's needed.
1146 """
1147 def __init__(self, capacity):
1148 """
1149 Initialize the handler with the buffer size.
1150 """
1151 logging.Handler.__init__(self)
1152 self.capacity = capacity
1153 self.buffer = []
1154
1155 def shouldFlush(self, record):
1156 """
1157 Should the handler flush its buffer?
1158
1159 Returns true if the buffer is up to capacity. This method can be
1160 overridden to implement custom flushing strategies.
1161 """
1162 return (len(self.buffer) >= self.capacity)
1163
1164 def emit(self, record):
1165 """
1166 Emit a record.
1167
1168 Append the record. If shouldFlush() tells us to, call flush() to process
1169 the buffer.
1170 """
1171 self.buffer.append(record)
1172 if self.shouldFlush(record):
1173 self.flush()
1174
1175 def flush(self):
1176 """
1177 Override to implement custom flushing behaviour.
1178
1179 This version just zaps the buffer to empty.
1180 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001181 self.acquire()
1182 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001183 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001184 finally:
1185 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001186
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001187 def close(self):
1188 """
1189 Close the handler.
1190
1191 This version just flushes and chains to the parent class' close().
1192 """
1193 self.flush()
1194 logging.Handler.close(self)
1195
Guido van Rossum57102f82002-11-13 16:15:58 +00001196class MemoryHandler(BufferingHandler):
1197 """
1198 A handler class which buffers logging records in memory, periodically
1199 flushing them to a target handler. Flushing occurs whenever the buffer
1200 is full, or when an event of a certain severity or greater is seen.
1201 """
1202 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1203 """
1204 Initialize the handler with the buffer size, the level at which
1205 flushing should occur and an optional target.
1206
1207 Note that without a target being set either here or via setTarget(),
1208 a MemoryHandler is no use to anyone!
1209 """
1210 BufferingHandler.__init__(self, capacity)
1211 self.flushLevel = flushLevel
1212 self.target = target
1213
1214 def shouldFlush(self, record):
1215 """
1216 Check for buffer full or a record at the flushLevel or higher.
1217 """
1218 return (len(self.buffer) >= self.capacity) or \
1219 (record.levelno >= self.flushLevel)
1220
1221 def setTarget(self, target):
1222 """
1223 Set the target handler for this handler.
1224 """
1225 self.target = target
1226
1227 def flush(self):
1228 """
1229 For a MemoryHandler, flushing means just sending the buffered
1230 records to the target, if there is one. Override if you want
1231 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001232
1233 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001234 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001235 self.acquire()
1236 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001237 if self.target:
1238 for record in self.buffer:
1239 self.target.handle(record)
1240 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001241 finally:
1242 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001243
1244 def close(self):
1245 """
1246 Flush, set the target to None and lose the buffer.
1247 """
1248 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001249 self.acquire()
1250 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001251 self.target = None
1252 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001253 finally:
1254 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001255
1256
1257class QueueHandler(logging.Handler):
1258 """
1259 This handler sends events to a queue. Typically, it would be used together
1260 with a multiprocessing Queue to centralise logging to file in one process
1261 (in a multi-process application), so as to avoid file write contention
1262 between processes.
1263
1264 This code is new in Python 3.2, but this class can be copy pasted into
1265 user code for use with earlier Python versions.
1266 """
1267
1268 def __init__(self, queue):
1269 """
1270 Initialise an instance, using the passed queue.
1271 """
1272 logging.Handler.__init__(self)
1273 self.queue = queue
1274
1275 def enqueue(self, record):
1276 """
1277 Enqueue a record.
1278
1279 The base implementation uses put_nowait. You may want to override
1280 this method if you want to use blocking, timeouts or custom queue
1281 implementations.
1282 """
1283 self.queue.put_nowait(record)
1284
Vinay Sajip0258ce82010-09-22 20:34:53 +00001285 def prepare(self, record):
1286 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001287 Prepares a record for queuing. The object returned by this method is
1288 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001289
1290 The base implementation formats the record to merge the message
1291 and arguments, and removes unpickleable items from the record
1292 in-place.
1293
1294 You might want to override this method if you want to convert
1295 the record to a dict or JSON string, or send a modified copy
1296 of the record while leaving the original intact.
1297 """
1298 # The format operation gets traceback text into record.exc_text
1299 # (if there's exception data), and also puts the message into
1300 # record.message. We can then use this to replace the original
1301 # msg + args, as these might be unpickleable. We also zap the
1302 # exc_info attribute, as it's no longer needed and, if not None,
1303 # will typically not be pickleable.
1304 self.format(record)
1305 record.msg = record.message
1306 record.args = None
1307 record.exc_info = None
1308 return record
1309
Vinay Sajip121a1c42010-09-08 10:46:15 +00001310 def emit(self, record):
1311 """
1312 Emit a record.
1313
Vinay Sajip0637d492010-09-23 08:15:54 +00001314 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001315 """
1316 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001317 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001318 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001319 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001320
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001321if threading:
1322 class QueueListener(object):
1323 """
1324 This class implements an internal threaded listener which watches for
1325 LogRecords being added to a queue, removes them and passes them to a
1326 list of handlers for processing.
1327 """
1328 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001329
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001330 def __init__(self, queue, *handlers):
1331 """
1332 Initialise an instance with the specified queue and
1333 handlers.
1334 """
1335 self.queue = queue
1336 self.handlers = handlers
1337 self._stop = threading.Event()
1338 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001339
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001340 def dequeue(self, block):
1341 """
1342 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001343
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001344 The base implementation uses get. You may want to override this method
1345 if you want to use timeouts or work with custom queue implementations.
1346 """
1347 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001348
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001349 def start(self):
1350 """
1351 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001352
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001353 This starts up a background thread to monitor the queue for
1354 LogRecords to process.
1355 """
1356 self._thread = t = threading.Thread(target=self._monitor)
1357 t.setDaemon(True)
1358 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001359
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001360 def prepare(self , record):
1361 """
1362 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001363
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001364 This method just returns the passed-in record. You may want to
1365 override this method if you need to do any custom marshalling or
1366 manipulation of the record before passing it to the handlers.
1367 """
1368 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001369
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001370 def handle(self, record):
1371 """
1372 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001373
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001374 This just loops through the handlers offering them the record
1375 to handle.
1376 """
1377 record = self.prepare(record)
1378 for handler in self.handlers:
1379 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001380
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001381 def _monitor(self):
1382 """
1383 Monitor the queue for records, and ask the handler
1384 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001385
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001386 This method runs on a separate, internal thread.
1387 The thread will terminate if it sees a sentinel object in the queue.
1388 """
1389 q = self.queue
1390 has_task_done = hasattr(q, 'task_done')
1391 while not self._stop.isSet():
1392 try:
1393 record = self.dequeue(True)
1394 if record is self._sentinel:
1395 break
1396 self.handle(record)
1397 if has_task_done:
1398 q.task_done()
1399 except queue.Empty:
1400 pass
1401 # There might still be records in the queue.
1402 while True:
1403 try:
1404 record = self.dequeue(False)
1405 if record is self._sentinel:
1406 break
1407 self.handle(record)
1408 if has_task_done:
1409 q.task_done()
1410 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001411 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001412
Victor Stinner59bec362011-05-02 16:14:16 +02001413 def enqueue_sentinel(self):
1414 """
1415 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001416
Victor Stinner59bec362011-05-02 16:14:16 +02001417 The base implementation uses put_nowait. You may want to override this
1418 method if you want to use timeouts or work with custom queue
1419 implementations.
1420 """
1421 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001422
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001423 def stop(self):
1424 """
1425 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001426
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001427 This asks the thread to terminate, and then waits for it to do so.
1428 Note that if you don't call this before your application exits, there
1429 may be some records still left on the queue, which won't be processed.
1430 """
1431 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001432 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001433 self._thread.join()
1434 self._thread = None