blob: 4e8f0a8cc2f469fd97810ac7343f3e1bcf98eec7 [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 """
Miss Islington (bot)5aca34f2021-10-11 03:01:12 -0700190 # See bpo-45401: Never rollover anything other than regular files
191 if os.path.exists(self.baseFilename) and not os.path.isfile(self.baseFilename):
192 return False
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000193 if self.stream is None: # delay was set...
194 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000195 if self.maxBytes > 0: # are we rolling over?
196 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000197 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000198 if self.stream.tell() + len(msg) >= self.maxBytes:
Miss Islington (bot)5aca34f2021-10-11 03:01:12 -0700199 return True
200 return False
Guido van Rossum57102f82002-11-13 16:15:58 +0000201
Vinay Sajip17c52d82004-07-03 11:48:34 +0000202class TimedRotatingFileHandler(BaseRotatingHandler):
203 """
204 Handler for logging to a file, rotating the log file at certain timed
205 intervals.
206
207 If backupCount is > 0, when rollover is done, no more than backupCount
208 files are kept - the oldest ones are deleted.
209 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100210 def __init__(self, filename, when='h', interval=1, backupCount=0,
211 encoding=None, delay=False, utc=False, atTime=None,
212 errors=None):
Inada Naokifb786922021-04-06 11:18:41 +0900213 encoding = io.text_encoding(encoding)
Vinay Sajipca7b5042019-06-17 17:40:52 +0100214 BaseRotatingHandler.__init__(self, filename, 'a', encoding=encoding,
215 delay=delay, errors=errors)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000216 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000217 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000218 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100219 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000220 # Calculate the real rollover interval, which is just the number of
221 # seconds between rollovers. Also set the filename suffix used when
222 # a rollover occurs. Current 'when' events supported:
223 # S - Seconds
224 # M - Minutes
225 # H - Hours
226 # D - Days
227 # midnight - roll over at midnight
228 # W{0-6} - roll over on a certain day; 0 - Monday
229 #
230 # Case of the 'when' specifier is not important; lower or upper case
231 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000232 if self.when == 'S':
233 self.interval = 1 # one second
234 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000235 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000236 elif self.when == 'M':
237 self.interval = 60 # one minute
238 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000239 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000240 elif self.when == 'H':
241 self.interval = 60 * 60 # one hour
242 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000243 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000244 elif self.when == 'D' or self.when == 'MIDNIGHT':
245 self.interval = 60 * 60 * 24 # one day
246 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000247 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000248 elif self.when.startswith('W'):
249 self.interval = 60 * 60 * 24 * 7 # one week
250 if len(self.when) != 2:
251 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
252 if self.when[1] < '0' or self.when[1] > '6':
253 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
254 self.dayOfWeek = int(self.when[1])
255 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000256 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000257 else:
258 raise ValueError("Invalid rollover interval specified: %s" % self.when)
259
Antoine Pitroufd036452008-08-19 17:56:33 +0000260 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000261 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip638e6222016-07-22 18:23:04 +0100262 # The following line added because the filename passed in could be a
263 # path object (see Issue #27493), but self.baseFilename will be a string
264 filename = self.baseFilename
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000265 if os.path.exists(filename):
266 t = os.stat(filename)[ST_MTIME]
267 else:
268 t = int(time.time())
269 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000270
Vinay Sajipaffbd872009-06-11 10:11:47 +0000271 def computeRollover(self, currentTime):
272 """
273 Work out the rollover time based on the specified time.
274 """
275 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000276 # If we are rolling over at midnight or weekly, then the interval is already known.
277 # What we need to figure out is WHEN the next interval is. In other words,
278 # if you are rolling over at midnight, then your base interval is 1 day,
279 # but you want to start that one day clock at midnight, not now. So, we
280 # have to fudge the rolloverAt value in order to trigger the first rollover
281 # at the right time. After that, the regular interval will take care of
282 # the rest. Note that this code doesn't care about leap seconds. :)
283 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
284 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000285 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000286 t = time.gmtime(currentTime)
287 else:
288 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000289 currentHour = t[3]
290 currentMinute = t[4]
291 currentSecond = t[5]
Vinay Sajipa7130792013-04-12 17:04:23 +0100292 currentDay = t[6]
293 # r is the number of seconds left between now and the next rotation
294 if self.atTime is None:
295 rotate_ts = _MIDNIGHT
296 else:
297 rotate_ts = ((self.atTime.hour * 60 + self.atTime.minute)*60 +
298 self.atTime.second)
299
300 r = rotate_ts - ((currentHour * 60 + currentMinute) * 60 +
301 currentSecond)
302 if r < 0:
303 # Rotate time is before the current time (for example when
304 # self.rotateAt is 13:45 and it now 14:15), rotation is
305 # tomorrow.
306 r += _MIDNIGHT
307 currentDay = (currentDay + 1) % 7
Vinay Sajipaffbd872009-06-11 10:11:47 +0000308 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000309 # If we are rolling over on a certain day, add in the number of days until
310 # the next rollover, but offset by 1 since we just calculated the time
311 # until the next day starts. There are three cases:
312 # Case 1) The day to rollover is today; in this case, do nothing
313 # Case 2) The day to rollover is further in the interval (i.e., today is
314 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
315 # next rollover is simply 6 - 2 - 1, or 3.
316 # Case 3) The day to rollover is behind us in the interval (i.e., today
317 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
318 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
319 # number of days left in the current week (1) plus the number
320 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000321 # The calculations described in 2) and 3) above need to have a day added.
322 # This is because the above time calculation takes us to midnight on this
323 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000324 if self.when.startswith('W'):
Vinay Sajipa7130792013-04-12 17:04:23 +0100325 day = currentDay # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000326 if day != self.dayOfWeek:
327 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000328 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000329 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000330 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000331 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
332 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000333 dstNow = t[-1]
334 dstAtRollover = time.localtime(newRolloverAt)[-1]
335 if dstNow != dstAtRollover:
336 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000337 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000338 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000339 addend = 3600
340 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000341 result = newRolloverAt
342 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000343
344 def shouldRollover(self, record):
345 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000346 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000347
348 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000349 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000350 """
Miss Islington (bot)5aca34f2021-10-11 03:01:12 -0700351 # See bpo-45401: Never rollover anything other than regular files
352 if os.path.exists(self.baseFilename) and not os.path.isfile(self.baseFilename):
353 return False
Vinay Sajip17c52d82004-07-03 11:48:34 +0000354 t = int(time.time())
355 if t >= self.rolloverAt:
Miss Islington (bot)5aca34f2021-10-11 03:01:12 -0700356 return True
357 return False
Vinay Sajip17c52d82004-07-03 11:48:34 +0000358
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000359 def getFilesToDelete(self):
360 """
361 Determine the files to delete when rolling over.
362
363 More specific than the earlier method, which just used glob.glob().
364 """
365 dirName, baseName = os.path.split(self.baseFilename)
366 fileNames = os.listdir(dirName)
367 result = []
Miss Islington (bot)6e6dc252021-07-30 09:20:59 -0700368 # See bpo-44753: Don't use the extension when computing the prefix.
369 prefix = os.path.splitext(baseName)[0] + "."
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000370 plen = len(prefix)
371 for fileName in fileNames:
372 if fileName[:plen] == prefix:
373 suffix = fileName[plen:]
Miss Islington (bot)191a9392021-10-29 08:25:31 -0700374 # See bpo-45628: The date/time suffix could be anywhere in the
375 # filename
376 parts = suffix.split('.')
377 for part in parts:
378 if self.extMatch.match(part):
379 result.append(os.path.join(dirName, fileName))
380 break
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000381 if len(result) < self.backupCount:
382 result = []
383 else:
Lovesh Harchandaniafad1472017-10-27 09:04:33 +0200384 result.sort()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000385 result = result[:len(result) - self.backupCount]
386 return result
387
Vinay Sajip17c52d82004-07-03 11:48:34 +0000388 def doRollover(self):
389 """
390 do a rollover; in this case, a date/time stamp is appended to the filename
391 when the rollover happens. However, you want the file to be named for the
392 start of the interval, not the current time. If there is a backup count,
393 then we have to get a list of matching filenames, sort them and remove
394 the one with the oldest suffix.
395 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000396 if self.stream:
397 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000398 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000399 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000400 currentTime = int(time.time())
401 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000402 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000403 if self.utc:
404 timeTuple = time.gmtime(t)
405 else:
406 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000407 dstThen = timeTuple[-1]
408 if dstNow != dstThen:
409 if dstNow:
410 addend = 3600
411 else:
412 addend = -3600
413 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000414 dfn = self.rotation_filename(self.baseFilename + "." +
415 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000416 if os.path.exists(dfn):
417 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000418 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000419 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000420 for s in self.getFilesToDelete():
421 os.remove(s)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100422 if not self.delay:
423 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000424 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000425 while newRolloverAt <= currentTime:
426 newRolloverAt = newRolloverAt + self.interval
427 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000428 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000429 dstAtRollover = time.localtime(newRolloverAt)[-1]
430 if dstNow != dstAtRollover:
431 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000432 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000433 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000434 addend = 3600
435 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000436 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000437
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000438class WatchedFileHandler(logging.FileHandler):
439 """
440 A handler for logging to a file, which watches the file
441 to see if it has changed while in use. This can happen because of
442 usage of programs such as newsyslog and logrotate which perform
443 log file rotation. This handler, intended for use under Unix,
444 watches the file to see if it has changed since the last emit.
445 (A file has changed if its device or inode have changed.)
446 If it has changed, the old file stream is closed, and the file
447 opened to get a new stream.
448
449 This handler is not appropriate for use under Windows, because
450 under Windows open files cannot be moved or renamed - logging
451 opens the files with exclusive locks - and so there is no need
452 for such a handler. Furthermore, ST_INO is not supported under
453 Windows; stat always returns zero for this value.
454
455 This handler is based on a suggestion and patch by Chad J.
456 Schroeder.
457 """
Vinay Sajipca7b5042019-06-17 17:40:52 +0100458 def __init__(self, filename, mode='a', encoding=None, delay=False,
459 errors=None):
Inada Naokifb786922021-04-06 11:18:41 +0900460 if "b" not in mode:
461 encoding = io.text_encoding(encoding)
Vinay Sajipca7b5042019-06-17 17:40:52 +0100462 logging.FileHandler.__init__(self, filename, mode=mode,
463 encoding=encoding, delay=delay,
464 errors=errors)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100465 self.dev, self.ino = -1, -1
466 self._statstream()
467
468 def _statstream(self):
469 if self.stream:
470 sres = os.fstat(self.stream.fileno())
471 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000472
Vinay Sajip29a14452015-10-01 20:54:41 +0100473 def reopenIfNeeded(self):
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000474 """
Vinay Sajip29a14452015-10-01 20:54:41 +0100475 Reopen log file if needed.
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000476
Vinay Sajip29a14452015-10-01 20:54:41 +0100477 Checks if the underlying file has changed, and if it
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000478 has, close the old stream and reopen the file to get the
479 current stream.
480 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100481 # Reduce the chance of race conditions by stat'ing by path only
482 # once and then fstat'ing our new fd if we opened a new log stream.
483 # See issue #14632: Thanks to John Mulligan for the problem report
484 # and patch.
485 try:
486 # stat the file by path, checking for existence
487 sres = os.stat(self.baseFilename)
Giampaolo Rodola'0166a282013-02-12 15:14:17 +0100488 except FileNotFoundError:
489 sres = None
Vinay Sajip66b8b082012-04-24 23:25:30 +0100490 # compare file system stat with that of our stream file handle
491 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
492 if self.stream is not None:
493 # we have an open file handle, clean it up
494 self.stream.flush()
495 self.stream.close()
Vinay Sajip02200482014-06-14 10:22:05 +0100496 self.stream = None # See Issue #21742: _open () might fail.
Vinay Sajip66b8b082012-04-24 23:25:30 +0100497 # open a new file handle and get new stat info from that fd
498 self.stream = self._open()
499 self._statstream()
Vinay Sajip29a14452015-10-01 20:54:41 +0100500
501 def emit(self, record):
502 """
503 Emit a record.
504
505 If underlying file has changed, reopen the file before emitting the
506 record to it.
507 """
508 self.reopenIfNeeded()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000509 logging.FileHandler.emit(self, record)
510
Vinay Sajip66b8b082012-04-24 23:25:30 +0100511
Guido van Rossum57102f82002-11-13 16:15:58 +0000512class SocketHandler(logging.Handler):
513 """
514 A handler class which writes logging records, in pickle format, to
515 a streaming socket. The socket is kept open across logging calls.
516 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000517 The pickle which is sent is that of the LogRecord's attribute dictionary
518 (__dict__), so that the receiver does not need to have the logging module
519 installed in order to process the logging event.
520
521 To unpickle the record at the receiving end into a LogRecord, use the
522 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000523 """
524
525 def __init__(self, host, port):
526 """
527 Initializes the handler with a specific host address and port.
528
Vinay Sajipde19e082011-04-30 21:52:26 +0100529 When the attribute *closeOnError* is set to True - if a socket error
530 occurs, the socket is silently closed and then reopened on the next
531 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000532 """
533 logging.Handler.__init__(self)
534 self.host = host
535 self.port = port
Vinay Sajip5421f352013-09-27 18:18:28 +0100536 if port is None:
537 self.address = host
538 else:
539 self.address = (host, port)
Guido van Rossum57102f82002-11-13 16:15:58 +0000540 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100541 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000542 self.retryTime = None
543 #
544 # Exponential backoff parameters.
545 #
546 self.retryStart = 1.0
547 self.retryMax = 30.0
548 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000549
Guido van Rossumd8faa362007-04-27 19:54:29 +0000550 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000551 """
552 A factory method which allows subclasses to define the precise
553 type of socket they want.
554 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100555 if self.port is not None:
556 result = socket.create_connection(self.address, timeout=timeout)
557 else:
558 result = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
559 result.settimeout(timeout)
Vinay Sajip38c741c2013-10-06 18:36:00 +0100560 try:
561 result.connect(self.address)
562 except OSError:
563 result.close() # Issue 19182
564 raise
Vinay Sajip5421f352013-09-27 18:18:28 +0100565 return result
Guido van Rossum57102f82002-11-13 16:15:58 +0000566
Vinay Sajip48cfe382004-02-20 13:17:27 +0000567 def createSocket(self):
568 """
569 Try to create a socket, using an exponential backoff with
570 a max retry time. Thanks to Robert Olson for the original patch
571 (SF #815911) which has been slightly refactored.
572 """
573 now = time.time()
574 # Either retryTime is None, in which case this
575 # is the first time back after a disconnect, or
576 # we've waited long enough.
577 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100578 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000579 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000580 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000581 if attempt:
582 try:
583 self.sock = self.makeSocket()
584 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200585 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000586 #Creation failed, so set the retry time and return.
587 if self.retryTime is None:
588 self.retryPeriod = self.retryStart
589 else:
590 self.retryPeriod = self.retryPeriod * self.retryFactor
591 if self.retryPeriod > self.retryMax:
592 self.retryPeriod = self.retryMax
593 self.retryTime = now + self.retryPeriod
594
Guido van Rossum57102f82002-11-13 16:15:58 +0000595 def send(self, s):
596 """
597 Send a pickled string to the socket.
598
599 This function allows for partial sends which can happen when the
600 network is busy.
601 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000602 if self.sock is None:
603 self.createSocket()
604 #self.sock can be None either because we haven't reached the retry
605 #time yet, or because we have reached the retry time and retried,
606 #but are still unable to connect.
607 if self.sock:
608 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200609 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200610 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000611 self.sock.close()
612 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000613
614 def makePickle(self, record):
615 """
616 Pickles the record in binary format with a length prefix, and
617 returns it ready for transmission across the socket.
618 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000619 ei = record.exc_info
620 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100621 # just to get traceback text into record.exc_text ...
622 dummy = self.format(record)
623 # See issue #14436: If msg or args are objects, they may not be
624 # available on the receiving end. So we convert the msg % args
625 # to a string, save it as msg and zap the args.
626 d = dict(record.__dict__)
627 d['msg'] = record.getMessage()
628 d['args'] = None
629 d['exc_info'] = None
Vinay Sajip9cdfd182015-12-26 12:48:44 +0000630 # Issue #25685: delete 'message' if present: redundant with 'msg'
631 d.pop('message', None)
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100632 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000633 slen = struct.pack(">L", len(s))
634 return slen + s
635
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000636 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000637 """
638 Handle an error during logging.
639
640 An error has occurred during logging. Most likely cause -
641 connection lost. Close the socket so that we can retry on the
642 next event.
643 """
644 if self.closeOnError and self.sock:
645 self.sock.close()
646 self.sock = None #try to reconnect next time
647 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000648 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000649
650 def emit(self, record):
651 """
652 Emit a record.
653
654 Pickles the record and writes it to the socket in binary format.
655 If there is an error with the socket, silently drop the packet.
656 If there was a problem with the socket, re-establishes the
657 socket.
658 """
659 try:
660 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000661 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100662 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000663 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000664
665 def close(self):
666 """
667 Closes the socket.
668 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000669 self.acquire()
670 try:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300671 sock = self.sock
672 if sock:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000673 self.sock = None
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300674 sock.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000675 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000676 finally:
677 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000678
679class DatagramHandler(SocketHandler):
680 """
681 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000682 a datagram socket. The pickle which is sent is that of the LogRecord's
683 attribute dictionary (__dict__), so that the receiver does not need to
684 have the logging module installed in order to process the logging event.
685
686 To unpickle the record at the receiving end into a LogRecord, use the
687 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000688
689 """
690 def __init__(self, host, port):
691 """
692 Initializes the handler with a specific host address and port.
693 """
694 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100695 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000696
697 def makeSocket(self):
698 """
699 The factory method of SocketHandler is here overridden to create
700 a UDP socket (SOCK_DGRAM).
701 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100702 if self.port is None:
703 family = socket.AF_UNIX
704 else:
705 family = socket.AF_INET
706 s = socket.socket(family, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000707 return s
708
709 def send(self, s):
710 """
711 Send a pickled string to a socket.
712
713 This function no longer allows for partial sends which can happen
714 when the network is busy - UDP does not guarantee delivery and
715 can deliver packets out of sequence.
716 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000717 if self.sock is None:
718 self.createSocket()
Vinay Sajip5421f352013-09-27 18:18:28 +0100719 self.sock.sendto(s, self.address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000720
721class SysLogHandler(logging.Handler):
722 """
723 A handler class which sends formatted logging records to a syslog
724 server. Based on Sam Rushing's syslog module:
725 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
726 Contributed by Nicolas Untz (after which minor refactoring changes
727 have been made).
728 """
729
730 # from <linux/sys/syslog.h>:
731 # ======================================================================
732 # priorities/facilities are encoded into a single 32-bit quantity, where
733 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
734 # facility (0-big number). Both the priorities and the facilities map
735 # roughly one-to-one to strings in the syslogd(8) source code. This
736 # mapping is included in this file.
737 #
738 # priorities (these are ordered)
739
740 LOG_EMERG = 0 # system is unusable
741 LOG_ALERT = 1 # action must be taken immediately
742 LOG_CRIT = 2 # critical conditions
743 LOG_ERR = 3 # error conditions
744 LOG_WARNING = 4 # warning conditions
745 LOG_NOTICE = 5 # normal but significant condition
746 LOG_INFO = 6 # informational
747 LOG_DEBUG = 7 # debug-level messages
748
749 # facility codes
750 LOG_KERN = 0 # kernel messages
751 LOG_USER = 1 # random user-level messages
752 LOG_MAIL = 2 # mail system
753 LOG_DAEMON = 3 # system daemons
754 LOG_AUTH = 4 # security/authorization messages
755 LOG_SYSLOG = 5 # messages generated internally by syslogd
756 LOG_LPR = 6 # line printer subsystem
757 LOG_NEWS = 7 # network news subsystem
758 LOG_UUCP = 8 # UUCP subsystem
759 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000760 LOG_AUTHPRIV = 10 # security/authorization messages (private)
761 LOG_FTP = 11 # FTP daemon
Vinay Sajipce545192020-01-10 19:37:48 +0000762 LOG_NTP = 12 # NTP subsystem
763 LOG_SECURITY = 13 # Log audit
764 LOG_CONSOLE = 14 # Log alert
765 LOG_SOLCRON = 15 # Scheduling daemon (Solaris)
Guido van Rossum57102f82002-11-13 16:15:58 +0000766
767 # other codes through 15 reserved for system use
768 LOG_LOCAL0 = 16 # reserved for local use
769 LOG_LOCAL1 = 17 # reserved for local use
770 LOG_LOCAL2 = 18 # reserved for local use
771 LOG_LOCAL3 = 19 # reserved for local use
772 LOG_LOCAL4 = 20 # reserved for local use
773 LOG_LOCAL5 = 21 # reserved for local use
774 LOG_LOCAL6 = 22 # reserved for local use
775 LOG_LOCAL7 = 23 # reserved for local use
776
777 priority_names = {
778 "alert": LOG_ALERT,
779 "crit": LOG_CRIT,
780 "critical": LOG_CRIT,
781 "debug": LOG_DEBUG,
782 "emerg": LOG_EMERG,
783 "err": LOG_ERR,
784 "error": LOG_ERR, # DEPRECATED
785 "info": LOG_INFO,
786 "notice": LOG_NOTICE,
787 "panic": LOG_EMERG, # DEPRECATED
788 "warn": LOG_WARNING, # DEPRECATED
789 "warning": LOG_WARNING,
790 }
791
792 facility_names = {
Vinay Sajipce545192020-01-10 19:37:48 +0000793 "auth": LOG_AUTH,
794 "authpriv": LOG_AUTHPRIV,
795 "console": LOG_CONSOLE,
796 "cron": LOG_CRON,
797 "daemon": LOG_DAEMON,
798 "ftp": LOG_FTP,
799 "kern": LOG_KERN,
800 "lpr": LOG_LPR,
801 "mail": LOG_MAIL,
802 "news": LOG_NEWS,
803 "ntp": LOG_NTP,
804 "security": LOG_SECURITY,
805 "solaris-cron": LOG_SOLCRON,
806 "syslog": LOG_SYSLOG,
807 "user": LOG_USER,
808 "uucp": LOG_UUCP,
809 "local0": LOG_LOCAL0,
810 "local1": LOG_LOCAL1,
811 "local2": LOG_LOCAL2,
812 "local3": LOG_LOCAL3,
813 "local4": LOG_LOCAL4,
814 "local5": LOG_LOCAL5,
815 "local6": LOG_LOCAL6,
816 "local7": LOG_LOCAL7,
Guido van Rossum57102f82002-11-13 16:15:58 +0000817 }
818
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000819 #The map below appears to be trivially lowercasing the key. However,
820 #there's more to it than meets the eye - in some locales, lowercasing
821 #gives unexpected results. See SF #1524081: in the Turkish locale,
822 #"INFO".lower() != "info"
823 priority_map = {
824 "DEBUG" : "debug",
825 "INFO" : "info",
826 "WARNING" : "warning",
827 "ERROR" : "error",
828 "CRITICAL" : "critical"
829 }
830
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000831 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100832 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000833 """
834 Initialize a handler.
835
Guido van Rossume7ba4952007-06-06 23:52:48 +0000836 If address is specified as a string, a UNIX socket is used. To log to a
837 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100838 If facility is not specified, LOG_USER is used. If socktype is
839 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
840 socket type will be used. For Unix sockets, you can also specify a
841 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
842 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000843 """
844 logging.Handler.__init__(self)
845
846 self.address = address
847 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000848 self.socktype = socktype
849
Guido van Rossum13257902007-06-07 23:15:56 +0000850 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100851 self.unixsocket = True
Коренберг Марк1b038e02017-03-17 20:25:05 +0500852 # Syslog server may be unavailable during handler initialisation.
853 # C's openlog() function also ignores connection errors.
854 # Moreover, we ignore these errors while logging, so it not worse
855 # to ignore it also here.
856 try:
857 self._connect_unixsocket(address)
858 except OSError:
859 pass
Guido van Rossum57102f82002-11-13 16:15:58 +0000860 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100861 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100862 if socktype is None:
863 socktype = socket.SOCK_DGRAM
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800864 host, port = address
865 ress = socket.getaddrinfo(host, port, 0, socktype)
866 if not ress:
867 raise OSError("getaddrinfo returns an empty list")
868 for res in ress:
869 af, socktype, proto, _, sa = res
870 err = sock = None
871 try:
872 sock = socket.socket(af, socktype, proto)
873 if socktype == socket.SOCK_STREAM:
874 sock.connect(sa)
875 break
876 except OSError as exc:
877 err = exc
878 if sock is not None:
879 sock.close()
880 if err is not None:
881 raise err
882 self.socket = sock
Vinay Sajipe9170522013-04-22 10:07:40 +0100883 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000884
Vinay Sajipa1974c12005-01-13 08:23:56 +0000885 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100886 use_socktype = self.socktype
887 if use_socktype is None:
888 use_socktype = socket.SOCK_DGRAM
889 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000890 try:
891 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100892 # it worked, so set self.socktype to the used type
893 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200894 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000895 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100896 if self.socktype is not None:
897 # user didn't specify falling back, so fail
898 raise
899 use_socktype = socket.SOCK_STREAM
900 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
901 try:
902 self.socket.connect(address)
903 # it worked, so set self.socktype to the used type
904 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100905 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100906 self.socket.close()
907 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000908
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000909 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000910 """
911 Encode the facility and priority. You can pass in strings or
912 integers - if strings are passed, the facility_names and
913 priority_names mapping dictionaries are used to convert them to
914 integers.
915 """
Guido van Rossum13257902007-06-07 23:15:56 +0000916 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000917 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000918 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000919 priority = self.priority_names[priority]
920 return (facility << 3) | priority
921
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800922 def close(self):
Guido van Rossum57102f82002-11-13 16:15:58 +0000923 """
924 Closes the socket.
925 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000926 self.acquire()
927 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000928 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000929 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000930 finally:
931 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000932
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000933 def mapPriority(self, levelName):
934 """
935 Map a logging level name to a key in the priority_names map.
936 This is useful in two scenarios: when custom levels are being
937 used, and in the case where you can't do a straightforward
938 mapping by lowercasing the logging level name because of locale-
939 specific issues (see SF #1524081).
940 """
941 return self.priority_map.get(levelName, "warning")
942
Vinay Sajip2353e352011-06-27 15:40:06 +0100943 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100944 append_nul = True # some old syslog daemons expect a NUL terminator
945
Guido van Rossum57102f82002-11-13 16:15:58 +0000946 def emit(self, record):
947 """
948 Emit a record.
949
950 The record is formatted, and then sent to the syslog server. If
951 exception information is present, it is NOT sent to the server.
952 """
Guido van Rossum57102f82002-11-13 16:15:58 +0000953 try:
Vinay Sajipc33a0cc2014-11-01 19:58:47 +0000954 msg = self.format(record)
955 if self.ident:
956 msg = self.ident + msg
957 if self.append_nul:
958 msg += '\000'
959
960 # We need to convert record level to lowercase, maybe this will
961 # change in the future.
962 prio = '<%d>' % self.encodePriority(self.facility,
963 self.mapPriority(record.levelname))
964 prio = prio.encode('utf-8')
965 # Message is a string. Convert to bytes as required by RFC 5424
966 msg = msg.encode('utf-8')
967 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000968 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000969 try:
970 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200971 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100972 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000973 self._connect_unixsocket(self.address)
974 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000975 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000976 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000977 else:
978 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100979 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000980 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000981
982class SMTPHandler(logging.Handler):
983 """
984 A handler class which sends an SMTP email for each logging event.
985 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000986 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100987 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000988 """
989 Initialize the handler.
990
991 Initialize the instance with the from and to addresses and subject
992 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000993 (host, port) tuple format for the mailhost argument. To specify
994 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000995 for the credentials argument. To specify the use of a secure
996 protocol (TLS), pass in a tuple for the secure argument. This will
997 only be used when authentication credentials are supplied. The tuple
998 will be either an empty tuple, or a single-value tuple with the name
999 of a keyfile, or a 2-value tuple with the names of the keyfile and
1000 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +00001001 A timeout in seconds can be specified for the SMTP connection (the
1002 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +00001003 """
1004 logging.Handler.__init__(self)
Vinay Sajip16c41ab2014-10-17 08:49:38 +01001005 if isinstance(mailhost, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001006 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +00001007 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +00001008 self.mailhost, self.mailport = mailhost, None
Vinay Sajip16c41ab2014-10-17 08:49:38 +01001009 if isinstance(credentials, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001010 self.username, self.password = credentials
1011 else:
1012 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +00001013 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +00001014 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001015 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +00001016 self.toaddrs = toaddrs
1017 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +00001018 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +00001019 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +00001020
1021 def getSubject(self, record):
1022 """
1023 Determine the subject for the email.
1024
1025 If you want to specify a subject line which is record-dependent,
1026 override this method.
1027 """
1028 return self.subject
1029
1030 def emit(self, record):
1031 """
1032 Emit a record.
1033
1034 Format the record and send it to the specified addressees.
1035 """
1036 try:
1037 import smtplib
Vinay Sajip277640a2015-10-17 16:13:10 +01001038 from email.message import EmailMessage
1039 import email.utils
1040
Guido van Rossum57102f82002-11-13 16:15:58 +00001041 port = self.mailport
1042 if not port:
1043 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +00001044 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Vinay Sajip277640a2015-10-17 16:13:10 +01001045 msg = EmailMessage()
1046 msg['From'] = self.fromaddr
1047 msg['To'] = ','.join(self.toaddrs)
1048 msg['Subject'] = self.getSubject(record)
1049 msg['Date'] = email.utils.localtime()
1050 msg.set_content(self.format(record))
Guido van Rossum360e4b82007-05-14 22:51:27 +00001051 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +00001052 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +00001053 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +00001054 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +00001055 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +00001056 smtp.login(self.username, self.password)
Vinay Sajip277640a2015-10-17 16:13:10 +01001057 smtp.send_message(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +00001058 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001059 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001060 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001061
1062class NTEventLogHandler(logging.Handler):
1063 """
1064 A handler class which sends events to the NT Event Log. Adds a
1065 registry entry for the specified application name. If no dllname is
1066 provided, win32service.pyd (which contains some basic message
1067 placeholders) is used. Note that use of these placeholders will make
1068 your event logs big, as the entire message source is held in the log.
1069 If you want slimmer logs, you have to pass in the name of your own DLL
1070 which contains the message definitions you want to use in the event log.
1071 """
1072 def __init__(self, appname, dllname=None, logtype="Application"):
1073 logging.Handler.__init__(self)
1074 try:
1075 import win32evtlogutil, win32evtlog
1076 self.appname = appname
1077 self._welu = win32evtlogutil
1078 if not dllname:
1079 dllname = os.path.split(self._welu.__file__)
1080 dllname = os.path.split(dllname[0])
1081 dllname = os.path.join(dllname[0], r'win32service.pyd')
1082 self.dllname = dllname
1083 self.logtype = logtype
1084 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1085 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1086 self.typemap = {
1087 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1088 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001089 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001090 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1091 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1092 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001093 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001094 print("The Python Win32 extensions for NT (service, event "\
1095 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001096 self._welu = None
1097
1098 def getMessageID(self, record):
1099 """
1100 Return the message ID for the event record. If you are using your
1101 own messages, you could do this by having the msg passed to the
1102 logger being an ID rather than a formatting string. Then, in here,
1103 you could use a dictionary lookup to get the message ID. This
1104 version returns 1, which is the base message ID in win32service.pyd.
1105 """
1106 return 1
1107
1108 def getEventCategory(self, record):
1109 """
1110 Return the event category for the record.
1111
1112 Override this if you want to specify your own categories. This version
1113 returns 0.
1114 """
1115 return 0
1116
1117 def getEventType(self, record):
1118 """
1119 Return the event type for the record.
1120
1121 Override this if you want to specify your own types. This version does
1122 a mapping using the handler's typemap attribute, which is set up in
1123 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001124 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001125 either need to override this method or place a suitable dictionary in
1126 the handler's typemap attribute.
1127 """
1128 return self.typemap.get(record.levelno, self.deftype)
1129
1130 def emit(self, record):
1131 """
1132 Emit a record.
1133
1134 Determine the message ID, event category and event type. Then
1135 log the message in the NT event log.
1136 """
1137 if self._welu:
1138 try:
1139 id = self.getMessageID(record)
1140 cat = self.getEventCategory(record)
1141 type = self.getEventType(record)
1142 msg = self.format(record)
1143 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001144 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001145 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001146
1147 def close(self):
1148 """
1149 Clean up this handler.
1150
1151 You can remove the application name from the registry as a
1152 source of event log entries. However, if you do this, you will
1153 not be able to see the events as you intended in the Event Log
1154 Viewer - it needs to be able to access the registry to get the
1155 DLL name.
1156 """
1157 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001158 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001159
1160class HTTPHandler(logging.Handler):
1161 """
Miss Islington (bot)6fc1efa2021-07-26 15:34:32 -07001162 A class which sends records to a web server, using either GET or
Guido van Rossum57102f82002-11-13 16:15:58 +00001163 POST semantics.
1164 """
Benjamin Peterson43052a12014-11-23 20:36:44 -06001165 def __init__(self, host, url, method="GET", secure=False, credentials=None,
1166 context=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001167 """
1168 Initialize the instance with the host, the request URL, and the method
1169 ("GET" or "POST")
1170 """
1171 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001172 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001173 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001174 raise ValueError("method must be GET or POST")
Benjamin Peterson43052a12014-11-23 20:36:44 -06001175 if not secure and context is not None:
1176 raise ValueError("context parameter only makes sense "
1177 "with secure=True")
Guido van Rossum57102f82002-11-13 16:15:58 +00001178 self.host = host
1179 self.url = url
1180 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001181 self.secure = secure
1182 self.credentials = credentials
Benjamin Peterson43052a12014-11-23 20:36:44 -06001183 self.context = context
Guido van Rossum57102f82002-11-13 16:15:58 +00001184
Neal Norwitzf297bd12003-04-23 03:49:43 +00001185 def mapLogRecord(self, record):
1186 """
1187 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001188 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001189 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001190 """
1191 return record.__dict__
1192
l0rb22a9a542020-03-04 11:49:51 +01001193 def getConnection(self, host, secure):
1194 """
1195 get a HTTP[S]Connection.
1196
1197 Override when a custom connection is required, for example if
1198 there is a proxy.
1199 """
1200 import http.client
1201 if secure:
1202 connection = http.client.HTTPSConnection(host, context=self.context)
1203 else:
1204 connection = http.client.HTTPConnection(host)
1205 return connection
1206
Guido van Rossum57102f82002-11-13 16:15:58 +00001207 def emit(self, record):
1208 """
1209 Emit a record.
1210
Miss Islington (bot)6fc1efa2021-07-26 15:34:32 -07001211 Send the record to the web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001212 """
1213 try:
l0rb22a9a542020-03-04 11:49:51 +01001214 import urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001215 host = self.host
l0rb22a9a542020-03-04 11:49:51 +01001216 h = self.getConnection(host, self.secure)
Guido van Rossum57102f82002-11-13 16:15:58 +00001217 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001218 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001219 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001220 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001221 sep = '&'
1222 else:
1223 sep = '?'
1224 url = url + "%c%s" % (sep, data)
1225 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001226 # support multiple hosts on one IP address...
1227 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001228 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001229 if i >= 0:
1230 host = host[:i]
Vinay Sajipe96ba182017-11-19 18:36:17 +00001231 # See issue #30904: putrequest call above already adds this header
1232 # on Python 3.x.
1233 # h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001234 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001235 h.putheader("Content-type",
1236 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001237 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001238 if self.credentials:
1239 import base64
Vinay Sajip1bf197e2016-06-07 21:19:55 +01001240 s = ('%s:%s' % self.credentials).encode('utf-8')
1241 s = 'Basic ' + base64.b64encode(s).strip().decode('ascii')
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001242 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001243 h.endheaders()
1244 if self.method == "POST":
1245 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001246 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001247 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001248 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001249
1250class BufferingHandler(logging.Handler):
1251 """
1252 A handler class which buffers logging records in memory. Whenever each
1253 record is added to the buffer, a check is made to see if the buffer should
1254 be flushed. If it should, then flush() is expected to do what's needed.
1255 """
1256 def __init__(self, capacity):
1257 """
1258 Initialize the handler with the buffer size.
1259 """
1260 logging.Handler.__init__(self)
1261 self.capacity = capacity
1262 self.buffer = []
1263
1264 def shouldFlush(self, record):
1265 """
1266 Should the handler flush its buffer?
1267
1268 Returns true if the buffer is up to capacity. This method can be
1269 overridden to implement custom flushing strategies.
1270 """
1271 return (len(self.buffer) >= self.capacity)
1272
1273 def emit(self, record):
1274 """
1275 Emit a record.
1276
1277 Append the record. If shouldFlush() tells us to, call flush() to process
1278 the buffer.
1279 """
1280 self.buffer.append(record)
1281 if self.shouldFlush(record):
1282 self.flush()
1283
1284 def flush(self):
1285 """
1286 Override to implement custom flushing behaviour.
1287
1288 This version just zaps the buffer to empty.
1289 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001290 self.acquire()
1291 try:
Daniel Anderssond89cea12019-11-13 10:03:45 +01001292 self.buffer.clear()
Vinay Sajipf0509032012-02-23 20:49:08 +00001293 finally:
1294 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001295
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001296 def close(self):
1297 """
1298 Close the handler.
1299
1300 This version just flushes and chains to the parent class' close().
1301 """
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001302 try:
1303 self.flush()
1304 finally:
1305 logging.Handler.close(self)
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001306
Guido van Rossum57102f82002-11-13 16:15:58 +00001307class MemoryHandler(BufferingHandler):
1308 """
1309 A handler class which buffers logging records in memory, periodically
1310 flushing them to a target handler. Flushing occurs whenever the buffer
1311 is full, or when an event of a certain severity or greater is seen.
1312 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001313 def __init__(self, capacity, flushLevel=logging.ERROR, target=None,
1314 flushOnClose=True):
Guido van Rossum57102f82002-11-13 16:15:58 +00001315 """
1316 Initialize the handler with the buffer size, the level at which
1317 flushing should occur and an optional target.
1318
1319 Note that without a target being set either here or via setTarget(),
1320 a MemoryHandler is no use to anyone!
Vinay Sajipcccf6062016-07-22 16:27:31 +01001321
1322 The ``flushOnClose`` argument is ``True`` for backward compatibility
1323 reasons - the old behaviour is that when the handler is closed, the
1324 buffer is flushed, even if the flush level hasn't been exceeded nor the
1325 capacity exceeded. To prevent this, set ``flushOnClose`` to ``False``.
Guido van Rossum57102f82002-11-13 16:15:58 +00001326 """
1327 BufferingHandler.__init__(self, capacity)
1328 self.flushLevel = flushLevel
1329 self.target = target
Vinay Sajipcccf6062016-07-22 16:27:31 +01001330 # See Issue #26559 for why this has been added
1331 self.flushOnClose = flushOnClose
Guido van Rossum57102f82002-11-13 16:15:58 +00001332
1333 def shouldFlush(self, record):
1334 """
1335 Check for buffer full or a record at the flushLevel or higher.
1336 """
1337 return (len(self.buffer) >= self.capacity) or \
1338 (record.levelno >= self.flushLevel)
1339
1340 def setTarget(self, target):
1341 """
1342 Set the target handler for this handler.
1343 """
Irit Katriel2353d772020-08-16 16:10:13 +01001344 self.acquire()
1345 try:
1346 self.target = target
1347 finally:
1348 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001349
1350 def flush(self):
1351 """
1352 For a MemoryHandler, flushing means just sending the buffered
1353 records to the target, if there is one. Override if you want
1354 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001355
1356 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001357 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001358 self.acquire()
1359 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001360 if self.target:
1361 for record in self.buffer:
1362 self.target.handle(record)
Daniel Anderssond89cea12019-11-13 10:03:45 +01001363 self.buffer.clear()
Vinay Sajipf0509032012-02-23 20:49:08 +00001364 finally:
1365 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001366
1367 def close(self):
1368 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001369 Flush, if appropriately configured, set the target to None and lose the
1370 buffer.
Guido van Rossum57102f82002-11-13 16:15:58 +00001371 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001372 try:
Vinay Sajipcccf6062016-07-22 16:27:31 +01001373 if self.flushOnClose:
1374 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001375 finally:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001376 self.acquire()
1377 try:
1378 self.target = None
1379 BufferingHandler.close(self)
1380 finally:
1381 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001382
1383
1384class QueueHandler(logging.Handler):
1385 """
1386 This handler sends events to a queue. Typically, it would be used together
1387 with a multiprocessing Queue to centralise logging to file in one process
1388 (in a multi-process application), so as to avoid file write contention
1389 between processes.
1390
1391 This code is new in Python 3.2, but this class can be copy pasted into
1392 user code for use with earlier Python versions.
1393 """
1394
1395 def __init__(self, queue):
1396 """
1397 Initialise an instance, using the passed queue.
1398 """
1399 logging.Handler.__init__(self)
1400 self.queue = queue
1401
1402 def enqueue(self, record):
1403 """
1404 Enqueue a record.
1405
1406 The base implementation uses put_nowait. You may want to override
1407 this method if you want to use blocking, timeouts or custom queue
1408 implementations.
1409 """
1410 self.queue.put_nowait(record)
1411
Vinay Sajip0258ce82010-09-22 20:34:53 +00001412 def prepare(self, record):
1413 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001414 Prepares a record for queuing. The object returned by this method is
1415 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001416
1417 The base implementation formats the record to merge the message
1418 and arguments, and removes unpickleable items from the record
1419 in-place.
1420
1421 You might want to override this method if you want to convert
1422 the record to a dict or JSON string, or send a modified copy
1423 of the record while leaving the original intact.
1424 """
1425 # The format operation gets traceback text into record.exc_text
favlladfe3442017-08-01 20:12:26 +02001426 # (if there's exception data), and also returns the formatted
1427 # message. We can then use this to replace the original
Vinay Sajip0258ce82010-09-22 20:34:53 +00001428 # msg + args, as these might be unpickleable. We also zap the
Cheryl Sabellad345bb42018-09-25 19:00:08 -04001429 # exc_info and exc_text attributes, as they are no longer
1430 # needed and, if not None, will typically not be pickleable.
favlladfe3442017-08-01 20:12:26 +02001431 msg = self.format(record)
Manjusakada6424e2019-01-23 15:08:38 +08001432 # bpo-35726: make copy of record to avoid affecting other handlers in the chain.
1433 record = copy.copy(record)
favlladfe3442017-08-01 20:12:26 +02001434 record.message = msg
1435 record.msg = msg
Vinay Sajip0258ce82010-09-22 20:34:53 +00001436 record.args = None
1437 record.exc_info = None
Cheryl Sabellad345bb42018-09-25 19:00:08 -04001438 record.exc_text = None
Vinay Sajip0258ce82010-09-22 20:34:53 +00001439 return record
1440
Vinay Sajip121a1c42010-09-08 10:46:15 +00001441 def emit(self, record):
1442 """
1443 Emit a record.
1444
Vinay Sajip0637d492010-09-23 08:15:54 +00001445 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001446 """
1447 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001448 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001449 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001450 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001451
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001452
1453class QueueListener(object):
1454 """
1455 This class implements an internal threaded listener which watches for
1456 LogRecords being added to a queue, removes them and passes them to a
1457 list of handlers for processing.
1458 """
1459 _sentinel = None
1460
1461 def __init__(self, queue, *handlers, respect_handler_level=False):
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001462 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001463 Initialise an instance with the specified queue and
1464 handlers.
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001465 """
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001466 self.queue = queue
1467 self.handlers = handlers
1468 self._thread = None
1469 self.respect_handler_level = respect_handler_level
Vinay Sajip0637d492010-09-23 08:15:54 +00001470
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001471 def dequeue(self, block):
1472 """
1473 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001474
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001475 The base implementation uses get. You may want to override this method
1476 if you want to use timeouts or work with custom queue implementations.
1477 """
1478 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001479
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001480 def start(self):
1481 """
1482 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001483
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001484 This starts up a background thread to monitor the queue for
1485 LogRecords to process.
1486 """
1487 self._thread = t = threading.Thread(target=self._monitor)
1488 t.daemon = True
1489 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001490
Boris Feld800d7862019-06-04 17:20:18 +02001491 def prepare(self, record):
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001492 """
1493 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001494
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001495 This method just returns the passed-in record. You may want to
1496 override this method if you need to do any custom marshalling or
1497 manipulation of the record before passing it to the handlers.
1498 """
1499 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001500
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001501 def handle(self, record):
1502 """
1503 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001504
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001505 This just loops through the handlers offering them the record
1506 to handle.
1507 """
1508 record = self.prepare(record)
1509 for handler in self.handlers:
1510 if not self.respect_handler_level:
1511 process = True
1512 else:
1513 process = record.levelno >= handler.level
1514 if process:
1515 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001516
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001517 def _monitor(self):
1518 """
1519 Monitor the queue for records, and ask the handler
1520 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001521
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001522 This method runs on a separate, internal thread.
1523 The thread will terminate if it sees a sentinel object in the queue.
1524 """
1525 q = self.queue
1526 has_task_done = hasattr(q, 'task_done')
1527 while True:
1528 try:
1529 record = self.dequeue(True)
1530 if record is self._sentinel:
Bar Harel6b282e12019-06-01 12:19:09 +03001531 if has_task_done:
1532 q.task_done()
Vinay Sajip0637d492010-09-23 08:15:54 +00001533 break
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001534 self.handle(record)
1535 if has_task_done:
1536 q.task_done()
1537 except queue.Empty:
1538 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001539
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001540 def enqueue_sentinel(self):
1541 """
1542 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001543
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001544 The base implementation uses put_nowait. You may want to override this
1545 method if you want to use timeouts or work with custom queue
1546 implementations.
1547 """
1548 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001549
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001550 def stop(self):
1551 """
1552 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001553
Antoine Pitroua6a4dc82017-09-07 18:56:24 +02001554 This asks the thread to terminate, and then waits for it to do so.
1555 Note that if you don't call this before your application exits, there
1556 may be some records still left on the queue, which won't be processed.
1557 """
1558 self.enqueue_sentinel()
1559 self._thread.join()
1560 self._thread = None