blob: 02a5fc1283d1182db881ec19dc5100f4a4445a78 [file] [log] [blame]
Vinay Sajip365701a2015-02-09 19:49:00 +00001# Copyright 2001-2015 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 Sajip365701a2015-02-09 19:49:00 +000021Copyright (C) 2001-2015 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
Victor Stinner7fa767e2014-03-20 09:16:38 +010026import logging, socket, os, pickle, struct, time, re
Benjamin Peterson9451a1c2010-03-13 22:30:34 +000027from stat import ST_DEV, ST_INO, ST_MTIME
Vinay Sajipe723e962011-04-15 22:27:17 +010028import queue
Victor Stinnercafa2ef2011-05-02 16:11:28 +020029try:
30 import threading
Brett Cannoncd171c82013-07-04 17:43:24 -040031except ImportError: #pragma: no cover
Victor Stinnercafa2ef2011-05-02 16:11:28 +020032 threading = None
Guido van Rossum57102f82002-11-13 16:15:58 +000033
Guido van Rossum57102f82002-11-13 16:15:58 +000034#
35# Some constants...
36#
37
38DEFAULT_TCP_LOGGING_PORT = 9020
39DEFAULT_UDP_LOGGING_PORT = 9021
40DEFAULT_HTTP_LOGGING_PORT = 9022
41DEFAULT_SOAP_LOGGING_PORT = 9023
42SYSLOG_UDP_PORT = 514
Vinay Sajipcbabd7e2009-10-10 20:32:36 +000043SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000044
Thomas Wouters477c8d52006-05-27 19:21:47 +000045_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
46
Vinay Sajip17c52d82004-07-03 11:48:34 +000047class BaseRotatingHandler(logging.FileHandler):
48 """
49 Base class for handlers that rotate log files at a certain point.
50 Not meant to be instantiated directly. Instead, use RotatingFileHandler
51 or TimedRotatingFileHandler.
52 """
Vinay Sajip23b94d02012-01-04 12:02:26 +000053 def __init__(self, filename, mode, encoding=None, delay=False):
Vinay Sajip17c52d82004-07-03 11:48:34 +000054 """
55 Use the specified filename for streamed logging
56 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000057 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip4600f112005-03-13 09:56:36 +000058 self.mode = mode
59 self.encoding = encoding
Vinay Sajip23b94d02012-01-04 12:02:26 +000060 self.namer = None
61 self.rotator = None
Guido van Rossum57102f82002-11-13 16:15:58 +000062
Vinay Sajip17c52d82004-07-03 11:48:34 +000063 def emit(self, record):
64 """
65 Emit a record.
66
67 Output the record to the file, catering for rollover as described
68 in doRollover().
69 """
Vinay Sajip3970c112004-07-08 10:24:04 +000070 try:
71 if self.shouldRollover(record):
72 self.doRollover()
73 logging.FileHandler.emit(self, record)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +010074 except Exception:
Vinay Sajip3970c112004-07-08 10:24:04 +000075 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000076
Vinay Sajip23b94d02012-01-04 12:02:26 +000077 def rotation_filename(self, default_name):
78 """
79 Modify the filename of a log file when rotating.
80
81 This is provided so that a custom filename can be provided.
82
83 The default implementation calls the 'namer' attribute of the
84 handler, if it's callable, passing the default name to
85 it. If the attribute isn't callable (the default is None), the name
86 is returned unchanged.
87
88 :param default_name: The default name for the log file.
89 """
90 if not callable(self.namer):
91 result = default_name
92 else:
93 result = self.namer(default_name)
94 return result
95
96 def rotate(self, source, dest):
97 """
98 When rotating, rotate the current log.
99
100 The default implementation calls the 'rotator' attribute of the
101 handler, if it's callable, passing the source and dest arguments to
102 it. If the attribute isn't callable (the default is None), the source
103 is simply renamed to the destination.
104
105 :param source: The source filename. This is normally the base
106 filename, e.g. 'test.log'
107 :param dest: The destination filename. This is normally
108 what the source is rotated to, e.g. 'test.log.1'.
109 """
110 if not callable(self.rotator):
Vinay Sajipd8599262013-09-06 10:10:22 +0100111 # Issue 18940: A file may not have been created if delay is True.
112 if os.path.exists(source):
113 os.rename(source, dest)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000114 else:
115 self.rotator(source, dest)
116
Vinay Sajip17c52d82004-07-03 11:48:34 +0000117class RotatingFileHandler(BaseRotatingHandler):
118 """
119 Handler for logging to a set of files, which switches from one file
120 to the next when the current file reaches a certain size.
121 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000122 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
Guido van Rossum57102f82002-11-13 16:15:58 +0000123 """
124 Open the specified file and use it as the stream for logging.
125
126 By default, the file grows indefinitely. You can specify particular
127 values of maxBytes and backupCount to allow the file to rollover at
128 a predetermined size.
129
130 Rollover occurs whenever the current log file is nearly maxBytes in
131 length. If backupCount is >= 1, the system will successively create
132 new files with the same pathname as the base file, but with extensions
133 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
134 and a base file name of "app.log", you would get "app.log",
135 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
136 written to is always "app.log" - when it gets filled up, it is closed
137 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
138 exist, then they are renamed to "app.log.2", "app.log.3" etc.
139 respectively.
140
141 If maxBytes is zero, rollover never occurs.
142 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000143 # If rotation/rollover is wanted, it doesn't make sense to use another
144 # mode. If for example 'w' were specified, then if there were multiple
145 # runs of the calling application, the logs from previous runs would be
146 # lost if the 'w' is respected, because the log file would be truncated
147 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000148 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000149 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000150 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000151 self.maxBytes = maxBytes
152 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000153
154 def doRollover(self):
155 """
156 Do a rollover, as described in __init__().
157 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000158 if self.stream:
159 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000160 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000161 if self.backupCount > 0:
162 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000163 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
164 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
165 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000166 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000167 if os.path.exists(dfn):
168 os.remove(dfn)
169 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000170 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000171 if os.path.exists(dfn):
172 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000173 self.rotate(self.baseFilename, dfn)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100174 if not self.delay:
175 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000176
Vinay Sajip17c52d82004-07-03 11:48:34 +0000177 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000178 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000179 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000180
Vinay Sajip17c52d82004-07-03 11:48:34 +0000181 Basically, see if the supplied record would cause the file to exceed
182 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000183 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000184 if self.stream is None: # delay was set...
185 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000186 if self.maxBytes > 0: # are we rolling over?
187 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000188 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000189 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000190 return 1
191 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000192
Vinay Sajip17c52d82004-07-03 11:48:34 +0000193class TimedRotatingFileHandler(BaseRotatingHandler):
194 """
195 Handler for logging to a file, rotating the log file at certain timed
196 intervals.
197
198 If backupCount is > 0, when rollover is done, no more than backupCount
199 files are kept - the oldest ones are deleted.
200 """
Vinay Sajipa7130792013-04-12 17:04:23 +0100201 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 +0000202 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000203 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000204 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000205 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100206 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000207 # Calculate the real rollover interval, which is just the number of
208 # seconds between rollovers. Also set the filename suffix used when
209 # a rollover occurs. Current 'when' events supported:
210 # S - Seconds
211 # M - Minutes
212 # H - Hours
213 # D - Days
214 # midnight - roll over at midnight
215 # W{0-6} - roll over on a certain day; 0 - Monday
216 #
217 # Case of the 'when' specifier is not important; lower or upper case
218 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000219 if self.when == 'S':
220 self.interval = 1 # one second
221 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000222 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000223 elif self.when == 'M':
224 self.interval = 60 # one minute
225 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000226 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000227 elif self.when == 'H':
228 self.interval = 60 * 60 # one hour
229 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000230 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000231 elif self.when == 'D' or self.when == 'MIDNIGHT':
232 self.interval = 60 * 60 * 24 # one day
233 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000234 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000235 elif self.when.startswith('W'):
236 self.interval = 60 * 60 * 24 * 7 # one week
237 if len(self.when) != 2:
238 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
239 if self.when[1] < '0' or self.when[1] > '6':
240 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
241 self.dayOfWeek = int(self.when[1])
242 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000243 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000244 else:
245 raise ValueError("Invalid rollover interval specified: %s" % self.when)
246
Antoine Pitroufd036452008-08-19 17:56:33 +0000247 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000248 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000249 if os.path.exists(filename):
250 t = os.stat(filename)[ST_MTIME]
251 else:
252 t = int(time.time())
253 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000254
Vinay Sajipaffbd872009-06-11 10:11:47 +0000255 def computeRollover(self, currentTime):
256 """
257 Work out the rollover time based on the specified time.
258 """
259 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000260 # If we are rolling over at midnight or weekly, then the interval is already known.
261 # What we need to figure out is WHEN the next interval is. In other words,
262 # if you are rolling over at midnight, then your base interval is 1 day,
263 # but you want to start that one day clock at midnight, not now. So, we
264 # have to fudge the rolloverAt value in order to trigger the first rollover
265 # at the right time. After that, the regular interval will take care of
266 # the rest. Note that this code doesn't care about leap seconds. :)
267 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
268 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000269 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000270 t = time.gmtime(currentTime)
271 else:
272 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000273 currentHour = t[3]
274 currentMinute = t[4]
275 currentSecond = t[5]
Vinay Sajipa7130792013-04-12 17:04:23 +0100276 currentDay = t[6]
277 # r is the number of seconds left between now and the next rotation
278 if self.atTime is None:
279 rotate_ts = _MIDNIGHT
280 else:
281 rotate_ts = ((self.atTime.hour * 60 + self.atTime.minute)*60 +
282 self.atTime.second)
283
284 r = rotate_ts - ((currentHour * 60 + currentMinute) * 60 +
285 currentSecond)
286 if r < 0:
287 # Rotate time is before the current time (for example when
288 # self.rotateAt is 13:45 and it now 14:15), rotation is
289 # tomorrow.
290 r += _MIDNIGHT
291 currentDay = (currentDay + 1) % 7
Vinay Sajipaffbd872009-06-11 10:11:47 +0000292 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000293 # If we are rolling over on a certain day, add in the number of days until
294 # the next rollover, but offset by 1 since we just calculated the time
295 # until the next day starts. There are three cases:
296 # Case 1) The day to rollover is today; in this case, do nothing
297 # Case 2) The day to rollover is further in the interval (i.e., today is
298 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
299 # next rollover is simply 6 - 2 - 1, or 3.
300 # Case 3) The day to rollover is behind us in the interval (i.e., today
301 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
302 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
303 # number of days left in the current week (1) plus the number
304 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000305 # The calculations described in 2) and 3) above need to have a day added.
306 # This is because the above time calculation takes us to midnight on this
307 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000308 if self.when.startswith('W'):
Vinay Sajipa7130792013-04-12 17:04:23 +0100309 day = currentDay # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000310 if day != self.dayOfWeek:
311 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000312 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000313 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000314 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000315 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
316 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000317 dstNow = t[-1]
318 dstAtRollover = time.localtime(newRolloverAt)[-1]
319 if dstNow != dstAtRollover:
320 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000321 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000322 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000323 addend = 3600
324 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000325 result = newRolloverAt
326 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000327
328 def shouldRollover(self, record):
329 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000330 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000331
332 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000333 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000334 """
335 t = int(time.time())
336 if t >= self.rolloverAt:
337 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000338 return 0
339
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000340 def getFilesToDelete(self):
341 """
342 Determine the files to delete when rolling over.
343
344 More specific than the earlier method, which just used glob.glob().
345 """
346 dirName, baseName = os.path.split(self.baseFilename)
347 fileNames = os.listdir(dirName)
348 result = []
349 prefix = baseName + "."
350 plen = len(prefix)
351 for fileName in fileNames:
352 if fileName[:plen] == prefix:
353 suffix = fileName[plen:]
354 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000355 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000356 result.sort()
357 if len(result) < self.backupCount:
358 result = []
359 else:
360 result = result[:len(result) - self.backupCount]
361 return result
362
Vinay Sajip17c52d82004-07-03 11:48:34 +0000363 def doRollover(self):
364 """
365 do a rollover; in this case, a date/time stamp is appended to the filename
366 when the rollover happens. However, you want the file to be named for the
367 start of the interval, not the current time. If there is a backup count,
368 then we have to get a list of matching filenames, sort them and remove
369 the one with the oldest suffix.
370 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000371 if self.stream:
372 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000373 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000374 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000375 currentTime = int(time.time())
376 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000377 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000378 if self.utc:
379 timeTuple = time.gmtime(t)
380 else:
381 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000382 dstThen = timeTuple[-1]
383 if dstNow != dstThen:
384 if dstNow:
385 addend = 3600
386 else:
387 addend = -3600
388 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000389 dfn = self.rotation_filename(self.baseFilename + "." +
390 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000391 if os.path.exists(dfn):
392 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000393 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000394 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000395 for s in self.getFilesToDelete():
396 os.remove(s)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100397 if not self.delay:
398 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000399 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000400 while newRolloverAt <= currentTime:
401 newRolloverAt = newRolloverAt + self.interval
402 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000403 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000404 dstAtRollover = time.localtime(newRolloverAt)[-1]
405 if dstNow != dstAtRollover:
406 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000407 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000408 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000409 addend = 3600
410 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000411 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000412
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000413class WatchedFileHandler(logging.FileHandler):
414 """
415 A handler for logging to a file, which watches the file
416 to see if it has changed while in use. This can happen because of
417 usage of programs such as newsyslog and logrotate which perform
418 log file rotation. This handler, intended for use under Unix,
419 watches the file to see if it has changed since the last emit.
420 (A file has changed if its device or inode have changed.)
421 If it has changed, the old file stream is closed, and the file
422 opened to get a new stream.
423
424 This handler is not appropriate for use under Windows, because
425 under Windows open files cannot be moved or renamed - logging
426 opens the files with exclusive locks - and so there is no need
427 for such a handler. Furthermore, ST_INO is not supported under
428 Windows; stat always returns zero for this value.
429
430 This handler is based on a suggestion and patch by Chad J.
431 Schroeder.
432 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000433 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000434 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100435 self.dev, self.ino = -1, -1
436 self._statstream()
437
438 def _statstream(self):
439 if self.stream:
440 sres = os.fstat(self.stream.fileno())
441 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000442
443 def emit(self, record):
444 """
445 Emit a record.
446
447 First check if the underlying file has changed, and if it
448 has, close the old stream and reopen the file to get the
449 current stream.
450 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100451 # Reduce the chance of race conditions by stat'ing by path only
452 # once and then fstat'ing our new fd if we opened a new log stream.
453 # See issue #14632: Thanks to John Mulligan for the problem report
454 # and patch.
455 try:
456 # stat the file by path, checking for existence
457 sres = os.stat(self.baseFilename)
Giampaolo Rodola'0166a282013-02-12 15:14:17 +0100458 except FileNotFoundError:
459 sres = None
Vinay Sajip66b8b082012-04-24 23:25:30 +0100460 # compare file system stat with that of our stream file handle
461 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
462 if self.stream is not None:
463 # we have an open file handle, clean it up
464 self.stream.flush()
465 self.stream.close()
Vinay Sajip02200482014-06-14 10:22:05 +0100466 self.stream = None # See Issue #21742: _open () might fail.
Vinay Sajip66b8b082012-04-24 23:25:30 +0100467 # 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)
Vinay Sajip38c741c2013-10-06 18:36:00 +0100521 try:
522 result.connect(self.address)
523 except OSError:
524 result.close() # Issue 19182
525 raise
Vinay Sajip5421f352013-09-27 18:18:28 +0100526 return result
Guido van Rossum57102f82002-11-13 16:15:58 +0000527
Vinay Sajip48cfe382004-02-20 13:17:27 +0000528 def createSocket(self):
529 """
530 Try to create a socket, using an exponential backoff with
531 a max retry time. Thanks to Robert Olson for the original patch
532 (SF #815911) which has been slightly refactored.
533 """
534 now = time.time()
535 # Either retryTime is None, in which case this
536 # is the first time back after a disconnect, or
537 # we've waited long enough.
538 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100539 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000540 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000541 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000542 if attempt:
543 try:
544 self.sock = self.makeSocket()
545 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200546 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000547 #Creation failed, so set the retry time and return.
548 if self.retryTime is None:
549 self.retryPeriod = self.retryStart
550 else:
551 self.retryPeriod = self.retryPeriod * self.retryFactor
552 if self.retryPeriod > self.retryMax:
553 self.retryPeriod = self.retryMax
554 self.retryTime = now + self.retryPeriod
555
Guido van Rossum57102f82002-11-13 16:15:58 +0000556 def send(self, s):
557 """
558 Send a pickled string to the socket.
559
560 This function allows for partial sends which can happen when the
561 network is busy.
562 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000563 if self.sock is None:
564 self.createSocket()
565 #self.sock can be None either because we haven't reached the retry
566 #time yet, or because we have reached the retry time and retried,
567 #but are still unable to connect.
568 if self.sock:
569 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200570 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200571 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000572 self.sock.close()
573 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000574
575 def makePickle(self, record):
576 """
577 Pickles the record in binary format with a length prefix, and
578 returns it ready for transmission across the socket.
579 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000580 ei = record.exc_info
581 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100582 # just to get traceback text into record.exc_text ...
583 dummy = self.format(record)
584 # See issue #14436: If msg or args are objects, they may not be
585 # available on the receiving end. So we convert the msg % args
586 # to a string, save it as msg and zap the args.
587 d = dict(record.__dict__)
588 d['msg'] = record.getMessage()
589 d['args'] = None
590 d['exc_info'] = None
591 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000592 slen = struct.pack(">L", len(s))
593 return slen + s
594
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000595 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000596 """
597 Handle an error during logging.
598
599 An error has occurred during logging. Most likely cause -
600 connection lost. Close the socket so that we can retry on the
601 next event.
602 """
603 if self.closeOnError and self.sock:
604 self.sock.close()
605 self.sock = None #try to reconnect next time
606 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000607 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000608
609 def emit(self, record):
610 """
611 Emit a record.
612
613 Pickles the record and writes it to the socket in binary format.
614 If there is an error with the socket, silently drop the packet.
615 If there was a problem with the socket, re-establishes the
616 socket.
617 """
618 try:
619 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000620 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100621 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000622 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000623
624 def close(self):
625 """
626 Closes the socket.
627 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000628 self.acquire()
629 try:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300630 sock = self.sock
631 if sock:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000632 self.sock = None
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300633 sock.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000634 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000635 finally:
636 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000637
638class DatagramHandler(SocketHandler):
639 """
640 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000641 a datagram socket. The pickle which is sent is that of the LogRecord's
642 attribute dictionary (__dict__), so that the receiver does not need to
643 have the logging module installed in order to process the logging event.
644
645 To unpickle the record at the receiving end into a LogRecord, use the
646 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000647
648 """
649 def __init__(self, host, port):
650 """
651 Initializes the handler with a specific host address and port.
652 """
653 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100654 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000655
656 def makeSocket(self):
657 """
658 The factory method of SocketHandler is here overridden to create
659 a UDP socket (SOCK_DGRAM).
660 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100661 if self.port is None:
662 family = socket.AF_UNIX
663 else:
664 family = socket.AF_INET
665 s = socket.socket(family, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000666 return s
667
668 def send(self, s):
669 """
670 Send a pickled string to a socket.
671
672 This function no longer allows for partial sends which can happen
673 when the network is busy - UDP does not guarantee delivery and
674 can deliver packets out of sequence.
675 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000676 if self.sock is None:
677 self.createSocket()
Vinay Sajip5421f352013-09-27 18:18:28 +0100678 self.sock.sendto(s, self.address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000679
680class SysLogHandler(logging.Handler):
681 """
682 A handler class which sends formatted logging records to a syslog
683 server. Based on Sam Rushing's syslog module:
684 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
685 Contributed by Nicolas Untz (after which minor refactoring changes
686 have been made).
687 """
688
689 # from <linux/sys/syslog.h>:
690 # ======================================================================
691 # priorities/facilities are encoded into a single 32-bit quantity, where
692 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
693 # facility (0-big number). Both the priorities and the facilities map
694 # roughly one-to-one to strings in the syslogd(8) source code. This
695 # mapping is included in this file.
696 #
697 # priorities (these are ordered)
698
699 LOG_EMERG = 0 # system is unusable
700 LOG_ALERT = 1 # action must be taken immediately
701 LOG_CRIT = 2 # critical conditions
702 LOG_ERR = 3 # error conditions
703 LOG_WARNING = 4 # warning conditions
704 LOG_NOTICE = 5 # normal but significant condition
705 LOG_INFO = 6 # informational
706 LOG_DEBUG = 7 # debug-level messages
707
708 # facility codes
709 LOG_KERN = 0 # kernel messages
710 LOG_USER = 1 # random user-level messages
711 LOG_MAIL = 2 # mail system
712 LOG_DAEMON = 3 # system daemons
713 LOG_AUTH = 4 # security/authorization messages
714 LOG_SYSLOG = 5 # messages generated internally by syslogd
715 LOG_LPR = 6 # line printer subsystem
716 LOG_NEWS = 7 # network news subsystem
717 LOG_UUCP = 8 # UUCP subsystem
718 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000719 LOG_AUTHPRIV = 10 # security/authorization messages (private)
720 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000721
722 # other codes through 15 reserved for system use
723 LOG_LOCAL0 = 16 # reserved for local use
724 LOG_LOCAL1 = 17 # reserved for local use
725 LOG_LOCAL2 = 18 # reserved for local use
726 LOG_LOCAL3 = 19 # reserved for local use
727 LOG_LOCAL4 = 20 # reserved for local use
728 LOG_LOCAL5 = 21 # reserved for local use
729 LOG_LOCAL6 = 22 # reserved for local use
730 LOG_LOCAL7 = 23 # reserved for local use
731
732 priority_names = {
733 "alert": LOG_ALERT,
734 "crit": LOG_CRIT,
735 "critical": LOG_CRIT,
736 "debug": LOG_DEBUG,
737 "emerg": LOG_EMERG,
738 "err": LOG_ERR,
739 "error": LOG_ERR, # DEPRECATED
740 "info": LOG_INFO,
741 "notice": LOG_NOTICE,
742 "panic": LOG_EMERG, # DEPRECATED
743 "warn": LOG_WARNING, # DEPRECATED
744 "warning": LOG_WARNING,
745 }
746
747 facility_names = {
748 "auth": LOG_AUTH,
749 "authpriv": LOG_AUTHPRIV,
750 "cron": LOG_CRON,
751 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000752 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000753 "kern": LOG_KERN,
754 "lpr": LOG_LPR,
755 "mail": LOG_MAIL,
756 "news": LOG_NEWS,
757 "security": LOG_AUTH, # DEPRECATED
758 "syslog": LOG_SYSLOG,
759 "user": LOG_USER,
760 "uucp": LOG_UUCP,
761 "local0": LOG_LOCAL0,
762 "local1": LOG_LOCAL1,
763 "local2": LOG_LOCAL2,
764 "local3": LOG_LOCAL3,
765 "local4": LOG_LOCAL4,
766 "local5": LOG_LOCAL5,
767 "local6": LOG_LOCAL6,
768 "local7": LOG_LOCAL7,
769 }
770
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000771 #The map below appears to be trivially lowercasing the key. However,
772 #there's more to it than meets the eye - in some locales, lowercasing
773 #gives unexpected results. See SF #1524081: in the Turkish locale,
774 #"INFO".lower() != "info"
775 priority_map = {
776 "DEBUG" : "debug",
777 "INFO" : "info",
778 "WARNING" : "warning",
779 "ERROR" : "error",
780 "CRITICAL" : "critical"
781 }
782
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000783 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100784 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000785 """
786 Initialize a handler.
787
Guido van Rossume7ba4952007-06-06 23:52:48 +0000788 If address is specified as a string, a UNIX socket is used. To log to a
789 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100790 If facility is not specified, LOG_USER is used. If socktype is
791 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
792 socket type will be used. For Unix sockets, you can also specify a
793 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
794 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000795 """
796 logging.Handler.__init__(self)
797
798 self.address = address
799 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000800 self.socktype = socktype
801
Guido van Rossum13257902007-06-07 23:15:56 +0000802 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100803 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000804 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000805 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100806 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100807 if socktype is None:
808 socktype = socket.SOCK_DGRAM
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000809 self.socket = socket.socket(socket.AF_INET, socktype)
810 if socktype == socket.SOCK_STREAM:
811 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100812 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000813 self.formatter = None
814
Vinay Sajipa1974c12005-01-13 08:23:56 +0000815 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100816 use_socktype = self.socktype
817 if use_socktype is None:
818 use_socktype = socket.SOCK_DGRAM
819 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000820 try:
821 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100822 # it worked, so set self.socktype to the used type
823 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200824 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000825 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100826 if self.socktype is not None:
827 # user didn't specify falling back, so fail
828 raise
829 use_socktype = socket.SOCK_STREAM
830 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
831 try:
832 self.socket.connect(address)
833 # it worked, so set self.socktype to the used type
834 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100835 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100836 self.socket.close()
837 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000838
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000839 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000840 """
841 Encode the facility and priority. You can pass in strings or
842 integers - if strings are passed, the facility_names and
843 priority_names mapping dictionaries are used to convert them to
844 integers.
845 """
Guido van Rossum13257902007-06-07 23:15:56 +0000846 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000847 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000848 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000849 priority = self.priority_names[priority]
850 return (facility << 3) | priority
851
852 def close (self):
853 """
854 Closes the socket.
855 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000856 self.acquire()
857 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000858 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000859 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000860 finally:
861 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000862
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000863 def mapPriority(self, levelName):
864 """
865 Map a logging level name to a key in the priority_names map.
866 This is useful in two scenarios: when custom levels are being
867 used, and in the case where you can't do a straightforward
868 mapping by lowercasing the logging level name because of locale-
869 specific issues (see SF #1524081).
870 """
871 return self.priority_map.get(levelName, "warning")
872
Vinay Sajip2353e352011-06-27 15:40:06 +0100873 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100874 append_nul = True # some old syslog daemons expect a NUL terminator
875
Guido van Rossum57102f82002-11-13 16:15:58 +0000876 def emit(self, record):
877 """
878 Emit a record.
879
880 The record is formatted, and then sent to the syslog server. If
881 exception information is present, it is NOT sent to the server.
882 """
Guido van Rossum57102f82002-11-13 16:15:58 +0000883 try:
Vinay Sajipc33a0cc2014-11-01 19:58:47 +0000884 msg = self.format(record)
885 if self.ident:
886 msg = self.ident + msg
887 if self.append_nul:
888 msg += '\000'
889
890 # We need to convert record level to lowercase, maybe this will
891 # change in the future.
892 prio = '<%d>' % self.encodePriority(self.facility,
893 self.mapPriority(record.levelname))
894 prio = prio.encode('utf-8')
895 # Message is a string. Convert to bytes as required by RFC 5424
896 msg = msg.encode('utf-8')
897 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000898 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000899 try:
900 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200901 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100902 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000903 self._connect_unixsocket(self.address)
904 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000905 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000906 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000907 else:
908 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100909 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000910 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000911
912class SMTPHandler(logging.Handler):
913 """
914 A handler class which sends an SMTP email for each logging event.
915 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000916 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100917 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000918 """
919 Initialize the handler.
920
921 Initialize the instance with the from and to addresses and subject
922 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000923 (host, port) tuple format for the mailhost argument. To specify
924 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000925 for the credentials argument. To specify the use of a secure
926 protocol (TLS), pass in a tuple for the secure argument. This will
927 only be used when authentication credentials are supplied. The tuple
928 will be either an empty tuple, or a single-value tuple with the name
929 of a keyfile, or a 2-value tuple with the names of the keyfile and
930 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000931 A timeout in seconds can be specified for the SMTP connection (the
932 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000933 """
934 logging.Handler.__init__(self)
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100935 if isinstance(mailhost, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000936 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000937 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000938 self.mailhost, self.mailport = mailhost, None
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100939 if isinstance(credentials, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000940 self.username, self.password = credentials
941 else:
942 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000943 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000944 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000945 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000946 self.toaddrs = toaddrs
947 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000948 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000949 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000950
951 def getSubject(self, record):
952 """
953 Determine the subject for the email.
954
955 If you want to specify a subject line which is record-dependent,
956 override this method.
957 """
958 return self.subject
959
960 def emit(self, record):
961 """
962 Emit a record.
963
964 Format the record and send it to the specified addressees.
965 """
966 try:
967 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000968 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000969 port = self.mailport
970 if not port:
971 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000972 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000973 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000974 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 +0000975 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000976 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000977 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000978 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000979 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000980 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000981 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000982 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000983 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000984 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000985 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
986 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100987 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000988 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000989
990class NTEventLogHandler(logging.Handler):
991 """
992 A handler class which sends events to the NT Event Log. Adds a
993 registry entry for the specified application name. If no dllname is
994 provided, win32service.pyd (which contains some basic message
995 placeholders) is used. Note that use of these placeholders will make
996 your event logs big, as the entire message source is held in the log.
997 If you want slimmer logs, you have to pass in the name of your own DLL
998 which contains the message definitions you want to use in the event log.
999 """
1000 def __init__(self, appname, dllname=None, logtype="Application"):
1001 logging.Handler.__init__(self)
1002 try:
1003 import win32evtlogutil, win32evtlog
1004 self.appname = appname
1005 self._welu = win32evtlogutil
1006 if not dllname:
1007 dllname = os.path.split(self._welu.__file__)
1008 dllname = os.path.split(dllname[0])
1009 dllname = os.path.join(dllname[0], r'win32service.pyd')
1010 self.dllname = dllname
1011 self.logtype = logtype
1012 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1013 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1014 self.typemap = {
1015 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1016 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001017 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001018 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1019 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1020 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001021 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001022 print("The Python Win32 extensions for NT (service, event "\
1023 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001024 self._welu = None
1025
1026 def getMessageID(self, record):
1027 """
1028 Return the message ID for the event record. If you are using your
1029 own messages, you could do this by having the msg passed to the
1030 logger being an ID rather than a formatting string. Then, in here,
1031 you could use a dictionary lookup to get the message ID. This
1032 version returns 1, which is the base message ID in win32service.pyd.
1033 """
1034 return 1
1035
1036 def getEventCategory(self, record):
1037 """
1038 Return the event category for the record.
1039
1040 Override this if you want to specify your own categories. This version
1041 returns 0.
1042 """
1043 return 0
1044
1045 def getEventType(self, record):
1046 """
1047 Return the event type for the record.
1048
1049 Override this if you want to specify your own types. This version does
1050 a mapping using the handler's typemap attribute, which is set up in
1051 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001052 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001053 either need to override this method or place a suitable dictionary in
1054 the handler's typemap attribute.
1055 """
1056 return self.typemap.get(record.levelno, self.deftype)
1057
1058 def emit(self, record):
1059 """
1060 Emit a record.
1061
1062 Determine the message ID, event category and event type. Then
1063 log the message in the NT event log.
1064 """
1065 if self._welu:
1066 try:
1067 id = self.getMessageID(record)
1068 cat = self.getEventCategory(record)
1069 type = self.getEventType(record)
1070 msg = self.format(record)
1071 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001072 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001073 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001074
1075 def close(self):
1076 """
1077 Clean up this handler.
1078
1079 You can remove the application name from the registry as a
1080 source of event log entries. However, if you do this, you will
1081 not be able to see the events as you intended in the Event Log
1082 Viewer - it needs to be able to access the registry to get the
1083 DLL name.
1084 """
1085 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001086 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001087
1088class HTTPHandler(logging.Handler):
1089 """
1090 A class which sends records to a Web server, using either GET or
1091 POST semantics.
1092 """
Benjamin Peterson43052a12014-11-23 20:36:44 -06001093 def __init__(self, host, url, method="GET", secure=False, credentials=None,
1094 context=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001095 """
1096 Initialize the instance with the host, the request URL, and the method
1097 ("GET" or "POST")
1098 """
1099 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001100 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001101 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001102 raise ValueError("method must be GET or POST")
Benjamin Peterson43052a12014-11-23 20:36:44 -06001103 if not secure and context is not None:
1104 raise ValueError("context parameter only makes sense "
1105 "with secure=True")
Guido van Rossum57102f82002-11-13 16:15:58 +00001106 self.host = host
1107 self.url = url
1108 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001109 self.secure = secure
1110 self.credentials = credentials
Benjamin Peterson43052a12014-11-23 20:36:44 -06001111 self.context = context
Guido van Rossum57102f82002-11-13 16:15:58 +00001112
Neal Norwitzf297bd12003-04-23 03:49:43 +00001113 def mapLogRecord(self, record):
1114 """
1115 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001116 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001117 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001118 """
1119 return record.__dict__
1120
Guido van Rossum57102f82002-11-13 16:15:58 +00001121 def emit(self, record):
1122 """
1123 Emit a record.
1124
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001125 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001126 """
1127 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001128 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001129 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001130 if self.secure:
Benjamin Peterson43052a12014-11-23 20:36:44 -06001131 h = http.client.HTTPSConnection(host, context=self.context)
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001132 else:
1133 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001134 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001135 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001136 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001137 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001138 sep = '&'
1139 else:
1140 sep = '?'
1141 url = url + "%c%s" % (sep, data)
1142 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001143 # support multiple hosts on one IP address...
1144 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001145 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001146 if i >= 0:
1147 host = host[:i]
1148 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001149 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001150 h.putheader("Content-type",
1151 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001152 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001153 if self.credentials:
1154 import base64
1155 s = ('u%s:%s' % self.credentials).encode('utf-8')
1156 s = 'Basic ' + base64.b64encode(s).strip()
1157 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001158 h.endheaders()
1159 if self.method == "POST":
1160 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001161 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001162 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001163 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001164
1165class BufferingHandler(logging.Handler):
1166 """
1167 A handler class which buffers logging records in memory. Whenever each
1168 record is added to the buffer, a check is made to see if the buffer should
1169 be flushed. If it should, then flush() is expected to do what's needed.
1170 """
1171 def __init__(self, capacity):
1172 """
1173 Initialize the handler with the buffer size.
1174 """
1175 logging.Handler.__init__(self)
1176 self.capacity = capacity
1177 self.buffer = []
1178
1179 def shouldFlush(self, record):
1180 """
1181 Should the handler flush its buffer?
1182
1183 Returns true if the buffer is up to capacity. This method can be
1184 overridden to implement custom flushing strategies.
1185 """
1186 return (len(self.buffer) >= self.capacity)
1187
1188 def emit(self, record):
1189 """
1190 Emit a record.
1191
1192 Append the record. If shouldFlush() tells us to, call flush() to process
1193 the buffer.
1194 """
1195 self.buffer.append(record)
1196 if self.shouldFlush(record):
1197 self.flush()
1198
1199 def flush(self):
1200 """
1201 Override to implement custom flushing behaviour.
1202
1203 This version just zaps the buffer to empty.
1204 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001205 self.acquire()
1206 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001207 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001208 finally:
1209 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001210
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001211 def close(self):
1212 """
1213 Close the handler.
1214
1215 This version just flushes and chains to the parent class' close().
1216 """
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001217 try:
1218 self.flush()
1219 finally:
1220 logging.Handler.close(self)
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001221
Guido van Rossum57102f82002-11-13 16:15:58 +00001222class MemoryHandler(BufferingHandler):
1223 """
1224 A handler class which buffers logging records in memory, periodically
1225 flushing them to a target handler. Flushing occurs whenever the buffer
1226 is full, or when an event of a certain severity or greater is seen.
1227 """
1228 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1229 """
1230 Initialize the handler with the buffer size, the level at which
1231 flushing should occur and an optional target.
1232
1233 Note that without a target being set either here or via setTarget(),
1234 a MemoryHandler is no use to anyone!
1235 """
1236 BufferingHandler.__init__(self, capacity)
1237 self.flushLevel = flushLevel
1238 self.target = target
1239
1240 def shouldFlush(self, record):
1241 """
1242 Check for buffer full or a record at the flushLevel or higher.
1243 """
1244 return (len(self.buffer) >= self.capacity) or \
1245 (record.levelno >= self.flushLevel)
1246
1247 def setTarget(self, target):
1248 """
1249 Set the target handler for this handler.
1250 """
1251 self.target = target
1252
1253 def flush(self):
1254 """
1255 For a MemoryHandler, flushing means just sending the buffered
1256 records to the target, if there is one. Override if you want
1257 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001258
1259 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001260 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001261 self.acquire()
1262 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001263 if self.target:
1264 for record in self.buffer:
1265 self.target.handle(record)
1266 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001267 finally:
1268 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001269
1270 def close(self):
1271 """
1272 Flush, set the target to None and lose the buffer.
1273 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001274 try:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001275 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001276 finally:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001277 self.acquire()
1278 try:
1279 self.target = None
1280 BufferingHandler.close(self)
1281 finally:
1282 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001283
1284
1285class QueueHandler(logging.Handler):
1286 """
1287 This handler sends events to a queue. Typically, it would be used together
1288 with a multiprocessing Queue to centralise logging to file in one process
1289 (in a multi-process application), so as to avoid file write contention
1290 between processes.
1291
1292 This code is new in Python 3.2, but this class can be copy pasted into
1293 user code for use with earlier Python versions.
1294 """
1295
1296 def __init__(self, queue):
1297 """
1298 Initialise an instance, using the passed queue.
1299 """
1300 logging.Handler.__init__(self)
1301 self.queue = queue
1302
1303 def enqueue(self, record):
1304 """
1305 Enqueue a record.
1306
1307 The base implementation uses put_nowait. You may want to override
1308 this method if you want to use blocking, timeouts or custom queue
1309 implementations.
1310 """
1311 self.queue.put_nowait(record)
1312
Vinay Sajip0258ce82010-09-22 20:34:53 +00001313 def prepare(self, record):
1314 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001315 Prepares a record for queuing. The object returned by this method is
1316 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001317
1318 The base implementation formats the record to merge the message
1319 and arguments, and removes unpickleable items from the record
1320 in-place.
1321
1322 You might want to override this method if you want to convert
1323 the record to a dict or JSON string, or send a modified copy
1324 of the record while leaving the original intact.
1325 """
1326 # The format operation gets traceback text into record.exc_text
1327 # (if there's exception data), and also puts the message into
1328 # record.message. We can then use this to replace the original
1329 # msg + args, as these might be unpickleable. We also zap the
1330 # exc_info attribute, as it's no longer needed and, if not None,
1331 # will typically not be pickleable.
1332 self.format(record)
1333 record.msg = record.message
1334 record.args = None
1335 record.exc_info = None
1336 return record
1337
Vinay Sajip121a1c42010-09-08 10:46:15 +00001338 def emit(self, record):
1339 """
1340 Emit a record.
1341
Vinay Sajip0637d492010-09-23 08:15:54 +00001342 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001343 """
1344 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001345 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001346 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001347 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001348
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001349if threading:
1350 class QueueListener(object):
1351 """
1352 This class implements an internal threaded listener which watches for
1353 LogRecords being added to a queue, removes them and passes them to a
1354 list of handlers for processing.
1355 """
1356 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001357
Vinay Sajip365701a2015-02-09 19:49:00 +00001358 def __init__(self, queue, *handlers, respect_handler_level=False):
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001359 """
1360 Initialise an instance with the specified queue and
1361 handlers.
1362 """
1363 self.queue = queue
1364 self.handlers = handlers
1365 self._stop = threading.Event()
1366 self._thread = None
Vinay Sajip365701a2015-02-09 19:49:00 +00001367 self.respect_handler_level = respect_handler_level
Vinay Sajip0637d492010-09-23 08:15:54 +00001368
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001369 def dequeue(self, block):
1370 """
1371 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001372
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001373 The base implementation uses get. You may want to override this method
1374 if you want to use timeouts or work with custom queue implementations.
1375 """
1376 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001377
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001378 def start(self):
1379 """
1380 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001381
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001382 This starts up a background thread to monitor the queue for
1383 LogRecords to process.
1384 """
1385 self._thread = t = threading.Thread(target=self._monitor)
1386 t.setDaemon(True)
1387 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001388
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001389 def prepare(self , record):
1390 """
1391 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001392
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001393 This method just returns the passed-in record. You may want to
1394 override this method if you need to do any custom marshalling or
1395 manipulation of the record before passing it to the handlers.
1396 """
1397 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001398
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001399 def handle(self, record):
1400 """
1401 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001402
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001403 This just loops through the handlers offering them the record
1404 to handle.
1405 """
1406 record = self.prepare(record)
1407 for handler in self.handlers:
Vinay Sajip365701a2015-02-09 19:49:00 +00001408 if not self.respect_handler_level:
1409 process = True
1410 else:
1411 process = record.levelno >= handler.level
1412 if process:
1413 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001414
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001415 def _monitor(self):
1416 """
1417 Monitor the queue for records, and ask the handler
1418 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001419
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001420 This method runs on a separate, internal thread.
1421 The thread will terminate if it sees a sentinel object in the queue.
1422 """
1423 q = self.queue
1424 has_task_done = hasattr(q, 'task_done')
1425 while not self._stop.isSet():
1426 try:
1427 record = self.dequeue(True)
1428 if record is self._sentinel:
1429 break
1430 self.handle(record)
1431 if has_task_done:
1432 q.task_done()
1433 except queue.Empty:
1434 pass
1435 # There might still be records in the queue.
1436 while True:
1437 try:
1438 record = self.dequeue(False)
1439 if record is self._sentinel:
1440 break
1441 self.handle(record)
1442 if has_task_done:
1443 q.task_done()
1444 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001445 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001446
Victor Stinner59bec362011-05-02 16:14:16 +02001447 def enqueue_sentinel(self):
1448 """
1449 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001450
Victor Stinner59bec362011-05-02 16:14:16 +02001451 The base implementation uses put_nowait. You may want to override this
1452 method if you want to use timeouts or work with custom queue
1453 implementations.
1454 """
1455 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001456
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001457 def stop(self):
1458 """
1459 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001460
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001461 This asks the thread to terminate, and then waits for it to do so.
1462 Note that if you don't call this before your application exits, there
1463 may be some records still left on the queue, which won't be processed.
1464 """
1465 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001466 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001467 self._thread.join()
1468 self._thread = None