blob: b5fdfbc68397b0dca707b2c66884b5276c46cff7 [file] [log] [blame]
Vinay Sajipcccf6062016-07-22 16:27:31 +01001# Copyright 2001-2016 by Vinay Sajip. All Rights Reserved.
Guido van Rossum57102f82002-11-13 16:15:58 +00002#
3# Permission to use, copy, modify, and distribute this software and its
4# documentation for any purpose and without fee is hereby granted,
5# provided that the above copyright notice appear in all copies and that
6# both that copyright notice and this permission notice appear in
7# supporting documentation, and that the name of Vinay Sajip
8# not be used in advertising or publicity pertaining to distribution
9# of the software without specific, written prior permission.
10# VINAY SAJIP DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING
11# ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
12# VINAY SAJIP BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR
13# ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER
14# IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
15# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
Guido van Rossum57102f82002-11-13 16:15:58 +000016
17"""
Vinay Sajip3f742842004-02-28 16:07:46 +000018Additional handlers for the logging package for Python. The core package is
Vinay Sajip0abf61d2012-02-23 19:45:52 +000019based on PEP 282 and comments thereto in comp.lang.python.
Guido van Rossum57102f82002-11-13 16:15:58 +000020
Vinay Sajipcccf6062016-07-22 16:27:31 +010021Copyright (C) 2001-2016 Vinay Sajip. All Rights Reserved.
Guido van Rossum57102f82002-11-13 16:15:58 +000022
Vinay Sajip6268cbc2009-01-21 00:19:28 +000023To use, simply 'import logging.handlers' and log away!
Guido van Rossum57102f82002-11-13 16:15:58 +000024"""
25
Victor Stinner7fa767e2014-03-20 09:16:38 +010026import logging, socket, os, pickle, struct, time, re
Benjamin Peterson9451a1c2010-03-13 22:30:34 +000027from stat import ST_DEV, ST_INO, ST_MTIME
Vinay Sajipe723e962011-04-15 22:27:17 +010028import queue
Victor Stinnercafa2ef2011-05-02 16:11:28 +020029try:
30 import threading
Brett Cannoncd171c82013-07-04 17:43:24 -040031except ImportError: #pragma: no cover
Victor Stinnercafa2ef2011-05-02 16:11:28 +020032 threading = None
Guido van Rossum57102f82002-11-13 16:15:58 +000033
Guido van Rossum57102f82002-11-13 16:15:58 +000034#
35# Some constants...
36#
37
38DEFAULT_TCP_LOGGING_PORT = 9020
39DEFAULT_UDP_LOGGING_PORT = 9021
40DEFAULT_HTTP_LOGGING_PORT = 9022
41DEFAULT_SOAP_LOGGING_PORT = 9023
42SYSLOG_UDP_PORT = 514
Vinay Sajipcbabd7e2009-10-10 20:32:36 +000043SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000044
Thomas Wouters477c8d52006-05-27 19:21:47 +000045_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
46
Vinay Sajip17c52d82004-07-03 11:48:34 +000047class BaseRotatingHandler(logging.FileHandler):
48 """
49 Base class for handlers that rotate log files at a certain point.
50 Not meant to be instantiated directly. Instead, use RotatingFileHandler
51 or TimedRotatingFileHandler.
52 """
Vinay Sajip23b94d02012-01-04 12:02:26 +000053 def __init__(self, filename, mode, encoding=None, delay=False):
Vinay Sajip17c52d82004-07-03 11:48:34 +000054 """
55 Use the specified filename for streamed logging
56 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000057 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip4600f112005-03-13 09:56:36 +000058 self.mode = mode
59 self.encoding = encoding
Vinay Sajip23b94d02012-01-04 12:02:26 +000060 self.namer = None
61 self.rotator = None
Guido van Rossum57102f82002-11-13 16:15:58 +000062
Vinay Sajip17c52d82004-07-03 11:48:34 +000063 def emit(self, record):
64 """
65 Emit a record.
66
67 Output the record to the file, catering for rollover as described
68 in doRollover().
69 """
Vinay Sajip3970c112004-07-08 10:24:04 +000070 try:
71 if self.shouldRollover(record):
72 self.doRollover()
73 logging.FileHandler.emit(self, record)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +010074 except Exception:
Vinay Sajip3970c112004-07-08 10:24:04 +000075 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000076
Vinay Sajip23b94d02012-01-04 12:02:26 +000077 def rotation_filename(self, default_name):
78 """
79 Modify the filename of a log file when rotating.
80
81 This is provided so that a custom filename can be provided.
82
83 The default implementation calls the 'namer' attribute of the
84 handler, if it's callable, passing the default name to
85 it. If the attribute isn't callable (the default is None), the name
86 is returned unchanged.
87
88 :param default_name: The default name for the log file.
89 """
90 if not callable(self.namer):
91 result = default_name
92 else:
93 result = self.namer(default_name)
94 return result
95
96 def rotate(self, source, dest):
97 """
98 When rotating, rotate the current log.
99
100 The default implementation calls the 'rotator' attribute of the
101 handler, if it's callable, passing the source and dest arguments to
102 it. If the attribute isn't callable (the default is None), the source
103 is simply renamed to the destination.
104
105 :param source: The source filename. This is normally the base
106 filename, e.g. 'test.log'
107 :param dest: The destination filename. This is normally
108 what the source is rotated to, e.g. 'test.log.1'.
109 """
110 if not callable(self.rotator):
Vinay Sajipd8599262013-09-06 10:10:22 +0100111 # Issue 18940: A file may not have been created if delay is True.
112 if os.path.exists(source):
113 os.rename(source, dest)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000114 else:
115 self.rotator(source, dest)
116
Vinay Sajip17c52d82004-07-03 11:48:34 +0000117class RotatingFileHandler(BaseRotatingHandler):
118 """
119 Handler for logging to a set of files, which switches from one file
120 to the next when the current file reaches a certain size.
121 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000122 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
Guido van Rossum57102f82002-11-13 16:15:58 +0000123 """
124 Open the specified file and use it as the stream for logging.
125
126 By default, the file grows indefinitely. You can specify particular
127 values of maxBytes and backupCount to allow the file to rollover at
128 a predetermined size.
129
130 Rollover occurs whenever the current log file is nearly maxBytes in
131 length. If backupCount is >= 1, the system will successively create
132 new files with the same pathname as the base file, but with extensions
133 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
134 and a base file name of "app.log", you would get "app.log",
135 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
136 written to is always "app.log" - when it gets filled up, it is closed
137 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
138 exist, then they are renamed to "app.log.2", "app.log.3" etc.
139 respectively.
140
141 If maxBytes is zero, rollover never occurs.
142 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000143 # If rotation/rollover is wanted, it doesn't make sense to use another
144 # mode. If for example 'w' were specified, then if there were multiple
145 # runs of the calling application, the logs from previous runs would be
146 # lost if the 'w' is respected, because the log file would be truncated
147 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000148 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000149 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000150 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000151 self.maxBytes = maxBytes
152 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000153
154 def doRollover(self):
155 """
156 Do a rollover, as described in __init__().
157 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000158 if self.stream:
159 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000160 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000161 if self.backupCount > 0:
162 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000163 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
164 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
165 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000166 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000167 if os.path.exists(dfn):
168 os.remove(dfn)
169 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000170 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000171 if os.path.exists(dfn):
172 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000173 self.rotate(self.baseFilename, dfn)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100174 if not self.delay:
175 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000176
Vinay Sajip17c52d82004-07-03 11:48:34 +0000177 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000178 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000179 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000180
Vinay Sajip17c52d82004-07-03 11:48:34 +0000181 Basically, see if the supplied record would cause the file to exceed
182 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000183 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000184 if self.stream is None: # delay was set...
185 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000186 if self.maxBytes > 0: # are we rolling over?
187 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000188 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000189 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000190 return 1
191 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000192
Vinay Sajip17c52d82004-07-03 11:48:34 +0000193class TimedRotatingFileHandler(BaseRotatingHandler):
194 """
195 Handler for logging to a file, rotating the log file at certain timed
196 intervals.
197
198 If backupCount is > 0, when rollover is done, no more than backupCount
199 files are kept - the oldest ones are deleted.
200 """
Vinay Sajipa7130792013-04-12 17:04:23 +0100201 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False, atTime=None):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000202 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000203 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000204 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000205 self.utc = utc
Vinay Sajipa7130792013-04-12 17:04:23 +0100206 self.atTime = atTime
Vinay Sajip17c52d82004-07-03 11:48:34 +0000207 # Calculate the real rollover interval, which is just the number of
208 # seconds between rollovers. Also set the filename suffix used when
209 # a rollover occurs. Current 'when' events supported:
210 # S - Seconds
211 # M - Minutes
212 # H - Hours
213 # D - Days
214 # midnight - roll over at midnight
215 # W{0-6} - roll over on a certain day; 0 - Monday
216 #
217 # Case of the 'when' specifier is not important; lower or upper case
218 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000219 if self.when == 'S':
220 self.interval = 1 # one second
221 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000222 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000223 elif self.when == 'M':
224 self.interval = 60 # one minute
225 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000226 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000227 elif self.when == 'H':
228 self.interval = 60 * 60 # one hour
229 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000230 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000231 elif self.when == 'D' or self.when == 'MIDNIGHT':
232 self.interval = 60 * 60 * 24 # one day
233 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000234 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000235 elif self.when.startswith('W'):
236 self.interval = 60 * 60 * 24 * 7 # one week
237 if len(self.when) != 2:
238 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
239 if self.when[1] < '0' or self.when[1] > '6':
240 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
241 self.dayOfWeek = int(self.when[1])
242 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000243 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000244 else:
245 raise ValueError("Invalid rollover interval specified: %s" % self.when)
246
Antoine Pitroufd036452008-08-19 17:56:33 +0000247 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000248 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip638e6222016-07-22 18:23:04 +0100249 # The following line added because the filename passed in could be a
250 # path object (see Issue #27493), but self.baseFilename will be a string
251 filename = self.baseFilename
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000252 if os.path.exists(filename):
253 t = os.stat(filename)[ST_MTIME]
254 else:
255 t = int(time.time())
256 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000257
Vinay Sajipaffbd872009-06-11 10:11:47 +0000258 def computeRollover(self, currentTime):
259 """
260 Work out the rollover time based on the specified time.
261 """
262 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000263 # If we are rolling over at midnight or weekly, then the interval is already known.
264 # What we need to figure out is WHEN the next interval is. In other words,
265 # if you are rolling over at midnight, then your base interval is 1 day,
266 # but you want to start that one day clock at midnight, not now. So, we
267 # have to fudge the rolloverAt value in order to trigger the first rollover
268 # at the right time. After that, the regular interval will take care of
269 # the rest. Note that this code doesn't care about leap seconds. :)
270 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
271 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000272 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000273 t = time.gmtime(currentTime)
274 else:
275 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000276 currentHour = t[3]
277 currentMinute = t[4]
278 currentSecond = t[5]
Vinay Sajipa7130792013-04-12 17:04:23 +0100279 currentDay = t[6]
280 # r is the number of seconds left between now and the next rotation
281 if self.atTime is None:
282 rotate_ts = _MIDNIGHT
283 else:
284 rotate_ts = ((self.atTime.hour * 60 + self.atTime.minute)*60 +
285 self.atTime.second)
286
287 r = rotate_ts - ((currentHour * 60 + currentMinute) * 60 +
288 currentSecond)
289 if r < 0:
290 # Rotate time is before the current time (for example when
291 # self.rotateAt is 13:45 and it now 14:15), rotation is
292 # tomorrow.
293 r += _MIDNIGHT
294 currentDay = (currentDay + 1) % 7
Vinay Sajipaffbd872009-06-11 10:11:47 +0000295 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000296 # If we are rolling over on a certain day, add in the number of days until
297 # the next rollover, but offset by 1 since we just calculated the time
298 # until the next day starts. There are three cases:
299 # Case 1) The day to rollover is today; in this case, do nothing
300 # Case 2) The day to rollover is further in the interval (i.e., today is
301 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
302 # next rollover is simply 6 - 2 - 1, or 3.
303 # Case 3) The day to rollover is behind us in the interval (i.e., today
304 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
305 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
306 # number of days left in the current week (1) plus the number
307 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000308 # The calculations described in 2) and 3) above need to have a day added.
309 # This is because the above time calculation takes us to midnight on this
310 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000311 if self.when.startswith('W'):
Vinay Sajipa7130792013-04-12 17:04:23 +0100312 day = currentDay # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000313 if day != self.dayOfWeek:
314 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000315 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000316 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000317 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000318 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
319 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000320 dstNow = t[-1]
321 dstAtRollover = time.localtime(newRolloverAt)[-1]
322 if dstNow != dstAtRollover:
323 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000324 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000325 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000326 addend = 3600
327 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000328 result = newRolloverAt
329 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000330
331 def shouldRollover(self, record):
332 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000333 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000334
335 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000336 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000337 """
338 t = int(time.time())
339 if t >= self.rolloverAt:
340 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000341 return 0
342
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000343 def getFilesToDelete(self):
344 """
345 Determine the files to delete when rolling over.
346
347 More specific than the earlier method, which just used glob.glob().
348 """
349 dirName, baseName = os.path.split(self.baseFilename)
350 fileNames = os.listdir(dirName)
351 result = []
352 prefix = baseName + "."
353 plen = len(prefix)
354 for fileName in fileNames:
355 if fileName[:plen] == prefix:
356 suffix = fileName[plen:]
357 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000358 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000359 result.sort()
360 if len(result) < self.backupCount:
361 result = []
362 else:
363 result = result[:len(result) - self.backupCount]
364 return result
365
Vinay Sajip17c52d82004-07-03 11:48:34 +0000366 def doRollover(self):
367 """
368 do a rollover; in this case, a date/time stamp is appended to the filename
369 when the rollover happens. However, you want the file to be named for the
370 start of the interval, not the current time. If there is a backup count,
371 then we have to get a list of matching filenames, sort them and remove
372 the one with the oldest suffix.
373 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000374 if self.stream:
375 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000376 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000377 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000378 currentTime = int(time.time())
379 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000380 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000381 if self.utc:
382 timeTuple = time.gmtime(t)
383 else:
384 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000385 dstThen = timeTuple[-1]
386 if dstNow != dstThen:
387 if dstNow:
388 addend = 3600
389 else:
390 addend = -3600
391 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000392 dfn = self.rotation_filename(self.baseFilename + "." +
393 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000394 if os.path.exists(dfn):
395 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000396 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000397 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000398 for s in self.getFilesToDelete():
399 os.remove(s)
Vinay Sajip43c6ef12013-09-06 10:25:31 +0100400 if not self.delay:
401 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000402 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000403 while newRolloverAt <= currentTime:
404 newRolloverAt = newRolloverAt + self.interval
405 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000406 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000407 dstAtRollover = time.localtime(newRolloverAt)[-1]
408 if dstNow != dstAtRollover:
409 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000410 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000411 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000412 addend = 3600
413 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000414 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000415
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000416class WatchedFileHandler(logging.FileHandler):
417 """
418 A handler for logging to a file, which watches the file
419 to see if it has changed while in use. This can happen because of
420 usage of programs such as newsyslog and logrotate which perform
421 log file rotation. This handler, intended for use under Unix,
422 watches the file to see if it has changed since the last emit.
423 (A file has changed if its device or inode have changed.)
424 If it has changed, the old file stream is closed, and the file
425 opened to get a new stream.
426
427 This handler is not appropriate for use under Windows, because
428 under Windows open files cannot be moved or renamed - logging
429 opens the files with exclusive locks - and so there is no need
430 for such a handler. Furthermore, ST_INO is not supported under
431 Windows; stat always returns zero for this value.
432
433 This handler is based on a suggestion and patch by Chad J.
434 Schroeder.
435 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000436 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000437 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100438 self.dev, self.ino = -1, -1
439 self._statstream()
440
441 def _statstream(self):
442 if self.stream:
443 sres = os.fstat(self.stream.fileno())
444 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000445
Vinay Sajip29a14452015-10-01 20:54:41 +0100446 def reopenIfNeeded(self):
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000447 """
Vinay Sajip29a14452015-10-01 20:54:41 +0100448 Reopen log file if needed.
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000449
Vinay Sajip29a14452015-10-01 20:54:41 +0100450 Checks if the underlying file has changed, and if it
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000451 has, close the old stream and reopen the file to get the
452 current stream.
453 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100454 # Reduce the chance of race conditions by stat'ing by path only
455 # once and then fstat'ing our new fd if we opened a new log stream.
456 # See issue #14632: Thanks to John Mulligan for the problem report
457 # and patch.
458 try:
459 # stat the file by path, checking for existence
460 sres = os.stat(self.baseFilename)
Giampaolo Rodola'0166a282013-02-12 15:14:17 +0100461 except FileNotFoundError:
462 sres = None
Vinay Sajip66b8b082012-04-24 23:25:30 +0100463 # compare file system stat with that of our stream file handle
464 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
465 if self.stream is not None:
466 # we have an open file handle, clean it up
467 self.stream.flush()
468 self.stream.close()
Vinay Sajip02200482014-06-14 10:22:05 +0100469 self.stream = None # See Issue #21742: _open () might fail.
Vinay Sajip66b8b082012-04-24 23:25:30 +0100470 # open a new file handle and get new stat info from that fd
471 self.stream = self._open()
472 self._statstream()
Vinay Sajip29a14452015-10-01 20:54:41 +0100473
474 def emit(self, record):
475 """
476 Emit a record.
477
478 If underlying file has changed, reopen the file before emitting the
479 record to it.
480 """
481 self.reopenIfNeeded()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000482 logging.FileHandler.emit(self, record)
483
Vinay Sajip66b8b082012-04-24 23:25:30 +0100484
Guido van Rossum57102f82002-11-13 16:15:58 +0000485class SocketHandler(logging.Handler):
486 """
487 A handler class which writes logging records, in pickle format, to
488 a streaming socket. The socket is kept open across logging calls.
489 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000490 The pickle which is sent is that of the LogRecord's attribute dictionary
491 (__dict__), so that the receiver does not need to have the logging module
492 installed in order to process the logging event.
493
494 To unpickle the record at the receiving end into a LogRecord, use the
495 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000496 """
497
498 def __init__(self, host, port):
499 """
500 Initializes the handler with a specific host address and port.
501
Vinay Sajipde19e082011-04-30 21:52:26 +0100502 When the attribute *closeOnError* is set to True - if a socket error
503 occurs, the socket is silently closed and then reopened on the next
504 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000505 """
506 logging.Handler.__init__(self)
507 self.host = host
508 self.port = port
Vinay Sajip5421f352013-09-27 18:18:28 +0100509 if port is None:
510 self.address = host
511 else:
512 self.address = (host, port)
Guido van Rossum57102f82002-11-13 16:15:58 +0000513 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100514 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000515 self.retryTime = None
516 #
517 # Exponential backoff parameters.
518 #
519 self.retryStart = 1.0
520 self.retryMax = 30.0
521 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000522
Guido van Rossumd8faa362007-04-27 19:54:29 +0000523 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000524 """
525 A factory method which allows subclasses to define the precise
526 type of socket they want.
527 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100528 if self.port is not None:
529 result = socket.create_connection(self.address, timeout=timeout)
530 else:
531 result = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
532 result.settimeout(timeout)
Vinay Sajip38c741c2013-10-06 18:36:00 +0100533 try:
534 result.connect(self.address)
535 except OSError:
536 result.close() # Issue 19182
537 raise
Vinay Sajip5421f352013-09-27 18:18:28 +0100538 return result
Guido van Rossum57102f82002-11-13 16:15:58 +0000539
Vinay Sajip48cfe382004-02-20 13:17:27 +0000540 def createSocket(self):
541 """
542 Try to create a socket, using an exponential backoff with
543 a max retry time. Thanks to Robert Olson for the original patch
544 (SF #815911) which has been slightly refactored.
545 """
546 now = time.time()
547 # Either retryTime is None, in which case this
548 # is the first time back after a disconnect, or
549 # we've waited long enough.
550 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100551 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000552 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000553 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000554 if attempt:
555 try:
556 self.sock = self.makeSocket()
557 self.retryTime = None # next time, no delay before trying
Andrew Svetlov0832af62012-12-18 23:10:48 +0200558 except OSError:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000559 #Creation failed, so set the retry time and return.
560 if self.retryTime is None:
561 self.retryPeriod = self.retryStart
562 else:
563 self.retryPeriod = self.retryPeriod * self.retryFactor
564 if self.retryPeriod > self.retryMax:
565 self.retryPeriod = self.retryMax
566 self.retryTime = now + self.retryPeriod
567
Guido van Rossum57102f82002-11-13 16:15:58 +0000568 def send(self, s):
569 """
570 Send a pickled string to the socket.
571
572 This function allows for partial sends which can happen when the
573 network is busy.
574 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000575 if self.sock is None:
576 self.createSocket()
577 #self.sock can be None either because we haven't reached the retry
578 #time yet, or because we have reached the retry time and retried,
579 #but are still unable to connect.
580 if self.sock:
581 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200582 self.sock.sendall(s)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200583 except OSError: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000584 self.sock.close()
585 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000586
587 def makePickle(self, record):
588 """
589 Pickles the record in binary format with a length prefix, and
590 returns it ready for transmission across the socket.
591 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000592 ei = record.exc_info
593 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100594 # just to get traceback text into record.exc_text ...
595 dummy = self.format(record)
596 # See issue #14436: If msg or args are objects, they may not be
597 # available on the receiving end. So we convert the msg % args
598 # to a string, save it as msg and zap the args.
599 d = dict(record.__dict__)
600 d['msg'] = record.getMessage()
601 d['args'] = None
602 d['exc_info'] = None
Vinay Sajip9cdfd182015-12-26 12:48:44 +0000603 # Issue #25685: delete 'message' if present: redundant with 'msg'
604 d.pop('message', None)
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100605 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000606 slen = struct.pack(">L", len(s))
607 return slen + s
608
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000609 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000610 """
611 Handle an error during logging.
612
613 An error has occurred during logging. Most likely cause -
614 connection lost. Close the socket so that we can retry on the
615 next event.
616 """
617 if self.closeOnError and self.sock:
618 self.sock.close()
619 self.sock = None #try to reconnect next time
620 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000621 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000622
623 def emit(self, record):
624 """
625 Emit a record.
626
627 Pickles the record and writes it to the socket in binary format.
628 If there is an error with the socket, silently drop the packet.
629 If there was a problem with the socket, re-establishes the
630 socket.
631 """
632 try:
633 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000634 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100635 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000636 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000637
638 def close(self):
639 """
640 Closes the socket.
641 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000642 self.acquire()
643 try:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300644 sock = self.sock
645 if sock:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000646 self.sock = None
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300647 sock.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000648 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000649 finally:
650 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000651
652class DatagramHandler(SocketHandler):
653 """
654 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000655 a datagram socket. The pickle which is sent is that of the LogRecord's
656 attribute dictionary (__dict__), so that the receiver does not need to
657 have the logging module installed in order to process the logging event.
658
659 To unpickle the record at the receiving end into a LogRecord, use the
660 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000661
662 """
663 def __init__(self, host, port):
664 """
665 Initializes the handler with a specific host address and port.
666 """
667 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100668 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000669
670 def makeSocket(self):
671 """
672 The factory method of SocketHandler is here overridden to create
673 a UDP socket (SOCK_DGRAM).
674 """
Vinay Sajip5421f352013-09-27 18:18:28 +0100675 if self.port is None:
676 family = socket.AF_UNIX
677 else:
678 family = socket.AF_INET
679 s = socket.socket(family, socket.SOCK_DGRAM)
Guido van Rossum57102f82002-11-13 16:15:58 +0000680 return s
681
682 def send(self, s):
683 """
684 Send a pickled string to a socket.
685
686 This function no longer allows for partial sends which can happen
687 when the network is busy - UDP does not guarantee delivery and
688 can deliver packets out of sequence.
689 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000690 if self.sock is None:
691 self.createSocket()
Vinay Sajip5421f352013-09-27 18:18:28 +0100692 self.sock.sendto(s, self.address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000693
694class SysLogHandler(logging.Handler):
695 """
696 A handler class which sends formatted logging records to a syslog
697 server. Based on Sam Rushing's syslog module:
698 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
699 Contributed by Nicolas Untz (after which minor refactoring changes
700 have been made).
701 """
702
703 # from <linux/sys/syslog.h>:
704 # ======================================================================
705 # priorities/facilities are encoded into a single 32-bit quantity, where
706 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
707 # facility (0-big number). Both the priorities and the facilities map
708 # roughly one-to-one to strings in the syslogd(8) source code. This
709 # mapping is included in this file.
710 #
711 # priorities (these are ordered)
712
713 LOG_EMERG = 0 # system is unusable
714 LOG_ALERT = 1 # action must be taken immediately
715 LOG_CRIT = 2 # critical conditions
716 LOG_ERR = 3 # error conditions
717 LOG_WARNING = 4 # warning conditions
718 LOG_NOTICE = 5 # normal but significant condition
719 LOG_INFO = 6 # informational
720 LOG_DEBUG = 7 # debug-level messages
721
722 # facility codes
723 LOG_KERN = 0 # kernel messages
724 LOG_USER = 1 # random user-level messages
725 LOG_MAIL = 2 # mail system
726 LOG_DAEMON = 3 # system daemons
727 LOG_AUTH = 4 # security/authorization messages
728 LOG_SYSLOG = 5 # messages generated internally by syslogd
729 LOG_LPR = 6 # line printer subsystem
730 LOG_NEWS = 7 # network news subsystem
731 LOG_UUCP = 8 # UUCP subsystem
732 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000733 LOG_AUTHPRIV = 10 # security/authorization messages (private)
734 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000735
736 # other codes through 15 reserved for system use
737 LOG_LOCAL0 = 16 # reserved for local use
738 LOG_LOCAL1 = 17 # reserved for local use
739 LOG_LOCAL2 = 18 # reserved for local use
740 LOG_LOCAL3 = 19 # reserved for local use
741 LOG_LOCAL4 = 20 # reserved for local use
742 LOG_LOCAL5 = 21 # reserved for local use
743 LOG_LOCAL6 = 22 # reserved for local use
744 LOG_LOCAL7 = 23 # reserved for local use
745
746 priority_names = {
747 "alert": LOG_ALERT,
748 "crit": LOG_CRIT,
749 "critical": LOG_CRIT,
750 "debug": LOG_DEBUG,
751 "emerg": LOG_EMERG,
752 "err": LOG_ERR,
753 "error": LOG_ERR, # DEPRECATED
754 "info": LOG_INFO,
755 "notice": LOG_NOTICE,
756 "panic": LOG_EMERG, # DEPRECATED
757 "warn": LOG_WARNING, # DEPRECATED
758 "warning": LOG_WARNING,
759 }
760
761 facility_names = {
762 "auth": LOG_AUTH,
763 "authpriv": LOG_AUTHPRIV,
764 "cron": LOG_CRON,
765 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000766 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000767 "kern": LOG_KERN,
768 "lpr": LOG_LPR,
769 "mail": LOG_MAIL,
770 "news": LOG_NEWS,
771 "security": LOG_AUTH, # DEPRECATED
772 "syslog": LOG_SYSLOG,
773 "user": LOG_USER,
774 "uucp": LOG_UUCP,
775 "local0": LOG_LOCAL0,
776 "local1": LOG_LOCAL1,
777 "local2": LOG_LOCAL2,
778 "local3": LOG_LOCAL3,
779 "local4": LOG_LOCAL4,
780 "local5": LOG_LOCAL5,
781 "local6": LOG_LOCAL6,
782 "local7": LOG_LOCAL7,
783 }
784
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000785 #The map below appears to be trivially lowercasing the key. However,
786 #there's more to it than meets the eye - in some locales, lowercasing
787 #gives unexpected results. See SF #1524081: in the Turkish locale,
788 #"INFO".lower() != "info"
789 priority_map = {
790 "DEBUG" : "debug",
791 "INFO" : "info",
792 "WARNING" : "warning",
793 "ERROR" : "error",
794 "CRITICAL" : "critical"
795 }
796
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000797 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
Vinay Sajipe9170522013-04-22 10:07:40 +0100798 facility=LOG_USER, socktype=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000799 """
800 Initialize a handler.
801
Guido van Rossume7ba4952007-06-06 23:52:48 +0000802 If address is specified as a string, a UNIX socket is used. To log to a
803 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Vinay Sajip40589f42013-04-22 10:14:12 +0100804 If facility is not specified, LOG_USER is used. If socktype is
805 specified as socket.SOCK_DGRAM or socket.SOCK_STREAM, that specific
806 socket type will be used. For Unix sockets, you can also specify a
807 socktype of None, in which case socket.SOCK_DGRAM will be used, falling
808 back to socket.SOCK_STREAM.
Guido van Rossum57102f82002-11-13 16:15:58 +0000809 """
810 logging.Handler.__init__(self)
811
812 self.address = address
813 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000814 self.socktype = socktype
815
Guido van Rossum13257902007-06-07 23:15:56 +0000816 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100817 self.unixsocket = True
Коренберг Марк1b038e02017-03-17 20:25:05 +0500818 # Syslog server may be unavailable during handler initialisation.
819 # C's openlog() function also ignores connection errors.
820 # Moreover, we ignore these errors while logging, so it not worse
821 # to ignore it also here.
822 try:
823 self._connect_unixsocket(address)
824 except OSError:
825 pass
Guido van Rossum57102f82002-11-13 16:15:58 +0000826 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100827 self.unixsocket = False
Vinay Sajipe9170522013-04-22 10:07:40 +0100828 if socktype is None:
829 socktype = socket.SOCK_DGRAM
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800830 host, port = address
831 ress = socket.getaddrinfo(host, port, 0, socktype)
832 if not ress:
833 raise OSError("getaddrinfo returns an empty list")
834 for res in ress:
835 af, socktype, proto, _, sa = res
836 err = sock = None
837 try:
838 sock = socket.socket(af, socktype, proto)
839 if socktype == socket.SOCK_STREAM:
840 sock.connect(sa)
841 break
842 except OSError as exc:
843 err = exc
844 if sock is not None:
845 sock.close()
846 if err is not None:
847 raise err
848 self.socket = sock
Vinay Sajipe9170522013-04-22 10:07:40 +0100849 self.socktype = socktype
Guido van Rossum57102f82002-11-13 16:15:58 +0000850
Vinay Sajipa1974c12005-01-13 08:23:56 +0000851 def _connect_unixsocket(self, address):
Vinay Sajipe9170522013-04-22 10:07:40 +0100852 use_socktype = self.socktype
853 if use_socktype is None:
854 use_socktype = socket.SOCK_DGRAM
855 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000856 try:
857 self.socket.connect(address)
Vinay Sajipe9170522013-04-22 10:07:40 +0100858 # it worked, so set self.socktype to the used type
859 self.socktype = use_socktype
Andrew Svetlov0832af62012-12-18 23:10:48 +0200860 except OSError:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000861 self.socket.close()
Vinay Sajipe9170522013-04-22 10:07:40 +0100862 if self.socktype is not None:
863 # user didn't specify falling back, so fail
864 raise
865 use_socktype = socket.SOCK_STREAM
866 self.socket = socket.socket(socket.AF_UNIX, use_socktype)
867 try:
868 self.socket.connect(address)
869 # it worked, so set self.socktype to the used type
870 self.socktype = use_socktype
Vinay Sajip40589f42013-04-22 10:14:12 +0100871 except OSError:
Vinay Sajipe9170522013-04-22 10:07:40 +0100872 self.socket.close()
873 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000874
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000875 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000876 """
877 Encode the facility and priority. You can pass in strings or
878 integers - if strings are passed, the facility_names and
879 priority_names mapping dictionaries are used to convert them to
880 integers.
881 """
Guido van Rossum13257902007-06-07 23:15:56 +0000882 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000883 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000884 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000885 priority = self.priority_names[priority]
886 return (facility << 3) | priority
887
Xiang Zhang0b4b57d2017-06-01 21:11:56 +0800888 def close(self):
Guido van Rossum57102f82002-11-13 16:15:58 +0000889 """
890 Closes the socket.
891 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000892 self.acquire()
893 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000894 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000895 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000896 finally:
897 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000898
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000899 def mapPriority(self, levelName):
900 """
901 Map a logging level name to a key in the priority_names map.
902 This is useful in two scenarios: when custom levels are being
903 used, and in the case where you can't do a straightforward
904 mapping by lowercasing the logging level name because of locale-
905 specific issues (see SF #1524081).
906 """
907 return self.priority_map.get(levelName, "warning")
908
Vinay Sajip2353e352011-06-27 15:40:06 +0100909 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100910 append_nul = True # some old syslog daemons expect a NUL terminator
911
Guido van Rossum57102f82002-11-13 16:15:58 +0000912 def emit(self, record):
913 """
914 Emit a record.
915
916 The record is formatted, and then sent to the syslog server. If
917 exception information is present, it is NOT sent to the server.
918 """
Guido van Rossum57102f82002-11-13 16:15:58 +0000919 try:
Vinay Sajipc33a0cc2014-11-01 19:58:47 +0000920 msg = self.format(record)
921 if self.ident:
922 msg = self.ident + msg
923 if self.append_nul:
924 msg += '\000'
925
926 # We need to convert record level to lowercase, maybe this will
927 # change in the future.
928 prio = '<%d>' % self.encodePriority(self.facility,
929 self.mapPriority(record.levelname))
930 prio = prio.encode('utf-8')
931 # Message is a string. Convert to bytes as required by RFC 5424
932 msg = msg.encode('utf-8')
933 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000934 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000935 try:
936 self.socket.send(msg)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200937 except OSError:
Vinay Sajip862b15e2013-05-16 22:57:02 +0100938 self.socket.close()
Vinay Sajipa1974c12005-01-13 08:23:56 +0000939 self._connect_unixsocket(self.address)
940 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000941 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000942 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000943 else:
944 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100945 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000946 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000947
948class SMTPHandler(logging.Handler):
949 """
950 A handler class which sends an SMTP email for each logging event.
951 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000952 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100953 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000954 """
955 Initialize the handler.
956
957 Initialize the instance with the from and to addresses and subject
958 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000959 (host, port) tuple format for the mailhost argument. To specify
960 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000961 for the credentials argument. To specify the use of a secure
962 protocol (TLS), pass in a tuple for the secure argument. This will
963 only be used when authentication credentials are supplied. The tuple
964 will be either an empty tuple, or a single-value tuple with the name
965 of a keyfile, or a 2-value tuple with the names of the keyfile and
966 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000967 A timeout in seconds can be specified for the SMTP connection (the
968 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000969 """
970 logging.Handler.__init__(self)
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100971 if isinstance(mailhost, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000972 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000973 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000974 self.mailhost, self.mailport = mailhost, None
Vinay Sajip16c41ab2014-10-17 08:49:38 +0100975 if isinstance(credentials, (list, tuple)):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000976 self.username, self.password = credentials
977 else:
978 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000979 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000980 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000981 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000982 self.toaddrs = toaddrs
983 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000984 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000985 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000986
987 def getSubject(self, record):
988 """
989 Determine the subject for the email.
990
991 If you want to specify a subject line which is record-dependent,
992 override this method.
993 """
994 return self.subject
995
996 def emit(self, record):
997 """
998 Emit a record.
999
1000 Format the record and send it to the specified addressees.
1001 """
1002 try:
1003 import smtplib
Vinay Sajip277640a2015-10-17 16:13:10 +01001004 from email.message import EmailMessage
1005 import email.utils
1006
Guido van Rossum57102f82002-11-13 16:15:58 +00001007 port = self.mailport
1008 if not port:
1009 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +00001010 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Vinay Sajip277640a2015-10-17 16:13:10 +01001011 msg = EmailMessage()
1012 msg['From'] = self.fromaddr
1013 msg['To'] = ','.join(self.toaddrs)
1014 msg['Subject'] = self.getSubject(record)
1015 msg['Date'] = email.utils.localtime()
1016 msg.set_content(self.format(record))
Guido van Rossum360e4b82007-05-14 22:51:27 +00001017 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +00001018 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +00001019 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +00001020 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +00001021 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +00001022 smtp.login(self.username, self.password)
Vinay Sajip277640a2015-10-17 16:13:10 +01001023 smtp.send_message(msg)
Guido van Rossum57102f82002-11-13 16:15:58 +00001024 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001025 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001026 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001027
1028class NTEventLogHandler(logging.Handler):
1029 """
1030 A handler class which sends events to the NT Event Log. Adds a
1031 registry entry for the specified application name. If no dllname is
1032 provided, win32service.pyd (which contains some basic message
1033 placeholders) is used. Note that use of these placeholders will make
1034 your event logs big, as the entire message source is held in the log.
1035 If you want slimmer logs, you have to pass in the name of your own DLL
1036 which contains the message definitions you want to use in the event log.
1037 """
1038 def __init__(self, appname, dllname=None, logtype="Application"):
1039 logging.Handler.__init__(self)
1040 try:
1041 import win32evtlogutil, win32evtlog
1042 self.appname = appname
1043 self._welu = win32evtlogutil
1044 if not dllname:
1045 dllname = os.path.split(self._welu.__file__)
1046 dllname = os.path.split(dllname[0])
1047 dllname = os.path.join(dllname[0], r'win32service.pyd')
1048 self.dllname = dllname
1049 self.logtype = logtype
1050 self._welu.AddSourceToRegistry(appname, dllname, logtype)
1051 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
1052 self.typemap = {
1053 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
1054 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001055 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +00001056 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
1057 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
1058 }
Brett Cannoncd171c82013-07-04 17:43:24 -04001059 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001060 print("The Python Win32 extensions for NT (service, event "\
1061 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +00001062 self._welu = None
1063
1064 def getMessageID(self, record):
1065 """
1066 Return the message ID for the event record. If you are using your
1067 own messages, you could do this by having the msg passed to the
1068 logger being an ID rather than a formatting string. Then, in here,
1069 you could use a dictionary lookup to get the message ID. This
1070 version returns 1, which is the base message ID in win32service.pyd.
1071 """
1072 return 1
1073
1074 def getEventCategory(self, record):
1075 """
1076 Return the event category for the record.
1077
1078 Override this if you want to specify your own categories. This version
1079 returns 0.
1080 """
1081 return 0
1082
1083 def getEventType(self, record):
1084 """
1085 Return the event type for the record.
1086
1087 Override this if you want to specify your own types. This version does
1088 a mapping using the handler's typemap attribute, which is set up in
1089 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001090 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +00001091 either need to override this method or place a suitable dictionary in
1092 the handler's typemap attribute.
1093 """
1094 return self.typemap.get(record.levelno, self.deftype)
1095
1096 def emit(self, record):
1097 """
1098 Emit a record.
1099
1100 Determine the message ID, event category and event type. Then
1101 log the message in the NT event log.
1102 """
1103 if self._welu:
1104 try:
1105 id = self.getMessageID(record)
1106 cat = self.getEventCategory(record)
1107 type = self.getEventType(record)
1108 msg = self.format(record)
1109 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001110 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001111 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001112
1113 def close(self):
1114 """
1115 Clean up this handler.
1116
1117 You can remove the application name from the registry as a
1118 source of event log entries. However, if you do this, you will
1119 not be able to see the events as you intended in the Event Log
1120 Viewer - it needs to be able to access the registry to get the
1121 DLL name.
1122 """
1123 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001124 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001125
1126class HTTPHandler(logging.Handler):
1127 """
1128 A class which sends records to a Web server, using either GET or
1129 POST semantics.
1130 """
Benjamin Peterson43052a12014-11-23 20:36:44 -06001131 def __init__(self, host, url, method="GET", secure=False, credentials=None,
1132 context=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001133 """
1134 Initialize the instance with the host, the request URL, and the method
1135 ("GET" or "POST")
1136 """
1137 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001138 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001139 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001140 raise ValueError("method must be GET or POST")
Benjamin Peterson43052a12014-11-23 20:36:44 -06001141 if not secure and context is not None:
1142 raise ValueError("context parameter only makes sense "
1143 "with secure=True")
Guido van Rossum57102f82002-11-13 16:15:58 +00001144 self.host = host
1145 self.url = url
1146 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001147 self.secure = secure
1148 self.credentials = credentials
Benjamin Peterson43052a12014-11-23 20:36:44 -06001149 self.context = context
Guido van Rossum57102f82002-11-13 16:15:58 +00001150
Neal Norwitzf297bd12003-04-23 03:49:43 +00001151 def mapLogRecord(self, record):
1152 """
1153 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001154 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001155 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001156 """
1157 return record.__dict__
1158
Guido van Rossum57102f82002-11-13 16:15:58 +00001159 def emit(self, record):
1160 """
1161 Emit a record.
1162
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001163 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001164 """
1165 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001166 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001167 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001168 if self.secure:
Benjamin Peterson43052a12014-11-23 20:36:44 -06001169 h = http.client.HTTPSConnection(host, context=self.context)
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001170 else:
1171 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001172 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001173 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001174 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001175 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001176 sep = '&'
1177 else:
1178 sep = '?'
1179 url = url + "%c%s" % (sep, data)
1180 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001181 # support multiple hosts on one IP address...
1182 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001183 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001184 if i >= 0:
1185 host = host[:i]
1186 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001187 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001188 h.putheader("Content-type",
1189 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001190 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001191 if self.credentials:
1192 import base64
Vinay Sajip1bf197e2016-06-07 21:19:55 +01001193 s = ('%s:%s' % self.credentials).encode('utf-8')
1194 s = 'Basic ' + base64.b64encode(s).strip().decode('ascii')
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001195 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001196 h.endheaders()
1197 if self.method == "POST":
1198 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001199 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001200 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001201 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001202
1203class BufferingHandler(logging.Handler):
1204 """
1205 A handler class which buffers logging records in memory. Whenever each
1206 record is added to the buffer, a check is made to see if the buffer should
1207 be flushed. If it should, then flush() is expected to do what's needed.
1208 """
1209 def __init__(self, capacity):
1210 """
1211 Initialize the handler with the buffer size.
1212 """
1213 logging.Handler.__init__(self)
1214 self.capacity = capacity
1215 self.buffer = []
1216
1217 def shouldFlush(self, record):
1218 """
1219 Should the handler flush its buffer?
1220
1221 Returns true if the buffer is up to capacity. This method can be
1222 overridden to implement custom flushing strategies.
1223 """
1224 return (len(self.buffer) >= self.capacity)
1225
1226 def emit(self, record):
1227 """
1228 Emit a record.
1229
1230 Append the record. If shouldFlush() tells us to, call flush() to process
1231 the buffer.
1232 """
1233 self.buffer.append(record)
1234 if self.shouldFlush(record):
1235 self.flush()
1236
1237 def flush(self):
1238 """
1239 Override to implement custom flushing behaviour.
1240
1241 This version just zaps the buffer to empty.
1242 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001243 self.acquire()
1244 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001245 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001246 finally:
1247 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001248
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001249 def close(self):
1250 """
1251 Close the handler.
1252
1253 This version just flushes and chains to the parent class' close().
1254 """
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001255 try:
1256 self.flush()
1257 finally:
1258 logging.Handler.close(self)
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001259
Guido van Rossum57102f82002-11-13 16:15:58 +00001260class MemoryHandler(BufferingHandler):
1261 """
1262 A handler class which buffers logging records in memory, periodically
1263 flushing them to a target handler. Flushing occurs whenever the buffer
1264 is full, or when an event of a certain severity or greater is seen.
1265 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001266 def __init__(self, capacity, flushLevel=logging.ERROR, target=None,
1267 flushOnClose=True):
Guido van Rossum57102f82002-11-13 16:15:58 +00001268 """
1269 Initialize the handler with the buffer size, the level at which
1270 flushing should occur and an optional target.
1271
1272 Note that without a target being set either here or via setTarget(),
1273 a MemoryHandler is no use to anyone!
Vinay Sajipcccf6062016-07-22 16:27:31 +01001274
1275 The ``flushOnClose`` argument is ``True`` for backward compatibility
1276 reasons - the old behaviour is that when the handler is closed, the
1277 buffer is flushed, even if the flush level hasn't been exceeded nor the
1278 capacity exceeded. To prevent this, set ``flushOnClose`` to ``False``.
Guido van Rossum57102f82002-11-13 16:15:58 +00001279 """
1280 BufferingHandler.__init__(self, capacity)
1281 self.flushLevel = flushLevel
1282 self.target = target
Vinay Sajipcccf6062016-07-22 16:27:31 +01001283 # See Issue #26559 for why this has been added
1284 self.flushOnClose = flushOnClose
Guido van Rossum57102f82002-11-13 16:15:58 +00001285
1286 def shouldFlush(self, record):
1287 """
1288 Check for buffer full or a record at the flushLevel or higher.
1289 """
1290 return (len(self.buffer) >= self.capacity) or \
1291 (record.levelno >= self.flushLevel)
1292
1293 def setTarget(self, target):
1294 """
1295 Set the target handler for this handler.
1296 """
1297 self.target = target
1298
1299 def flush(self):
1300 """
1301 For a MemoryHandler, flushing means just sending the buffered
1302 records to the target, if there is one. Override if you want
1303 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001304
1305 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001306 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001307 self.acquire()
1308 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001309 if self.target:
1310 for record in self.buffer:
1311 self.target.handle(record)
1312 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001313 finally:
1314 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001315
1316 def close(self):
1317 """
Vinay Sajipcccf6062016-07-22 16:27:31 +01001318 Flush, if appropriately configured, set the target to None and lose the
1319 buffer.
Guido van Rossum57102f82002-11-13 16:15:58 +00001320 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001321 try:
Vinay Sajipcccf6062016-07-22 16:27:31 +01001322 if self.flushOnClose:
1323 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001324 finally:
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001325 self.acquire()
1326 try:
1327 self.target = None
1328 BufferingHandler.close(self)
1329 finally:
1330 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001331
1332
1333class QueueHandler(logging.Handler):
1334 """
1335 This handler sends events to a queue. Typically, it would be used together
1336 with a multiprocessing Queue to centralise logging to file in one process
1337 (in a multi-process application), so as to avoid file write contention
1338 between processes.
1339
1340 This code is new in Python 3.2, but this class can be copy pasted into
1341 user code for use with earlier Python versions.
1342 """
1343
1344 def __init__(self, queue):
1345 """
1346 Initialise an instance, using the passed queue.
1347 """
1348 logging.Handler.__init__(self)
1349 self.queue = queue
1350
1351 def enqueue(self, record):
1352 """
1353 Enqueue a record.
1354
1355 The base implementation uses put_nowait. You may want to override
1356 this method if you want to use blocking, timeouts or custom queue
1357 implementations.
1358 """
1359 self.queue.put_nowait(record)
1360
Vinay Sajip0258ce82010-09-22 20:34:53 +00001361 def prepare(self, record):
1362 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001363 Prepares a record for queuing. The object returned by this method is
1364 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001365
1366 The base implementation formats the record to merge the message
1367 and arguments, and removes unpickleable items from the record
1368 in-place.
1369
1370 You might want to override this method if you want to convert
1371 the record to a dict or JSON string, or send a modified copy
1372 of the record while leaving the original intact.
1373 """
1374 # The format operation gets traceback text into record.exc_text
favlladfe3442017-08-01 20:12:26 +02001375 # (if there's exception data), and also returns the formatted
1376 # message. We can then use this to replace the original
Vinay Sajip0258ce82010-09-22 20:34:53 +00001377 # msg + args, as these might be unpickleable. We also zap the
1378 # exc_info attribute, as it's no longer needed and, if not None,
1379 # will typically not be pickleable.
favlladfe3442017-08-01 20:12:26 +02001380 msg = self.format(record)
1381 record.message = msg
1382 record.msg = msg
Vinay Sajip0258ce82010-09-22 20:34:53 +00001383 record.args = None
1384 record.exc_info = None
1385 return record
1386
Vinay Sajip121a1c42010-09-08 10:46:15 +00001387 def emit(self, record):
1388 """
1389 Emit a record.
1390
Vinay Sajip0637d492010-09-23 08:15:54 +00001391 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001392 """
1393 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001394 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001395 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001396 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001397
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001398if threading:
1399 class QueueListener(object):
1400 """
1401 This class implements an internal threaded listener which watches for
1402 LogRecords being added to a queue, removes them and passes them to a
1403 list of handlers for processing.
1404 """
1405 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001406
Vinay Sajip365701a2015-02-09 19:49:00 +00001407 def __init__(self, queue, *handlers, respect_handler_level=False):
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001408 """
1409 Initialise an instance with the specified queue and
1410 handlers.
1411 """
1412 self.queue = queue
1413 self.handlers = handlers
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001414 self._thread = None
Vinay Sajip365701a2015-02-09 19:49:00 +00001415 self.respect_handler_level = respect_handler_level
Vinay Sajip0637d492010-09-23 08:15:54 +00001416
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001417 def dequeue(self, block):
1418 """
1419 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001420
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001421 The base implementation uses get. You may want to override this method
1422 if you want to use timeouts or work with custom queue implementations.
1423 """
1424 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001425
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001426 def start(self):
1427 """
1428 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001429
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001430 This starts up a background thread to monitor the queue for
1431 LogRecords to process.
1432 """
1433 self._thread = t = threading.Thread(target=self._monitor)
Vinay Sajipd61910c2016-09-08 01:13:39 +01001434 t.daemon = True
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001435 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001436
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001437 def prepare(self , record):
1438 """
1439 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001440
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001441 This method just returns the passed-in record. You may want to
1442 override this method if you need to do any custom marshalling or
1443 manipulation of the record before passing it to the handlers.
1444 """
1445 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001446
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001447 def handle(self, record):
1448 """
1449 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001450
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001451 This just loops through the handlers offering them the record
1452 to handle.
1453 """
1454 record = self.prepare(record)
1455 for handler in self.handlers:
Vinay Sajip365701a2015-02-09 19:49:00 +00001456 if not self.respect_handler_level:
1457 process = True
1458 else:
1459 process = record.levelno >= handler.level
1460 if process:
1461 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001462
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001463 def _monitor(self):
1464 """
1465 Monitor the queue for records, and ask the handler
1466 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001467
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001468 This method runs on a separate, internal thread.
1469 The thread will terminate if it sees a sentinel object in the queue.
1470 """
1471 q = self.queue
1472 has_task_done = hasattr(q, 'task_done')
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001473 while True:
1474 try:
Vinay Sajipd61910c2016-09-08 01:13:39 +01001475 record = self.dequeue(True)
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001476 if record is self._sentinel:
1477 break
1478 self.handle(record)
1479 if has_task_done:
1480 q.task_done()
1481 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001482 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001483
Victor Stinner59bec362011-05-02 16:14:16 +02001484 def enqueue_sentinel(self):
1485 """
1486 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001487
Victor Stinner59bec362011-05-02 16:14:16 +02001488 The base implementation uses put_nowait. You may want to override this
1489 method if you want to use timeouts or work with custom queue
1490 implementations.
1491 """
1492 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001493
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001494 def stop(self):
1495 """
1496 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001497
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001498 This asks the thread to terminate, and then waits for it to do so.
1499 Note that if you don't call this before your application exits, there
1500 may be some records still left on the queue, which won't be processed.
1501 """
Victor Stinner59bec362011-05-02 16:14:16 +02001502 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001503 self._thread.join()
1504 self._thread = None