blob: 4cabc0d740c29f5d8a6213a42a0ead9c745c2b0f [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
Inada Naokifb786922021-04-06 11:18:41 +090026import io, 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'
Inada Naokifb786922021-04-06 11:18:41 +0900153 if "b" not in mode:
154 encoding = io.text_encoding(encoding)
Vinay Sajipca7b5042019-06-17 17:40:52 +0100155 BaseRotatingHandler.__init__(self, filename, mode, encoding=encoding,
156 delay=delay, errors=errors)
Guido van Rossum57102f82002-11-13 16:15:58 +0000157 self.maxBytes = maxBytes
158 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000159
160 def doRollover(self):
161 """
162 Do a rollover, as described in __init__().
163 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000164 if self.stream:
165 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000166 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000167 if self.backupCount > 0:
168 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000169 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
170 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
171 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000172 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000173 if os.path.exists(dfn):
174 os.remove(dfn)
175 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000176 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000177 if os.path.exists(dfn):
178 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000179 self.rotate(self.baseFilename, dfn)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100180 if not self.delay:
181 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000182
Vinay Sajip17c52d82004-07-03 11:48:34 +0000183 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000184 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000185 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000186
Vinay Sajip17c52d82004-07-03 11:48:34 +0000187 Basically, see if the supplied record would cause the file to exceed
188 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000189 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000190 if self.stream is None: # delay was set...
191 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000192 if self.maxBytes > 0: # are we rolling over?
193 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000194 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000195 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000196 return 1
197 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000198
Vinay Sajip17c52d82004-07-03 11:48:34 +0000199class TimedRotatingFileHandler(BaseRotatingHandler):
200 """
201 Handler for logging to a file, rotating the log file at certain timed
202 intervals.
203
204 If backupCount is > 0, when rollover is done, no more than backupCount
205 files are kept - the oldest ones are deleted.
206 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100207 def __init__(self, filename, when='h', interval=1, backupCount=0,
208 encoding=None, delay=False, utc=False, atTime=None,
209 errors=None):
Inada Naokifb786922021-04-06 11:18:41 +0900210 encoding = io.text_encoding(encoding)
Vinay Sajipca7b5042019-06-17 17:40:52 +0100211 BaseRotatingHandler.__init__(self, filename, 'a', encoding=encoding,
212 delay=delay, errors=errors)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000213 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000214 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000215 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100216 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000217 # Calculate the real rollover interval, which is just the number of
218 # seconds between rollovers. Also set the filename suffix used when
219 # a rollover occurs. Current 'when' events supported:
220 # S - Seconds
221 # M - Minutes
222 # H - Hours
223 # D - Days
224 # midnight - roll over at midnight
225 # W{0-6} - roll over on a certain day; 0 - Monday
226 #
227 # Case of the 'when' specifier is not important; lower or upper case
228 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000229 if self.when == 'S':
230 self.interval = 1 # one second
231 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000232 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000233 elif self.when == 'M':
234 self.interval = 60 # one minute
235 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000236 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000237 elif self.when == 'H':
238 self.interval = 60 * 60 # one hour
239 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000240 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000241 elif self.when == 'D' or self.when == 'MIDNIGHT':
242 self.interval = 60 * 60 * 24 # one day
243 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000244 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000245 elif self.when.startswith('W'):
246 self.interval = 60 * 60 * 24 * 7 # one week
247 if len(self.when) != 2:
248 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
249 if self.when[1] < '0' or self.when[1] > '6':
250 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
251 self.dayOfWeek = int(self.when[1])
252 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000253 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000254 else:
255 raise ValueError("Invalid rollover interval specified: %s" % self.when)
256
Antoine Pitroufd036452008-08-19 17:56:33 +0000257 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000258 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip638e6222016-07-22 18:23:04 +0100259 # The following line added because the filename passed in could be a
260 # path object (see Issue #27493), but self.baseFilename will be a string
261 filename = self.baseFilename
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000262 if os.path.exists(filename):
263 t = os.stat(filename)[ST_MTIME]
264 else:
265 t = int(time.time())
266 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000267
Vinay Sajipaffbd872009-06-11 10:11:47 +0000268 def computeRollover(self, currentTime):
269 """
270 Work out the rollover time based on the specified time.
271 """
272 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000273 # If we are rolling over at midnight or weekly, then the interval is already known.
274 # What we need to figure out is WHEN the next interval is. In other words,
275 # if you are rolling over at midnight, then your base interval is 1 day,
276 # but you want to start that one day clock at midnight, not now. So, we
277 # have to fudge the rolloverAt value in order to trigger the first rollover
278 # at the right time. After that, the regular interval will take care of
279 # the rest. Note that this code doesn't care about leap seconds. :)
280 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
281 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000282 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000283 t = time.gmtime(currentTime)
284 else:
285 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000286 currentHour = t[3]
287 currentMinute = t[4]
288 currentSecond = t[5]
Vinay Sajipa7130792013-04-12 17:04:23 +0100289 currentDay = t[6]
290 # r is the number of seconds left between now and the next rotation
291 if self.atTime is None:
292 rotate_ts = _MIDNIGHT
293 else:
294 rotate_ts = ((self.atTime.hour * 60 + self.atTime.minute)*60 +
295 self.atTime.second)
296
297 r = rotate_ts - ((currentHour * 60 + currentMinute) * 60 +
298 currentSecond)
299 if r < 0:
300 # Rotate time is before the current time (for example when
301 # self.rotateAt is 13:45 and it now 14:15), rotation is
302 # tomorrow.
303 r += _MIDNIGHT
304 currentDay = (currentDay + 1) % 7
Vinay Sajipaffbd872009-06-11 10:11:47 +0000305 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000306 # If we are rolling over on a certain day, add in the number of days until
307 # the next rollover, but offset by 1 since we just calculated the time
308 # until the next day starts. There are three cases:
309 # Case 1) The day to rollover is today; in this case, do nothing
310 # Case 2) The day to rollover is further in the interval (i.e., today is
311 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
312 # next rollover is simply 6 - 2 - 1, or 3.
313 # Case 3) The day to rollover is behind us in the interval (i.e., today
314 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
315 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
316 # number of days left in the current week (1) plus the number
317 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000318 # The calculations described in 2) and 3) above need to have a day added.
319 # This is because the above time calculation takes us to midnight on this
320 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000321 if self.when.startswith('W'):
Vinay Sajipa7130792013-04-12 17:04:23 +0100322 day = currentDay # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000323 if day != self.dayOfWeek:
324 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000325 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000326 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000327 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000328 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
329 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000330 dstNow = t[-1]
331 dstAtRollover = time.localtime(newRolloverAt)[-1]
332 if dstNow != dstAtRollover:
333 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000334 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000335 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000336 addend = 3600
337 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000338 result = newRolloverAt
339 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000340
341 def shouldRollover(self, record):
342 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000343 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000344
345 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000346 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000347 """
348 t = int(time.time())
349 if t >= self.rolloverAt:
350 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000351 return 0
352
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000353 def getFilesToDelete(self):
354 """
355 Determine the files to delete when rolling over.
356
357 More specific than the earlier method, which just used glob.glob().
358 """
359 dirName, baseName = os.path.split(self.baseFilename)
360 fileNames = os.listdir(dirName)
361 result = []
Miss Islington (bot)6e6dc252021-07-30 09:20:59 -0700362 # See bpo-44753: Don't use the extension when computing the prefix.
363 prefix = os.path.splitext(baseName)[0] + "."
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000364 plen = len(prefix)
365 for fileName in fileNames:
366 if fileName[:plen] == prefix:
367 suffix = fileName[plen:]
368 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000369 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000370 if len(result) < self.backupCount:
371 result = []
372 else:
Lovesh Harchandaniafad1472017-10-27 09:04:33 +0200373 result.sort()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000374 result = result[:len(result) - self.backupCount]
375 return result
376
Vinay Sajip17c52d82004-07-03 11:48:34 +0000377 def doRollover(self):
378 """
379 do a rollover; in this case, a date/time stamp is appended to the filename
380 when the rollover happens. However, you want the file to be named for the
381 start of the interval, not the current time. If there is a backup count,
382 then we have to get a list of matching filenames, sort them and remove
383 the one with the oldest suffix.
384 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000385 if self.stream:
386 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000387 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000388 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000389 currentTime = int(time.time())
390 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000391 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000392 if self.utc:
393 timeTuple = time.gmtime(t)
394 else:
395 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000396 dstThen = timeTuple[-1]
397 if dstNow != dstThen:
398 if dstNow:
399 addend = 3600
400 else:
401 addend = -3600
402 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000403 dfn = self.rotation_filename(self.baseFilename + "." +
404 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000405 if os.path.exists(dfn):
406 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000407 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000408 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000409 for s in self.getFilesToDelete():
410 os.remove(s)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100411 if not self.delay:
412 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000413 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000414 while newRolloverAt <= currentTime:
415 newRolloverAt = newRolloverAt + self.interval
416 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000417 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000418 dstAtRollover = time.localtime(newRolloverAt)[-1]
419 if dstNow != dstAtRollover:
420 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000421 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000422 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000423 addend = 3600
424 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000425 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000426
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000427class WatchedFileHandler(logging.FileHandler):
428 """
429 A handler for logging to a file, which watches the file
430 to see if it has changed while in use. This can happen because of
431 usage of programs such as newsyslog and logrotate which perform
432 log file rotation. This handler, intended for use under Unix,
433 watches the file to see if it has changed since the last emit.
434 (A file has changed if its device or inode have changed.)
435 If it has changed, the old file stream is closed, and the file
436 opened to get a new stream.
437
438 This handler is not appropriate for use under Windows, because
439 under Windows open files cannot be moved or renamed - logging
440 opens the files with exclusive locks - and so there is no need
441 for such a handler. Furthermore, ST_INO is not supported under
442 Windows; stat always returns zero for this value.
443
444 This handler is based on a suggestion and patch by Chad J.
445 Schroeder.
446 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100447 def __init__(self, filename, mode='a', encoding=None, delay=False,
448 errors=None):
Inada Naokifb786922021-04-06 11:18:41 +0900449 if "b" not in mode:
450 encoding = io.text_encoding(encoding)
Vinay Sajipca7b5042019-06-17 17:40:52 +0100451 logging.FileHandler.__init__(self, filename, mode=mode,
452 encoding=encoding, delay=delay,
453 errors=errors)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100454 self.dev, self.ino = -1, -1
455 self._statstream()
456
457 def _statstream(self):
458 if self.stream:
459 sres = os.fstat(self.stream.fileno())
460 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000461
Vinay Sajip29a14452015-10-01 20:54:41 +0100462 def reopenIfNeeded(self):
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000463 """
Vinay Sajip29a14452015-10-01 20:54:41 +0100464 Reopen log file if needed.
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000465
Vinay Sajip29a14452015-10-01 20:54:41 +0100466 Checks if the underlying file has changed, and if it
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000467 has, close the old stream and reopen the file to get the
468 current stream.
469 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100470 # Reduce the chance of race conditions by stat'ing by path only
471 # once and then fstat'ing our new fd if we opened a new log stream.
472 # See issue #14632: Thanks to John Mulligan for the problem report
473 # and patch.
474 try:
475 # stat the file by path, checking for existence
476 sres = os.stat(self.baseFilename)
Giampaolo Rodola'0166a282013-02-12 15:14:17 +0100477 except FileNotFoundError:
478 sres = None
Vinay Sajip66b8b082012-04-24 23:25:30 +0100479 # compare file system stat with that of our stream file handle
480 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
481 if self.stream is not None:
482 # we have an open file handle, clean it up
483 self.stream.flush()
484 self.stream.close()
Vinay Sajip02200482014-06-14 10:22:05 +0100485 self.stream = None # See Issue #21742: _open () might fail.
Vinay Sajip66b8b082012-04-24 23:25:30 +0100486 # open a new file handle and get new stat info from that fd
487 self.stream = self._open()
488 self._statstream()
Vinay Sajip29a14452015-10-01 20:54:41 +0100489
490 def emit(self, record):
491 """
492 Emit a record.
493
494 If underlying file has changed, reopen the file before emitting the
495 record to it.
496 """
497 self.reopenIfNeeded()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000498 logging.FileHandler.emit(self, record)
499
Vinay Sajip66b8b082012-04-24 23:25:30 +0100500
Guido van Rossum57102f82002-11-13 16:15:58 +0000501class SocketHandler(logging.Handler):
502 """
503 A handler class which writes logging records, in pickle format, to
504 a streaming socket. The socket is kept open across logging calls.
505 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000506 The pickle which is sent is that of the LogRecord's attribute dictionary
507 (__dict__), so that the receiver does not need to have the logging module
508 installed in order to process the logging event.
509
510 To unpickle the record at the receiving end into a LogRecord, use the
511 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000512 """
513
514 def __init__(self, host, port):
515 """
516 Initializes the handler with a specific host address and port.
517
Vinay Sajipde19e082011-04-30 21:52:26 +0100518 When the attribute *closeOnError* is set to True - if a socket error
519 occurs, the socket is silently closed and then reopened on the next
520 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000521 """
522 logging.Handler.__init__(self)
523 self.host = host
524 self.port = port
Vinay Sajip5421f352013-09-27 18:18:28 +0100525 if port is None:
526 self.address = host
527 else:
528 self.address = (host, port)
Guido van Rossum57102f82002-11-13 16:15:58 +0000529 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100530 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000531 self.retryTime = None
532 #
533 # Exponential backoff parameters.
534 #
535 self.retryStart = 1.0
536 self.retryMax = 30.0
537 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000538
Guido van Rossumd8faa362007-04-27 19:54:29 +0000539 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000540 """
541 A factory method which allows subclasses to define the precise
542 type of socket they want.
543 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100544 if self.port is not None:
545 result = socket.create_connection(self.address, timeout=timeout)
546 else:
547 result = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
548 result.settimeout(timeout)
Vinay Sajip38c741c2013-10-06 18:36:00 +0100549 try:
550 result.connect(self.address)
551 except OSError:
552 result.close() # Issue 19182
553 raise
Vinay Sajip5421f352013-09-27 18:18:28 +0100554 return result
Guido van Rossum57102f82002-11-13 16:15:58 +0000555
Vinay Sajip48cfe382004-02-20 13:17:27 +0000556 def createSocket(self):
557 """
558 Try to create a socket, using an exponential backoff with
559 a max retry time. Thanks to Robert Olson for the original patch
560 (SF #815911) which has been slightly refactored.
561 """
562 now = time.time()
563 # Either retryTime is None, in which case this
564 # is the first time back after a disconnect, or
565 # we've waited long enough.
566 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100567 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000568 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000569 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000570 if attempt:
571 try:
572 self.sock = self.makeSocket()
573 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200574 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000575 #Creation failed, so set the retry time and return.
576 if self.retryTime is None:
577 self.retryPeriod = self.retryStart
578 else:
579 self.retryPeriod = self.retryPeriod * self.retryFactor
580 if self.retryPeriod > self.retryMax:
581 self.retryPeriod = self.retryMax
582 self.retryTime = now + self.retryPeriod
583
Guido van Rossum57102f82002-11-13 16:15:58 +0000584 def send(self, s):
585 """
586 Send a pickled string to the socket.
587
588 This function allows for partial sends which can happen when the
589 network is busy.
590 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000591 if self.sock is None:
592 self.createSocket()
593 #self.sock can be None either because we haven't reached the retry
594 #time yet, or because we have reached the retry time and retried,
595 #but are still unable to connect.
596 if self.sock:
597 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200598 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200599 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000600 self.sock.close()
601 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000602
603 def makePickle(self, record):
604 """
605 Pickles the record in binary format with a length prefix, and
606 returns it ready for transmission across the socket.
607 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000608 ei = record.exc_info
609 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100610 # just to get traceback text into record.exc_text ...
611 dummy = self.format(record)
612 # See issue #14436: If msg or args are objects, they may not be
613 # available on the receiving end. So we convert the msg % args
614 # to a string, save it as msg and zap the args.
615 d = dict(record.__dict__)
616 d['msg'] = record.getMessage()
617 d['args'] = None
618 d['exc_info'] = None
Vinay Sajip9cdfd182015-12-26 12:48:44 +0000619 # Issue #25685: delete 'message' if present: redundant with 'msg'
620 d.pop('message', None)
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100621 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000622 slen = struct.pack(">L", len(s))
623 return slen + s
624
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000625 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000626 """
627 Handle an error during logging.
628
629 An error has occurred during logging. Most likely cause -
630 connection lost. Close the socket so that we can retry on the
631 next event.
632 """
633 if self.closeOnError and self.sock:
634 self.sock.close()
635 self.sock = None #try to reconnect next time
636 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000637 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000638
639 def emit(self, record):
640 """
641 Emit a record.
642
643 Pickles the record and writes it to the socket in binary format.
644 If there is an error with the socket, silently drop the packet.
645 If there was a problem with the socket, re-establishes the
646 socket.
647 """
648 try:
649 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000650 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100651 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000652 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000653
654 def close(self):
655 """
656 Closes the socket.
657 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000658 self.acquire()
659 try:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300660 sock = self.sock
661 if sock:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000662 self.sock = None
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300663 sock.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000664 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000665 finally:
666 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000667
668class DatagramHandler(SocketHandler):
669 """
670 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000671 a datagram socket. The pickle which is sent is that of the LogRecord's
672 attribute dictionary (__dict__), so that the receiver does not need to
673 have the logging module installed in order to process the logging event.
674
675 To unpickle the record at the receiving end into a LogRecord, use the
676 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000677
678 """
679 def __init__(self, host, port):
680 """
681 Initializes the handler with a specific host address and port.
682 """
683 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100684 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000685
686 def makeSocket(self):
687 """
688 The factory method of SocketHandler is here overridden to create
689 a UDP socket (SOCK_DGRAM).
690 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100691 if self.port is None:
692 family = socket.AF_UNIX
693 else:
694 family = socket.AF_INET
695 s = socket.socket(family, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000696 return s
697
698 def send(self, s):
699 """
700 Send a pickled string to a socket.
701
702 This function no longer allows for partial sends which can happen
703 when the network is busy - UDP does not guarantee delivery and
704 can deliver packets out of sequence.
705 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000706 if self.sock is None:
707 self.createSocket()
Vinay Sajip5421f352013-09-27 18:18:28 +0100708 self.sock.sendto(s, self.address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000709
710class SysLogHandler(logging.Handler):
711 """
712 A handler class which sends formatted logging records to a syslog
713 server. Based on Sam Rushing's syslog module:
714 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
715 Contributed by Nicolas Untz (after which minor refactoring changes
716 have been made).
717 """
718
719 # from <linux/sys/syslog.h>:
720 # ======================================================================
721 # priorities/facilities are encoded into a single 32-bit quantity, where
722 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
723 # facility (0-big number). Both the priorities and the facilities map
724 # roughly one-to-one to strings in the syslogd(8) source code. This
725 # mapping is included in this file.
726 #
727 # priorities (these are ordered)
728
729 LOG_EMERG = 0 # system is unusable
730 LOG_ALERT = 1 # action must be taken immediately
731 LOG_CRIT = 2 # critical conditions
732 LOG_ERR = 3 # error conditions
733 LOG_WARNING = 4 # warning conditions
734 LOG_NOTICE = 5 # normal but significant condition
735 LOG_INFO = 6 # informational
736 LOG_DEBUG = 7 # debug-level messages
737
738 # facility codes
739 LOG_KERN = 0 # kernel messages
740 LOG_USER = 1 # random user-level messages
741 LOG_MAIL = 2 # mail system
742 LOG_DAEMON = 3 # system daemons
743 LOG_AUTH = 4 # security/authorization messages
744 LOG_SYSLOG = 5 # messages generated internally by syslogd
745 LOG_LPR = 6 # line printer subsystem
746 LOG_NEWS = 7 # network news subsystem
747 LOG_UUCP = 8 # UUCP subsystem
748 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000749 LOG_AUTHPRIV = 10 # security/authorization messages (private)
750 LOG_FTP = 11 # FTP daemon
Vinay Sajipce545192020-01-10 19:37:48 +0000751 LOG_NTP = 12 # NTP subsystem
752 LOG_SECURITY = 13 # Log audit
753 LOG_CONSOLE = 14 # Log alert
754 LOG_SOLCRON = 15 # Scheduling daemon (Solaris)
Guido van Rossum57102f82002-11-13 16:15:58 +0000755
756 # other codes through 15 reserved for system use
757 LOG_LOCAL0 = 16 # reserved for local use
758 LOG_LOCAL1 = 17 # reserved for local use
759 LOG_LOCAL2 = 18 # reserved for local use
760 LOG_LOCAL3 = 19 # reserved for local use
761 LOG_LOCAL4 = 20 # reserved for local use
762 LOG_LOCAL5 = 21 # reserved for local use
763 LOG_LOCAL6 = 22 # reserved for local use
764 LOG_LOCAL7 = 23 # reserved for local use
765
766 priority_names = {
767 "alert": LOG_ALERT,
768 "crit": LOG_CRIT,
769 "critical": LOG_CRIT,
770 "debug": LOG_DEBUG,
771 "emerg": LOG_EMERG,
772 "err": LOG_ERR,
773 "error": LOG_ERR, # DEPRECATED
774 "info": LOG_INFO,
775 "notice": LOG_NOTICE,
776 "panic": LOG_EMERG, # DEPRECATED
777 "warn": LOG_WARNING, # DEPRECATED
778 "warning": LOG_WARNING,
779 }
780
781 facility_names = {
Vinay Sajipce545192020-01-10 19:37:48 +0000782 "auth": LOG_AUTH,
783 "authpriv": LOG_AUTHPRIV,
784 "console": LOG_CONSOLE,
785 "cron": LOG_CRON,
786 "daemon": LOG_DAEMON,
787 "ftp": LOG_FTP,
788 "kern": LOG_KERN,
789 "lpr": LOG_LPR,
790 "mail": LOG_MAIL,
791 "news": LOG_NEWS,
792 "ntp": LOG_NTP,
793 "security": LOG_SECURITY,
794 "solaris-cron": LOG_SOLCRON,
795 "syslog": LOG_SYSLOG,
796 "user": LOG_USER,
797 "uucp": LOG_UUCP,
798 "local0": LOG_LOCAL0,
799 "local1": LOG_LOCAL1,
800 "local2": LOG_LOCAL2,
801 "local3": LOG_LOCAL3,
802 "local4": LOG_LOCAL4,
803 "local5": LOG_LOCAL5,
804 "local6": LOG_LOCAL6,
805 "local7": LOG_LOCAL7,
Guido van Rossum57102f82002-11-13 16:15:58 +0000806 }
807
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000808 #The map below appears to be trivially lowercasing the key. However,
809 #there's more to it than meets the eye - in some locales, lowercasing
810 #gives unexpected results. See SF #1524081: in the Turkish locale,
811 #"INFO".lower() != "info"
812 priority_map = {
813 "DEBUG" : "debug",
814 "INFO" : "info",
815 "WARNING" : "warning",
816 "ERROR" : "error",
817 "CRITICAL" : "critical"
818 }
819
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000820 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100821 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000822 """
823 Initialize a handler.
824
Guido van Rossume7ba4952007-06-06 23:52:48 +0000825 If address is specified as a string, a UNIX socket is used. To log to a
826 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100827 If facility is not specified, LOG_USER is used. If socktype is
828 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
829 socket type will be used. For Unix sockets, you can also specify a
830 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
831 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000832 """
833 logging.Handler.__init__(self)
834
835 self.address = address
836 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000837 self.socktype = socktype
838
Guido van Rossum13257902007-06-07 23:15:56 +0000839 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100840 self.unixsocket = True
Коренберг Марк1b038e02017-03-17 20:25:05 +0500841 # Syslog server may be unavailable during handler initialisation.
842 # C's openlog() function also ignores connection errors.
843 # Moreover, we ignore these errors while logging, so it not worse
844 # to ignore it also here.
845 try:
846 self._connect_unixsocket(address)
847 except OSError:
848 pass
Guido van Rossum57102f82002-11-13 16:15:58 +0000849 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100850 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100851 if socktype is None:
852 socktype = socket.SOCK_DGRAM
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800853 host, port = address
854 ress = socket.getaddrinfo(host, port, 0, socktype)
855 if not ress:
856 raise OSError("getaddrinfo returns an empty list")
857 for res in ress:
858 af, socktype, proto, _, sa = res
859 err = sock = None
860 try:
861 sock = socket.socket(af, socktype, proto)
862 if socktype == socket.SOCK_STREAM:
863 sock.connect(sa)
864 break
865 except OSError as exc:
866 err = exc
867 if sock is not None:
868 sock.close()
869 if err is not None:
870 raise err
871 self.socket = sock
Vinay Sajipe9170522013-04-22 10:07:40 +0100872 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000873
Vinay Sajipa1974c12005-01-13 08:23:56 +0000874 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100875 use_socktype = self.socktype
876 if use_socktype is None:
877 use_socktype = socket.SOCK_DGRAM
878 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000879 try:
880 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100881 # it worked, so set self.socktype to the used type
882 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200883 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000884 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100885 if self.socktype is not None:
886 # user didn't specify falling back, so fail
887 raise
888 use_socktype = socket.SOCK_STREAM
889 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
890 try:
891 self.socket.connect(address)
892 # it worked, so set self.socktype to the used type
893 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100894 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100895 self.socket.close()
896 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000897
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000898 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000899 """
900 Encode the facility and priority. You can pass in strings or
901 integers - if strings are passed, the facility_names and
902 priority_names mapping dictionaries are used to convert them to
903 integers.
904 """
Guido van Rossum13257902007-06-07 23:15:56 +0000905 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000906 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000907 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000908 priority = self.priority_names[priority]
909 return (facility << 3) | priority
910
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800911 def close(self):
Guido van Rossum57102f82002-11-13 16:15:58 +0000912 """
913 Closes the socket.
914 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000915 self.acquire()
916 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000917 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000918 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000919 finally:
920 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000921
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000922 def mapPriority(self, levelName):
923 """
924 Map a logging level name to a key in the priority_names map.
925 This is useful in two scenarios: when custom levels are being
926 used, and in the case where you can't do a straightforward
927 mapping by lowercasing the logging level name because of locale-
928 specific issues (see SF #1524081).
929 """
930 return self.priority_map.get(levelName, "warning")
931
Vinay Sajip2353e352011-06-27 15:40:06 +0100932 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100933 append_nul = True # some old syslog daemons expect a NUL terminator
934
Guido van Rossum57102f82002-11-13 16:15:58 +0000935 def emit(self, record):
936 """
937 Emit a record.
938
939 The record is formatted, and then sent to the syslog server. If
940 exception information is present, it is NOT sent to the server.
941 """
Guido van Rossum57102f82002-11-13 16:15:58 +0000942 try:
Vinay Sajipc33a0cc2014-11-01 19:58:47 +0000943 msg = self.format(record)
944 if self.ident:
945 msg = self.ident + msg
946 if self.append_nul:
947 msg += '\000'
948
949 # We need to convert record level to lowercase, maybe this will
950 # change in the future.
951 prio = '<%d>' % self.encodePriority(self.facility,
952 self.mapPriority(record.levelname))
953 prio = prio.encode('utf-8')
954 # Message is a string. Convert to bytes as required by RFC 5424
955 msg = msg.encode('utf-8')
956 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000957 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000958 try:
959 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200960 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100961 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000962 self._connect_unixsocket(self.address)
963 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000964 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000965 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000966 else:
967 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100968 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000969 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000970
971class SMTPHandler(logging.Handler):
972 """
973 A handler class which sends an SMTP email for each logging event.
974 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000975 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100976 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000977 """
978 Initialize the handler.
979
980 Initialize the instance with the from and to addresses and subject
981 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000982 (host, port) tuple format for the mailhost argument. To specify
983 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000984 for the credentials argument. To specify the use of a secure
985 protocol (TLS), pass in a tuple for the secure argument. This will
986 only be used when authentication credentials are supplied. The tuple
987 will be either an empty tuple, or a single-value tuple with the name
988 of a keyfile, or a 2-value tuple with the names of the keyfile and
989 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000990 A timeout in seconds can be specified for the SMTP connection (the
991 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000992 """
993 logging.Handler.__init__(self)
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100994 if isinstance(mailhost, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000995 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000996 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000997 self.mailhost, self.mailport = mailhost, None
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100998 if isinstance(credentials, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000999 self.username, self.password = credentials
1000 else:
1001 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +00001002 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +00001003 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001004 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +00001005 self.toaddrs = toaddrs
1006 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +00001007 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +00001008 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +00001009
1010 def getSubject(self, record):
1011 """
1012 Determine the subject for the email.
1013
1014 If you want to specify a subject line which is record-dependent,
1015 override this method.
1016 """
1017 return self.subject
1018
1019 def emit(self, record):
1020 """
1021 Emit a record.
1022
1023 Format the record and send it to the specified addressees.
1024 """
1025 try:
1026 import smtplib
Vinay Sajip277640a2015-10-17 16:13:10 +01001027 from email.message import EmailMessage
1028 import email.utils
1029
Guido van Rossum57102f82002-11-13 16:15:58 +00001030 port = self.mailport
1031 if not port:
1032 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +00001033 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Vinay Sajip277640a2015-10-17 16:13:10 +01001034 msg = EmailMessage()
1035 msg['From'] = self.fromaddr
1036 msg['To'] = ','.join(self.toaddrs)
1037 msg['Subject'] = self.getSubject(record)
1038 msg['Date'] = email.utils.localtime()
1039 msg.set_content(self.format(record))
Guido van Rossum360e4b82007-05-14 22:51:27 +00001040 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +00001041 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +00001042 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +00001043 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +00001044 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +00001045 smtp.login(self.username, self.password)
Vinay Sajip277640a2015-10-17 16:13:10 +01001046 smtp.send_message(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +00001047 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001048 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001049 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001050
1051class NTEventLogHandler(logging.Handler):
1052 """
1053 A handler class which sends events to the NT Event Log. Adds a
1054 registry entry for the specified application name. If no dllname is
1055 provided, win32service.pyd (which contains some basic message
1056 placeholders) is used. Note that use of these placeholders will make
1057 your event logs big, as the entire message source is held in the log.
1058 If you want slimmer logs, you have to pass in the name of your own DLL
1059 which contains the message definitions you want to use in the event log.
1060 """
1061 def __init__(self, appname, dllname=None, logtype="Application"):
1062 logging.Handler.__init__(self)
1063 try:
1064 import win32evtlogutil, win32evtlog
1065 self.appname = appname
1066 self._welu = win32evtlogutil
1067 if not dllname:
1068 dllname = os.path.split(self._welu.__file__)
1069 dllname = os.path.split(dllname[0])
1070 dllname = os.path.join(dllname[0], r'win32service.pyd')
1071 self.dllname = dllname
1072 self.logtype = logtype
1073 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1074 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1075 self.typemap = {
1076 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1077 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001078 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001079 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1080 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1081 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001082 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001083 print("The Python Win32 extensions for NT (service, event "\
1084 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001085 self._welu = None
1086
1087 def getMessageID(self, record):
1088 """
1089 Return the message ID for the event record. If you are using your
1090 own messages, you could do this by having the msg passed to the
1091 logger being an ID rather than a formatting string. Then, in here,
1092 you could use a dictionary lookup to get the message ID. This
1093 version returns 1, which is the base message ID in win32service.pyd.
1094 """
1095 return 1
1096
1097 def getEventCategory(self, record):
1098 """
1099 Return the event category for the record.
1100
1101 Override this if you want to specify your own categories. This version
1102 returns 0.
1103 """
1104 return 0
1105
1106 def getEventType(self, record):
1107 """
1108 Return the event type for the record.
1109
1110 Override this if you want to specify your own types. This version does
1111 a mapping using the handler's typemap attribute, which is set up in
1112 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001113 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001114 either need to override this method or place a suitable dictionary in
1115 the handler's typemap attribute.
1116 """
1117 return self.typemap.get(record.levelno, self.deftype)
1118
1119 def emit(self, record):
1120 """
1121 Emit a record.
1122
1123 Determine the message ID, event category and event type. Then
1124 log the message in the NT event log.
1125 """
1126 if self._welu:
1127 try:
1128 id = self.getMessageID(record)
1129 cat = self.getEventCategory(record)
1130 type = self.getEventType(record)
1131 msg = self.format(record)
1132 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001133 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001134 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001135
1136 def close(self):
1137 """
1138 Clean up this handler.
1139
1140 You can remove the application name from the registry as a
1141 source of event log entries. However, if you do this, you will
1142 not be able to see the events as you intended in the Event Log
1143 Viewer - it needs to be able to access the registry to get the
1144 DLL name.
1145 """
1146 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001147 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001148
1149class HTTPHandler(logging.Handler):
1150 """
Miss Islington (bot)6fc1efa2021-07-26 15:34:32 -07001151 A class which sends records to a web server, using either GET or
Guido van Rossum57102f82002-11-13 16:15:58 +00001152 POST semantics.
1153 """
Benjamin Peterson43052a12014-11-23 20:36:44 -06001154 def __init__(self, host, url, method="GET", secure=False, credentials=None,
1155 context=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001156 """
1157 Initialize the instance with the host, the request URL, and the method
1158 ("GET" or "POST")
1159 """
1160 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001161 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001162 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001163 raise ValueError("method must be GET or POST")
Benjamin Peterson43052a12014-11-23 20:36:44 -06001164 if not secure and context is not None:
1165 raise ValueError("context parameter only makes sense "
1166 "with secure=True")
Guido van Rossum57102f82002-11-13 16:15:58 +00001167 self.host = host
1168 self.url = url
1169 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001170 self.secure = secure
1171 self.credentials = credentials
Benjamin Peterson43052a12014-11-23 20:36:44 -06001172 self.context = context
Guido van Rossum57102f82002-11-13 16:15:58 +00001173
Neal Norwitzf297bd12003-04-23 03:49:43 +00001174 def mapLogRecord(self, record):
1175 """
1176 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001177 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001178 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001179 """
1180 return record.__dict__
1181
l0rb22a9a542020-03-04 11:49:51 +01001182 def getConnection(self, host, secure):
1183 """
1184 get a HTTP[S]Connection.
1185
1186 Override when a custom connection is required, for example if
1187 there is a proxy.
1188 """
1189 import http.client
1190 if secure:
1191 connection = http.client.HTTPSConnection(host, context=self.context)
1192 else:
1193 connection = http.client.HTTPConnection(host)
1194 return connection
1195
Guido van Rossum57102f82002-11-13 16:15:58 +00001196 def emit(self, record):
1197 """
1198 Emit a record.
1199
Miss Islington (bot)6fc1efa2021-07-26 15:34:32 -07001200 Send the record to the web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001201 """
1202 try:
l0rb22a9a542020-03-04 11:49:51 +01001203 import urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001204 host = self.host
l0rb22a9a542020-03-04 11:49:51 +01001205 h = self.getConnection(host, self.secure)
Guido van Rossum57102f82002-11-13 16:15:58 +00001206 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001207 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001208 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001209 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001210 sep = '&'
1211 else:
1212 sep = '?'
1213 url = url + "%c%s" % (sep, data)
1214 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001215 # support multiple hosts on one IP address...
1216 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001217 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001218 if i >= 0:
1219 host = host[:i]
Vinay Sajipe96ba182017-11-19 18:36:17 +00001220 # See issue #30904: putrequest call above already adds this header
1221 # on Python 3.x.
1222 # h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001223 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001224 h.putheader("Content-type",
1225 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001226 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001227 if self.credentials:
1228 import base64
Vinay Sajip1bf197e2016-06-07 21:19:55 +01001229 s = ('%s:%s' % self.credentials).encode('utf-8')
1230 s = 'Basic ' + base64.b64encode(s).strip().decode('ascii')
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001231 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001232 h.endheaders()
1233 if self.method == "POST":
1234 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001235 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001236 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001237 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001238
1239class BufferingHandler(logging.Handler):
1240 """
1241 A handler class which buffers logging records in memory. Whenever each
1242 record is added to the buffer, a check is made to see if the buffer should
1243 be flushed. If it should, then flush() is expected to do what's needed.
1244 """
1245 def __init__(self, capacity):
1246 """
1247 Initialize the handler with the buffer size.
1248 """
1249 logging.Handler.__init__(self)
1250 self.capacity = capacity
1251 self.buffer = []
1252
1253 def shouldFlush(self, record):
1254 """
1255 Should the handler flush its buffer?
1256
1257 Returns true if the buffer is up to capacity. This method can be
1258 overridden to implement custom flushing strategies.
1259 """
1260 return (len(self.buffer) >= self.capacity)
1261
1262 def emit(self, record):
1263 """
1264 Emit a record.
1265
1266 Append the record. If shouldFlush() tells us to, call flush() to process
1267 the buffer.
1268 """
1269 self.buffer.append(record)
1270 if self.shouldFlush(record):
1271 self.flush()
1272
1273 def flush(self):
1274 """
1275 Override to implement custom flushing behaviour.
1276
1277 This version just zaps the buffer to empty.
1278 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001279 self.acquire()
1280 try:
Daniel Anderssond89cea12019-11-13 10:03:45 +01001281 self.buffer.clear()
Vinay Sajipf0509032012-02-23 20:49:08 +00001282 finally:
1283 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001284
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001285 def close(self):
1286 """
1287 Close the handler.
1288
1289 This version just flushes and chains to the parent class' close().
1290 """
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001291 try:
1292 self.flush()
1293 finally:
1294 logging.Handler.close(self)
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001295
Guido van Rossum57102f82002-11-13 16:15:58 +00001296class MemoryHandler(BufferingHandler):
1297 """
1298 A handler class which buffers logging records in memory, periodically
1299 flushing them to a target handler. Flushing occurs whenever the buffer
1300 is full, or when an event of a certain severity or greater is seen.
1301 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001302 def __init__(self, capacity, flushLevel=logging.ERROR, target=None,
1303 flushOnClose=True):
Guido van Rossum57102f82002-11-13 16:15:58 +00001304 """
1305 Initialize the handler with the buffer size, the level at which
1306 flushing should occur and an optional target.
1307
1308 Note that without a target being set either here or via setTarget(),
1309 a MemoryHandler is no use to anyone!
Vinay Sajipcccf6062016-07-22 16:27:31 +01001310
1311 The ``flushOnClose`` argument is ``True`` for backward compatibility
1312 reasons - the old behaviour is that when the handler is closed, the
1313 buffer is flushed, even if the flush level hasn't been exceeded nor the
1314 capacity exceeded. To prevent this, set ``flushOnClose`` to ``False``.
Guido van Rossum57102f82002-11-13 16:15:58 +00001315 """
1316 BufferingHandler.__init__(self, capacity)
1317 self.flushLevel = flushLevel
1318 self.target = target
Vinay Sajipcccf6062016-07-22 16:27:31 +01001319 # See Issue #26559 for why this has been added
1320 self.flushOnClose = flushOnClose
Guido van Rossum57102f82002-11-13 16:15:58 +00001321
1322 def shouldFlush(self, record):
1323 """
1324 Check for buffer full or a record at the flushLevel or higher.
1325 """
1326 return (len(self.buffer) >= self.capacity) or \
1327 (record.levelno >= self.flushLevel)
1328
1329 def setTarget(self, target):
1330 """
1331 Set the target handler for this handler.
1332 """
Irit Katriel2353d772020-08-16 16:10:13 +01001333 self.acquire()
1334 try:
1335 self.target = target
1336 finally:
1337 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001338
1339 def flush(self):
1340 """
1341 For a MemoryHandler, flushing means just sending the buffered
1342 records to the target, if there is one. Override if you want
1343 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001344
1345 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001346 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001347 self.acquire()
1348 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001349 if self.target:
1350 for record in self.buffer:
1351 self.target.handle(record)
Daniel Anderssond89cea12019-11-13 10:03:45 +01001352 self.buffer.clear()
Vinay Sajipf0509032012-02-23 20:49:08 +00001353 finally:
1354 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001355
1356 def close(self):
1357 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001358 Flush, if appropriately configured, set the target to None and lose the
1359 buffer.
Guido van Rossum57102f82002-11-13 16:15:58 +00001360 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001361 try:
Vinay Sajipcccf6062016-07-22 16:27:31 +01001362 if self.flushOnClose:
1363 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001364 finally:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001365 self.acquire()
1366 try:
1367 self.target = None
1368 BufferingHandler.close(self)
1369 finally:
1370 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001371
1372
1373class QueueHandler(logging.Handler):
1374 """
1375 This handler sends events to a queue. Typically, it would be used together
1376 with a multiprocessing Queue to centralise logging to file in one process
1377 (in a multi-process application), so as to avoid file write contention
1378 between processes.
1379
1380 This code is new in Python 3.2, but this class can be copy pasted into
1381 user code for use with earlier Python versions.
1382 """
1383
1384 def __init__(self, queue):
1385 """
1386 Initialise an instance, using the passed queue.
1387 """
1388 logging.Handler.__init__(self)
1389 self.queue = queue
1390
1391 def enqueue(self, record):
1392 """
1393 Enqueue a record.
1394
1395 The base implementation uses put_nowait. You may want to override
1396 this method if you want to use blocking, timeouts or custom queue
1397 implementations.
1398 """
1399 self.queue.put_nowait(record)
1400
Vinay Sajip0258ce82010-09-22 20:34:53 +00001401 def prepare(self, record):
1402 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001403 Prepares a record for queuing. The object returned by this method is
1404 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001405
1406 The base implementation formats the record to merge the message
1407 and arguments, and removes unpickleable items from the record
1408 in-place.
1409
1410 You might want to override this method if you want to convert
1411 the record to a dict or JSON string, or send a modified copy
1412 of the record while leaving the original intact.
1413 """
1414 # The format operation gets traceback text into record.exc_text
favlladfe3442017-08-01 20:12:26 +02001415 # (if there's exception data), and also returns the formatted
1416 # message. We can then use this to replace the original
Vinay Sajip0258ce82010-09-22 20:34:53 +00001417 # msg + args, as these might be unpickleable. We also zap the
Cheryl Sabellad345bb42018-09-25 19:00:08 -04001418 # exc_info and exc_text attributes, as they are no longer
1419 # needed and, if not None, will typically not be pickleable.
favlladfe3442017-08-01 20:12:26 +02001420 msg = self.format(record)
Manjusakada6424e2019-01-23 15:08:38 +08001421 # bpo-35726: make copy of record to avoid affecting other handlers in the chain.
1422 record = copy.copy(record)
favlladfe3442017-08-01 20:12:26 +02001423 record.message = msg
1424 record.msg = msg
Vinay Sajip0258ce82010-09-22 20:34:53 +00001425 record.args = None
1426 record.exc_info = None
Cheryl Sabellad345bb42018-09-25 19:00:08 -04001427 record.exc_text = None
Vinay Sajip0258ce82010-09-22 20:34:53 +00001428 return record
1429
Vinay Sajip121a1c42010-09-08 10:46:15 +00001430 def emit(self, record):
1431 """
1432 Emit a record.
1433
Vinay Sajip0637d492010-09-23 08:15:54 +00001434 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001435 """
1436 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001437 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001438 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001439 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001440
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001441
1442class QueueListener(object):
1443 """
1444 This class implements an internal threaded listener which watches for
1445 LogRecords being added to a queue, removes them and passes them to a
1446 list of handlers for processing.
1447 """
1448 _sentinel = None
1449
1450 def __init__(self, queue, *handlers, respect_handler_level=False):
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001451 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001452 Initialise an instance with the specified queue and
1453 handlers.
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001454 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001455 self.queue = queue
1456 self.handlers = handlers
1457 self._thread = None
1458 self.respect_handler_level = respect_handler_level
Vinay Sajip0637d492010-09-23 08:15:54 +00001459
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001460 def dequeue(self, block):
1461 """
1462 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001463
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001464 The base implementation uses get. You may want to override this method
1465 if you want to use timeouts or work with custom queue implementations.
1466 """
1467 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001468
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001469 def start(self):
1470 """
1471 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001472
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001473 This starts up a background thread to monitor the queue for
1474 LogRecords to process.
1475 """
1476 self._thread = t = threading.Thread(target=self._monitor)
1477 t.daemon = True
1478 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001479
Boris Feld800d7862019-06-04 17:20:18 +02001480 def prepare(self, record):
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001481 """
1482 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001483
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001484 This method just returns the passed-in record. You may want to
1485 override this method if you need to do any custom marshalling or
1486 manipulation of the record before passing it to the handlers.
1487 """
1488 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001489
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001490 def handle(self, record):
1491 """
1492 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001493
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001494 This just loops through the handlers offering them the record
1495 to handle.
1496 """
1497 record = self.prepare(record)
1498 for handler in self.handlers:
1499 if not self.respect_handler_level:
1500 process = True
1501 else:
1502 process = record.levelno >= handler.level
1503 if process:
1504 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001505
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001506 def _monitor(self):
1507 """
1508 Monitor the queue for records, and ask the handler
1509 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001510
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001511 This method runs on a separate, internal thread.
1512 The thread will terminate if it sees a sentinel object in the queue.
1513 """
1514 q = self.queue
1515 has_task_done = hasattr(q, 'task_done')
1516 while True:
1517 try:
1518 record = self.dequeue(True)
1519 if record is self._sentinel:
Bar Harel6b282e12019-06-01 12:19:09 +03001520 if has_task_done:
1521 q.task_done()
Vinay Sajip0637d492010-09-23 08:15:54 +00001522 break
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001523 self.handle(record)
1524 if has_task_done:
1525 q.task_done()
1526 except queue.Empty:
1527 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001528
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001529 def enqueue_sentinel(self):
1530 """
1531 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001532
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001533 The base implementation uses put_nowait. You may want to override this
1534 method if you want to use timeouts or work with custom queue
1535 implementations.
1536 """
1537 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001538
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001539 def stop(self):
1540 """
1541 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001542
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001543 This asks the thread to terminate, and then waits for it to do so.
1544 Note that if you don't call this before your application exits, there
1545 may be some records still left on the queue, which won't be processed.
1546 """
1547 self.enqueue_sentinel()
1548 self._thread.join()
1549 self._thread = None