blob: 974c089d40ec346eebd62dd063a232dcd4efd3c0 [file] [log] [blame]
Vinay Sajipcccf6062016-07-22 16:27:31 +01001# Copyright 2001-2016 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 Sajipcccf6062016-07-22 16:27:31 +010021Copyright (C) 2001-2016 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
Antoine Pitroua6a4dc82017-09-07 18:56:24 +020029import threading
Guido van Rossum57102f82002-11-13 16:15:58 +000030
Guido van Rossum57102f82002-11-13 16:15:58 +000031#
32# Some constants...
33#
34
35DEFAULT_TCP_LOGGING_PORT = 9020
36DEFAULT_UDP_LOGGING_PORT = 9021
37DEFAULT_HTTP_LOGGING_PORT = 9022
38DEFAULT_SOAP_LOGGING_PORT = 9023
39SYSLOG_UDP_PORT = 514
Vinay Sajipcbabd7e2009-10-10 20:32:36 +000040SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000041
Thomas Wouters477c8d52006-05-27 19:21:47 +000042_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
43
Vinay Sajip17c52d82004-07-03 11:48:34 +000044class BaseRotatingHandler(logging.FileHandler):
45 """
46 Base class for handlers that rotate log files at a certain point.
47 Not meant to be instantiated directly. Instead, use RotatingFileHandler
48 or TimedRotatingFileHandler.
49 """
Vinay Sajip23b94d02012-01-04 12:02:26 +000050 def __init__(self, filename, mode, encoding=None, delay=False):
Vinay Sajip17c52d82004-07-03 11:48:34 +000051 """
52 Use the specified filename for streamed logging
53 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000054 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip4600f112005-03-13 09:56:36 +000055 self.mode = mode
56 self.encoding = encoding
Vinay Sajip23b94d02012-01-04 12:02:26 +000057 self.namer = None
58 self.rotator = None
Guido van Rossum57102f82002-11-13 16:15:58 +000059
Vinay Sajip17c52d82004-07-03 11:48:34 +000060 def emit(self, record):
61 """
62 Emit a record.
63
64 Output the record to the file, catering for rollover as described
65 in doRollover().
66 """
Vinay Sajip3970c112004-07-08 10:24:04 +000067 try:
68 if self.shouldRollover(record):
69 self.doRollover()
70 logging.FileHandler.emit(self, record)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +010071 except Exception:
Vinay Sajip3970c112004-07-08 10:24:04 +000072 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000073
Vinay Sajip23b94d02012-01-04 12:02:26 +000074 def rotation_filename(self, default_name):
75 """
76 Modify the filename of a log file when rotating.
77
78 This is provided so that a custom filename can be provided.
79
80 The default implementation calls the 'namer' attribute of the
81 handler, if it's callable, passing the default name to
82 it. If the attribute isn't callable (the default is None), the name
83 is returned unchanged.
84
85 :param default_name: The default name for the log file.
86 """
87 if not callable(self.namer):
88 result = default_name
89 else:
90 result = self.namer(default_name)
91 return result
92
93 def rotate(self, source, dest):
94 """
95 When rotating, rotate the current log.
96
97 The default implementation calls the 'rotator' attribute of the
98 handler, if it's callable, passing the source and dest arguments to
99 it. If the attribute isn't callable (the default is None), the source
100 is simply renamed to the destination.
101
102 :param source: The source filename. This is normally the base
103 filename, e.g. 'test.log'
104 :param dest: The destination filename. This is normally
105 what the source is rotated to, e.g. 'test.log.1'.
106 """
107 if not callable(self.rotator):
Vinay Sajipd8599262013-09-06 10:10:22 +0100108 # Issue 18940: A file may not have been created if delay is True.
109 if os.path.exists(source):
110 os.rename(source, dest)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000111 else:
112 self.rotator(source, dest)
113
Vinay Sajip17c52d82004-07-03 11:48:34 +0000114class RotatingFileHandler(BaseRotatingHandler):
115 """
116 Handler for logging to a set of files, which switches from one file
117 to the next when the current file reaches a certain size.
118 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000119 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
Guido van Rossum57102f82002-11-13 16:15:58 +0000120 """
121 Open the specified file and use it as the stream for logging.
122
123 By default, the file grows indefinitely. You can specify particular
124 values of maxBytes and backupCount to allow the file to rollover at
125 a predetermined size.
126
127 Rollover occurs whenever the current log file is nearly maxBytes in
128 length. If backupCount is >= 1, the system will successively create
129 new files with the same pathname as the base file, but with extensions
130 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
131 and a base file name of "app.log", you would get "app.log",
132 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
133 written to is always "app.log" - when it gets filled up, it is closed
134 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
135 exist, then they are renamed to "app.log.2", "app.log.3" etc.
136 respectively.
137
138 If maxBytes is zero, rollover never occurs.
139 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000140 # If rotation/rollover is wanted, it doesn't make sense to use another
141 # mode. If for example 'w' were specified, then if there were multiple
142 # runs of the calling application, the logs from previous runs would be
143 # lost if the 'w' is respected, because the log file would be truncated
144 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000145 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000146 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000147 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000148 self.maxBytes = maxBytes
149 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000150
151 def doRollover(self):
152 """
153 Do a rollover, as described in __init__().
154 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000155 if self.stream:
156 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000157 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000158 if self.backupCount > 0:
159 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000160 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
161 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
162 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000163 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000164 if os.path.exists(dfn):
165 os.remove(dfn)
166 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000167 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000168 if os.path.exists(dfn):
169 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000170 self.rotate(self.baseFilename, dfn)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100171 if not self.delay:
172 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000173
Vinay Sajip17c52d82004-07-03 11:48:34 +0000174 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000175 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000176 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000177
Vinay Sajip17c52d82004-07-03 11:48:34 +0000178 Basically, see if the supplied record would cause the file to exceed
179 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000180 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000181 if self.stream is None: # delay was set...
182 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000183 if self.maxBytes > 0: # are we rolling over?
184 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000185 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000186 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000187 return 1
188 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000189
Vinay Sajip17c52d82004-07-03 11:48:34 +0000190class TimedRotatingFileHandler(BaseRotatingHandler):
191 """
192 Handler for logging to a file, rotating the log file at certain timed
193 intervals.
194
195 If backupCount is > 0, when rollover is done, no more than backupCount
196 files are kept - the oldest ones are deleted.
197 """
Vinay Sajipa7130792013-04-12 17:04:23 +0100198 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 +0000199 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000200 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000201 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000202 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100203 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000204 # Calculate the real rollover interval, which is just the number of
205 # seconds between rollovers. Also set the filename suffix used when
206 # a rollover occurs. Current 'when' events supported:
207 # S - Seconds
208 # M - Minutes
209 # H - Hours
210 # D - Days
211 # midnight - roll over at midnight
212 # W{0-6} - roll over on a certain day; 0 - Monday
213 #
214 # Case of the 'when' specifier is not important; lower or upper case
215 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000216 if self.when == 'S':
217 self.interval = 1 # one second
218 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000219 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000220 elif self.when == 'M':
221 self.interval = 60 # one minute
222 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000223 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000224 elif self.when == 'H':
225 self.interval = 60 * 60 # one hour
226 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000227 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000228 elif self.when == 'D' or self.when == 'MIDNIGHT':
229 self.interval = 60 * 60 * 24 # one day
230 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000231 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000232 elif self.when.startswith('W'):
233 self.interval = 60 * 60 * 24 * 7 # one week
234 if len(self.when) != 2:
235 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
236 if self.when[1] < '0' or self.when[1] > '6':
237 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
238 self.dayOfWeek = int(self.when[1])
239 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000240 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000241 else:
242 raise ValueError("Invalid rollover interval specified: %s" % self.when)
243
Antoine Pitroufd036452008-08-19 17:56:33 +0000244 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000245 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip638e6222016-07-22 18:23:04 +0100246 # The following line added because the filename passed in could be a
247 # path object (see Issue #27493), but self.baseFilename will be a string
248 filename = self.baseFilename
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 if len(result) < self.backupCount:
357 result = []
358 else:
Lovesh Harchandaniafad1472017-10-27 09:04:33 +0200359 result.sort()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000360 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
Vinay Sajip29a14452015-10-01 20:54:41 +0100443 def reopenIfNeeded(self):
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000444 """
Vinay Sajip29a14452015-10-01 20:54:41 +0100445 Reopen log file if needed.
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000446
Vinay Sajip29a14452015-10-01 20:54:41 +0100447 Checks if the underlying file has changed, and if it
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000448 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()
Vinay Sajip29a14452015-10-01 20:54:41 +0100470
471 def emit(self, record):
472 """
473 Emit a record.
474
475 If underlying file has changed, reopen the file before emitting the
476 record to it.
477 """
478 self.reopenIfNeeded()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000479 logging.FileHandler.emit(self, record)
480
Vinay Sajip66b8b082012-04-24 23:25:30 +0100481
Guido van Rossum57102f82002-11-13 16:15:58 +0000482class SocketHandler(logging.Handler):
483 """
484 A handler class which writes logging records, in pickle format, to
485 a streaming socket. The socket is kept open across logging calls.
486 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000487 The pickle which is sent is that of the LogRecord's attribute dictionary
488 (__dict__), so that the receiver does not need to have the logging module
489 installed in order to process the logging event.
490
491 To unpickle the record at the receiving end into a LogRecord, use the
492 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000493 """
494
495 def __init__(self, host, port):
496 """
497 Initializes the handler with a specific host address and port.
498
Vinay Sajipde19e082011-04-30 21:52:26 +0100499 When the attribute *closeOnError* is set to True - if a socket error
500 occurs, the socket is silently closed and then reopened on the next
501 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000502 """
503 logging.Handler.__init__(self)
504 self.host = host
505 self.port = port
Vinay Sajip5421f352013-09-27 18:18:28 +0100506 if port is None:
507 self.address = host
508 else:
509 self.address = (host, port)
Guido van Rossum57102f82002-11-13 16:15:58 +0000510 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100511 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000512 self.retryTime = None
513 #
514 # Exponential backoff parameters.
515 #
516 self.retryStart = 1.0
517 self.retryMax = 30.0
518 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000519
Guido van Rossumd8faa362007-04-27 19:54:29 +0000520 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000521 """
522 A factory method which allows subclasses to define the precise
523 type of socket they want.
524 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100525 if self.port is not None:
526 result = socket.create_connection(self.address, timeout=timeout)
527 else:
528 result = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
529 result.settimeout(timeout)
Vinay Sajip38c741c2013-10-06 18:36:00 +0100530 try:
531 result.connect(self.address)
532 except OSError:
533 result.close() # Issue 19182
534 raise
Vinay Sajip5421f352013-09-27 18:18:28 +0100535 return result
Guido van Rossum57102f82002-11-13 16:15:58 +0000536
Vinay Sajip48cfe382004-02-20 13:17:27 +0000537 def createSocket(self):
538 """
539 Try to create a socket, using an exponential backoff with
540 a max retry time. Thanks to Robert Olson for the original patch
541 (SF #815911) which has been slightly refactored.
542 """
543 now = time.time()
544 # Either retryTime is None, in which case this
545 # is the first time back after a disconnect, or
546 # we've waited long enough.
547 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100548 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000549 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000550 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000551 if attempt:
552 try:
553 self.sock = self.makeSocket()
554 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200555 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000556 #Creation failed, so set the retry time and return.
557 if self.retryTime is None:
558 self.retryPeriod = self.retryStart
559 else:
560 self.retryPeriod = self.retryPeriod * self.retryFactor
561 if self.retryPeriod > self.retryMax:
562 self.retryPeriod = self.retryMax
563 self.retryTime = now + self.retryPeriod
564
Guido van Rossum57102f82002-11-13 16:15:58 +0000565 def send(self, s):
566 """
567 Send a pickled string to the socket.
568
569 This function allows for partial sends which can happen when the
570 network is busy.
571 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000572 if self.sock is None:
573 self.createSocket()
574 #self.sock can be None either because we haven't reached the retry
575 #time yet, or because we have reached the retry time and retried,
576 #but are still unable to connect.
577 if self.sock:
578 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200579 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200580 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000581 self.sock.close()
582 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000583
584 def makePickle(self, record):
585 """
586 Pickles the record in binary format with a length prefix, and
587 returns it ready for transmission across the socket.
588 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000589 ei = record.exc_info
590 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100591 # just to get traceback text into record.exc_text ...
592 dummy = self.format(record)
593 # See issue #14436: If msg or args are objects, they may not be
594 # available on the receiving end. So we convert the msg % args
595 # to a string, save it as msg and zap the args.
596 d = dict(record.__dict__)
597 d['msg'] = record.getMessage()
598 d['args'] = None
599 d['exc_info'] = None
Vinay Sajip9cdfd182015-12-26 12:48:44 +0000600 # Issue #25685: delete 'message' if present: redundant with 'msg'
601 d.pop('message', None)
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100602 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000603 slen = struct.pack(">L", len(s))
604 return slen + s
605
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000606 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000607 """
608 Handle an error during logging.
609
610 An error has occurred during logging. Most likely cause -
611 connection lost. Close the socket so that we can retry on the
612 next event.
613 """
614 if self.closeOnError and self.sock:
615 self.sock.close()
616 self.sock = None #try to reconnect next time
617 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000618 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000619
620 def emit(self, record):
621 """
622 Emit a record.
623
624 Pickles the record and writes it to the socket in binary format.
625 If there is an error with the socket, silently drop the packet.
626 If there was a problem with the socket, re-establishes the
627 socket.
628 """
629 try:
630 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000631 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100632 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000633 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000634
635 def close(self):
636 """
637 Closes the socket.
638 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000639 self.acquire()
640 try:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300641 sock = self.sock
642 if sock:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000643 self.sock = None
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300644 sock.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000645 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000646 finally:
647 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000648
649class DatagramHandler(SocketHandler):
650 """
651 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000652 a datagram socket. The pickle which is sent is that of the LogRecord's
653 attribute dictionary (__dict__), so that the receiver does not need to
654 have the logging module installed in order to process the logging event.
655
656 To unpickle the record at the receiving end into a LogRecord, use the
657 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000658
659 """
660 def __init__(self, host, port):
661 """
662 Initializes the handler with a specific host address and port.
663 """
664 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100665 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000666
667 def makeSocket(self):
668 """
669 The factory method of SocketHandler is here overridden to create
670 a UDP socket (SOCK_DGRAM).
671 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100672 if self.port is None:
673 family = socket.AF_UNIX
674 else:
675 family = socket.AF_INET
676 s = socket.socket(family, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000677 return s
678
679 def send(self, s):
680 """
681 Send a pickled string to a socket.
682
683 This function no longer allows for partial sends which can happen
684 when the network is busy - UDP does not guarantee delivery and
685 can deliver packets out of sequence.
686 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000687 if self.sock is None:
688 self.createSocket()
Vinay Sajip5421f352013-09-27 18:18:28 +0100689 self.sock.sendto(s, self.address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000690
691class SysLogHandler(logging.Handler):
692 """
693 A handler class which sends formatted logging records to a syslog
694 server. Based on Sam Rushing's syslog module:
695 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
696 Contributed by Nicolas Untz (after which minor refactoring changes
697 have been made).
698 """
699
700 # from <linux/sys/syslog.h>:
701 # ======================================================================
702 # priorities/facilities are encoded into a single 32-bit quantity, where
703 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
704 # facility (0-big number). Both the priorities and the facilities map
705 # roughly one-to-one to strings in the syslogd(8) source code. This
706 # mapping is included in this file.
707 #
708 # priorities (these are ordered)
709
710 LOG_EMERG = 0 # system is unusable
711 LOG_ALERT = 1 # action must be taken immediately
712 LOG_CRIT = 2 # critical conditions
713 LOG_ERR = 3 # error conditions
714 LOG_WARNING = 4 # warning conditions
715 LOG_NOTICE = 5 # normal but significant condition
716 LOG_INFO = 6 # informational
717 LOG_DEBUG = 7 # debug-level messages
718
719 # facility codes
720 LOG_KERN = 0 # kernel messages
721 LOG_USER = 1 # random user-level messages
722 LOG_MAIL = 2 # mail system
723 LOG_DAEMON = 3 # system daemons
724 LOG_AUTH = 4 # security/authorization messages
725 LOG_SYSLOG = 5 # messages generated internally by syslogd
726 LOG_LPR = 6 # line printer subsystem
727 LOG_NEWS = 7 # network news subsystem
728 LOG_UUCP = 8 # UUCP subsystem
729 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000730 LOG_AUTHPRIV = 10 # security/authorization messages (private)
731 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000732
733 # other codes through 15 reserved for system use
734 LOG_LOCAL0 = 16 # reserved for local use
735 LOG_LOCAL1 = 17 # reserved for local use
736 LOG_LOCAL2 = 18 # reserved for local use
737 LOG_LOCAL3 = 19 # reserved for local use
738 LOG_LOCAL4 = 20 # reserved for local use
739 LOG_LOCAL5 = 21 # reserved for local use
740 LOG_LOCAL6 = 22 # reserved for local use
741 LOG_LOCAL7 = 23 # reserved for local use
742
743 priority_names = {
744 "alert": LOG_ALERT,
745 "crit": LOG_CRIT,
746 "critical": LOG_CRIT,
747 "debug": LOG_DEBUG,
748 "emerg": LOG_EMERG,
749 "err": LOG_ERR,
750 "error": LOG_ERR, # DEPRECATED
751 "info": LOG_INFO,
752 "notice": LOG_NOTICE,
753 "panic": LOG_EMERG, # DEPRECATED
754 "warn": LOG_WARNING, # DEPRECATED
755 "warning": LOG_WARNING,
756 }
757
758 facility_names = {
759 "auth": LOG_AUTH,
760 "authpriv": LOG_AUTHPRIV,
761 "cron": LOG_CRON,
762 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000763 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000764 "kern": LOG_KERN,
765 "lpr": LOG_LPR,
766 "mail": LOG_MAIL,
767 "news": LOG_NEWS,
768 "security": LOG_AUTH, # DEPRECATED
769 "syslog": LOG_SYSLOG,
770 "user": LOG_USER,
771 "uucp": LOG_UUCP,
772 "local0": LOG_LOCAL0,
773 "local1": LOG_LOCAL1,
774 "local2": LOG_LOCAL2,
775 "local3": LOG_LOCAL3,
776 "local4": LOG_LOCAL4,
777 "local5": LOG_LOCAL5,
778 "local6": LOG_LOCAL6,
779 "local7": LOG_LOCAL7,
780 }
781
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000782 #The map below appears to be trivially lowercasing the key. However,
783 #there's more to it than meets the eye - in some locales, lowercasing
784 #gives unexpected results. See SF #1524081: in the Turkish locale,
785 #"INFO".lower() != "info"
786 priority_map = {
787 "DEBUG" : "debug",
788 "INFO" : "info",
789 "WARNING" : "warning",
790 "ERROR" : "error",
791 "CRITICAL" : "critical"
792 }
793
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000794 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100795 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000796 """
797 Initialize a handler.
798
Guido van Rossume7ba4952007-06-06 23:52:48 +0000799 If address is specified as a string, a UNIX socket is used. To log to a
800 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100801 If facility is not specified, LOG_USER is used. If socktype is
802 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
803 socket type will be used. For Unix sockets, you can also specify a
804 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
805 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000806 """
807 logging.Handler.__init__(self)
808
809 self.address = address
810 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000811 self.socktype = socktype
812
Guido van Rossum13257902007-06-07 23:15:56 +0000813 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100814 self.unixsocket = True
Коренберг Марк1b038e02017-03-17 20:25:05 +0500815 # Syslog server may be unavailable during handler initialisation.
816 # C's openlog() function also ignores connection errors.
817 # Moreover, we ignore these errors while logging, so it not worse
818 # to ignore it also here.
819 try:
820 self._connect_unixsocket(address)
821 except OSError:
822 pass
Guido van Rossum57102f82002-11-13 16:15:58 +0000823 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100824 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100825 if socktype is None:
826 socktype = socket.SOCK_DGRAM
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800827 host, port = address
828 ress = socket.getaddrinfo(host, port, 0, socktype)
829 if not ress:
830 raise OSError("getaddrinfo returns an empty list")
831 for res in ress:
832 af, socktype, proto, _, sa = res
833 err = sock = None
834 try:
835 sock = socket.socket(af, socktype, proto)
836 if socktype == socket.SOCK_STREAM:
837 sock.connect(sa)
838 break
839 except OSError as exc:
840 err = exc
841 if sock is not None:
842 sock.close()
843 if err is not None:
844 raise err
845 self.socket = sock
Vinay Sajipe9170522013-04-22 10:07:40 +0100846 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000847
Vinay Sajipa1974c12005-01-13 08:23:56 +0000848 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100849 use_socktype = self.socktype
850 if use_socktype is None:
851 use_socktype = socket.SOCK_DGRAM
852 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000853 try:
854 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100855 # it worked, so set self.socktype to the used type
856 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200857 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000858 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100859 if self.socktype is not None:
860 # user didn't specify falling back, so fail
861 raise
862 use_socktype = socket.SOCK_STREAM
863 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
864 try:
865 self.socket.connect(address)
866 # it worked, so set self.socktype to the used type
867 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100868 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100869 self.socket.close()
870 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000871
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000872 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000873 """
874 Encode the facility and priority. You can pass in strings or
875 integers - if strings are passed, the facility_names and
876 priority_names mapping dictionaries are used to convert them to
877 integers.
878 """
Guido van Rossum13257902007-06-07 23:15:56 +0000879 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000880 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000881 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000882 priority = self.priority_names[priority]
883 return (facility << 3) | priority
884
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800885 def close(self):
Guido van Rossum57102f82002-11-13 16:15:58 +0000886 """
887 Closes the socket.
888 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000889 self.acquire()
890 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000891 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000892 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000893 finally:
894 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000895
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000896 def mapPriority(self, levelName):
897 """
898 Map a logging level name to a key in the priority_names map.
899 This is useful in two scenarios: when custom levels are being
900 used, and in the case where you can't do a straightforward
901 mapping by lowercasing the logging level name because of locale-
902 specific issues (see SF #1524081).
903 """
904 return self.priority_map.get(levelName, "warning")
905
Vinay Sajip2353e352011-06-27 15:40:06 +0100906 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100907 append_nul = True # some old syslog daemons expect a NUL terminator
908
Guido van Rossum57102f82002-11-13 16:15:58 +0000909 def emit(self, record):
910 """
911 Emit a record.
912
913 The record is formatted, and then sent to the syslog server. If
914 exception information is present, it is NOT sent to the server.
915 """
Guido van Rossum57102f82002-11-13 16:15:58 +0000916 try:
Vinay Sajipc33a0cc2014-11-01 19:58:47 +0000917 msg = self.format(record)
918 if self.ident:
919 msg = self.ident + msg
920 if self.append_nul:
921 msg += '\000'
922
923 # We need to convert record level to lowercase, maybe this will
924 # change in the future.
925 prio = '<%d>' % self.encodePriority(self.facility,
926 self.mapPriority(record.levelname))
927 prio = prio.encode('utf-8')
928 # Message is a string. Convert to bytes as required by RFC 5424
929 msg = msg.encode('utf-8')
930 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000931 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000932 try:
933 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200934 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100935 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000936 self._connect_unixsocket(self.address)
937 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000938 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000939 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000940 else:
941 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100942 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000943 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000944
945class SMTPHandler(logging.Handler):
946 """
947 A handler class which sends an SMTP email for each logging event.
948 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000949 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100950 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000951 """
952 Initialize the handler.
953
954 Initialize the instance with the from and to addresses and subject
955 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000956 (host, port) tuple format for the mailhost argument. To specify
957 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000958 for the credentials argument. To specify the use of a secure
959 protocol (TLS), pass in a tuple for the secure argument. This will
960 only be used when authentication credentials are supplied. The tuple
961 will be either an empty tuple, or a single-value tuple with the name
962 of a keyfile, or a 2-value tuple with the names of the keyfile and
963 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000964 A timeout in seconds can be specified for the SMTP connection (the
965 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000966 """
967 logging.Handler.__init__(self)
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100968 if isinstance(mailhost, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000969 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000970 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000971 self.mailhost, self.mailport = mailhost, None
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100972 if isinstance(credentials, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000973 self.username, self.password = credentials
974 else:
975 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000976 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000977 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000978 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000979 self.toaddrs = toaddrs
980 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000981 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000982 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000983
984 def getSubject(self, record):
985 """
986 Determine the subject for the email.
987
988 If you want to specify a subject line which is record-dependent,
989 override this method.
990 """
991 return self.subject
992
993 def emit(self, record):
994 """
995 Emit a record.
996
997 Format the record and send it to the specified addressees.
998 """
999 try:
1000 import smtplib
Vinay Sajip277640a2015-10-17 16:13:10 +01001001 from email.message import EmailMessage
1002 import email.utils
1003
Guido van Rossum57102f82002-11-13 16:15:58 +00001004 port = self.mailport
1005 if not port:
1006 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +00001007 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Vinay Sajip277640a2015-10-17 16:13:10 +01001008 msg = EmailMessage()
1009 msg['From'] = self.fromaddr
1010 msg['To'] = ','.join(self.toaddrs)
1011 msg['Subject'] = self.getSubject(record)
1012 msg['Date'] = email.utils.localtime()
1013 msg.set_content(self.format(record))
Guido van Rossum360e4b82007-05-14 22:51:27 +00001014 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +00001015 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +00001016 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +00001017 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +00001018 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +00001019 smtp.login(self.username, self.password)
Vinay Sajip277640a2015-10-17 16:13:10 +01001020 smtp.send_message(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +00001021 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001022 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001023 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001024
1025class NTEventLogHandler(logging.Handler):
1026 """
1027 A handler class which sends events to the NT Event Log. Adds a
1028 registry entry for the specified application name. If no dllname is
1029 provided, win32service.pyd (which contains some basic message
1030 placeholders) is used. Note that use of these placeholders will make
1031 your event logs big, as the entire message source is held in the log.
1032 If you want slimmer logs, you have to pass in the name of your own DLL
1033 which contains the message definitions you want to use in the event log.
1034 """
1035 def __init__(self, appname, dllname=None, logtype="Application"):
1036 logging.Handler.__init__(self)
1037 try:
1038 import win32evtlogutil, win32evtlog
1039 self.appname = appname
1040 self._welu = win32evtlogutil
1041 if not dllname:
1042 dllname = os.path.split(self._welu.__file__)
1043 dllname = os.path.split(dllname[0])
1044 dllname = os.path.join(dllname[0], r'win32service.pyd')
1045 self.dllname = dllname
1046 self.logtype = logtype
1047 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1048 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1049 self.typemap = {
1050 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1051 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001052 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001053 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1054 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1055 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001056 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001057 print("The Python Win32 extensions for NT (service, event "\
1058 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001059 self._welu = None
1060
1061 def getMessageID(self, record):
1062 """
1063 Return the message ID for the event record. If you are using your
1064 own messages, you could do this by having the msg passed to the
1065 logger being an ID rather than a formatting string. Then, in here,
1066 you could use a dictionary lookup to get the message ID. This
1067 version returns 1, which is the base message ID in win32service.pyd.
1068 """
1069 return 1
1070
1071 def getEventCategory(self, record):
1072 """
1073 Return the event category for the record.
1074
1075 Override this if you want to specify your own categories. This version
1076 returns 0.
1077 """
1078 return 0
1079
1080 def getEventType(self, record):
1081 """
1082 Return the event type for the record.
1083
1084 Override this if you want to specify your own types. This version does
1085 a mapping using the handler's typemap attribute, which is set up in
1086 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001087 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001088 either need to override this method or place a suitable dictionary in
1089 the handler's typemap attribute.
1090 """
1091 return self.typemap.get(record.levelno, self.deftype)
1092
1093 def emit(self, record):
1094 """
1095 Emit a record.
1096
1097 Determine the message ID, event category and event type. Then
1098 log the message in the NT event log.
1099 """
1100 if self._welu:
1101 try:
1102 id = self.getMessageID(record)
1103 cat = self.getEventCategory(record)
1104 type = self.getEventType(record)
1105 msg = self.format(record)
1106 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001107 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001108 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001109
1110 def close(self):
1111 """
1112 Clean up this handler.
1113
1114 You can remove the application name from the registry as a
1115 source of event log entries. However, if you do this, you will
1116 not be able to see the events as you intended in the Event Log
1117 Viewer - it needs to be able to access the registry to get the
1118 DLL name.
1119 """
1120 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001121 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001122
1123class HTTPHandler(logging.Handler):
1124 """
1125 A class which sends records to a Web server, using either GET or
1126 POST semantics.
1127 """
Benjamin Peterson43052a12014-11-23 20:36:44 -06001128 def __init__(self, host, url, method="GET", secure=False, credentials=None,
1129 context=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001130 """
1131 Initialize the instance with the host, the request URL, and the method
1132 ("GET" or "POST")
1133 """
1134 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001135 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001136 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001137 raise ValueError("method must be GET or POST")
Benjamin Peterson43052a12014-11-23 20:36:44 -06001138 if not secure and context is not None:
1139 raise ValueError("context parameter only makes sense "
1140 "with secure=True")
Guido van Rossum57102f82002-11-13 16:15:58 +00001141 self.host = host
1142 self.url = url
1143 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001144 self.secure = secure
1145 self.credentials = credentials
Benjamin Peterson43052a12014-11-23 20:36:44 -06001146 self.context = context
Guido van Rossum57102f82002-11-13 16:15:58 +00001147
Neal Norwitzf297bd12003-04-23 03:49:43 +00001148 def mapLogRecord(self, record):
1149 """
1150 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001151 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001152 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001153 """
1154 return record.__dict__
1155
Guido van Rossum57102f82002-11-13 16:15:58 +00001156 def emit(self, record):
1157 """
1158 Emit a record.
1159
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001160 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001161 """
1162 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001163 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001164 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001165 if self.secure:
Benjamin Peterson43052a12014-11-23 20:36:44 -06001166 h = http.client.HTTPSConnection(host, context=self.context)
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001167 else:
1168 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001169 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001170 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001171 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001172 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001173 sep = '&'
1174 else:
1175 sep = '?'
1176 url = url + "%c%s" % (sep, data)
1177 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001178 # support multiple hosts on one IP address...
1179 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001180 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001181 if i >= 0:
1182 host = host[:i]
Vinay Sajipe96ba182017-11-19 18:36:17 +00001183 # See issue #30904: putrequest call above already adds this header
1184 # on Python 3.x.
1185 # h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001186 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001187 h.putheader("Content-type",
1188 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001189 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001190 if self.credentials:
1191 import base64
Vinay Sajip1bf197e2016-06-07 21:19:55 +01001192 s = ('%s:%s' % self.credentials).encode('utf-8')
1193 s = 'Basic ' + base64.b64encode(s).strip().decode('ascii')
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001194 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001195 h.endheaders()
1196 if self.method == "POST":
1197 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001198 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001199 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001200 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001201
1202class BufferingHandler(logging.Handler):
1203 """
1204 A handler class which buffers logging records in memory. Whenever each
1205 record is added to the buffer, a check is made to see if the buffer should
1206 be flushed. If it should, then flush() is expected to do what's needed.
1207 """
1208 def __init__(self, capacity):
1209 """
1210 Initialize the handler with the buffer size.
1211 """
1212 logging.Handler.__init__(self)
1213 self.capacity = capacity
1214 self.buffer = []
1215
1216 def shouldFlush(self, record):
1217 """
1218 Should the handler flush its buffer?
1219
1220 Returns true if the buffer is up to capacity. This method can be
1221 overridden to implement custom flushing strategies.
1222 """
1223 return (len(self.buffer) >= self.capacity)
1224
1225 def emit(self, record):
1226 """
1227 Emit a record.
1228
1229 Append the record. If shouldFlush() tells us to, call flush() to process
1230 the buffer.
1231 """
1232 self.buffer.append(record)
1233 if self.shouldFlush(record):
1234 self.flush()
1235
1236 def flush(self):
1237 """
1238 Override to implement custom flushing behaviour.
1239
1240 This version just zaps the buffer to empty.
1241 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001242 self.acquire()
1243 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001244 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001245 finally:
1246 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001247
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001248 def close(self):
1249 """
1250 Close the handler.
1251
1252 This version just flushes and chains to the parent class' close().
1253 """
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001254 try:
1255 self.flush()
1256 finally:
1257 logging.Handler.close(self)
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001258
Guido van Rossum57102f82002-11-13 16:15:58 +00001259class MemoryHandler(BufferingHandler):
1260 """
1261 A handler class which buffers logging records in memory, periodically
1262 flushing them to a target handler. Flushing occurs whenever the buffer
1263 is full, or when an event of a certain severity or greater is seen.
1264 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001265 def __init__(self, capacity, flushLevel=logging.ERROR, target=None,
1266 flushOnClose=True):
Guido van Rossum57102f82002-11-13 16:15:58 +00001267 """
1268 Initialize the handler with the buffer size, the level at which
1269 flushing should occur and an optional target.
1270
1271 Note that without a target being set either here or via setTarget(),
1272 a MemoryHandler is no use to anyone!
Vinay Sajipcccf6062016-07-22 16:27:31 +01001273
1274 The ``flushOnClose`` argument is ``True`` for backward compatibility
1275 reasons - the old behaviour is that when the handler is closed, the
1276 buffer is flushed, even if the flush level hasn't been exceeded nor the
1277 capacity exceeded. To prevent this, set ``flushOnClose`` to ``False``.
Guido van Rossum57102f82002-11-13 16:15:58 +00001278 """
1279 BufferingHandler.__init__(self, capacity)
1280 self.flushLevel = flushLevel
1281 self.target = target
Vinay Sajipcccf6062016-07-22 16:27:31 +01001282 # See Issue #26559 for why this has been added
1283 self.flushOnClose = flushOnClose
Guido van Rossum57102f82002-11-13 16:15:58 +00001284
1285 def shouldFlush(self, record):
1286 """
1287 Check for buffer full or a record at the flushLevel or higher.
1288 """
1289 return (len(self.buffer) >= self.capacity) or \
1290 (record.levelno >= self.flushLevel)
1291
1292 def setTarget(self, target):
1293 """
1294 Set the target handler for this handler.
1295 """
1296 self.target = target
1297
1298 def flush(self):
1299 """
1300 For a MemoryHandler, flushing means just sending the buffered
1301 records to the target, if there is one. Override if you want
1302 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001303
1304 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001305 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001306 self.acquire()
1307 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001308 if self.target:
1309 for record in self.buffer:
1310 self.target.handle(record)
1311 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001312 finally:
1313 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001314
1315 def close(self):
1316 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001317 Flush, if appropriately configured, set the target to None and lose the
1318 buffer.
Guido van Rossum57102f82002-11-13 16:15:58 +00001319 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001320 try:
Vinay Sajipcccf6062016-07-22 16:27:31 +01001321 if self.flushOnClose:
1322 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001323 finally:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001324 self.acquire()
1325 try:
1326 self.target = None
1327 BufferingHandler.close(self)
1328 finally:
1329 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001330
1331
1332class QueueHandler(logging.Handler):
1333 """
1334 This handler sends events to a queue. Typically, it would be used together
1335 with a multiprocessing Queue to centralise logging to file in one process
1336 (in a multi-process application), so as to avoid file write contention
1337 between processes.
1338
1339 This code is new in Python 3.2, but this class can be copy pasted into
1340 user code for use with earlier Python versions.
1341 """
1342
1343 def __init__(self, queue):
1344 """
1345 Initialise an instance, using the passed queue.
1346 """
1347 logging.Handler.__init__(self)
1348 self.queue = queue
1349
1350 def enqueue(self, record):
1351 """
1352 Enqueue a record.
1353
1354 The base implementation uses put_nowait. You may want to override
1355 this method if you want to use blocking, timeouts or custom queue
1356 implementations.
1357 """
1358 self.queue.put_nowait(record)
1359
Vinay Sajip0258ce82010-09-22 20:34:53 +00001360 def prepare(self, record):
1361 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001362 Prepares a record for queuing. The object returned by this method is
1363 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001364
1365 The base implementation formats the record to merge the message
1366 and arguments, and removes unpickleable items from the record
1367 in-place.
1368
1369 You might want to override this method if you want to convert
1370 the record to a dict or JSON string, or send a modified copy
1371 of the record while leaving the original intact.
1372 """
1373 # The format operation gets traceback text into record.exc_text
favlladfe3442017-08-01 20:12:26 +02001374 # (if there's exception data), and also returns the formatted
1375 # message. We can then use this to replace the original
Vinay Sajip0258ce82010-09-22 20:34:53 +00001376 # msg + args, as these might be unpickleable. We also zap the
1377 # exc_info attribute, as it's no longer needed and, if not None,
1378 # will typically not be pickleable.
favlladfe3442017-08-01 20:12:26 +02001379 msg = self.format(record)
1380 record.message = msg
1381 record.msg = msg
Vinay Sajip0258ce82010-09-22 20:34:53 +00001382 record.args = None
1383 record.exc_info = None
1384 return record
1385
Vinay Sajip121a1c42010-09-08 10:46:15 +00001386 def emit(self, record):
1387 """
1388 Emit a record.
1389
Vinay Sajip0637d492010-09-23 08:15:54 +00001390 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001391 """
1392 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001393 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001394 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001395 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001396
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001397
1398class QueueListener(object):
1399 """
1400 This class implements an internal threaded listener which watches for
1401 LogRecords being added to a queue, removes them and passes them to a
1402 list of handlers for processing.
1403 """
1404 _sentinel = None
1405
1406 def __init__(self, queue, *handlers, respect_handler_level=False):
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001407 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001408 Initialise an instance with the specified queue and
1409 handlers.
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001410 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001411 self.queue = queue
1412 self.handlers = handlers
1413 self._thread = None
1414 self.respect_handler_level = respect_handler_level
Vinay Sajip0637d492010-09-23 08:15:54 +00001415
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001416 def dequeue(self, block):
1417 """
1418 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001419
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001420 The base implementation uses get. You may want to override this method
1421 if you want to use timeouts or work with custom queue implementations.
1422 """
1423 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001424
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001425 def start(self):
1426 """
1427 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001428
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001429 This starts up a background thread to monitor the queue for
1430 LogRecords to process.
1431 """
1432 self._thread = t = threading.Thread(target=self._monitor)
1433 t.daemon = True
1434 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001435
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001436 def prepare(self , record):
1437 """
1438 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001439
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001440 This method just returns the passed-in record. You may want to
1441 override this method if you need to do any custom marshalling or
1442 manipulation of the record before passing it to the handlers.
1443 """
1444 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001445
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001446 def handle(self, record):
1447 """
1448 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001449
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001450 This just loops through the handlers offering them the record
1451 to handle.
1452 """
1453 record = self.prepare(record)
1454 for handler in self.handlers:
1455 if not self.respect_handler_level:
1456 process = True
1457 else:
1458 process = record.levelno >= handler.level
1459 if process:
1460 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001461
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001462 def _monitor(self):
1463 """
1464 Monitor the queue for records, and ask the handler
1465 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001466
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001467 This method runs on a separate, internal thread.
1468 The thread will terminate if it sees a sentinel object in the queue.
1469 """
1470 q = self.queue
1471 has_task_done = hasattr(q, 'task_done')
1472 while True:
1473 try:
1474 record = self.dequeue(True)
1475 if record is self._sentinel:
Vinay Sajip0637d492010-09-23 08:15:54 +00001476 break
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001477 self.handle(record)
1478 if has_task_done:
1479 q.task_done()
1480 except queue.Empty:
1481 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001482
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001483 def enqueue_sentinel(self):
1484 """
1485 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001486
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001487 The base implementation uses put_nowait. You may want to override this
1488 method if you want to use timeouts or work with custom queue
1489 implementations.
1490 """
1491 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001492
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001493 def stop(self):
1494 """
1495 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001496
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001497 This asks the thread to terminate, and then waits for it to do so.
1498 Note that if you don't call this before your application exits, there
1499 may be some records still left on the queue, which won't be processed.
1500 """
1501 self.enqueue_sentinel()
1502 self._thread.join()
1503 self._thread = None