blob: 5641fee573556794653ad29e829b4907d55dd3a4 [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
Manjusakada6424e2019-01-23 15:08:38 +080030import copy
Guido van Rossum57102f82002-11-13 16:15:58 +000031
Guido van Rossum57102f82002-11-13 16:15:58 +000032#
33# Some constants...
34#
35
36DEFAULT_TCP_LOGGING_PORT = 9020
37DEFAULT_UDP_LOGGING_PORT = 9021
38DEFAULT_HTTP_LOGGING_PORT = 9022
39DEFAULT_SOAP_LOGGING_PORT = 9023
40SYSLOG_UDP_PORT = 514
Vinay Sajipcbabd7e2009-10-10 20:32:36 +000041SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000042
Thomas Wouters477c8d52006-05-27 19:21:47 +000043_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
44
Vinay Sajip17c52d82004-07-03 11:48:34 +000045class BaseRotatingHandler(logging.FileHandler):
46 """
47 Base class for handlers that rotate log files at a certain point.
48 Not meant to be instantiated directly. Instead, use RotatingFileHandler
49 or TimedRotatingFileHandler.
50 """
Vinay Sajipca7b5042019-06-17 17:40:52 +010051 def __init__(self, filename, mode, encoding=None, delay=False, errors=None):
Vinay Sajip17c52d82004-07-03 11:48:34 +000052 """
53 Use the specified filename for streamed logging
54 """
Vinay Sajipca7b5042019-06-17 17:40:52 +010055 logging.FileHandler.__init__(self, filename, mode=mode,
56 encoding=encoding, delay=delay,
57 errors=errors)
Vinay Sajip4600f112005-03-13 09:56:36 +000058 self.mode = mode
59 self.encoding = encoding
Vinay Sajipca7b5042019-06-17 17:40:52 +010060 self.errors = errors
Vinay Sajip23b94d02012-01-04 12:02:26 +000061 self.namer = None
62 self.rotator = None
Guido van Rossum57102f82002-11-13 16:15:58 +000063
Vinay Sajip17c52d82004-07-03 11:48:34 +000064 def emit(self, record):
65 """
66 Emit a record.
67
68 Output the record to the file, catering for rollover as described
69 in doRollover().
70 """
Vinay Sajip3970c112004-07-08 10:24:04 +000071 try:
72 if self.shouldRollover(record):
73 self.doRollover()
74 logging.FileHandler.emit(self, record)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +010075 except Exception:
Vinay Sajip3970c112004-07-08 10:24:04 +000076 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000077
Vinay Sajip23b94d02012-01-04 12:02:26 +000078 def rotation_filename(self, default_name):
79 """
80 Modify the filename of a log file when rotating.
81
82 This is provided so that a custom filename can be provided.
83
84 The default implementation calls the 'namer' attribute of the
85 handler, if it's callable, passing the default name to
86 it. If the attribute isn't callable (the default is None), the name
87 is returned unchanged.
88
89 :param default_name: The default name for the log file.
90 """
91 if not callable(self.namer):
92 result = default_name
93 else:
94 result = self.namer(default_name)
95 return result
96
97 def rotate(self, source, dest):
98 """
99 When rotating, rotate the current log.
100
101 The default implementation calls the 'rotator' attribute of the
102 handler, if it's callable, passing the source and dest arguments to
103 it. If the attribute isn't callable (the default is None), the source
104 is simply renamed to the destination.
105
106 :param source: The source filename. This is normally the base
107 filename, e.g. 'test.log'
108 :param dest: The destination filename. This is normally
109 what the source is rotated to, e.g. 'test.log.1'.
110 """
111 if not callable(self.rotator):
Vinay Sajipd8599262013-09-06 10:10:22 +0100112 # Issue 18940: A file may not have been created if delay is True.
113 if os.path.exists(source):
114 os.rename(source, dest)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000115 else:
116 self.rotator(source, dest)
117
Vinay Sajip17c52d82004-07-03 11:48:34 +0000118class RotatingFileHandler(BaseRotatingHandler):
119 """
120 Handler for logging to a set of files, which switches from one file
121 to the next when the current file reaches a certain size.
122 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100123 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0,
124 encoding=None, delay=False, errors=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000125 """
126 Open the specified file and use it as the stream for logging.
127
128 By default, the file grows indefinitely. You can specify particular
129 values of maxBytes and backupCount to allow the file to rollover at
130 a predetermined size.
131
132 Rollover occurs whenever the current log file is nearly maxBytes in
133 length. If backupCount is >= 1, the system will successively create
134 new files with the same pathname as the base file, but with extensions
135 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
136 and a base file name of "app.log", you would get "app.log",
137 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
138 written to is always "app.log" - when it gets filled up, it is closed
139 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
140 exist, then they are renamed to "app.log.2", "app.log.3" etc.
141 respectively.
142
143 If maxBytes is zero, rollover never occurs.
144 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000145 # If rotation/rollover is wanted, it doesn't make sense to use another
146 # mode. If for example 'w' were specified, then if there were multiple
147 # runs of the calling application, the logs from previous runs would be
148 # lost if the 'w' is respected, because the log file would be truncated
149 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000150 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000151 mode = 'a'
Vinay Sajipca7b5042019-06-17 17:40:52 +0100152 BaseRotatingHandler.__init__(self, filename, mode, encoding=encoding,
153 delay=delay, errors=errors)
Guido van Rossum57102f82002-11-13 16:15:58 +0000154 self.maxBytes = maxBytes
155 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000156
157 def doRollover(self):
158 """
159 Do a rollover, as described in __init__().
160 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000161 if self.stream:
162 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000163 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000164 if self.backupCount > 0:
165 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000166 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
167 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
168 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000169 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000170 if os.path.exists(dfn):
171 os.remove(dfn)
172 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000173 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000174 if os.path.exists(dfn):
175 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000176 self.rotate(self.baseFilename, dfn)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100177 if not self.delay:
178 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000179
Vinay Sajip17c52d82004-07-03 11:48:34 +0000180 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000181 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000182 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000183
Vinay Sajip17c52d82004-07-03 11:48:34 +0000184 Basically, see if the supplied record would cause the file to exceed
185 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000186 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000187 if self.stream is None: # delay was set...
188 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000189 if self.maxBytes > 0: # are we rolling over?
190 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000191 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000192 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000193 return 1
194 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000195
Vinay Sajip17c52d82004-07-03 11:48:34 +0000196class TimedRotatingFileHandler(BaseRotatingHandler):
197 """
198 Handler for logging to a file, rotating the log file at certain timed
199 intervals.
200
201 If backupCount is > 0, when rollover is done, no more than backupCount
202 files are kept - the oldest ones are deleted.
203 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100204 def __init__(self, filename, when='h', interval=1, backupCount=0,
205 encoding=None, delay=False, utc=False, atTime=None,
206 errors=None):
207 BaseRotatingHandler.__init__(self, filename, 'a', encoding=encoding,
208 delay=delay, errors=errors)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000209 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000210 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000211 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100212 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000213 # Calculate the real rollover interval, which is just the number of
214 # seconds between rollovers. Also set the filename suffix used when
215 # a rollover occurs. Current 'when' events supported:
216 # S - Seconds
217 # M - Minutes
218 # H - Hours
219 # D - Days
220 # midnight - roll over at midnight
221 # W{0-6} - roll over on a certain day; 0 - Monday
222 #
223 # Case of the 'when' specifier is not important; lower or upper case
224 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000225 if self.when == 'S':
226 self.interval = 1 # one second
227 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000228 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000229 elif self.when == 'M':
230 self.interval = 60 # one minute
231 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000232 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000233 elif self.when == 'H':
234 self.interval = 60 * 60 # one hour
235 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000236 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000237 elif self.when == 'D' or self.when == 'MIDNIGHT':
238 self.interval = 60 * 60 * 24 # one day
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 elif self.when.startswith('W'):
242 self.interval = 60 * 60 * 24 * 7 # one week
243 if len(self.when) != 2:
244 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
245 if self.when[1] < '0' or self.when[1] > '6':
246 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
247 self.dayOfWeek = int(self.when[1])
248 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000249 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000250 else:
251 raise ValueError("Invalid rollover interval specified: %s" % self.when)
252
Antoine Pitroufd036452008-08-19 17:56:33 +0000253 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000254 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip638e6222016-07-22 18:23:04 +0100255 # The following line added because the filename passed in could be a
256 # path object (see Issue #27493), but self.baseFilename will be a string
257 filename = self.baseFilename
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000258 if os.path.exists(filename):
259 t = os.stat(filename)[ST_MTIME]
260 else:
261 t = int(time.time())
262 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000263
Vinay Sajipaffbd872009-06-11 10:11:47 +0000264 def computeRollover(self, currentTime):
265 """
266 Work out the rollover time based on the specified time.
267 """
268 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000269 # If we are rolling over at midnight or weekly, then the interval is already known.
270 # What we need to figure out is WHEN the next interval is. In other words,
271 # if you are rolling over at midnight, then your base interval is 1 day,
272 # but you want to start that one day clock at midnight, not now. So, we
273 # have to fudge the rolloverAt value in order to trigger the first rollover
274 # at the right time. After that, the regular interval will take care of
275 # the rest. Note that this code doesn't care about leap seconds. :)
276 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
277 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000278 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000279 t = time.gmtime(currentTime)
280 else:
281 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000282 currentHour = t[3]
283 currentMinute = t[4]
284 currentSecond = t[5]
Vinay Sajipa7130792013-04-12 17:04:23 +0100285 currentDay = t[6]
286 # r is the number of seconds left between now and the next rotation
287 if self.atTime is None:
288 rotate_ts = _MIDNIGHT
289 else:
290 rotate_ts = ((self.atTime.hour * 60 + self.atTime.minute)*60 +
291 self.atTime.second)
292
293 r = rotate_ts - ((currentHour * 60 + currentMinute) * 60 +
294 currentSecond)
295 if r < 0:
296 # Rotate time is before the current time (for example when
297 # self.rotateAt is 13:45 and it now 14:15), rotation is
298 # tomorrow.
299 r += _MIDNIGHT
300 currentDay = (currentDay + 1) % 7
Vinay Sajipaffbd872009-06-11 10:11:47 +0000301 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000302 # If we are rolling over on a certain day, add in the number of days until
303 # the next rollover, but offset by 1 since we just calculated the time
304 # until the next day starts. There are three cases:
305 # Case 1) The day to rollover is today; in this case, do nothing
306 # Case 2) The day to rollover is further in the interval (i.e., today is
307 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
308 # next rollover is simply 6 - 2 - 1, or 3.
309 # Case 3) The day to rollover is behind us in the interval (i.e., today
310 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
311 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
312 # number of days left in the current week (1) plus the number
313 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000314 # The calculations described in 2) and 3) above need to have a day added.
315 # This is because the above time calculation takes us to midnight on this
316 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000317 if self.when.startswith('W'):
Vinay Sajipa7130792013-04-12 17:04:23 +0100318 day = currentDay # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000319 if day != self.dayOfWeek:
320 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000321 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000322 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000323 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000324 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
325 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000326 dstNow = t[-1]
327 dstAtRollover = time.localtime(newRolloverAt)[-1]
328 if dstNow != dstAtRollover:
329 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000330 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000331 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000332 addend = 3600
333 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000334 result = newRolloverAt
335 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000336
337 def shouldRollover(self, record):
338 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000339 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000340
341 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000342 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000343 """
344 t = int(time.time())
345 if t >= self.rolloverAt:
346 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000347 return 0
348
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000349 def getFilesToDelete(self):
350 """
351 Determine the files to delete when rolling over.
352
353 More specific than the earlier method, which just used glob.glob().
354 """
355 dirName, baseName = os.path.split(self.baseFilename)
356 fileNames = os.listdir(dirName)
357 result = []
358 prefix = baseName + "."
359 plen = len(prefix)
360 for fileName in fileNames:
361 if fileName[:plen] == prefix:
362 suffix = fileName[plen:]
363 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000364 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000365 if len(result) < self.backupCount:
366 result = []
367 else:
Lovesh Harchandaniafad1472017-10-27 09:04:33 +0200368 result.sort()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000369 result = result[:len(result) - self.backupCount]
370 return result
371
Vinay Sajip17c52d82004-07-03 11:48:34 +0000372 def doRollover(self):
373 """
374 do a rollover; in this case, a date/time stamp is appended to the filename
375 when the rollover happens. However, you want the file to be named for the
376 start of the interval, not the current time. If there is a backup count,
377 then we have to get a list of matching filenames, sort them and remove
378 the one with the oldest suffix.
379 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000380 if self.stream:
381 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000382 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000383 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000384 currentTime = int(time.time())
385 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000386 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000387 if self.utc:
388 timeTuple = time.gmtime(t)
389 else:
390 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000391 dstThen = timeTuple[-1]
392 if dstNow != dstThen:
393 if dstNow:
394 addend = 3600
395 else:
396 addend = -3600
397 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000398 dfn = self.rotation_filename(self.baseFilename + "." +
399 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000400 if os.path.exists(dfn):
401 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000402 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000403 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000404 for s in self.getFilesToDelete():
405 os.remove(s)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100406 if not self.delay:
407 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000408 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000409 while newRolloverAt <= currentTime:
410 newRolloverAt = newRolloverAt + self.interval
411 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000412 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000413 dstAtRollover = time.localtime(newRolloverAt)[-1]
414 if dstNow != dstAtRollover:
415 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000416 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000417 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000418 addend = 3600
419 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000420 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000421
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000422class WatchedFileHandler(logging.FileHandler):
423 """
424 A handler for logging to a file, which watches the file
425 to see if it has changed while in use. This can happen because of
426 usage of programs such as newsyslog and logrotate which perform
427 log file rotation. This handler, intended for use under Unix,
428 watches the file to see if it has changed since the last emit.
429 (A file has changed if its device or inode have changed.)
430 If it has changed, the old file stream is closed, and the file
431 opened to get a new stream.
432
433 This handler is not appropriate for use under Windows, because
434 under Windows open files cannot be moved or renamed - logging
435 opens the files with exclusive locks - and so there is no need
436 for such a handler. Furthermore, ST_INO is not supported under
437 Windows; stat always returns zero for this value.
438
439 This handler is based on a suggestion and patch by Chad J.
440 Schroeder.
441 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100442 def __init__(self, filename, mode='a', encoding=None, delay=False,
443 errors=None):
444 logging.FileHandler.__init__(self, filename, mode=mode,
445 encoding=encoding, delay=delay,
446 errors=errors)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100447 self.dev, self.ino = -1, -1
448 self._statstream()
449
450 def _statstream(self):
451 if self.stream:
452 sres = os.fstat(self.stream.fileno())
453 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000454
Vinay Sajip29a14452015-10-01 20:54:41 +0100455 def reopenIfNeeded(self):
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000456 """
Vinay Sajip29a14452015-10-01 20:54:41 +0100457 Reopen log file if needed.
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000458
Vinay Sajip29a14452015-10-01 20:54:41 +0100459 Checks if the underlying file has changed, and if it
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000460 has, close the old stream and reopen the file to get the
461 current stream.
462 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100463 # Reduce the chance of race conditions by stat'ing by path only
464 # once and then fstat'ing our new fd if we opened a new log stream.
465 # See issue #14632: Thanks to John Mulligan for the problem report
466 # and patch.
467 try:
468 # stat the file by path, checking for existence
469 sres = os.stat(self.baseFilename)
Giampaolo Rodola'0166a282013-02-12 15:14:17 +0100470 except FileNotFoundError:
471 sres = None
Vinay Sajip66b8b082012-04-24 23:25:30 +0100472 # compare file system stat with that of our stream file handle
473 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
474 if self.stream is not None:
475 # we have an open file handle, clean it up
476 self.stream.flush()
477 self.stream.close()
Vinay Sajip02200482014-06-14 10:22:05 +0100478 self.stream = None # See Issue #21742: _open () might fail.
Vinay Sajip66b8b082012-04-24 23:25:30 +0100479 # open a new file handle and get new stat info from that fd
480 self.stream = self._open()
481 self._statstream()
Vinay Sajip29a14452015-10-01 20:54:41 +0100482
483 def emit(self, record):
484 """
485 Emit a record.
486
487 If underlying file has changed, reopen the file before emitting the
488 record to it.
489 """
490 self.reopenIfNeeded()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000491 logging.FileHandler.emit(self, record)
492
Vinay Sajip66b8b082012-04-24 23:25:30 +0100493
Guido van Rossum57102f82002-11-13 16:15:58 +0000494class SocketHandler(logging.Handler):
495 """
496 A handler class which writes logging records, in pickle format, to
497 a streaming socket. The socket is kept open across logging calls.
498 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000499 The pickle which is sent is that of the LogRecord's attribute dictionary
500 (__dict__), so that the receiver does not need to have the logging module
501 installed in order to process the logging event.
502
503 To unpickle the record at the receiving end into a LogRecord, use the
504 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000505 """
506
507 def __init__(self, host, port):
508 """
509 Initializes the handler with a specific host address and port.
510
Vinay Sajipde19e082011-04-30 21:52:26 +0100511 When the attribute *closeOnError* is set to True - if a socket error
512 occurs, the socket is silently closed and then reopened on the next
513 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000514 """
515 logging.Handler.__init__(self)
516 self.host = host
517 self.port = port
Vinay Sajip5421f352013-09-27 18:18:28 +0100518 if port is None:
519 self.address = host
520 else:
521 self.address = (host, port)
Guido van Rossum57102f82002-11-13 16:15:58 +0000522 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100523 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000524 self.retryTime = None
525 #
526 # Exponential backoff parameters.
527 #
528 self.retryStart = 1.0
529 self.retryMax = 30.0
530 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000531
Guido van Rossumd8faa362007-04-27 19:54:29 +0000532 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000533 """
534 A factory method which allows subclasses to define the precise
535 type of socket they want.
536 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100537 if self.port is not None:
538 result = socket.create_connection(self.address, timeout=timeout)
539 else:
540 result = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
541 result.settimeout(timeout)
Vinay Sajip38c741c2013-10-06 18:36:00 +0100542 try:
543 result.connect(self.address)
544 except OSError:
545 result.close() # Issue 19182
546 raise
Vinay Sajip5421f352013-09-27 18:18:28 +0100547 return result
Guido van Rossum57102f82002-11-13 16:15:58 +0000548
Vinay Sajip48cfe382004-02-20 13:17:27 +0000549 def createSocket(self):
550 """
551 Try to create a socket, using an exponential backoff with
552 a max retry time. Thanks to Robert Olson for the original patch
553 (SF #815911) which has been slightly refactored.
554 """
555 now = time.time()
556 # Either retryTime is None, in which case this
557 # is the first time back after a disconnect, or
558 # we've waited long enough.
559 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100560 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000561 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000562 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000563 if attempt:
564 try:
565 self.sock = self.makeSocket()
566 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200567 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000568 #Creation failed, so set the retry time and return.
569 if self.retryTime is None:
570 self.retryPeriod = self.retryStart
571 else:
572 self.retryPeriod = self.retryPeriod * self.retryFactor
573 if self.retryPeriod > self.retryMax:
574 self.retryPeriod = self.retryMax
575 self.retryTime = now + self.retryPeriod
576
Guido van Rossum57102f82002-11-13 16:15:58 +0000577 def send(self, s):
578 """
579 Send a pickled string to the socket.
580
581 This function allows for partial sends which can happen when the
582 network is busy.
583 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000584 if self.sock is None:
585 self.createSocket()
586 #self.sock can be None either because we haven't reached the retry
587 #time yet, or because we have reached the retry time and retried,
588 #but are still unable to connect.
589 if self.sock:
590 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200591 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200592 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000593 self.sock.close()
594 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000595
596 def makePickle(self, record):
597 """
598 Pickles the record in binary format with a length prefix, and
599 returns it ready for transmission across the socket.
600 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000601 ei = record.exc_info
602 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100603 # just to get traceback text into record.exc_text ...
604 dummy = self.format(record)
605 # See issue #14436: If msg or args are objects, they may not be
606 # available on the receiving end. So we convert the msg % args
607 # to a string, save it as msg and zap the args.
608 d = dict(record.__dict__)
609 d['msg'] = record.getMessage()
610 d['args'] = None
611 d['exc_info'] = None
Vinay Sajip9cdfd182015-12-26 12:48:44 +0000612 # Issue #25685: delete 'message' if present: redundant with 'msg'
613 d.pop('message', None)
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100614 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000615 slen = struct.pack(">L", len(s))
616 return slen + s
617
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000618 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000619 """
620 Handle an error during logging.
621
622 An error has occurred during logging. Most likely cause -
623 connection lost. Close the socket so that we can retry on the
624 next event.
625 """
626 if self.closeOnError and self.sock:
627 self.sock.close()
628 self.sock = None #try to reconnect next time
629 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000630 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000631
632 def emit(self, record):
633 """
634 Emit a record.
635
636 Pickles the record and writes it to the socket in binary format.
637 If there is an error with the socket, silently drop the packet.
638 If there was a problem with the socket, re-establishes the
639 socket.
640 """
641 try:
642 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000643 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100644 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000645 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000646
647 def close(self):
648 """
649 Closes the socket.
650 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000651 self.acquire()
652 try:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300653 sock = self.sock
654 if sock:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000655 self.sock = None
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300656 sock.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000657 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000658 finally:
659 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000660
661class DatagramHandler(SocketHandler):
662 """
663 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000664 a datagram socket. The pickle which is sent is that of the LogRecord's
665 attribute dictionary (__dict__), so that the receiver does not need to
666 have the logging module installed in order to process the logging event.
667
668 To unpickle the record at the receiving end into a LogRecord, use the
669 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000670
671 """
672 def __init__(self, host, port):
673 """
674 Initializes the handler with a specific host address and port.
675 """
676 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100677 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000678
679 def makeSocket(self):
680 """
681 The factory method of SocketHandler is here overridden to create
682 a UDP socket (SOCK_DGRAM).
683 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100684 if self.port is None:
685 family = socket.AF_UNIX
686 else:
687 family = socket.AF_INET
688 s = socket.socket(family, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000689 return s
690
691 def send(self, s):
692 """
693 Send a pickled string to a socket.
694
695 This function no longer allows for partial sends which can happen
696 when the network is busy - UDP does not guarantee delivery and
697 can deliver packets out of sequence.
698 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000699 if self.sock is None:
700 self.createSocket()
Vinay Sajip5421f352013-09-27 18:18:28 +0100701 self.sock.sendto(s, self.address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000702
703class SysLogHandler(logging.Handler):
704 """
705 A handler class which sends formatted logging records to a syslog
706 server. Based on Sam Rushing's syslog module:
707 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
708 Contributed by Nicolas Untz (after which minor refactoring changes
709 have been made).
710 """
711
712 # from <linux/sys/syslog.h>:
713 # ======================================================================
714 # priorities/facilities are encoded into a single 32-bit quantity, where
715 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
716 # facility (0-big number). Both the priorities and the facilities map
717 # roughly one-to-one to strings in the syslogd(8) source code. This
718 # mapping is included in this file.
719 #
720 # priorities (these are ordered)
721
722 LOG_EMERG = 0 # system is unusable
723 LOG_ALERT = 1 # action must be taken immediately
724 LOG_CRIT = 2 # critical conditions
725 LOG_ERR = 3 # error conditions
726 LOG_WARNING = 4 # warning conditions
727 LOG_NOTICE = 5 # normal but significant condition
728 LOG_INFO = 6 # informational
729 LOG_DEBUG = 7 # debug-level messages
730
731 # facility codes
732 LOG_KERN = 0 # kernel messages
733 LOG_USER = 1 # random user-level messages
734 LOG_MAIL = 2 # mail system
735 LOG_DAEMON = 3 # system daemons
736 LOG_AUTH = 4 # security/authorization messages
737 LOG_SYSLOG = 5 # messages generated internally by syslogd
738 LOG_LPR = 6 # line printer subsystem
739 LOG_NEWS = 7 # network news subsystem
740 LOG_UUCP = 8 # UUCP subsystem
741 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000742 LOG_AUTHPRIV = 10 # security/authorization messages (private)
743 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000744
745 # other codes through 15 reserved for system use
746 LOG_LOCAL0 = 16 # reserved for local use
747 LOG_LOCAL1 = 17 # reserved for local use
748 LOG_LOCAL2 = 18 # reserved for local use
749 LOG_LOCAL3 = 19 # reserved for local use
750 LOG_LOCAL4 = 20 # reserved for local use
751 LOG_LOCAL5 = 21 # reserved for local use
752 LOG_LOCAL6 = 22 # reserved for local use
753 LOG_LOCAL7 = 23 # reserved for local use
754
755 priority_names = {
756 "alert": LOG_ALERT,
757 "crit": LOG_CRIT,
758 "critical": LOG_CRIT,
759 "debug": LOG_DEBUG,
760 "emerg": LOG_EMERG,
761 "err": LOG_ERR,
762 "error": LOG_ERR, # DEPRECATED
763 "info": LOG_INFO,
764 "notice": LOG_NOTICE,
765 "panic": LOG_EMERG, # DEPRECATED
766 "warn": LOG_WARNING, # DEPRECATED
767 "warning": LOG_WARNING,
768 }
769
770 facility_names = {
771 "auth": LOG_AUTH,
772 "authpriv": LOG_AUTHPRIV,
773 "cron": LOG_CRON,
774 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000775 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000776 "kern": LOG_KERN,
777 "lpr": LOG_LPR,
778 "mail": LOG_MAIL,
779 "news": LOG_NEWS,
780 "security": LOG_AUTH, # DEPRECATED
781 "syslog": LOG_SYSLOG,
782 "user": LOG_USER,
783 "uucp": LOG_UUCP,
784 "local0": LOG_LOCAL0,
785 "local1": LOG_LOCAL1,
786 "local2": LOG_LOCAL2,
787 "local3": LOG_LOCAL3,
788 "local4": LOG_LOCAL4,
789 "local5": LOG_LOCAL5,
790 "local6": LOG_LOCAL6,
791 "local7": LOG_LOCAL7,
792 }
793
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000794 #The map below appears to be trivially lowercasing the key. However,
795 #there's more to it than meets the eye - in some locales, lowercasing
796 #gives unexpected results. See SF #1524081: in the Turkish locale,
797 #"INFO".lower() != "info"
798 priority_map = {
799 "DEBUG" : "debug",
800 "INFO" : "info",
801 "WARNING" : "warning",
802 "ERROR" : "error",
803 "CRITICAL" : "critical"
804 }
805
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000806 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100807 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000808 """
809 Initialize a handler.
810
Guido van Rossume7ba4952007-06-06 23:52:48 +0000811 If address is specified as a string, a UNIX socket is used. To log to a
812 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100813 If facility is not specified, LOG_USER is used. If socktype is
814 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
815 socket type will be used. For Unix sockets, you can also specify a
816 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
817 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000818 """
819 logging.Handler.__init__(self)
820
821 self.address = address
822 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000823 self.socktype = socktype
824
Guido van Rossum13257902007-06-07 23:15:56 +0000825 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100826 self.unixsocket = True
Коренберг Марк1b038e02017-03-17 20:25:05 +0500827 # Syslog server may be unavailable during handler initialisation.
828 # C's openlog() function also ignores connection errors.
829 # Moreover, we ignore these errors while logging, so it not worse
830 # to ignore it also here.
831 try:
832 self._connect_unixsocket(address)
833 except OSError:
834 pass
Guido van Rossum57102f82002-11-13 16:15:58 +0000835 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100836 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100837 if socktype is None:
838 socktype = socket.SOCK_DGRAM
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800839 host, port = address
840 ress = socket.getaddrinfo(host, port, 0, socktype)
841 if not ress:
842 raise OSError("getaddrinfo returns an empty list")
843 for res in ress:
844 af, socktype, proto, _, sa = res
845 err = sock = None
846 try:
847 sock = socket.socket(af, socktype, proto)
848 if socktype == socket.SOCK_STREAM:
849 sock.connect(sa)
850 break
851 except OSError as exc:
852 err = exc
853 if sock is not None:
854 sock.close()
855 if err is not None:
856 raise err
857 self.socket = sock
Vinay Sajipe9170522013-04-22 10:07:40 +0100858 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000859
Vinay Sajipa1974c12005-01-13 08:23:56 +0000860 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100861 use_socktype = self.socktype
862 if use_socktype is None:
863 use_socktype = socket.SOCK_DGRAM
864 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000865 try:
866 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100867 # it worked, so set self.socktype to the used type
868 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200869 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000870 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100871 if self.socktype is not None:
872 # user didn't specify falling back, so fail
873 raise
874 use_socktype = socket.SOCK_STREAM
875 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
876 try:
877 self.socket.connect(address)
878 # it worked, so set self.socktype to the used type
879 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100880 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100881 self.socket.close()
882 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000883
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000884 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000885 """
886 Encode the facility and priority. You can pass in strings or
887 integers - if strings are passed, the facility_names and
888 priority_names mapping dictionaries are used to convert them to
889 integers.
890 """
Guido van Rossum13257902007-06-07 23:15:56 +0000891 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000892 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000893 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000894 priority = self.priority_names[priority]
895 return (facility << 3) | priority
896
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800897 def close(self):
Guido van Rossum57102f82002-11-13 16:15:58 +0000898 """
899 Closes the socket.
900 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000901 self.acquire()
902 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000903 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000904 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000905 finally:
906 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000907
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000908 def mapPriority(self, levelName):
909 """
910 Map a logging level name to a key in the priority_names map.
911 This is useful in two scenarios: when custom levels are being
912 used, and in the case where you can't do a straightforward
913 mapping by lowercasing the logging level name because of locale-
914 specific issues (see SF #1524081).
915 """
916 return self.priority_map.get(levelName, "warning")
917
Vinay Sajip2353e352011-06-27 15:40:06 +0100918 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100919 append_nul = True # some old syslog daemons expect a NUL terminator
920
Guido van Rossum57102f82002-11-13 16:15:58 +0000921 def emit(self, record):
922 """
923 Emit a record.
924
925 The record is formatted, and then sent to the syslog server. If
926 exception information is present, it is NOT sent to the server.
927 """
Guido van Rossum57102f82002-11-13 16:15:58 +0000928 try:
Vinay Sajipc33a0cc2014-11-01 19:58:47 +0000929 msg = self.format(record)
930 if self.ident:
931 msg = self.ident + msg
932 if self.append_nul:
933 msg += '\000'
934
935 # We need to convert record level to lowercase, maybe this will
936 # change in the future.
937 prio = '<%d>' % self.encodePriority(self.facility,
938 self.mapPriority(record.levelname))
939 prio = prio.encode('utf-8')
940 # Message is a string. Convert to bytes as required by RFC 5424
941 msg = msg.encode('utf-8')
942 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000943 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000944 try:
945 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200946 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100947 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000948 self._connect_unixsocket(self.address)
949 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000950 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000951 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000952 else:
953 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100954 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000955 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000956
957class SMTPHandler(logging.Handler):
958 """
959 A handler class which sends an SMTP email for each logging event.
960 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000961 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100962 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000963 """
964 Initialize the handler.
965
966 Initialize the instance with the from and to addresses and subject
967 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000968 (host, port) tuple format for the mailhost argument. To specify
969 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000970 for the credentials argument. To specify the use of a secure
971 protocol (TLS), pass in a tuple for the secure argument. This will
972 only be used when authentication credentials are supplied. The tuple
973 will be either an empty tuple, or a single-value tuple with the name
974 of a keyfile, or a 2-value tuple with the names of the keyfile and
975 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000976 A timeout in seconds can be specified for the SMTP connection (the
977 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000978 """
979 logging.Handler.__init__(self)
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100980 if isinstance(mailhost, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000981 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000982 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000983 self.mailhost, self.mailport = mailhost, None
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100984 if isinstance(credentials, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000985 self.username, self.password = credentials
986 else:
987 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000988 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000989 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000990 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000991 self.toaddrs = toaddrs
992 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000993 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000994 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000995
996 def getSubject(self, record):
997 """
998 Determine the subject for the email.
999
1000 If you want to specify a subject line which is record-dependent,
1001 override this method.
1002 """
1003 return self.subject
1004
1005 def emit(self, record):
1006 """
1007 Emit a record.
1008
1009 Format the record and send it to the specified addressees.
1010 """
1011 try:
1012 import smtplib
Vinay Sajip277640a2015-10-17 16:13:10 +01001013 from email.message import EmailMessage
1014 import email.utils
1015
Guido van Rossum57102f82002-11-13 16:15:58 +00001016 port = self.mailport
1017 if not port:
1018 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +00001019 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Vinay Sajip277640a2015-10-17 16:13:10 +01001020 msg = EmailMessage()
1021 msg['From'] = self.fromaddr
1022 msg['To'] = ','.join(self.toaddrs)
1023 msg['Subject'] = self.getSubject(record)
1024 msg['Date'] = email.utils.localtime()
1025 msg.set_content(self.format(record))
Guido van Rossum360e4b82007-05-14 22:51:27 +00001026 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +00001027 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +00001028 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +00001029 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +00001030 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +00001031 smtp.login(self.username, self.password)
Vinay Sajip277640a2015-10-17 16:13:10 +01001032 smtp.send_message(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +00001033 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001034 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001035 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001036
1037class NTEventLogHandler(logging.Handler):
1038 """
1039 A handler class which sends events to the NT Event Log. Adds a
1040 registry entry for the specified application name. If no dllname is
1041 provided, win32service.pyd (which contains some basic message
1042 placeholders) is used. Note that use of these placeholders will make
1043 your event logs big, as the entire message source is held in the log.
1044 If you want slimmer logs, you have to pass in the name of your own DLL
1045 which contains the message definitions you want to use in the event log.
1046 """
1047 def __init__(self, appname, dllname=None, logtype="Application"):
1048 logging.Handler.__init__(self)
1049 try:
1050 import win32evtlogutil, win32evtlog
1051 self.appname = appname
1052 self._welu = win32evtlogutil
1053 if not dllname:
1054 dllname = os.path.split(self._welu.__file__)
1055 dllname = os.path.split(dllname[0])
1056 dllname = os.path.join(dllname[0], r'win32service.pyd')
1057 self.dllname = dllname
1058 self.logtype = logtype
1059 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1060 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1061 self.typemap = {
1062 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1063 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001064 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001065 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1066 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1067 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001068 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001069 print("The Python Win32 extensions for NT (service, event "\
1070 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001071 self._welu = None
1072
1073 def getMessageID(self, record):
1074 """
1075 Return the message ID for the event record. If you are using your
1076 own messages, you could do this by having the msg passed to the
1077 logger being an ID rather than a formatting string. Then, in here,
1078 you could use a dictionary lookup to get the message ID. This
1079 version returns 1, which is the base message ID in win32service.pyd.
1080 """
1081 return 1
1082
1083 def getEventCategory(self, record):
1084 """
1085 Return the event category for the record.
1086
1087 Override this if you want to specify your own categories. This version
1088 returns 0.
1089 """
1090 return 0
1091
1092 def getEventType(self, record):
1093 """
1094 Return the event type for the record.
1095
1096 Override this if you want to specify your own types. This version does
1097 a mapping using the handler's typemap attribute, which is set up in
1098 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001099 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001100 either need to override this method or place a suitable dictionary in
1101 the handler's typemap attribute.
1102 """
1103 return self.typemap.get(record.levelno, self.deftype)
1104
1105 def emit(self, record):
1106 """
1107 Emit a record.
1108
1109 Determine the message ID, event category and event type. Then
1110 log the message in the NT event log.
1111 """
1112 if self._welu:
1113 try:
1114 id = self.getMessageID(record)
1115 cat = self.getEventCategory(record)
1116 type = self.getEventType(record)
1117 msg = self.format(record)
1118 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001119 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001120 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001121
1122 def close(self):
1123 """
1124 Clean up this handler.
1125
1126 You can remove the application name from the registry as a
1127 source of event log entries. However, if you do this, you will
1128 not be able to see the events as you intended in the Event Log
1129 Viewer - it needs to be able to access the registry to get the
1130 DLL name.
1131 """
1132 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001133 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001134
1135class HTTPHandler(logging.Handler):
1136 """
1137 A class which sends records to a Web server, using either GET or
1138 POST semantics.
1139 """
Benjamin Peterson43052a12014-11-23 20:36:44 -06001140 def __init__(self, host, url, method="GET", secure=False, credentials=None,
1141 context=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001142 """
1143 Initialize the instance with the host, the request URL, and the method
1144 ("GET" or "POST")
1145 """
1146 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001147 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001148 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001149 raise ValueError("method must be GET or POST")
Benjamin Peterson43052a12014-11-23 20:36:44 -06001150 if not secure and context is not None:
1151 raise ValueError("context parameter only makes sense "
1152 "with secure=True")
Guido van Rossum57102f82002-11-13 16:15:58 +00001153 self.host = host
1154 self.url = url
1155 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001156 self.secure = secure
1157 self.credentials = credentials
Benjamin Peterson43052a12014-11-23 20:36:44 -06001158 self.context = context
Guido van Rossum57102f82002-11-13 16:15:58 +00001159
Neal Norwitzf297bd12003-04-23 03:49:43 +00001160 def mapLogRecord(self, record):
1161 """
1162 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001163 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001164 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001165 """
1166 return record.__dict__
1167
Guido van Rossum57102f82002-11-13 16:15:58 +00001168 def emit(self, record):
1169 """
1170 Emit a record.
1171
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001172 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001173 """
1174 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001175 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001176 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001177 if self.secure:
Benjamin Peterson43052a12014-11-23 20:36:44 -06001178 h = http.client.HTTPSConnection(host, context=self.context)
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001179 else:
1180 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001181 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001182 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001183 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001184 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001185 sep = '&'
1186 else:
1187 sep = '?'
1188 url = url + "%c%s" % (sep, data)
1189 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001190 # support multiple hosts on one IP address...
1191 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001192 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001193 if i >= 0:
1194 host = host[:i]
Vinay Sajipe96ba182017-11-19 18:36:17 +00001195 # See issue #30904: putrequest call above already adds this header
1196 # on Python 3.x.
1197 # h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001198 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001199 h.putheader("Content-type",
1200 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001201 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001202 if self.credentials:
1203 import base64
Vinay Sajip1bf197e2016-06-07 21:19:55 +01001204 s = ('%s:%s' % self.credentials).encode('utf-8')
1205 s = 'Basic ' + base64.b64encode(s).strip().decode('ascii')
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001206 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001207 h.endheaders()
1208 if self.method == "POST":
1209 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001210 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001211 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001212 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001213
1214class BufferingHandler(logging.Handler):
1215 """
1216 A handler class which buffers logging records in memory. Whenever each
1217 record is added to the buffer, a check is made to see if the buffer should
1218 be flushed. If it should, then flush() is expected to do what's needed.
1219 """
1220 def __init__(self, capacity):
1221 """
1222 Initialize the handler with the buffer size.
1223 """
1224 logging.Handler.__init__(self)
1225 self.capacity = capacity
1226 self.buffer = []
1227
1228 def shouldFlush(self, record):
1229 """
1230 Should the handler flush its buffer?
1231
1232 Returns true if the buffer is up to capacity. This method can be
1233 overridden to implement custom flushing strategies.
1234 """
1235 return (len(self.buffer) >= self.capacity)
1236
1237 def emit(self, record):
1238 """
1239 Emit a record.
1240
1241 Append the record. If shouldFlush() tells us to, call flush() to process
1242 the buffer.
1243 """
1244 self.buffer.append(record)
1245 if self.shouldFlush(record):
1246 self.flush()
1247
1248 def flush(self):
1249 """
1250 Override to implement custom flushing behaviour.
1251
1252 This version just zaps the buffer to empty.
1253 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001254 self.acquire()
1255 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001256 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001257 finally:
1258 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001259
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001260 def close(self):
1261 """
1262 Close the handler.
1263
1264 This version just flushes and chains to the parent class' close().
1265 """
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001266 try:
1267 self.flush()
1268 finally:
1269 logging.Handler.close(self)
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001270
Guido van Rossum57102f82002-11-13 16:15:58 +00001271class MemoryHandler(BufferingHandler):
1272 """
1273 A handler class which buffers logging records in memory, periodically
1274 flushing them to a target handler. Flushing occurs whenever the buffer
1275 is full, or when an event of a certain severity or greater is seen.
1276 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001277 def __init__(self, capacity, flushLevel=logging.ERROR, target=None,
1278 flushOnClose=True):
Guido van Rossum57102f82002-11-13 16:15:58 +00001279 """
1280 Initialize the handler with the buffer size, the level at which
1281 flushing should occur and an optional target.
1282
1283 Note that without a target being set either here or via setTarget(),
1284 a MemoryHandler is no use to anyone!
Vinay Sajipcccf6062016-07-22 16:27:31 +01001285
1286 The ``flushOnClose`` argument is ``True`` for backward compatibility
1287 reasons - the old behaviour is that when the handler is closed, the
1288 buffer is flushed, even if the flush level hasn't been exceeded nor the
1289 capacity exceeded. To prevent this, set ``flushOnClose`` to ``False``.
Guido van Rossum57102f82002-11-13 16:15:58 +00001290 """
1291 BufferingHandler.__init__(self, capacity)
1292 self.flushLevel = flushLevel
1293 self.target = target
Vinay Sajipcccf6062016-07-22 16:27:31 +01001294 # See Issue #26559 for why this has been added
1295 self.flushOnClose = flushOnClose
Guido van Rossum57102f82002-11-13 16:15:58 +00001296
1297 def shouldFlush(self, record):
1298 """
1299 Check for buffer full or a record at the flushLevel or higher.
1300 """
1301 return (len(self.buffer) >= self.capacity) or \
1302 (record.levelno >= self.flushLevel)
1303
1304 def setTarget(self, target):
1305 """
1306 Set the target handler for this handler.
1307 """
1308 self.target = target
1309
1310 def flush(self):
1311 """
1312 For a MemoryHandler, flushing means just sending the buffered
1313 records to the target, if there is one. Override if you want
1314 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001315
1316 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001317 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001318 self.acquire()
1319 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001320 if self.target:
1321 for record in self.buffer:
1322 self.target.handle(record)
1323 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001324 finally:
1325 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001326
1327 def close(self):
1328 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001329 Flush, if appropriately configured, set the target to None and lose the
1330 buffer.
Guido van Rossum57102f82002-11-13 16:15:58 +00001331 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001332 try:
Vinay Sajipcccf6062016-07-22 16:27:31 +01001333 if self.flushOnClose:
1334 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001335 finally:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001336 self.acquire()
1337 try:
1338 self.target = None
1339 BufferingHandler.close(self)
1340 finally:
1341 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001342
1343
1344class QueueHandler(logging.Handler):
1345 """
1346 This handler sends events to a queue. Typically, it would be used together
1347 with a multiprocessing Queue to centralise logging to file in one process
1348 (in a multi-process application), so as to avoid file write contention
1349 between processes.
1350
1351 This code is new in Python 3.2, but this class can be copy pasted into
1352 user code for use with earlier Python versions.
1353 """
1354
1355 def __init__(self, queue):
1356 """
1357 Initialise an instance, using the passed queue.
1358 """
1359 logging.Handler.__init__(self)
1360 self.queue = queue
1361
1362 def enqueue(self, record):
1363 """
1364 Enqueue a record.
1365
1366 The base implementation uses put_nowait. You may want to override
1367 this method if you want to use blocking, timeouts or custom queue
1368 implementations.
1369 """
1370 self.queue.put_nowait(record)
1371
Vinay Sajip0258ce82010-09-22 20:34:53 +00001372 def prepare(self, record):
1373 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001374 Prepares a record for queuing. The object returned by this method is
1375 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001376
1377 The base implementation formats the record to merge the message
1378 and arguments, and removes unpickleable items from the record
1379 in-place.
1380
1381 You might want to override this method if you want to convert
1382 the record to a dict or JSON string, or send a modified copy
1383 of the record while leaving the original intact.
1384 """
1385 # The format operation gets traceback text into record.exc_text
favlladfe3442017-08-01 20:12:26 +02001386 # (if there's exception data), and also returns the formatted
1387 # message. We can then use this to replace the original
Vinay Sajip0258ce82010-09-22 20:34:53 +00001388 # msg + args, as these might be unpickleable. We also zap the
Cheryl Sabellad345bb42018-09-25 19:00:08 -04001389 # exc_info and exc_text attributes, as they are no longer
1390 # needed and, if not None, will typically not be pickleable.
favlladfe3442017-08-01 20:12:26 +02001391 msg = self.format(record)
Manjusakada6424e2019-01-23 15:08:38 +08001392 # bpo-35726: make copy of record to avoid affecting other handlers in the chain.
1393 record = copy.copy(record)
favlladfe3442017-08-01 20:12:26 +02001394 record.message = msg
1395 record.msg = msg
Vinay Sajip0258ce82010-09-22 20:34:53 +00001396 record.args = None
1397 record.exc_info = None
Cheryl Sabellad345bb42018-09-25 19:00:08 -04001398 record.exc_text = None
Vinay Sajip0258ce82010-09-22 20:34:53 +00001399 return record
1400
Vinay Sajip121a1c42010-09-08 10:46:15 +00001401 def emit(self, record):
1402 """
1403 Emit a record.
1404
Vinay Sajip0637d492010-09-23 08:15:54 +00001405 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001406 """
1407 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001408 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001409 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001410 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001411
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001412
1413class QueueListener(object):
1414 """
1415 This class implements an internal threaded listener which watches for
1416 LogRecords being added to a queue, removes them and passes them to a
1417 list of handlers for processing.
1418 """
1419 _sentinel = None
1420
1421 def __init__(self, queue, *handlers, respect_handler_level=False):
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001422 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001423 Initialise an instance with the specified queue and
1424 handlers.
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001425 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001426 self.queue = queue
1427 self.handlers = handlers
1428 self._thread = None
1429 self.respect_handler_level = respect_handler_level
Vinay Sajip0637d492010-09-23 08:15:54 +00001430
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001431 def dequeue(self, block):
1432 """
1433 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001434
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001435 The base implementation uses get. You may want to override this method
1436 if you want to use timeouts or work with custom queue implementations.
1437 """
1438 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001439
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001440 def start(self):
1441 """
1442 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001443
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001444 This starts up a background thread to monitor the queue for
1445 LogRecords to process.
1446 """
1447 self._thread = t = threading.Thread(target=self._monitor)
1448 t.daemon = True
1449 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001450
Boris Feld800d7862019-06-04 17:20:18 +02001451 def prepare(self, record):
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001452 """
1453 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001454
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001455 This method just returns the passed-in record. You may want to
1456 override this method if you need to do any custom marshalling or
1457 manipulation of the record before passing it to the handlers.
1458 """
1459 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001460
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001461 def handle(self, record):
1462 """
1463 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001464
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001465 This just loops through the handlers offering them the record
1466 to handle.
1467 """
1468 record = self.prepare(record)
1469 for handler in self.handlers:
1470 if not self.respect_handler_level:
1471 process = True
1472 else:
1473 process = record.levelno >= handler.level
1474 if process:
1475 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001476
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001477 def _monitor(self):
1478 """
1479 Monitor the queue for records, and ask the handler
1480 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001481
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001482 This method runs on a separate, internal thread.
1483 The thread will terminate if it sees a sentinel object in the queue.
1484 """
1485 q = self.queue
1486 has_task_done = hasattr(q, 'task_done')
1487 while True:
1488 try:
1489 record = self.dequeue(True)
1490 if record is self._sentinel:
Bar Harel6b282e12019-06-01 12:19:09 +03001491 if has_task_done:
1492 q.task_done()
Vinay Sajip0637d492010-09-23 08:15:54 +00001493 break
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001494 self.handle(record)
1495 if has_task_done:
1496 q.task_done()
1497 except queue.Empty:
1498 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001499
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001500 def enqueue_sentinel(self):
1501 """
1502 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001503
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001504 The base implementation uses put_nowait. You may want to override this
1505 method if you want to use timeouts or work with custom queue
1506 implementations.
1507 """
1508 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001509
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001510 def stop(self):
1511 """
1512 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001513
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001514 This asks the thread to terminate, and then waits for it to do so.
1515 Note that if you don't call this before your application exits, there
1516 may be some records still left on the queue, which won't be processed.
1517 """
1518 self.enqueue_sentinel()
1519 self._thread.join()
1520 self._thread = None