blob: c1aec9880d72abf9e15d586fecceb27531155ee8 [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 """
l0rb519cb872019-11-06 22:21:40 +010051 namer = None
52 rotator = None
53
Vinay Sajipca7b5042019-06-17 17:40:52 +010054 def __init__(self, filename, mode, encoding=None, delay=False, errors=None):
Vinay Sajip17c52d82004-07-03 11:48:34 +000055 """
56 Use the specified filename for streamed logging
57 """
Vinay Sajipca7b5042019-06-17 17:40:52 +010058 logging.FileHandler.__init__(self, filename, mode=mode,
59 encoding=encoding, delay=delay,
60 errors=errors)
Vinay Sajip4600f112005-03-13 09:56:36 +000061 self.mode = mode
62 self.encoding = encoding
Vinay Sajipca7b5042019-06-17 17:40:52 +010063 self.errors = errors
Guido van Rossum57102f82002-11-13 16:15:58 +000064
Vinay Sajip17c52d82004-07-03 11:48:34 +000065 def emit(self, record):
66 """
67 Emit a record.
68
69 Output the record to the file, catering for rollover as described
70 in doRollover().
71 """
Vinay Sajip3970c112004-07-08 10:24:04 +000072 try:
73 if self.shouldRollover(record):
74 self.doRollover()
75 logging.FileHandler.emit(self, record)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +010076 except Exception:
Vinay Sajip3970c112004-07-08 10:24:04 +000077 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000078
Vinay Sajip23b94d02012-01-04 12:02:26 +000079 def rotation_filename(self, default_name):
80 """
81 Modify the filename of a log file when rotating.
82
83 This is provided so that a custom filename can be provided.
84
85 The default implementation calls the 'namer' attribute of the
86 handler, if it's callable, passing the default name to
87 it. If the attribute isn't callable (the default is None), the name
88 is returned unchanged.
89
90 :param default_name: The default name for the log file.
91 """
92 if not callable(self.namer):
93 result = default_name
94 else:
95 result = self.namer(default_name)
96 return result
97
98 def rotate(self, source, dest):
99 """
100 When rotating, rotate the current log.
101
102 The default implementation calls the 'rotator' attribute of the
103 handler, if it's callable, passing the source and dest arguments to
104 it. If the attribute isn't callable (the default is None), the source
105 is simply renamed to the destination.
106
107 :param source: The source filename. This is normally the base
108 filename, e.g. 'test.log'
109 :param dest: The destination filename. This is normally
110 what the source is rotated to, e.g. 'test.log.1'.
111 """
112 if not callable(self.rotator):
Vinay Sajipd8599262013-09-06 10:10:22 +0100113 # Issue 18940: A file may not have been created if delay is True.
114 if os.path.exists(source):
115 os.rename(source, dest)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000116 else:
117 self.rotator(source, dest)
118
Vinay Sajip17c52d82004-07-03 11:48:34 +0000119class RotatingFileHandler(BaseRotatingHandler):
120 """
121 Handler for logging to a set of files, which switches from one file
122 to the next when the current file reaches a certain size.
123 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100124 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0,
125 encoding=None, delay=False, errors=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000126 """
127 Open the specified file and use it as the stream for logging.
128
129 By default, the file grows indefinitely. You can specify particular
130 values of maxBytes and backupCount to allow the file to rollover at
131 a predetermined size.
132
133 Rollover occurs whenever the current log file is nearly maxBytes in
134 length. If backupCount is >= 1, the system will successively create
135 new files with the same pathname as the base file, but with extensions
136 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
137 and a base file name of "app.log", you would get "app.log",
138 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
139 written to is always "app.log" - when it gets filled up, it is closed
140 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
141 exist, then they are renamed to "app.log.2", "app.log.3" etc.
142 respectively.
143
144 If maxBytes is zero, rollover never occurs.
145 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000146 # If rotation/rollover is wanted, it doesn't make sense to use another
147 # mode. If for example 'w' were specified, then if there were multiple
148 # runs of the calling application, the logs from previous runs would be
149 # lost if the 'w' is respected, because the log file would be truncated
150 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000151 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000152 mode = 'a'
Vinay Sajipca7b5042019-06-17 17:40:52 +0100153 BaseRotatingHandler.__init__(self, filename, mode, encoding=encoding,
154 delay=delay, errors=errors)
Guido van Rossum57102f82002-11-13 16:15:58 +0000155 self.maxBytes = maxBytes
156 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000157
158 def doRollover(self):
159 """
160 Do a rollover, as described in __init__().
161 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000162 if self.stream:
163 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000164 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000165 if self.backupCount > 0:
166 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000167 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
168 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
169 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000170 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000171 if os.path.exists(dfn):
172 os.remove(dfn)
173 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000174 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000175 if os.path.exists(dfn):
176 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000177 self.rotate(self.baseFilename, dfn)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100178 if not self.delay:
179 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000180
Vinay Sajip17c52d82004-07-03 11:48:34 +0000181 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000182 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000183 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000184
Vinay Sajip17c52d82004-07-03 11:48:34 +0000185 Basically, see if the supplied record would cause the file to exceed
186 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000187 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000188 if self.stream is None: # delay was set...
189 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000190 if self.maxBytes > 0: # are we rolling over?
191 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000192 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000193 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000194 return 1
195 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000196
Vinay Sajip17c52d82004-07-03 11:48:34 +0000197class TimedRotatingFileHandler(BaseRotatingHandler):
198 """
199 Handler for logging to a file, rotating the log file at certain timed
200 intervals.
201
202 If backupCount is > 0, when rollover is done, no more than backupCount
203 files are kept - the oldest ones are deleted.
204 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100205 def __init__(self, filename, when='h', interval=1, backupCount=0,
206 encoding=None, delay=False, utc=False, atTime=None,
207 errors=None):
208 BaseRotatingHandler.__init__(self, filename, 'a', encoding=encoding,
209 delay=delay, errors=errors)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000210 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000211 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000212 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100213 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000214 # Calculate the real rollover interval, which is just the number of
215 # seconds between rollovers. Also set the filename suffix used when
216 # a rollover occurs. Current 'when' events supported:
217 # S - Seconds
218 # M - Minutes
219 # H - Hours
220 # D - Days
221 # midnight - roll over at midnight
222 # W{0-6} - roll over on a certain day; 0 - Monday
223 #
224 # Case of the 'when' specifier is not important; lower or upper case
225 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000226 if self.when == 'S':
227 self.interval = 1 # one second
228 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000229 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000230 elif self.when == 'M':
231 self.interval = 60 # one minute
232 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000233 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000234 elif self.when == 'H':
235 self.interval = 60 * 60 # one hour
236 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000237 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000238 elif self.when == 'D' or self.when == 'MIDNIGHT':
239 self.interval = 60 * 60 * 24 # one day
240 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000241 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000242 elif self.when.startswith('W'):
243 self.interval = 60 * 60 * 24 * 7 # one week
244 if len(self.when) != 2:
245 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
246 if self.when[1] < '0' or self.when[1] > '6':
247 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
248 self.dayOfWeek = int(self.when[1])
249 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000250 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000251 else:
252 raise ValueError("Invalid rollover interval specified: %s" % self.when)
253
Antoine Pitroufd036452008-08-19 17:56:33 +0000254 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000255 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip638e6222016-07-22 18:23:04 +0100256 # The following line added because the filename passed in could be a
257 # path object (see Issue #27493), but self.baseFilename will be a string
258 filename = self.baseFilename
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000259 if os.path.exists(filename):
260 t = os.stat(filename)[ST_MTIME]
261 else:
262 t = int(time.time())
263 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000264
Vinay Sajipaffbd872009-06-11 10:11:47 +0000265 def computeRollover(self, currentTime):
266 """
267 Work out the rollover time based on the specified time.
268 """
269 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000270 # If we are rolling over at midnight or weekly, then the interval is already known.
271 # What we need to figure out is WHEN the next interval is. In other words,
272 # if you are rolling over at midnight, then your base interval is 1 day,
273 # but you want to start that one day clock at midnight, not now. So, we
274 # have to fudge the rolloverAt value in order to trigger the first rollover
275 # at the right time. After that, the regular interval will take care of
276 # the rest. Note that this code doesn't care about leap seconds. :)
277 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
278 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000279 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000280 t = time.gmtime(currentTime)
281 else:
282 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000283 currentHour = t[3]
284 currentMinute = t[4]
285 currentSecond = t[5]
Vinay Sajipa7130792013-04-12 17:04:23 +0100286 currentDay = t[6]
287 # r is the number of seconds left between now and the next rotation
288 if self.atTime is None:
289 rotate_ts = _MIDNIGHT
290 else:
291 rotate_ts = ((self.atTime.hour * 60 + self.atTime.minute)*60 +
292 self.atTime.second)
293
294 r = rotate_ts - ((currentHour * 60 + currentMinute) * 60 +
295 currentSecond)
296 if r < 0:
297 # Rotate time is before the current time (for example when
298 # self.rotateAt is 13:45 and it now 14:15), rotation is
299 # tomorrow.
300 r += _MIDNIGHT
301 currentDay = (currentDay + 1) % 7
Vinay Sajipaffbd872009-06-11 10:11:47 +0000302 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000303 # If we are rolling over on a certain day, add in the number of days until
304 # the next rollover, but offset by 1 since we just calculated the time
305 # until the next day starts. There are three cases:
306 # Case 1) The day to rollover is today; in this case, do nothing
307 # Case 2) The day to rollover is further in the interval (i.e., today is
308 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
309 # next rollover is simply 6 - 2 - 1, or 3.
310 # Case 3) The day to rollover is behind us in the interval (i.e., today
311 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
312 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
313 # number of days left in the current week (1) plus the number
314 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000315 # The calculations described in 2) and 3) above need to have a day added.
316 # This is because the above time calculation takes us to midnight on this
317 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000318 if self.when.startswith('W'):
Vinay Sajipa7130792013-04-12 17:04:23 +0100319 day = currentDay # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000320 if day != self.dayOfWeek:
321 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000322 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000323 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000324 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000325 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
326 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000327 dstNow = t[-1]
328 dstAtRollover = time.localtime(newRolloverAt)[-1]
329 if dstNow != dstAtRollover:
330 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000331 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000332 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000333 addend = 3600
334 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000335 result = newRolloverAt
336 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000337
338 def shouldRollover(self, record):
339 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000340 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000341
342 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000343 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000344 """
345 t = int(time.time())
346 if t >= self.rolloverAt:
347 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000348 return 0
349
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000350 def getFilesToDelete(self):
351 """
352 Determine the files to delete when rolling over.
353
354 More specific than the earlier method, which just used glob.glob().
355 """
356 dirName, baseName = os.path.split(self.baseFilename)
357 fileNames = os.listdir(dirName)
358 result = []
359 prefix = baseName + "."
360 plen = len(prefix)
361 for fileName in fileNames:
362 if fileName[:plen] == prefix:
363 suffix = fileName[plen:]
364 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000365 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000366 if len(result) < self.backupCount:
367 result = []
368 else:
Lovesh Harchandaniafad1472017-10-27 09:04:33 +0200369 result.sort()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000370 result = result[:len(result) - self.backupCount]
371 return result
372
Vinay Sajip17c52d82004-07-03 11:48:34 +0000373 def doRollover(self):
374 """
375 do a rollover; in this case, a date/time stamp is appended to the filename
376 when the rollover happens. However, you want the file to be named for the
377 start of the interval, not the current time. If there is a backup count,
378 then we have to get a list of matching filenames, sort them and remove
379 the one with the oldest suffix.
380 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000381 if self.stream:
382 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000383 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000384 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000385 currentTime = int(time.time())
386 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000387 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000388 if self.utc:
389 timeTuple = time.gmtime(t)
390 else:
391 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000392 dstThen = timeTuple[-1]
393 if dstNow != dstThen:
394 if dstNow:
395 addend = 3600
396 else:
397 addend = -3600
398 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000399 dfn = self.rotation_filename(self.baseFilename + "." +
400 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000401 if os.path.exists(dfn):
402 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000403 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000404 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000405 for s in self.getFilesToDelete():
406 os.remove(s)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100407 if not self.delay:
408 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000409 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000410 while newRolloverAt <= currentTime:
411 newRolloverAt = newRolloverAt + self.interval
412 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000413 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000414 dstAtRollover = time.localtime(newRolloverAt)[-1]
415 if dstNow != dstAtRollover:
416 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000417 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000418 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000419 addend = 3600
420 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000421 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000422
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000423class WatchedFileHandler(logging.FileHandler):
424 """
425 A handler for logging to a file, which watches the file
426 to see if it has changed while in use. This can happen because of
427 usage of programs such as newsyslog and logrotate which perform
428 log file rotation. This handler, intended for use under Unix,
429 watches the file to see if it has changed since the last emit.
430 (A file has changed if its device or inode have changed.)
431 If it has changed, the old file stream is closed, and the file
432 opened to get a new stream.
433
434 This handler is not appropriate for use under Windows, because
435 under Windows open files cannot be moved or renamed - logging
436 opens the files with exclusive locks - and so there is no need
437 for such a handler. Furthermore, ST_INO is not supported under
438 Windows; stat always returns zero for this value.
439
440 This handler is based on a suggestion and patch by Chad J.
441 Schroeder.
442 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100443 def __init__(self, filename, mode='a', encoding=None, delay=False,
444 errors=None):
445 logging.FileHandler.__init__(self, filename, mode=mode,
446 encoding=encoding, delay=delay,
447 errors=errors)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100448 self.dev, self.ino = -1, -1
449 self._statstream()
450
451 def _statstream(self):
452 if self.stream:
453 sres = os.fstat(self.stream.fileno())
454 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000455
Vinay Sajip29a14452015-10-01 20:54:41 +0100456 def reopenIfNeeded(self):
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000457 """
Vinay Sajip29a14452015-10-01 20:54:41 +0100458 Reopen log file if needed.
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000459
Vinay Sajip29a14452015-10-01 20:54:41 +0100460 Checks if the underlying file has changed, and if it
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000461 has, close the old stream and reopen the file to get the
462 current stream.
463 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100464 # Reduce the chance of race conditions by stat'ing by path only
465 # once and then fstat'ing our new fd if we opened a new log stream.
466 # See issue #14632: Thanks to John Mulligan for the problem report
467 # and patch.
468 try:
469 # stat the file by path, checking for existence
470 sres = os.stat(self.baseFilename)
Giampaolo Rodola'0166a282013-02-12 15:14:17 +0100471 except FileNotFoundError:
472 sres = None
Vinay Sajip66b8b082012-04-24 23:25:30 +0100473 # compare file system stat with that of our stream file handle
474 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
475 if self.stream is not None:
476 # we have an open file handle, clean it up
477 self.stream.flush()
478 self.stream.close()
Vinay Sajip02200482014-06-14 10:22:05 +0100479 self.stream = None # See Issue #21742: _open () might fail.
Vinay Sajip66b8b082012-04-24 23:25:30 +0100480 # open a new file handle and get new stat info from that fd
481 self.stream = self._open()
482 self._statstream()
Vinay Sajip29a14452015-10-01 20:54:41 +0100483
484 def emit(self, record):
485 """
486 Emit a record.
487
488 If underlying file has changed, reopen the file before emitting the
489 record to it.
490 """
491 self.reopenIfNeeded()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000492 logging.FileHandler.emit(self, record)
493
Vinay Sajip66b8b082012-04-24 23:25:30 +0100494
Guido van Rossum57102f82002-11-13 16:15:58 +0000495class SocketHandler(logging.Handler):
496 """
497 A handler class which writes logging records, in pickle format, to
498 a streaming socket. The socket is kept open across logging calls.
499 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000500 The pickle which is sent is that of the LogRecord's attribute dictionary
501 (__dict__), so that the receiver does not need to have the logging module
502 installed in order to process the logging event.
503
504 To unpickle the record at the receiving end into a LogRecord, use the
505 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000506 """
507
508 def __init__(self, host, port):
509 """
510 Initializes the handler with a specific host address and port.
511
Vinay Sajipde19e082011-04-30 21:52:26 +0100512 When the attribute *closeOnError* is set to True - if a socket error
513 occurs, the socket is silently closed and then reopened on the next
514 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000515 """
516 logging.Handler.__init__(self)
517 self.host = host
518 self.port = port
Vinay Sajip5421f352013-09-27 18:18:28 +0100519 if port is None:
520 self.address = host
521 else:
522 self.address = (host, port)
Guido van Rossum57102f82002-11-13 16:15:58 +0000523 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100524 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000525 self.retryTime = None
526 #
527 # Exponential backoff parameters.
528 #
529 self.retryStart = 1.0
530 self.retryMax = 30.0
531 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000532
Guido van Rossumd8faa362007-04-27 19:54:29 +0000533 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000534 """
535 A factory method which allows subclasses to define the precise
536 type of socket they want.
537 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100538 if self.port is not None:
539 result = socket.create_connection(self.address, timeout=timeout)
540 else:
541 result = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
542 result.settimeout(timeout)
Vinay Sajip38c741c2013-10-06 18:36:00 +0100543 try:
544 result.connect(self.address)
545 except OSError:
546 result.close() # Issue 19182
547 raise
Vinay Sajip5421f352013-09-27 18:18:28 +0100548 return result
Guido van Rossum57102f82002-11-13 16:15:58 +0000549
Vinay Sajip48cfe382004-02-20 13:17:27 +0000550 def createSocket(self):
551 """
552 Try to create a socket, using an exponential backoff with
553 a max retry time. Thanks to Robert Olson for the original patch
554 (SF #815911) which has been slightly refactored.
555 """
556 now = time.time()
557 # Either retryTime is None, in which case this
558 # is the first time back after a disconnect, or
559 # we've waited long enough.
560 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100561 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000562 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000563 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000564 if attempt:
565 try:
566 self.sock = self.makeSocket()
567 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200568 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000569 #Creation failed, so set the retry time and return.
570 if self.retryTime is None:
571 self.retryPeriod = self.retryStart
572 else:
573 self.retryPeriod = self.retryPeriod * self.retryFactor
574 if self.retryPeriod > self.retryMax:
575 self.retryPeriod = self.retryMax
576 self.retryTime = now + self.retryPeriod
577
Guido van Rossum57102f82002-11-13 16:15:58 +0000578 def send(self, s):
579 """
580 Send a pickled string to the socket.
581
582 This function allows for partial sends which can happen when the
583 network is busy.
584 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000585 if self.sock is None:
586 self.createSocket()
587 #self.sock can be None either because we haven't reached the retry
588 #time yet, or because we have reached the retry time and retried,
589 #but are still unable to connect.
590 if self.sock:
591 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200592 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200593 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000594 self.sock.close()
595 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000596
597 def makePickle(self, record):
598 """
599 Pickles the record in binary format with a length prefix, and
600 returns it ready for transmission across the socket.
601 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000602 ei = record.exc_info
603 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100604 # just to get traceback text into record.exc_text ...
605 dummy = self.format(record)
606 # See issue #14436: If msg or args are objects, they may not be
607 # available on the receiving end. So we convert the msg % args
608 # to a string, save it as msg and zap the args.
609 d = dict(record.__dict__)
610 d['msg'] = record.getMessage()
611 d['args'] = None
612 d['exc_info'] = None
Vinay Sajip9cdfd182015-12-26 12:48:44 +0000613 # Issue #25685: delete 'message' if present: redundant with 'msg'
614 d.pop('message', None)
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100615 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000616 slen = struct.pack(">L", len(s))
617 return slen + s
618
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000619 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000620 """
621 Handle an error during logging.
622
623 An error has occurred during logging. Most likely cause -
624 connection lost. Close the socket so that we can retry on the
625 next event.
626 """
627 if self.closeOnError and self.sock:
628 self.sock.close()
629 self.sock = None #try to reconnect next time
630 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000631 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000632
633 def emit(self, record):
634 """
635 Emit a record.
636
637 Pickles the record and writes it to the socket in binary format.
638 If there is an error with the socket, silently drop the packet.
639 If there was a problem with the socket, re-establishes the
640 socket.
641 """
642 try:
643 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000644 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100645 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000646 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000647
648 def close(self):
649 """
650 Closes the socket.
651 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000652 self.acquire()
653 try:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300654 sock = self.sock
655 if sock:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000656 self.sock = None
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300657 sock.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000658 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000659 finally:
660 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000661
662class DatagramHandler(SocketHandler):
663 """
664 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000665 a datagram socket. The pickle which is sent is that of the LogRecord's
666 attribute dictionary (__dict__), so that the receiver does not need to
667 have the logging module installed in order to process the logging event.
668
669 To unpickle the record at the receiving end into a LogRecord, use the
670 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000671
672 """
673 def __init__(self, host, port):
674 """
675 Initializes the handler with a specific host address and port.
676 """
677 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100678 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000679
680 def makeSocket(self):
681 """
682 The factory method of SocketHandler is here overridden to create
683 a UDP socket (SOCK_DGRAM).
684 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100685 if self.port is None:
686 family = socket.AF_UNIX
687 else:
688 family = socket.AF_INET
689 s = socket.socket(family, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000690 return s
691
692 def send(self, s):
693 """
694 Send a pickled string to a socket.
695
696 This function no longer allows for partial sends which can happen
697 when the network is busy - UDP does not guarantee delivery and
698 can deliver packets out of sequence.
699 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000700 if self.sock is None:
701 self.createSocket()
Vinay Sajip5421f352013-09-27 18:18:28 +0100702 self.sock.sendto(s, self.address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000703
704class SysLogHandler(logging.Handler):
705 """
706 A handler class which sends formatted logging records to a syslog
707 server. Based on Sam Rushing's syslog module:
708 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
709 Contributed by Nicolas Untz (after which minor refactoring changes
710 have been made).
711 """
712
713 # from <linux/sys/syslog.h>:
714 # ======================================================================
715 # priorities/facilities are encoded into a single 32-bit quantity, where
716 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
717 # facility (0-big number). Both the priorities and the facilities map
718 # roughly one-to-one to strings in the syslogd(8) source code. This
719 # mapping is included in this file.
720 #
721 # priorities (these are ordered)
722
723 LOG_EMERG = 0 # system is unusable
724 LOG_ALERT = 1 # action must be taken immediately
725 LOG_CRIT = 2 # critical conditions
726 LOG_ERR = 3 # error conditions
727 LOG_WARNING = 4 # warning conditions
728 LOG_NOTICE = 5 # normal but significant condition
729 LOG_INFO = 6 # informational
730 LOG_DEBUG = 7 # debug-level messages
731
732 # facility codes
733 LOG_KERN = 0 # kernel messages
734 LOG_USER = 1 # random user-level messages
735 LOG_MAIL = 2 # mail system
736 LOG_DAEMON = 3 # system daemons
737 LOG_AUTH = 4 # security/authorization messages
738 LOG_SYSLOG = 5 # messages generated internally by syslogd
739 LOG_LPR = 6 # line printer subsystem
740 LOG_NEWS = 7 # network news subsystem
741 LOG_UUCP = 8 # UUCP subsystem
742 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000743 LOG_AUTHPRIV = 10 # security/authorization messages (private)
744 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000745
746 # other codes through 15 reserved for system use
747 LOG_LOCAL0 = 16 # reserved for local use
748 LOG_LOCAL1 = 17 # reserved for local use
749 LOG_LOCAL2 = 18 # reserved for local use
750 LOG_LOCAL3 = 19 # reserved for local use
751 LOG_LOCAL4 = 20 # reserved for local use
752 LOG_LOCAL5 = 21 # reserved for local use
753 LOG_LOCAL6 = 22 # reserved for local use
754 LOG_LOCAL7 = 23 # reserved for local use
755
756 priority_names = {
757 "alert": LOG_ALERT,
758 "crit": LOG_CRIT,
759 "critical": LOG_CRIT,
760 "debug": LOG_DEBUG,
761 "emerg": LOG_EMERG,
762 "err": LOG_ERR,
763 "error": LOG_ERR, # DEPRECATED
764 "info": LOG_INFO,
765 "notice": LOG_NOTICE,
766 "panic": LOG_EMERG, # DEPRECATED
767 "warn": LOG_WARNING, # DEPRECATED
768 "warning": LOG_WARNING,
769 }
770
771 facility_names = {
772 "auth": LOG_AUTH,
773 "authpriv": LOG_AUTHPRIV,
774 "cron": LOG_CRON,
775 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000776 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000777 "kern": LOG_KERN,
778 "lpr": LOG_LPR,
779 "mail": LOG_MAIL,
780 "news": LOG_NEWS,
781 "security": LOG_AUTH, # DEPRECATED
782 "syslog": LOG_SYSLOG,
783 "user": LOG_USER,
784 "uucp": LOG_UUCP,
785 "local0": LOG_LOCAL0,
786 "local1": LOG_LOCAL1,
787 "local2": LOG_LOCAL2,
788 "local3": LOG_LOCAL3,
789 "local4": LOG_LOCAL4,
790 "local5": LOG_LOCAL5,
791 "local6": LOG_LOCAL6,
792 "local7": LOG_LOCAL7,
793 }
794
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000795 #The map below appears to be trivially lowercasing the key. However,
796 #there's more to it than meets the eye - in some locales, lowercasing
797 #gives unexpected results. See SF #1524081: in the Turkish locale,
798 #"INFO".lower() != "info"
799 priority_map = {
800 "DEBUG" : "debug",
801 "INFO" : "info",
802 "WARNING" : "warning",
803 "ERROR" : "error",
804 "CRITICAL" : "critical"
805 }
806
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000807 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100808 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000809 """
810 Initialize a handler.
811
Guido van Rossume7ba4952007-06-06 23:52:48 +0000812 If address is specified as a string, a UNIX socket is used. To log to a
813 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100814 If facility is not specified, LOG_USER is used. If socktype is
815 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
816 socket type will be used. For Unix sockets, you can also specify a
817 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
818 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000819 """
820 logging.Handler.__init__(self)
821
822 self.address = address
823 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000824 self.socktype = socktype
825
Guido van Rossum13257902007-06-07 23:15:56 +0000826 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100827 self.unixsocket = True
Коренберг Марк1b038e02017-03-17 20:25:05 +0500828 # Syslog server may be unavailable during handler initialisation.
829 # C's openlog() function also ignores connection errors.
830 # Moreover, we ignore these errors while logging, so it not worse
831 # to ignore it also here.
832 try:
833 self._connect_unixsocket(address)
834 except OSError:
835 pass
Guido van Rossum57102f82002-11-13 16:15:58 +0000836 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100837 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100838 if socktype is None:
839 socktype = socket.SOCK_DGRAM
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800840 host, port = address
841 ress = socket.getaddrinfo(host, port, 0, socktype)
842 if not ress:
843 raise OSError("getaddrinfo returns an empty list")
844 for res in ress:
845 af, socktype, proto, _, sa = res
846 err = sock = None
847 try:
848 sock = socket.socket(af, socktype, proto)
849 if socktype == socket.SOCK_STREAM:
850 sock.connect(sa)
851 break
852 except OSError as exc:
853 err = exc
854 if sock is not None:
855 sock.close()
856 if err is not None:
857 raise err
858 self.socket = sock
Vinay Sajipe9170522013-04-22 10:07:40 +0100859 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000860
Vinay Sajipa1974c12005-01-13 08:23:56 +0000861 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100862 use_socktype = self.socktype
863 if use_socktype is None:
864 use_socktype = socket.SOCK_DGRAM
865 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000866 try:
867 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100868 # it worked, so set self.socktype to the used type
869 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200870 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000871 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100872 if self.socktype is not None:
873 # user didn't specify falling back, so fail
874 raise
875 use_socktype = socket.SOCK_STREAM
876 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
877 try:
878 self.socket.connect(address)
879 # it worked, so set self.socktype to the used type
880 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100881 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100882 self.socket.close()
883 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000884
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000885 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000886 """
887 Encode the facility and priority. You can pass in strings or
888 integers - if strings are passed, the facility_names and
889 priority_names mapping dictionaries are used to convert them to
890 integers.
891 """
Guido van Rossum13257902007-06-07 23:15:56 +0000892 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000893 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000894 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000895 priority = self.priority_names[priority]
896 return (facility << 3) | priority
897
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800898 def close(self):
Guido van Rossum57102f82002-11-13 16:15:58 +0000899 """
900 Closes the socket.
901 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000902 self.acquire()
903 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000904 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000905 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000906 finally:
907 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000908
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000909 def mapPriority(self, levelName):
910 """
911 Map a logging level name to a key in the priority_names map.
912 This is useful in two scenarios: when custom levels are being
913 used, and in the case where you can't do a straightforward
914 mapping by lowercasing the logging level name because of locale-
915 specific issues (see SF #1524081).
916 """
917 return self.priority_map.get(levelName, "warning")
918
Vinay Sajip2353e352011-06-27 15:40:06 +0100919 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100920 append_nul = True # some old syslog daemons expect a NUL terminator
921
Guido van Rossum57102f82002-11-13 16:15:58 +0000922 def emit(self, record):
923 """
924 Emit a record.
925
926 The record is formatted, and then sent to the syslog server. If
927 exception information is present, it is NOT sent to the server.
928 """
Guido van Rossum57102f82002-11-13 16:15:58 +0000929 try:
Vinay Sajipc33a0cc2014-11-01 19:58:47 +0000930 msg = self.format(record)
931 if self.ident:
932 msg = self.ident + msg
933 if self.append_nul:
934 msg += '\000'
935
936 # We need to convert record level to lowercase, maybe this will
937 # change in the future.
938 prio = '<%d>' % self.encodePriority(self.facility,
939 self.mapPriority(record.levelname))
940 prio = prio.encode('utf-8')
941 # Message is a string. Convert to bytes as required by RFC 5424
942 msg = msg.encode('utf-8')
943 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000944 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000945 try:
946 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200947 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100948 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000949 self._connect_unixsocket(self.address)
950 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000951 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000952 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000953 else:
954 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100955 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000956 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000957
958class SMTPHandler(logging.Handler):
959 """
960 A handler class which sends an SMTP email for each logging event.
961 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000962 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100963 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000964 """
965 Initialize the handler.
966
967 Initialize the instance with the from and to addresses and subject
968 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000969 (host, port) tuple format for the mailhost argument. To specify
970 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000971 for the credentials argument. To specify the use of a secure
972 protocol (TLS), pass in a tuple for the secure argument. This will
973 only be used when authentication credentials are supplied. The tuple
974 will be either an empty tuple, or a single-value tuple with the name
975 of a keyfile, or a 2-value tuple with the names of the keyfile and
976 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000977 A timeout in seconds can be specified for the SMTP connection (the
978 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000979 """
980 logging.Handler.__init__(self)
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100981 if isinstance(mailhost, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000982 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000983 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000984 self.mailhost, self.mailport = mailhost, None
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100985 if isinstance(credentials, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000986 self.username, self.password = credentials
987 else:
988 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000989 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000990 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000991 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000992 self.toaddrs = toaddrs
993 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000994 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000995 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000996
997 def getSubject(self, record):
998 """
999 Determine the subject for the email.
1000
1001 If you want to specify a subject line which is record-dependent,
1002 override this method.
1003 """
1004 return self.subject
1005
1006 def emit(self, record):
1007 """
1008 Emit a record.
1009
1010 Format the record and send it to the specified addressees.
1011 """
1012 try:
1013 import smtplib
Vinay Sajip277640a2015-10-17 16:13:10 +01001014 from email.message import EmailMessage
1015 import email.utils
1016
Guido van Rossum57102f82002-11-13 16:15:58 +00001017 port = self.mailport
1018 if not port:
1019 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +00001020 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Vinay Sajip277640a2015-10-17 16:13:10 +01001021 msg = EmailMessage()
1022 msg['From'] = self.fromaddr
1023 msg['To'] = ','.join(self.toaddrs)
1024 msg['Subject'] = self.getSubject(record)
1025 msg['Date'] = email.utils.localtime()
1026 msg.set_content(self.format(record))
Guido van Rossum360e4b82007-05-14 22:51:27 +00001027 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +00001028 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +00001029 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +00001030 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +00001031 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +00001032 smtp.login(self.username, self.password)
Vinay Sajip277640a2015-10-17 16:13:10 +01001033 smtp.send_message(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +00001034 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001035 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001036 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001037
1038class NTEventLogHandler(logging.Handler):
1039 """
1040 A handler class which sends events to the NT Event Log. Adds a
1041 registry entry for the specified application name. If no dllname is
1042 provided, win32service.pyd (which contains some basic message
1043 placeholders) is used. Note that use of these placeholders will make
1044 your event logs big, as the entire message source is held in the log.
1045 If you want slimmer logs, you have to pass in the name of your own DLL
1046 which contains the message definitions you want to use in the event log.
1047 """
1048 def __init__(self, appname, dllname=None, logtype="Application"):
1049 logging.Handler.__init__(self)
1050 try:
1051 import win32evtlogutil, win32evtlog
1052 self.appname = appname
1053 self._welu = win32evtlogutil
1054 if not dllname:
1055 dllname = os.path.split(self._welu.__file__)
1056 dllname = os.path.split(dllname[0])
1057 dllname = os.path.join(dllname[0], r'win32service.pyd')
1058 self.dllname = dllname
1059 self.logtype = logtype
1060 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1061 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1062 self.typemap = {
1063 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1064 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001065 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001066 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1067 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1068 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001069 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001070 print("The Python Win32 extensions for NT (service, event "\
1071 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001072 self._welu = None
1073
1074 def getMessageID(self, record):
1075 """
1076 Return the message ID for the event record. If you are using your
1077 own messages, you could do this by having the msg passed to the
1078 logger being an ID rather than a formatting string. Then, in here,
1079 you could use a dictionary lookup to get the message ID. This
1080 version returns 1, which is the base message ID in win32service.pyd.
1081 """
1082 return 1
1083
1084 def getEventCategory(self, record):
1085 """
1086 Return the event category for the record.
1087
1088 Override this if you want to specify your own categories. This version
1089 returns 0.
1090 """
1091 return 0
1092
1093 def getEventType(self, record):
1094 """
1095 Return the event type for the record.
1096
1097 Override this if you want to specify your own types. This version does
1098 a mapping using the handler's typemap attribute, which is set up in
1099 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001100 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001101 either need to override this method or place a suitable dictionary in
1102 the handler's typemap attribute.
1103 """
1104 return self.typemap.get(record.levelno, self.deftype)
1105
1106 def emit(self, record):
1107 """
1108 Emit a record.
1109
1110 Determine the message ID, event category and event type. Then
1111 log the message in the NT event log.
1112 """
1113 if self._welu:
1114 try:
1115 id = self.getMessageID(record)
1116 cat = self.getEventCategory(record)
1117 type = self.getEventType(record)
1118 msg = self.format(record)
1119 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001120 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001121 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001122
1123 def close(self):
1124 """
1125 Clean up this handler.
1126
1127 You can remove the application name from the registry as a
1128 source of event log entries. However, if you do this, you will
1129 not be able to see the events as you intended in the Event Log
1130 Viewer - it needs to be able to access the registry to get the
1131 DLL name.
1132 """
1133 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001134 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001135
1136class HTTPHandler(logging.Handler):
1137 """
1138 A class which sends records to a Web server, using either GET or
1139 POST semantics.
1140 """
Benjamin Peterson43052a12014-11-23 20:36:44 -06001141 def __init__(self, host, url, method="GET", secure=False, credentials=None,
1142 context=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001143 """
1144 Initialize the instance with the host, the request URL, and the method
1145 ("GET" or "POST")
1146 """
1147 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001148 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001149 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001150 raise ValueError("method must be GET or POST")
Benjamin Peterson43052a12014-11-23 20:36:44 -06001151 if not secure and context is not None:
1152 raise ValueError("context parameter only makes sense "
1153 "with secure=True")
Guido van Rossum57102f82002-11-13 16:15:58 +00001154 self.host = host
1155 self.url = url
1156 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001157 self.secure = secure
1158 self.credentials = credentials
Benjamin Peterson43052a12014-11-23 20:36:44 -06001159 self.context = context
Guido van Rossum57102f82002-11-13 16:15:58 +00001160
Neal Norwitzf297bd12003-04-23 03:49:43 +00001161 def mapLogRecord(self, record):
1162 """
1163 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001164 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001165 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001166 """
1167 return record.__dict__
1168
Guido van Rossum57102f82002-11-13 16:15:58 +00001169 def emit(self, record):
1170 """
1171 Emit a record.
1172
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001173 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001174 """
1175 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001176 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001177 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001178 if self.secure:
Benjamin Peterson43052a12014-11-23 20:36:44 -06001179 h = http.client.HTTPSConnection(host, context=self.context)
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001180 else:
1181 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001182 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001183 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001184 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001185 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001186 sep = '&'
1187 else:
1188 sep = '?'
1189 url = url + "%c%s" % (sep, data)
1190 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001191 # support multiple hosts on one IP address...
1192 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001193 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001194 if i >= 0:
1195 host = host[:i]
Vinay Sajipe96ba182017-11-19 18:36:17 +00001196 # See issue #30904: putrequest call above already adds this header
1197 # on Python 3.x.
1198 # h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001199 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001200 h.putheader("Content-type",
1201 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001202 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001203 if self.credentials:
1204 import base64
Vinay Sajip1bf197e2016-06-07 21:19:55 +01001205 s = ('%s:%s' % self.credentials).encode('utf-8')
1206 s = 'Basic ' + base64.b64encode(s).strip().decode('ascii')
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001207 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001208 h.endheaders()
1209 if self.method == "POST":
1210 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001211 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001212 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001213 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001214
1215class BufferingHandler(logging.Handler):
1216 """
1217 A handler class which buffers logging records in memory. Whenever each
1218 record is added to the buffer, a check is made to see if the buffer should
1219 be flushed. If it should, then flush() is expected to do what's needed.
1220 """
1221 def __init__(self, capacity):
1222 """
1223 Initialize the handler with the buffer size.
1224 """
1225 logging.Handler.__init__(self)
1226 self.capacity = capacity
1227 self.buffer = []
1228
1229 def shouldFlush(self, record):
1230 """
1231 Should the handler flush its buffer?
1232
1233 Returns true if the buffer is up to capacity. This method can be
1234 overridden to implement custom flushing strategies.
1235 """
1236 return (len(self.buffer) >= self.capacity)
1237
1238 def emit(self, record):
1239 """
1240 Emit a record.
1241
1242 Append the record. If shouldFlush() tells us to, call flush() to process
1243 the buffer.
1244 """
1245 self.buffer.append(record)
1246 if self.shouldFlush(record):
1247 self.flush()
1248
1249 def flush(self):
1250 """
1251 Override to implement custom flushing behaviour.
1252
1253 This version just zaps the buffer to empty.
1254 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001255 self.acquire()
1256 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001257 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001258 finally:
1259 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001260
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001261 def close(self):
1262 """
1263 Close the handler.
1264
1265 This version just flushes and chains to the parent class' close().
1266 """
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001267 try:
1268 self.flush()
1269 finally:
1270 logging.Handler.close(self)
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001271
Guido van Rossum57102f82002-11-13 16:15:58 +00001272class MemoryHandler(BufferingHandler):
1273 """
1274 A handler class which buffers logging records in memory, periodically
1275 flushing them to a target handler. Flushing occurs whenever the buffer
1276 is full, or when an event of a certain severity or greater is seen.
1277 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001278 def __init__(self, capacity, flushLevel=logging.ERROR, target=None,
1279 flushOnClose=True):
Guido van Rossum57102f82002-11-13 16:15:58 +00001280 """
1281 Initialize the handler with the buffer size, the level at which
1282 flushing should occur and an optional target.
1283
1284 Note that without a target being set either here or via setTarget(),
1285 a MemoryHandler is no use to anyone!
Vinay Sajipcccf6062016-07-22 16:27:31 +01001286
1287 The ``flushOnClose`` argument is ``True`` for backward compatibility
1288 reasons - the old behaviour is that when the handler is closed, the
1289 buffer is flushed, even if the flush level hasn't been exceeded nor the
1290 capacity exceeded. To prevent this, set ``flushOnClose`` to ``False``.
Guido van Rossum57102f82002-11-13 16:15:58 +00001291 """
1292 BufferingHandler.__init__(self, capacity)
1293 self.flushLevel = flushLevel
1294 self.target = target
Vinay Sajipcccf6062016-07-22 16:27:31 +01001295 # See Issue #26559 for why this has been added
1296 self.flushOnClose = flushOnClose
Guido van Rossum57102f82002-11-13 16:15:58 +00001297
1298 def shouldFlush(self, record):
1299 """
1300 Check for buffer full or a record at the flushLevel or higher.
1301 """
1302 return (len(self.buffer) >= self.capacity) or \
1303 (record.levelno >= self.flushLevel)
1304
1305 def setTarget(self, target):
1306 """
1307 Set the target handler for this handler.
1308 """
1309 self.target = target
1310
1311 def flush(self):
1312 """
1313 For a MemoryHandler, flushing means just sending the buffered
1314 records to the target, if there is one. Override if you want
1315 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001316
1317 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001318 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001319 self.acquire()
1320 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001321 if self.target:
1322 for record in self.buffer:
1323 self.target.handle(record)
1324 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001325 finally:
1326 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001327
1328 def close(self):
1329 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001330 Flush, if appropriately configured, set the target to None and lose the
1331 buffer.
Guido van Rossum57102f82002-11-13 16:15:58 +00001332 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001333 try:
Vinay Sajipcccf6062016-07-22 16:27:31 +01001334 if self.flushOnClose:
1335 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001336 finally:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001337 self.acquire()
1338 try:
1339 self.target = None
1340 BufferingHandler.close(self)
1341 finally:
1342 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001343
1344
1345class QueueHandler(logging.Handler):
1346 """
1347 This handler sends events to a queue. Typically, it would be used together
1348 with a multiprocessing Queue to centralise logging to file in one process
1349 (in a multi-process application), so as to avoid file write contention
1350 between processes.
1351
1352 This code is new in Python 3.2, but this class can be copy pasted into
1353 user code for use with earlier Python versions.
1354 """
1355
1356 def __init__(self, queue):
1357 """
1358 Initialise an instance, using the passed queue.
1359 """
1360 logging.Handler.__init__(self)
1361 self.queue = queue
1362
1363 def enqueue(self, record):
1364 """
1365 Enqueue a record.
1366
1367 The base implementation uses put_nowait. You may want to override
1368 this method if you want to use blocking, timeouts or custom queue
1369 implementations.
1370 """
1371 self.queue.put_nowait(record)
1372
Vinay Sajip0258ce82010-09-22 20:34:53 +00001373 def prepare(self, record):
1374 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001375 Prepares a record for queuing. The object returned by this method is
1376 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001377
1378 The base implementation formats the record to merge the message
1379 and arguments, and removes unpickleable items from the record
1380 in-place.
1381
1382 You might want to override this method if you want to convert
1383 the record to a dict or JSON string, or send a modified copy
1384 of the record while leaving the original intact.
1385 """
1386 # The format operation gets traceback text into record.exc_text
favlladfe3442017-08-01 20:12:26 +02001387 # (if there's exception data), and also returns the formatted
1388 # message. We can then use this to replace the original
Vinay Sajip0258ce82010-09-22 20:34:53 +00001389 # msg + args, as these might be unpickleable. We also zap the
Cheryl Sabellad345bb42018-09-25 19:00:08 -04001390 # exc_info and exc_text attributes, as they are no longer
1391 # needed and, if not None, will typically not be pickleable.
favlladfe3442017-08-01 20:12:26 +02001392 msg = self.format(record)
Manjusakada6424e2019-01-23 15:08:38 +08001393 # bpo-35726: make copy of record to avoid affecting other handlers in the chain.
1394 record = copy.copy(record)
favlladfe3442017-08-01 20:12:26 +02001395 record.message = msg
1396 record.msg = msg
Vinay Sajip0258ce82010-09-22 20:34:53 +00001397 record.args = None
1398 record.exc_info = None
Cheryl Sabellad345bb42018-09-25 19:00:08 -04001399 record.exc_text = None
Vinay Sajip0258ce82010-09-22 20:34:53 +00001400 return record
1401
Vinay Sajip121a1c42010-09-08 10:46:15 +00001402 def emit(self, record):
1403 """
1404 Emit a record.
1405
Vinay Sajip0637d492010-09-23 08:15:54 +00001406 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001407 """
1408 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001409 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001410 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001411 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001412
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001413
1414class QueueListener(object):
1415 """
1416 This class implements an internal threaded listener which watches for
1417 LogRecords being added to a queue, removes them and passes them to a
1418 list of handlers for processing.
1419 """
1420 _sentinel = None
1421
1422 def __init__(self, queue, *handlers, respect_handler_level=False):
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001423 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001424 Initialise an instance with the specified queue and
1425 handlers.
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001426 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001427 self.queue = queue
1428 self.handlers = handlers
1429 self._thread = None
1430 self.respect_handler_level = respect_handler_level
Vinay Sajip0637d492010-09-23 08:15:54 +00001431
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001432 def dequeue(self, block):
1433 """
1434 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001435
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001436 The base implementation uses get. You may want to override this method
1437 if you want to use timeouts or work with custom queue implementations.
1438 """
1439 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001440
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001441 def start(self):
1442 """
1443 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001444
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001445 This starts up a background thread to monitor the queue for
1446 LogRecords to process.
1447 """
1448 self._thread = t = threading.Thread(target=self._monitor)
1449 t.daemon = True
1450 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001451
Boris Feld800d7862019-06-04 17:20:18 +02001452 def prepare(self, record):
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001453 """
1454 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001455
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001456 This method just returns the passed-in record. You may want to
1457 override this method if you need to do any custom marshalling or
1458 manipulation of the record before passing it to the handlers.
1459 """
1460 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001461
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001462 def handle(self, record):
1463 """
1464 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001465
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001466 This just loops through the handlers offering them the record
1467 to handle.
1468 """
1469 record = self.prepare(record)
1470 for handler in self.handlers:
1471 if not self.respect_handler_level:
1472 process = True
1473 else:
1474 process = record.levelno >= handler.level
1475 if process:
1476 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001477
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001478 def _monitor(self):
1479 """
1480 Monitor the queue for records, and ask the handler
1481 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001482
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001483 This method runs on a separate, internal thread.
1484 The thread will terminate if it sees a sentinel object in the queue.
1485 """
1486 q = self.queue
1487 has_task_done = hasattr(q, 'task_done')
1488 while True:
1489 try:
1490 record = self.dequeue(True)
1491 if record is self._sentinel:
Bar Harel6b282e12019-06-01 12:19:09 +03001492 if has_task_done:
1493 q.task_done()
Vinay Sajip0637d492010-09-23 08:15:54 +00001494 break
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001495 self.handle(record)
1496 if has_task_done:
1497 q.task_done()
1498 except queue.Empty:
1499 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001500
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001501 def enqueue_sentinel(self):
1502 """
1503 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001504
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001505 The base implementation uses put_nowait. You may want to override this
1506 method if you want to use timeouts or work with custom queue
1507 implementations.
1508 """
1509 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001510
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001511 def stop(self):
1512 """
1513 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001514
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001515 This asks the thread to terminate, and then waits for it to do so.
1516 Note that if you don't call this before your application exits, there
1517 may be some records still left on the queue, which won't be processed.
1518 """
1519 self.enqueue_sentinel()
1520 self._thread.join()
1521 self._thread = None