blob: 0a11b8232b42d9de797083bc0fc011426197b86b [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 Cannon0a140662013-06-13 20:57:26 -040032except ModuleNotFoundError: #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):
112 os.rename(source, dest)
113 else:
114 self.rotator(source, dest)
115
Vinay Sajip17c52d82004-07-03 11:48:34 +0000116class RotatingFileHandler(BaseRotatingHandler):
117 """
118 Handler for logging to a set of files, which switches from one file
119 to the next when the current file reaches a certain size.
120 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000121 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
Guido van Rossum57102f82002-11-13 16:15:58 +0000122 """
123 Open the specified file and use it as the stream for logging.
124
125 By default, the file grows indefinitely. You can specify particular
126 values of maxBytes and backupCount to allow the file to rollover at
127 a predetermined size.
128
129 Rollover occurs whenever the current log file is nearly maxBytes in
130 length. If backupCount is >= 1, the system will successively create
131 new files with the same pathname as the base file, but with extensions
132 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
133 and a base file name of "app.log", you would get "app.log",
134 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
135 written to is always "app.log" - when it gets filled up, it is closed
136 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
137 exist, then they are renamed to "app.log.2", "app.log.3" etc.
138 respectively.
139
140 If maxBytes is zero, rollover never occurs.
141 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000142 # If rotation/rollover is wanted, it doesn't make sense to use another
143 # mode. If for example 'w' were specified, then if there were multiple
144 # runs of the calling application, the logs from previous runs would be
145 # lost if the 'w' is respected, because the log file would be truncated
146 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000147 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000148 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000149 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000150 self.maxBytes = maxBytes
151 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000152
153 def doRollover(self):
154 """
155 Do a rollover, as described in __init__().
156 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000157 if self.stream:
158 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000159 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000160 if self.backupCount > 0:
161 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000162 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
163 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
164 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000165 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000166 if os.path.exists(dfn):
167 os.remove(dfn)
168 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000169 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000170 if os.path.exists(dfn):
171 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000172 self.rotate(self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000173 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000174
Vinay Sajip17c52d82004-07-03 11:48:34 +0000175 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000176 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000177 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000178
Vinay Sajip17c52d82004-07-03 11:48:34 +0000179 Basically, see if the supplied record would cause the file to exceed
180 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000181 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000182 if self.stream is None: # delay was set...
183 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000184 if self.maxBytes > 0: # are we rolling over?
185 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000186 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000187 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000188 return 1
189 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000190
Vinay Sajip17c52d82004-07-03 11:48:34 +0000191class TimedRotatingFileHandler(BaseRotatingHandler):
192 """
193 Handler for logging to a file, rotating the log file at certain timed
194 intervals.
195
196 If backupCount is > 0, when rollover is done, no more than backupCount
197 files are kept - the oldest ones are deleted.
198 """
Vinay Sajipa7130792013-04-12 17:04:23 +0100199 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 +0000200 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000201 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000202 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000203 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100204 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000205 # Calculate the real rollover interval, which is just the number of
206 # seconds between rollovers. Also set the filename suffix used when
207 # a rollover occurs. Current 'when' events supported:
208 # S - Seconds
209 # M - Minutes
210 # H - Hours
211 # D - Days
212 # midnight - roll over at midnight
213 # W{0-6} - roll over on a certain day; 0 - Monday
214 #
215 # Case of the 'when' specifier is not important; lower or upper case
216 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000217 if self.when == 'S':
218 self.interval = 1 # one second
219 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000220 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000221 elif self.when == 'M':
222 self.interval = 60 # one minute
223 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000224 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000225 elif self.when == 'H':
226 self.interval = 60 * 60 # one hour
227 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000228 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000229 elif self.when == 'D' or self.when == 'MIDNIGHT':
230 self.interval = 60 * 60 * 24 # one day
231 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000232 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000233 elif self.when.startswith('W'):
234 self.interval = 60 * 60 * 24 * 7 # one week
235 if len(self.when) != 2:
236 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
237 if self.when[1] < '0' or self.when[1] > '6':
238 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
239 self.dayOfWeek = int(self.when[1])
240 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000241 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000242 else:
243 raise ValueError("Invalid rollover interval specified: %s" % self.when)
244
Antoine Pitroufd036452008-08-19 17:56:33 +0000245 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000246 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000247 if os.path.exists(filename):
248 t = os.stat(filename)[ST_MTIME]
249 else:
250 t = int(time.time())
251 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000252
Vinay Sajipaffbd872009-06-11 10:11:47 +0000253 def computeRollover(self, currentTime):
254 """
255 Work out the rollover time based on the specified time.
256 """
257 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000258 # If we are rolling over at midnight or weekly, then the interval is already known.
259 # What we need to figure out is WHEN the next interval is. In other words,
260 # if you are rolling over at midnight, then your base interval is 1 day,
261 # but you want to start that one day clock at midnight, not now. So, we
262 # have to fudge the rolloverAt value in order to trigger the first rollover
263 # at the right time. After that, the regular interval will take care of
264 # the rest. Note that this code doesn't care about leap seconds. :)
265 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
266 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000267 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000268 t = time.gmtime(currentTime)
269 else:
270 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000271 currentHour = t[3]
272 currentMinute = t[4]
273 currentSecond = t[5]
Vinay Sajipa7130792013-04-12 17:04:23 +0100274 currentDay = t[6]
275 # r is the number of seconds left between now and the next rotation
276 if self.atTime is None:
277 rotate_ts = _MIDNIGHT
278 else:
279 rotate_ts = ((self.atTime.hour * 60 + self.atTime.minute)*60 +
280 self.atTime.second)
281
282 r = rotate_ts - ((currentHour * 60 + currentMinute) * 60 +
283 currentSecond)
284 if r < 0:
285 # Rotate time is before the current time (for example when
286 # self.rotateAt is 13:45 and it now 14:15), rotation is
287 # tomorrow.
288 r += _MIDNIGHT
289 currentDay = (currentDay + 1) % 7
Vinay Sajipaffbd872009-06-11 10:11:47 +0000290 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000291 # If we are rolling over on a certain day, add in the number of days until
292 # the next rollover, but offset by 1 since we just calculated the time
293 # until the next day starts. There are three cases:
294 # Case 1) The day to rollover is today; in this case, do nothing
295 # Case 2) The day to rollover is further in the interval (i.e., today is
296 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
297 # next rollover is simply 6 - 2 - 1, or 3.
298 # Case 3) The day to rollover is behind us in the interval (i.e., today
299 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
300 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
301 # number of days left in the current week (1) plus the number
302 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000303 # The calculations described in 2) and 3) above need to have a day added.
304 # This is because the above time calculation takes us to midnight on this
305 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000306 if self.when.startswith('W'):
Vinay Sajipa7130792013-04-12 17:04:23 +0100307 day = currentDay # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000308 if day != self.dayOfWeek:
309 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000310 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000311 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000312 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000313 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
314 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000315 dstNow = t[-1]
316 dstAtRollover = time.localtime(newRolloverAt)[-1]
317 if dstNow != dstAtRollover:
318 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000319 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000320 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000321 addend = 3600
322 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000323 result = newRolloverAt
324 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000325
326 def shouldRollover(self, record):
327 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000328 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000329
330 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000331 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000332 """
333 t = int(time.time())
334 if t >= self.rolloverAt:
335 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000336 return 0
337
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000338 def getFilesToDelete(self):
339 """
340 Determine the files to delete when rolling over.
341
342 More specific than the earlier method, which just used glob.glob().
343 """
344 dirName, baseName = os.path.split(self.baseFilename)
345 fileNames = os.listdir(dirName)
346 result = []
347 prefix = baseName + "."
348 plen = len(prefix)
349 for fileName in fileNames:
350 if fileName[:plen] == prefix:
351 suffix = fileName[plen:]
352 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000353 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000354 result.sort()
355 if len(result) < self.backupCount:
356 result = []
357 else:
358 result = result[:len(result) - self.backupCount]
359 return result
360
Vinay Sajip17c52d82004-07-03 11:48:34 +0000361 def doRollover(self):
362 """
363 do a rollover; in this case, a date/time stamp is appended to the filename
364 when the rollover happens. However, you want the file to be named for the
365 start of the interval, not the current time. If there is a backup count,
366 then we have to get a list of matching filenames, sort them and remove
367 the one with the oldest suffix.
368 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000369 if self.stream:
370 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000371 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000372 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000373 currentTime = int(time.time())
374 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000375 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000376 if self.utc:
377 timeTuple = time.gmtime(t)
378 else:
379 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000380 dstThen = timeTuple[-1]
381 if dstNow != dstThen:
382 if dstNow:
383 addend = 3600
384 else:
385 addend = -3600
386 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000387 dfn = self.rotation_filename(self.baseFilename + "." +
388 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000389 if os.path.exists(dfn):
390 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000391 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000392 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000393 for s in self.getFilesToDelete():
394 os.remove(s)
Thomas Woutersb2137042007-02-01 18:02:27 +0000395 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000396 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000397 while newRolloverAt <= currentTime:
398 newRolloverAt = newRolloverAt + self.interval
399 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000400 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000401 dstAtRollover = time.localtime(newRolloverAt)[-1]
402 if dstNow != dstAtRollover:
403 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000404 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000405 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000406 addend = 3600
407 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000408 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000409
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000410class WatchedFileHandler(logging.FileHandler):
411 """
412 A handler for logging to a file, which watches the file
413 to see if it has changed while in use. This can happen because of
414 usage of programs such as newsyslog and logrotate which perform
415 log file rotation. This handler, intended for use under Unix,
416 watches the file to see if it has changed since the last emit.
417 (A file has changed if its device or inode have changed.)
418 If it has changed, the old file stream is closed, and the file
419 opened to get a new stream.
420
421 This handler is not appropriate for use under Windows, because
422 under Windows open files cannot be moved or renamed - logging
423 opens the files with exclusive locks - and so there is no need
424 for such a handler. Furthermore, ST_INO is not supported under
425 Windows; stat always returns zero for this value.
426
427 This handler is based on a suggestion and patch by Chad J.
428 Schroeder.
429 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000430 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000431 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100432 self.dev, self.ino = -1, -1
433 self._statstream()
434
435 def _statstream(self):
436 if self.stream:
437 sres = os.fstat(self.stream.fileno())
438 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000439
440 def emit(self, record):
441 """
442 Emit a record.
443
444 First check if the underlying file has changed, and if it
445 has, close the old stream and reopen the file to get the
446 current stream.
447 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100448 # Reduce the chance of race conditions by stat'ing by path only
449 # once and then fstat'ing our new fd if we opened a new log stream.
450 # See issue #14632: Thanks to John Mulligan for the problem report
451 # and patch.
452 try:
453 # stat the file by path, checking for existence
454 sres = os.stat(self.baseFilename)
Giampaolo Rodola'0166a282013-02-12 15:14:17 +0100455 except FileNotFoundError:
456 sres = None
Vinay Sajip66b8b082012-04-24 23:25:30 +0100457 # compare file system stat with that of our stream file handle
458 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
459 if self.stream is not None:
460 # we have an open file handle, clean it up
461 self.stream.flush()
462 self.stream.close()
463 # open a new file handle and get new stat info from that fd
464 self.stream = self._open()
465 self._statstream()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000466 logging.FileHandler.emit(self, record)
467
Vinay Sajip66b8b082012-04-24 23:25:30 +0100468
Guido van Rossum57102f82002-11-13 16:15:58 +0000469class SocketHandler(logging.Handler):
470 """
471 A handler class which writes logging records, in pickle format, to
472 a streaming socket. The socket is kept open across logging calls.
473 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000474 The pickle which is sent is that of the LogRecord's attribute dictionary
475 (__dict__), so that the receiver does not need to have the logging module
476 installed in order to process the logging event.
477
478 To unpickle the record at the receiving end into a LogRecord, use the
479 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000480 """
481
482 def __init__(self, host, port):
483 """
484 Initializes the handler with a specific host address and port.
485
Vinay Sajipde19e082011-04-30 21:52:26 +0100486 When the attribute *closeOnError* is set to True - if a socket error
487 occurs, the socket is silently closed and then reopened on the next
488 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000489 """
490 logging.Handler.__init__(self)
491 self.host = host
492 self.port = port
493 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100494 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000495 self.retryTime = None
496 #
497 # Exponential backoff parameters.
498 #
499 self.retryStart = 1.0
500 self.retryMax = 30.0
501 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000502
Guido van Rossumd8faa362007-04-27 19:54:29 +0000503 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000504 """
505 A factory method which allows subclasses to define the precise
506 type of socket they want.
507 """
Giampaolo Rodola'35eeb302012-10-15 16:57:37 +0200508 return socket.create_connection((self.host, self.port), timeout=timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000509
Vinay Sajip48cfe382004-02-20 13:17:27 +0000510 def createSocket(self):
511 """
512 Try to create a socket, using an exponential backoff with
513 a max retry time. Thanks to Robert Olson for the original patch
514 (SF #815911) which has been slightly refactored.
515 """
516 now = time.time()
517 # Either retryTime is None, in which case this
518 # is the first time back after a disconnect, or
519 # we've waited long enough.
520 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100521 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000522 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000523 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000524 if attempt:
525 try:
526 self.sock = self.makeSocket()
527 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200528 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000529 #Creation failed, so set the retry time and return.
530 if self.retryTime is None:
531 self.retryPeriod = self.retryStart
532 else:
533 self.retryPeriod = self.retryPeriod * self.retryFactor
534 if self.retryPeriod > self.retryMax:
535 self.retryPeriod = self.retryMax
536 self.retryTime = now + self.retryPeriod
537
Guido van Rossum57102f82002-11-13 16:15:58 +0000538 def send(self, s):
539 """
540 Send a pickled string to the socket.
541
542 This function allows for partial sends which can happen when the
543 network is busy.
544 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000545 if self.sock is None:
546 self.createSocket()
547 #self.sock can be None either because we haven't reached the retry
548 #time yet, or because we have reached the retry time and retried,
549 #but are still unable to connect.
550 if self.sock:
551 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200552 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200553 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000554 self.sock.close()
555 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000556
557 def makePickle(self, record):
558 """
559 Pickles the record in binary format with a length prefix, and
560 returns it ready for transmission across the socket.
561 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000562 ei = record.exc_info
563 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100564 # just to get traceback text into record.exc_text ...
565 dummy = self.format(record)
566 # See issue #14436: If msg or args are objects, they may not be
567 # available on the receiving end. So we convert the msg % args
568 # to a string, save it as msg and zap the args.
569 d = dict(record.__dict__)
570 d['msg'] = record.getMessage()
571 d['args'] = None
572 d['exc_info'] = None
573 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000574 slen = struct.pack(">L", len(s))
575 return slen + s
576
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000577 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000578 """
579 Handle an error during logging.
580
581 An error has occurred during logging. Most likely cause -
582 connection lost. Close the socket so that we can retry on the
583 next event.
584 """
585 if self.closeOnError and self.sock:
586 self.sock.close()
587 self.sock = None #try to reconnect next time
588 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000589 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000590
591 def emit(self, record):
592 """
593 Emit a record.
594
595 Pickles the record and writes it to the socket in binary format.
596 If there is an error with the socket, silently drop the packet.
597 If there was a problem with the socket, re-establishes the
598 socket.
599 """
600 try:
601 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000602 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100603 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000604 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000605
606 def close(self):
607 """
608 Closes the socket.
609 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000610 self.acquire()
611 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000612 if self.sock:
613 self.sock.close()
614 self.sock = None
615 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000616 finally:
617 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000618
619class DatagramHandler(SocketHandler):
620 """
621 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000622 a datagram socket. The pickle which is sent is that of the LogRecord's
623 attribute dictionary (__dict__), so that the receiver does not need to
624 have the logging module installed in order to process the logging event.
625
626 To unpickle the record at the receiving end into a LogRecord, use the
627 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000628
629 """
630 def __init__(self, host, port):
631 """
632 Initializes the handler with a specific host address and port.
633 """
634 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100635 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000636
637 def makeSocket(self):
638 """
639 The factory method of SocketHandler is here overridden to create
640 a UDP socket (SOCK_DGRAM).
641 """
642 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
643 return s
644
645 def send(self, s):
646 """
647 Send a pickled string to a socket.
648
649 This function no longer allows for partial sends which can happen
650 when the network is busy - UDP does not guarantee delivery and
651 can deliver packets out of sequence.
652 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000653 if self.sock is None:
654 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000655 self.sock.sendto(s, (self.host, self.port))
656
657class SysLogHandler(logging.Handler):
658 """
659 A handler class which sends formatted logging records to a syslog
660 server. Based on Sam Rushing's syslog module:
661 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
662 Contributed by Nicolas Untz (after which minor refactoring changes
663 have been made).
664 """
665
666 # from <linux/sys/syslog.h>:
667 # ======================================================================
668 # priorities/facilities are encoded into a single 32-bit quantity, where
669 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
670 # facility (0-big number). Both the priorities and the facilities map
671 # roughly one-to-one to strings in the syslogd(8) source code. This
672 # mapping is included in this file.
673 #
674 # priorities (these are ordered)
675
676 LOG_EMERG = 0 # system is unusable
677 LOG_ALERT = 1 # action must be taken immediately
678 LOG_CRIT = 2 # critical conditions
679 LOG_ERR = 3 # error conditions
680 LOG_WARNING = 4 # warning conditions
681 LOG_NOTICE = 5 # normal but significant condition
682 LOG_INFO = 6 # informational
683 LOG_DEBUG = 7 # debug-level messages
684
685 # facility codes
686 LOG_KERN = 0 # kernel messages
687 LOG_USER = 1 # random user-level messages
688 LOG_MAIL = 2 # mail system
689 LOG_DAEMON = 3 # system daemons
690 LOG_AUTH = 4 # security/authorization messages
691 LOG_SYSLOG = 5 # messages generated internally by syslogd
692 LOG_LPR = 6 # line printer subsystem
693 LOG_NEWS = 7 # network news subsystem
694 LOG_UUCP = 8 # UUCP subsystem
695 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000696 LOG_AUTHPRIV = 10 # security/authorization messages (private)
697 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000698
699 # other codes through 15 reserved for system use
700 LOG_LOCAL0 = 16 # reserved for local use
701 LOG_LOCAL1 = 17 # reserved for local use
702 LOG_LOCAL2 = 18 # reserved for local use
703 LOG_LOCAL3 = 19 # reserved for local use
704 LOG_LOCAL4 = 20 # reserved for local use
705 LOG_LOCAL5 = 21 # reserved for local use
706 LOG_LOCAL6 = 22 # reserved for local use
707 LOG_LOCAL7 = 23 # reserved for local use
708
709 priority_names = {
710 "alert": LOG_ALERT,
711 "crit": LOG_CRIT,
712 "critical": LOG_CRIT,
713 "debug": LOG_DEBUG,
714 "emerg": LOG_EMERG,
715 "err": LOG_ERR,
716 "error": LOG_ERR, # DEPRECATED
717 "info": LOG_INFO,
718 "notice": LOG_NOTICE,
719 "panic": LOG_EMERG, # DEPRECATED
720 "warn": LOG_WARNING, # DEPRECATED
721 "warning": LOG_WARNING,
722 }
723
724 facility_names = {
725 "auth": LOG_AUTH,
726 "authpriv": LOG_AUTHPRIV,
727 "cron": LOG_CRON,
728 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000729 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000730 "kern": LOG_KERN,
731 "lpr": LOG_LPR,
732 "mail": LOG_MAIL,
733 "news": LOG_NEWS,
734 "security": LOG_AUTH, # DEPRECATED
735 "syslog": LOG_SYSLOG,
736 "user": LOG_USER,
737 "uucp": LOG_UUCP,
738 "local0": LOG_LOCAL0,
739 "local1": LOG_LOCAL1,
740 "local2": LOG_LOCAL2,
741 "local3": LOG_LOCAL3,
742 "local4": LOG_LOCAL4,
743 "local5": LOG_LOCAL5,
744 "local6": LOG_LOCAL6,
745 "local7": LOG_LOCAL7,
746 }
747
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000748 #The map below appears to be trivially lowercasing the key. However,
749 #there's more to it than meets the eye - in some locales, lowercasing
750 #gives unexpected results. See SF #1524081: in the Turkish locale,
751 #"INFO".lower() != "info"
752 priority_map = {
753 "DEBUG" : "debug",
754 "INFO" : "info",
755 "WARNING" : "warning",
756 "ERROR" : "error",
757 "CRITICAL" : "critical"
758 }
759
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000760 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100761 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000762 """
763 Initialize a handler.
764
Guido van Rossume7ba4952007-06-06 23:52:48 +0000765 If address is specified as a string, a UNIX socket is used. To log to a
766 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100767 If facility is not specified, LOG_USER is used. If socktype is
768 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
769 socket type will be used. For Unix sockets, you can also specify a
770 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
771 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000772 """
773 logging.Handler.__init__(self)
774
775 self.address = address
776 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000777 self.socktype = socktype
778
Guido van Rossum13257902007-06-07 23:15:56 +0000779 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100780 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000781 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000782 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100783 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100784 if socktype is None:
785 socktype = socket.SOCK_DGRAM
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000786 self.socket = socket.socket(socket.AF_INET, socktype)
787 if socktype == socket.SOCK_STREAM:
788 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100789 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000790 self.formatter = None
791
Vinay Sajipa1974c12005-01-13 08:23:56 +0000792 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100793 use_socktype = self.socktype
794 if use_socktype is None:
795 use_socktype = socket.SOCK_DGRAM
796 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000797 try:
798 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100799 # it worked, so set self.socktype to the used type
800 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200801 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000802 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100803 if self.socktype is not None:
804 # user didn't specify falling back, so fail
805 raise
806 use_socktype = socket.SOCK_STREAM
807 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
808 try:
809 self.socket.connect(address)
810 # it worked, so set self.socktype to the used type
811 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100812 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100813 self.socket.close()
814 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000815
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000816 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000817 """
818 Encode the facility and priority. You can pass in strings or
819 integers - if strings are passed, the facility_names and
820 priority_names mapping dictionaries are used to convert them to
821 integers.
822 """
Guido van Rossum13257902007-06-07 23:15:56 +0000823 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000824 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000825 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000826 priority = self.priority_names[priority]
827 return (facility << 3) | priority
828
829 def close (self):
830 """
831 Closes the socket.
832 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000833 self.acquire()
834 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000835 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000836 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000837 finally:
838 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000839
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000840 def mapPriority(self, levelName):
841 """
842 Map a logging level name to a key in the priority_names map.
843 This is useful in two scenarios: when custom levels are being
844 used, and in the case where you can't do a straightforward
845 mapping by lowercasing the logging level name because of locale-
846 specific issues (see SF #1524081).
847 """
848 return self.priority_map.get(levelName, "warning")
849
Vinay Sajip2353e352011-06-27 15:40:06 +0100850 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100851 append_nul = True # some old syslog daemons expect a NUL terminator
852
Guido van Rossum57102f82002-11-13 16:15:58 +0000853 def emit(self, record):
854 """
855 Emit a record.
856
857 The record is formatted, and then sent to the syslog server. If
858 exception information is present, it is NOT sent to the server.
859 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100860 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100861 if self.ident:
862 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100863 if self.append_nul:
864 msg += '\000'
Victor Stinner765531d2013-03-26 01:11:54 +0100865
866 # We need to convert record level to lowercase, maybe this will
867 # change in the future.
Vinay Sajip467d12f2010-08-19 22:20:22 +0000868 prio = '<%d>' % self.encodePriority(self.facility,
869 self.mapPriority(record.levelname))
870 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000871 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000872 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000873 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000874 try:
875 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000876 try:
877 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200878 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100879 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000880 self._connect_unixsocket(self.address)
881 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000882 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000883 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000884 else:
885 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100886 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000887 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000888
889class SMTPHandler(logging.Handler):
890 """
891 A handler class which sends an SMTP email for each logging event.
892 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000893 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100894 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000895 """
896 Initialize the handler.
897
898 Initialize the instance with the from and to addresses and subject
899 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000900 (host, port) tuple format for the mailhost argument. To specify
901 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000902 for the credentials argument. To specify the use of a secure
903 protocol (TLS), pass in a tuple for the secure argument. This will
904 only be used when authentication credentials are supplied. The tuple
905 will be either an empty tuple, or a single-value tuple with the name
906 of a keyfile, or a 2-value tuple with the names of the keyfile and
907 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000908 A timeout in seconds can be specified for the SMTP connection (the
909 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000910 """
911 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000912 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000913 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000914 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000915 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000916 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000917 self.username, self.password = credentials
918 else:
919 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000920 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000921 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000922 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000923 self.toaddrs = toaddrs
924 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000925 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000926 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000927
928 def getSubject(self, record):
929 """
930 Determine the subject for the email.
931
932 If you want to specify a subject line which is record-dependent,
933 override this method.
934 """
935 return self.subject
936
937 def emit(self, record):
938 """
939 Emit a record.
940
941 Format the record and send it to the specified addressees.
942 """
943 try:
944 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000945 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000946 port = self.mailport
947 if not port:
948 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000949 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000950 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000951 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 +0000952 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000953 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000954 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000955 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000956 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000957 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000958 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000959 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000960 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000961 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000962 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
963 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100964 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000965 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000966
967class NTEventLogHandler(logging.Handler):
968 """
969 A handler class which sends events to the NT Event Log. Adds a
970 registry entry for the specified application name. If no dllname is
971 provided, win32service.pyd (which contains some basic message
972 placeholders) is used. Note that use of these placeholders will make
973 your event logs big, as the entire message source is held in the log.
974 If you want slimmer logs, you have to pass in the name of your own DLL
975 which contains the message definitions you want to use in the event log.
976 """
977 def __init__(self, appname, dllname=None, logtype="Application"):
978 logging.Handler.__init__(self)
979 try:
980 import win32evtlogutil, win32evtlog
981 self.appname = appname
982 self._welu = win32evtlogutil
983 if not dllname:
984 dllname = os.path.split(self._welu.__file__)
985 dllname = os.path.split(dllname[0])
986 dllname = os.path.join(dllname[0], r'win32service.pyd')
987 self.dllname = dllname
988 self.logtype = logtype
989 self._welu.AddSourceToRegistry(appname, dllname, logtype)
990 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
991 self.typemap = {
992 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
993 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000994 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000995 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
996 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
997 }
Brett Cannon0a140662013-06-13 20:57:26 -0400998 except ModuleNotFoundError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000999 print("The Python Win32 extensions for NT (service, event "\
1000 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001001 self._welu = None
1002
1003 def getMessageID(self, record):
1004 """
1005 Return the message ID for the event record. If you are using your
1006 own messages, you could do this by having the msg passed to the
1007 logger being an ID rather than a formatting string. Then, in here,
1008 you could use a dictionary lookup to get the message ID. This
1009 version returns 1, which is the base message ID in win32service.pyd.
1010 """
1011 return 1
1012
1013 def getEventCategory(self, record):
1014 """
1015 Return the event category for the record.
1016
1017 Override this if you want to specify your own categories. This version
1018 returns 0.
1019 """
1020 return 0
1021
1022 def getEventType(self, record):
1023 """
1024 Return the event type for the record.
1025
1026 Override this if you want to specify your own types. This version does
1027 a mapping using the handler's typemap attribute, which is set up in
1028 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001029 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001030 either need to override this method or place a suitable dictionary in
1031 the handler's typemap attribute.
1032 """
1033 return self.typemap.get(record.levelno, self.deftype)
1034
1035 def emit(self, record):
1036 """
1037 Emit a record.
1038
1039 Determine the message ID, event category and event type. Then
1040 log the message in the NT event log.
1041 """
1042 if self._welu:
1043 try:
1044 id = self.getMessageID(record)
1045 cat = self.getEventCategory(record)
1046 type = self.getEventType(record)
1047 msg = self.format(record)
1048 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001049 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001050 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001051
1052 def close(self):
1053 """
1054 Clean up this handler.
1055
1056 You can remove the application name from the registry as a
1057 source of event log entries. However, if you do this, you will
1058 not be able to see the events as you intended in the Event Log
1059 Viewer - it needs to be able to access the registry to get the
1060 DLL name.
1061 """
1062 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001063 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001064
1065class HTTPHandler(logging.Handler):
1066 """
1067 A class which sends records to a Web server, using either GET or
1068 POST semantics.
1069 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001070 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001071 """
1072 Initialize the instance with the host, the request URL, and the method
1073 ("GET" or "POST")
1074 """
1075 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001076 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001077 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001078 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001079 self.host = host
1080 self.url = url
1081 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001082 self.secure = secure
1083 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001084
Neal Norwitzf297bd12003-04-23 03:49:43 +00001085 def mapLogRecord(self, record):
1086 """
1087 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001088 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001089 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001090 """
1091 return record.__dict__
1092
Guido van Rossum57102f82002-11-13 16:15:58 +00001093 def emit(self, record):
1094 """
1095 Emit a record.
1096
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001097 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001098 """
1099 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001100 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001101 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001102 if self.secure:
1103 h = http.client.HTTPSConnection(host)
1104 else:
1105 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001106 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001107 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001108 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001109 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001110 sep = '&'
1111 else:
1112 sep = '?'
1113 url = url + "%c%s" % (sep, data)
1114 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001115 # support multiple hosts on one IP address...
1116 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001117 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001118 if i >= 0:
1119 host = host[:i]
1120 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001121 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001122 h.putheader("Content-type",
1123 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001124 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001125 if self.credentials:
1126 import base64
1127 s = ('u%s:%s' % self.credentials).encode('utf-8')
1128 s = 'Basic ' + base64.b64encode(s).strip()
1129 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001130 h.endheaders()
1131 if self.method == "POST":
1132 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001133 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001134 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001135 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001136
1137class BufferingHandler(logging.Handler):
1138 """
1139 A handler class which buffers logging records in memory. Whenever each
1140 record is added to the buffer, a check is made to see if the buffer should
1141 be flushed. If it should, then flush() is expected to do what's needed.
1142 """
1143 def __init__(self, capacity):
1144 """
1145 Initialize the handler with the buffer size.
1146 """
1147 logging.Handler.__init__(self)
1148 self.capacity = capacity
1149 self.buffer = []
1150
1151 def shouldFlush(self, record):
1152 """
1153 Should the handler flush its buffer?
1154
1155 Returns true if the buffer is up to capacity. This method can be
1156 overridden to implement custom flushing strategies.
1157 """
1158 return (len(self.buffer) >= self.capacity)
1159
1160 def emit(self, record):
1161 """
1162 Emit a record.
1163
1164 Append the record. If shouldFlush() tells us to, call flush() to process
1165 the buffer.
1166 """
1167 self.buffer.append(record)
1168 if self.shouldFlush(record):
1169 self.flush()
1170
1171 def flush(self):
1172 """
1173 Override to implement custom flushing behaviour.
1174
1175 This version just zaps the buffer to empty.
1176 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001177 self.acquire()
1178 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001179 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001180 finally:
1181 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001182
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001183 def close(self):
1184 """
1185 Close the handler.
1186
1187 This version just flushes and chains to the parent class' close().
1188 """
1189 self.flush()
1190 logging.Handler.close(self)
1191
Guido van Rossum57102f82002-11-13 16:15:58 +00001192class MemoryHandler(BufferingHandler):
1193 """
1194 A handler class which buffers logging records in memory, periodically
1195 flushing them to a target handler. Flushing occurs whenever the buffer
1196 is full, or when an event of a certain severity or greater is seen.
1197 """
1198 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1199 """
1200 Initialize the handler with the buffer size, the level at which
1201 flushing should occur and an optional target.
1202
1203 Note that without a target being set either here or via setTarget(),
1204 a MemoryHandler is no use to anyone!
1205 """
1206 BufferingHandler.__init__(self, capacity)
1207 self.flushLevel = flushLevel
1208 self.target = target
1209
1210 def shouldFlush(self, record):
1211 """
1212 Check for buffer full or a record at the flushLevel or higher.
1213 """
1214 return (len(self.buffer) >= self.capacity) or \
1215 (record.levelno >= self.flushLevel)
1216
1217 def setTarget(self, target):
1218 """
1219 Set the target handler for this handler.
1220 """
1221 self.target = target
1222
1223 def flush(self):
1224 """
1225 For a MemoryHandler, flushing means just sending the buffered
1226 records to the target, if there is one. Override if you want
1227 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001228
1229 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001230 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001231 self.acquire()
1232 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001233 if self.target:
1234 for record in self.buffer:
1235 self.target.handle(record)
1236 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001237 finally:
1238 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001239
1240 def close(self):
1241 """
1242 Flush, set the target to None and lose the buffer.
1243 """
1244 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001245 self.acquire()
1246 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001247 self.target = None
1248 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001249 finally:
1250 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001251
1252
1253class QueueHandler(logging.Handler):
1254 """
1255 This handler sends events to a queue. Typically, it would be used together
1256 with a multiprocessing Queue to centralise logging to file in one process
1257 (in a multi-process application), so as to avoid file write contention
1258 between processes.
1259
1260 This code is new in Python 3.2, but this class can be copy pasted into
1261 user code for use with earlier Python versions.
1262 """
1263
1264 def __init__(self, queue):
1265 """
1266 Initialise an instance, using the passed queue.
1267 """
1268 logging.Handler.__init__(self)
1269 self.queue = queue
1270
1271 def enqueue(self, record):
1272 """
1273 Enqueue a record.
1274
1275 The base implementation uses put_nowait. You may want to override
1276 this method if you want to use blocking, timeouts or custom queue
1277 implementations.
1278 """
1279 self.queue.put_nowait(record)
1280
Vinay Sajip0258ce82010-09-22 20:34:53 +00001281 def prepare(self, record):
1282 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001283 Prepares a record for queuing. The object returned by this method is
1284 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001285
1286 The base implementation formats the record to merge the message
1287 and arguments, and removes unpickleable items from the record
1288 in-place.
1289
1290 You might want to override this method if you want to convert
1291 the record to a dict or JSON string, or send a modified copy
1292 of the record while leaving the original intact.
1293 """
1294 # The format operation gets traceback text into record.exc_text
1295 # (if there's exception data), and also puts the message into
1296 # record.message. We can then use this to replace the original
1297 # msg + args, as these might be unpickleable. We also zap the
1298 # exc_info attribute, as it's no longer needed and, if not None,
1299 # will typically not be pickleable.
1300 self.format(record)
1301 record.msg = record.message
1302 record.args = None
1303 record.exc_info = None
1304 return record
1305
Vinay Sajip121a1c42010-09-08 10:46:15 +00001306 def emit(self, record):
1307 """
1308 Emit a record.
1309
Vinay Sajip0637d492010-09-23 08:15:54 +00001310 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001311 """
1312 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001313 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001314 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001315 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001316
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001317if threading:
1318 class QueueListener(object):
1319 """
1320 This class implements an internal threaded listener which watches for
1321 LogRecords being added to a queue, removes them and passes them to a
1322 list of handlers for processing.
1323 """
1324 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001325
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001326 def __init__(self, queue, *handlers):
1327 """
1328 Initialise an instance with the specified queue and
1329 handlers.
1330 """
1331 self.queue = queue
1332 self.handlers = handlers
1333 self._stop = threading.Event()
1334 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001335
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001336 def dequeue(self, block):
1337 """
1338 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001339
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001340 The base implementation uses get. You may want to override this method
1341 if you want to use timeouts or work with custom queue implementations.
1342 """
1343 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001344
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001345 def start(self):
1346 """
1347 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001348
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001349 This starts up a background thread to monitor the queue for
1350 LogRecords to process.
1351 """
1352 self._thread = t = threading.Thread(target=self._monitor)
1353 t.setDaemon(True)
1354 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001355
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001356 def prepare(self , record):
1357 """
1358 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001359
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001360 This method just returns the passed-in record. You may want to
1361 override this method if you need to do any custom marshalling or
1362 manipulation of the record before passing it to the handlers.
1363 """
1364 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001365
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001366 def handle(self, record):
1367 """
1368 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001369
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001370 This just loops through the handlers offering them the record
1371 to handle.
1372 """
1373 record = self.prepare(record)
1374 for handler in self.handlers:
1375 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001376
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001377 def _monitor(self):
1378 """
1379 Monitor the queue for records, and ask the handler
1380 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001381
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001382 This method runs on a separate, internal thread.
1383 The thread will terminate if it sees a sentinel object in the queue.
1384 """
1385 q = self.queue
1386 has_task_done = hasattr(q, 'task_done')
1387 while not self._stop.isSet():
1388 try:
1389 record = self.dequeue(True)
1390 if record is self._sentinel:
1391 break
1392 self.handle(record)
1393 if has_task_done:
1394 q.task_done()
1395 except queue.Empty:
1396 pass
1397 # There might still be records in the queue.
1398 while True:
1399 try:
1400 record = self.dequeue(False)
1401 if record is self._sentinel:
1402 break
1403 self.handle(record)
1404 if has_task_done:
1405 q.task_done()
1406 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001407 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001408
Victor Stinner59bec362011-05-02 16:14:16 +02001409 def enqueue_sentinel(self):
1410 """
1411 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001412
Victor Stinner59bec362011-05-02 16:14:16 +02001413 The base implementation uses put_nowait. You may want to override this
1414 method if you want to use timeouts or work with custom queue
1415 implementations.
1416 """
1417 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001418
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001419 def stop(self):
1420 """
1421 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001422
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001423 This asks the thread to terminate, and then waits for it to do so.
1424 Note that if you don't call this before your application exits, there
1425 may be some records still left on the queue, which won't be processed.
1426 """
1427 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001428 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001429 self._thread.join()
1430 self._thread = None