blob: 0107853cfa8b01673e97b16674d91e25c218c435 [file] [log] [blame]
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001# Copyright 2001-2012 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 Sajip0abf61d2012-02-23 19:45:52 +000021Copyright (C) 2001-2012 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
Benjamin Petersonad9d48d2008-04-02 21:49:44 +000026import logging, socket, os, pickle, struct, time, re
Florent Xicluna5252f9f2011-11-07 19:43:05 +010027from codecs import BOM_UTF8
Benjamin Peterson9451a1c2010-03-13 22:30:34 +000028from stat import ST_DEV, ST_INO, ST_MTIME
Vinay Sajipe723e962011-04-15 22:27:17 +010029import queue
Victor Stinnercafa2ef2011-05-02 16:11:28 +020030try:
31 import threading
Vinay Sajip0372e102011-05-05 12:59:14 +010032except ImportError: #pragma: no cover
Victor Stinnercafa2ef2011-05-02 16:11:28 +020033 threading = None
Guido van Rossum57102f82002-11-13 16:15:58 +000034
Guido van Rossum57102f82002-11-13 16:15:58 +000035#
36# Some constants...
37#
38
39DEFAULT_TCP_LOGGING_PORT = 9020
40DEFAULT_UDP_LOGGING_PORT = 9021
41DEFAULT_HTTP_LOGGING_PORT = 9022
42DEFAULT_SOAP_LOGGING_PORT = 9023
43SYSLOG_UDP_PORT = 514
Vinay Sajipcbabd7e2009-10-10 20:32:36 +000044SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000045
Thomas Wouters477c8d52006-05-27 19:21:47 +000046_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
47
Vinay Sajip17c52d82004-07-03 11:48:34 +000048class BaseRotatingHandler(logging.FileHandler):
49 """
50 Base class for handlers that rotate log files at a certain point.
51 Not meant to be instantiated directly. Instead, use RotatingFileHandler
52 or TimedRotatingFileHandler.
53 """
Vinay Sajip23b94d02012-01-04 12:02:26 +000054 def __init__(self, filename, mode, encoding=None, delay=False):
Vinay Sajip17c52d82004-07-03 11:48:34 +000055 """
56 Use the specified filename for streamed logging
57 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000058 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip4600f112005-03-13 09:56:36 +000059 self.mode = mode
60 self.encoding = encoding
Vinay Sajip23b94d02012-01-04 12:02:26 +000061 self.namer = None
62 self.rotator = None
Guido van Rossum57102f82002-11-13 16:15:58 +000063
Vinay Sajip17c52d82004-07-03 11:48:34 +000064 def emit(self, record):
65 """
66 Emit a record.
67
68 Output the record to the file, catering for rollover as described
69 in doRollover().
70 """
Vinay Sajip3970c112004-07-08 10:24:04 +000071 try:
72 if self.shouldRollover(record):
73 self.doRollover()
74 logging.FileHandler.emit(self, record)
Vinay Sajip985ef872011-04-26 19:34:04 +010075 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +000076 raise
Vinay Sajip3970c112004-07-08 10:24:04 +000077 except:
78 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000079
Vinay Sajip23b94d02012-01-04 12:02:26 +000080 def rotation_filename(self, default_name):
81 """
82 Modify the filename of a log file when rotating.
83
84 This is provided so that a custom filename can be provided.
85
86 The default implementation calls the 'namer' attribute of the
87 handler, if it's callable, passing the default name to
88 it. If the attribute isn't callable (the default is None), the name
89 is returned unchanged.
90
91 :param default_name: The default name for the log file.
92 """
93 if not callable(self.namer):
94 result = default_name
95 else:
96 result = self.namer(default_name)
97 return result
98
99 def rotate(self, source, dest):
100 """
101 When rotating, rotate the current log.
102
103 The default implementation calls the 'rotator' attribute of the
104 handler, if it's callable, passing the source and dest arguments to
105 it. If the attribute isn't callable (the default is None), the source
106 is simply renamed to the destination.
107
108 :param source: The source filename. This is normally the base
109 filename, e.g. 'test.log'
110 :param dest: The destination filename. This is normally
111 what the source is rotated to, e.g. 'test.log.1'.
112 """
113 if not callable(self.rotator):
114 os.rename(source, dest)
115 else:
116 self.rotator(source, dest)
117
Vinay Sajip17c52d82004-07-03 11:48:34 +0000118class RotatingFileHandler(BaseRotatingHandler):
119 """
120 Handler for logging to a set of files, which switches from one file
121 to the next when the current file reaches a certain size.
122 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000123 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
Guido van Rossum57102f82002-11-13 16:15:58 +0000124 """
125 Open the specified file and use it as the stream for logging.
126
127 By default, the file grows indefinitely. You can specify particular
128 values of maxBytes and backupCount to allow the file to rollover at
129 a predetermined size.
130
131 Rollover occurs whenever the current log file is nearly maxBytes in
132 length. If backupCount is >= 1, the system will successively create
133 new files with the same pathname as the base file, but with extensions
134 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
135 and a base file name of "app.log", you would get "app.log",
136 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
137 written to is always "app.log" - when it gets filled up, it is closed
138 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
139 exist, then they are renamed to "app.log.2", "app.log.3" etc.
140 respectively.
141
142 If maxBytes is zero, rollover never occurs.
143 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000144 # If rotation/rollover is wanted, it doesn't make sense to use another
145 # mode. If for example 'w' were specified, then if there were multiple
146 # runs of the calling application, the logs from previous runs would be
147 # lost if the 'w' is respected, because the log file would be truncated
148 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000149 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000150 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000151 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000152 self.maxBytes = maxBytes
153 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000154
155 def doRollover(self):
156 """
157 Do a rollover, as described in __init__().
158 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000159 if self.stream:
160 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000161 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000162 if self.backupCount > 0:
163 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000164 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
165 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
166 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000167 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000168 if os.path.exists(dfn):
169 os.remove(dfn)
170 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000171 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000172 if os.path.exists(dfn):
173 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000174 self.rotate(self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000175 self.mode = 'w'
176 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000177
Vinay Sajip17c52d82004-07-03 11:48:34 +0000178 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000179 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000180 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000181
Vinay Sajip17c52d82004-07-03 11:48:34 +0000182 Basically, see if the supplied record would cause the file to exceed
183 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000184 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000185 if self.stream is None: # delay was set...
186 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000187 if self.maxBytes > 0: # are we rolling over?
188 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000189 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000190 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000191 return 1
192 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000193
Vinay Sajip17c52d82004-07-03 11:48:34 +0000194class TimedRotatingFileHandler(BaseRotatingHandler):
195 """
196 Handler for logging to a file, rotating the log file at certain timed
197 intervals.
198
199 If backupCount is > 0, when rollover is done, no more than backupCount
200 files are kept - the oldest ones are deleted.
201 """
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000202 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000203 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000204 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000205 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000206 self.utc = utc
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
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000249 if os.path.exists(filename):
250 t = os.stat(filename)[ST_MTIME]
251 else:
252 t = int(time.time())
253 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000254
Vinay Sajipaffbd872009-06-11 10:11:47 +0000255 def computeRollover(self, currentTime):
256 """
257 Work out the rollover time based on the specified time.
258 """
259 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000260 # If we are rolling over at midnight or weekly, then the interval is already known.
261 # What we need to figure out is WHEN the next interval is. In other words,
262 # if you are rolling over at midnight, then your base interval is 1 day,
263 # but you want to start that one day clock at midnight, not now. So, we
264 # have to fudge the rolloverAt value in order to trigger the first rollover
265 # at the right time. After that, the regular interval will take care of
266 # the rest. Note that this code doesn't care about leap seconds. :)
267 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
268 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000269 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000270 t = time.gmtime(currentTime)
271 else:
272 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000273 currentHour = t[3]
274 currentMinute = t[4]
275 currentSecond = t[5]
276 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000277 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
278 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000279 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000280 # If we are rolling over on a certain day, add in the number of days until
281 # the next rollover, but offset by 1 since we just calculated the time
282 # until the next day starts. There are three cases:
283 # Case 1) The day to rollover is today; in this case, do nothing
284 # Case 2) The day to rollover is further in the interval (i.e., today is
285 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
286 # next rollover is simply 6 - 2 - 1, or 3.
287 # Case 3) The day to rollover is behind us in the interval (i.e., today
288 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
289 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
290 # number of days left in the current week (1) plus the number
291 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000292 # The calculations described in 2) and 3) above need to have a day added.
293 # This is because the above time calculation takes us to midnight on this
294 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000295 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000296 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000297 if day != self.dayOfWeek:
298 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000299 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000300 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000301 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000302 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
303 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000304 dstNow = t[-1]
305 dstAtRollover = time.localtime(newRolloverAt)[-1]
306 if dstNow != dstAtRollover:
307 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
308 newRolloverAt = newRolloverAt - 3600
309 else: # DST bows out before next rollover, so we need to add an hour
310 newRolloverAt = newRolloverAt + 3600
Vinay Sajipaffbd872009-06-11 10:11:47 +0000311 result = newRolloverAt
312 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000313
314 def shouldRollover(self, record):
315 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000316 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000317
318 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000319 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000320 """
321 t = int(time.time())
322 if t >= self.rolloverAt:
323 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000324 return 0
325
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000326 def getFilesToDelete(self):
327 """
328 Determine the files to delete when rolling over.
329
330 More specific than the earlier method, which just used glob.glob().
331 """
332 dirName, baseName = os.path.split(self.baseFilename)
333 fileNames = os.listdir(dirName)
334 result = []
335 prefix = baseName + "."
336 plen = len(prefix)
337 for fileName in fileNames:
338 if fileName[:plen] == prefix:
339 suffix = fileName[plen:]
340 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000341 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000342 result.sort()
343 if len(result) < self.backupCount:
344 result = []
345 else:
346 result = result[:len(result) - self.backupCount]
347 return result
348
Vinay Sajip17c52d82004-07-03 11:48:34 +0000349 def doRollover(self):
350 """
351 do a rollover; in this case, a date/time stamp is appended to the filename
352 when the rollover happens. However, you want the file to be named for the
353 start of the interval, not the current time. If there is a backup count,
354 then we have to get a list of matching filenames, sort them and remove
355 the one with the oldest suffix.
356 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000357 if self.stream:
358 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000359 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000360 # get the time that this sequence started at and make it a TimeTuple
361 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000362 if self.utc:
363 timeTuple = time.gmtime(t)
364 else:
365 timeTuple = time.localtime(t)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000366 dfn = self.rotation_filename(self.baseFilename + "." +
367 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000368 if os.path.exists(dfn):
369 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000370 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000371 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000372 for s in self.getFilesToDelete():
373 os.remove(s)
Thomas Woutersb2137042007-02-01 18:02:27 +0000374 self.mode = 'w'
375 self.stream = self._open()
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000376 currentTime = int(time.time())
Vinay Sajipaffbd872009-06-11 10:11:47 +0000377 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000378 while newRolloverAt <= currentTime:
379 newRolloverAt = newRolloverAt + self.interval
380 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000381 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000382 dstNow = time.localtime(currentTime)[-1]
383 dstAtRollover = time.localtime(newRolloverAt)[-1]
384 if dstNow != dstAtRollover:
385 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
386 newRolloverAt = newRolloverAt - 3600
387 else: # DST bows out before next rollover, so we need to add an hour
388 newRolloverAt = newRolloverAt + 3600
389 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000390
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000391class WatchedFileHandler(logging.FileHandler):
392 """
393 A handler for logging to a file, which watches the file
394 to see if it has changed while in use. This can happen because of
395 usage of programs such as newsyslog and logrotate which perform
396 log file rotation. This handler, intended for use under Unix,
397 watches the file to see if it has changed since the last emit.
398 (A file has changed if its device or inode have changed.)
399 If it has changed, the old file stream is closed, and the file
400 opened to get a new stream.
401
402 This handler is not appropriate for use under Windows, because
403 under Windows open files cannot be moved or renamed - logging
404 opens the files with exclusive locks - and so there is no need
405 for such a handler. Furthermore, ST_INO is not supported under
406 Windows; stat always returns zero for this value.
407
408 This handler is based on a suggestion and patch by Chad J.
409 Schroeder.
410 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000411 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000412 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
413 if not os.path.exists(self.baseFilename):
414 self.dev, self.ino = -1, -1
415 else:
416 stat = os.stat(self.baseFilename)
417 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000418
419 def emit(self, record):
420 """
421 Emit a record.
422
423 First check if the underlying file has changed, and if it
424 has, close the old stream and reopen the file to get the
425 current stream.
426 """
427 if not os.path.exists(self.baseFilename):
428 stat = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100429 changed = True
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000430 else:
431 stat = os.stat(self.baseFilename)
432 changed = (stat[ST_DEV] != self.dev) or (stat[ST_INO] != self.ino)
Christian Heimese7a15bb2008-01-24 16:21:45 +0000433 if changed and self.stream is not None:
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000434 self.stream.flush()
435 self.stream.close()
436 self.stream = self._open()
437 if stat is None:
438 stat = os.stat(self.baseFilename)
439 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
440 logging.FileHandler.emit(self, record)
441
Guido van Rossum57102f82002-11-13 16:15:58 +0000442class SocketHandler(logging.Handler):
443 """
444 A handler class which writes logging records, in pickle format, to
445 a streaming socket. The socket is kept open across logging calls.
446 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000447 The pickle which is sent is that of the LogRecord's attribute dictionary
448 (__dict__), so that the receiver does not need to have the logging module
449 installed in order to process the logging event.
450
451 To unpickle the record at the receiving end into a LogRecord, use the
452 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000453 """
454
455 def __init__(self, host, port):
456 """
457 Initializes the handler with a specific host address and port.
458
Vinay Sajipde19e082011-04-30 21:52:26 +0100459 When the attribute *closeOnError* is set to True - if a socket error
460 occurs, the socket is silently closed and then reopened on the next
461 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000462 """
463 logging.Handler.__init__(self)
464 self.host = host
465 self.port = port
466 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100467 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000468 self.retryTime = None
469 #
470 # Exponential backoff parameters.
471 #
472 self.retryStart = 1.0
473 self.retryMax = 30.0
474 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000475
Guido van Rossumd8faa362007-04-27 19:54:29 +0000476 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000477 """
478 A factory method which allows subclasses to define the precise
479 type of socket they want.
480 """
481 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000482 if hasattr(s, 'settimeout'):
483 s.settimeout(timeout)
Vinay Sajipb37cd392011-05-07 15:55:47 +0100484 try:
485 s.connect((self.host, self.port))
486 return s
487 except socket.error:
488 s.close()
489 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000490
Vinay Sajip48cfe382004-02-20 13:17:27 +0000491 def createSocket(self):
492 """
493 Try to create a socket, using an exponential backoff with
494 a max retry time. Thanks to Robert Olson for the original patch
495 (SF #815911) which has been slightly refactored.
496 """
497 now = time.time()
498 # Either retryTime is None, in which case this
499 # is the first time back after a disconnect, or
500 # we've waited long enough.
501 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100502 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000503 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000504 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000505 if attempt:
506 try:
507 self.sock = self.makeSocket()
508 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000509 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000510 #Creation failed, so set the retry time and return.
511 if self.retryTime is None:
512 self.retryPeriod = self.retryStart
513 else:
514 self.retryPeriod = self.retryPeriod * self.retryFactor
515 if self.retryPeriod > self.retryMax:
516 self.retryPeriod = self.retryMax
517 self.retryTime = now + self.retryPeriod
518
Guido van Rossum57102f82002-11-13 16:15:58 +0000519 def send(self, s):
520 """
521 Send a pickled string to the socket.
522
523 This function allows for partial sends which can happen when the
524 network is busy.
525 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000526 if self.sock is None:
527 self.createSocket()
528 #self.sock can be None either because we haven't reached the retry
529 #time yet, or because we have reached the retry time and retried,
530 #but are still unable to connect.
531 if self.sock:
532 try:
533 if hasattr(self.sock, "sendall"):
534 self.sock.sendall(s)
Vinay Sajip7fe1d512011-04-28 12:04:58 +0100535 else: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000536 sentsofar = 0
537 left = len(s)
538 while left > 0:
539 sent = self.sock.send(s[sentsofar:])
540 sentsofar = sentsofar + sent
541 left = left - sent
Vinay Sajip7fe1d512011-04-28 12:04:58 +0100542 except socket.error: #pragma: no cover
Vinay Sajip48cfe382004-02-20 13:17:27 +0000543 self.sock.close()
544 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000545
546 def makePickle(self, record):
547 """
548 Pickles the record in binary format with a length prefix, and
549 returns it ready for transmission across the socket.
550 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000551 ei = record.exc_info
552 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000553 dummy = self.format(record) # just to get traceback text into record.exc_text
554 record.exc_info = None # to avoid Unpickleable error
Guido van Rossumba205d62006-08-17 08:57:26 +0000555 s = pickle.dumps(record.__dict__, 1)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000556 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000557 record.exc_info = ei # for next handler
Guido van Rossum57102f82002-11-13 16:15:58 +0000558 slen = struct.pack(">L", len(s))
559 return slen + s
560
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000561 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000562 """
563 Handle an error during logging.
564
565 An error has occurred during logging. Most likely cause -
566 connection lost. Close the socket so that we can retry on the
567 next event.
568 """
569 if self.closeOnError and self.sock:
570 self.sock.close()
571 self.sock = None #try to reconnect next time
572 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000573 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000574
575 def emit(self, record):
576 """
577 Emit a record.
578
579 Pickles the record and writes it to the socket in binary format.
580 If there is an error with the socket, silently drop the packet.
581 If there was a problem with the socket, re-establishes the
582 socket.
583 """
584 try:
585 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000586 self.send(s)
Vinay Sajip985ef872011-04-26 19:34:04 +0100587 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000588 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000589 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000590 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000591
592 def close(self):
593 """
594 Closes the socket.
595 """
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000596 with self.lock:
597 if self.sock:
598 self.sock.close()
599 self.sock = None
600 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000601
602class DatagramHandler(SocketHandler):
603 """
604 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000605 a datagram socket. The pickle which is sent is that of the LogRecord's
606 attribute dictionary (__dict__), so that the receiver does not need to
607 have the logging module installed in order to process the logging event.
608
609 To unpickle the record at the receiving end into a LogRecord, use the
610 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000611
612 """
613 def __init__(self, host, port):
614 """
615 Initializes the handler with a specific host address and port.
616 """
617 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100618 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000619
620 def makeSocket(self):
621 """
622 The factory method of SocketHandler is here overridden to create
623 a UDP socket (SOCK_DGRAM).
624 """
625 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
626 return s
627
628 def send(self, s):
629 """
630 Send a pickled string to a socket.
631
632 This function no longer allows for partial sends which can happen
633 when the network is busy - UDP does not guarantee delivery and
634 can deliver packets out of sequence.
635 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000636 if self.sock is None:
637 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000638 self.sock.sendto(s, (self.host, self.port))
639
640class SysLogHandler(logging.Handler):
641 """
642 A handler class which sends formatted logging records to a syslog
643 server. Based on Sam Rushing's syslog module:
644 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
645 Contributed by Nicolas Untz (after which minor refactoring changes
646 have been made).
647 """
648
649 # from <linux/sys/syslog.h>:
650 # ======================================================================
651 # priorities/facilities are encoded into a single 32-bit quantity, where
652 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
653 # facility (0-big number). Both the priorities and the facilities map
654 # roughly one-to-one to strings in the syslogd(8) source code. This
655 # mapping is included in this file.
656 #
657 # priorities (these are ordered)
658
659 LOG_EMERG = 0 # system is unusable
660 LOG_ALERT = 1 # action must be taken immediately
661 LOG_CRIT = 2 # critical conditions
662 LOG_ERR = 3 # error conditions
663 LOG_WARNING = 4 # warning conditions
664 LOG_NOTICE = 5 # normal but significant condition
665 LOG_INFO = 6 # informational
666 LOG_DEBUG = 7 # debug-level messages
667
668 # facility codes
669 LOG_KERN = 0 # kernel messages
670 LOG_USER = 1 # random user-level messages
671 LOG_MAIL = 2 # mail system
672 LOG_DAEMON = 3 # system daemons
673 LOG_AUTH = 4 # security/authorization messages
674 LOG_SYSLOG = 5 # messages generated internally by syslogd
675 LOG_LPR = 6 # line printer subsystem
676 LOG_NEWS = 7 # network news subsystem
677 LOG_UUCP = 8 # UUCP subsystem
678 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000679 LOG_AUTHPRIV = 10 # security/authorization messages (private)
680 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000681
682 # other codes through 15 reserved for system use
683 LOG_LOCAL0 = 16 # reserved for local use
684 LOG_LOCAL1 = 17 # reserved for local use
685 LOG_LOCAL2 = 18 # reserved for local use
686 LOG_LOCAL3 = 19 # reserved for local use
687 LOG_LOCAL4 = 20 # reserved for local use
688 LOG_LOCAL5 = 21 # reserved for local use
689 LOG_LOCAL6 = 22 # reserved for local use
690 LOG_LOCAL7 = 23 # reserved for local use
691
692 priority_names = {
693 "alert": LOG_ALERT,
694 "crit": LOG_CRIT,
695 "critical": LOG_CRIT,
696 "debug": LOG_DEBUG,
697 "emerg": LOG_EMERG,
698 "err": LOG_ERR,
699 "error": LOG_ERR, # DEPRECATED
700 "info": LOG_INFO,
701 "notice": LOG_NOTICE,
702 "panic": LOG_EMERG, # DEPRECATED
703 "warn": LOG_WARNING, # DEPRECATED
704 "warning": LOG_WARNING,
705 }
706
707 facility_names = {
708 "auth": LOG_AUTH,
709 "authpriv": LOG_AUTHPRIV,
710 "cron": LOG_CRON,
711 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000712 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000713 "kern": LOG_KERN,
714 "lpr": LOG_LPR,
715 "mail": LOG_MAIL,
716 "news": LOG_NEWS,
717 "security": LOG_AUTH, # DEPRECATED
718 "syslog": LOG_SYSLOG,
719 "user": LOG_USER,
720 "uucp": LOG_UUCP,
721 "local0": LOG_LOCAL0,
722 "local1": LOG_LOCAL1,
723 "local2": LOG_LOCAL2,
724 "local3": LOG_LOCAL3,
725 "local4": LOG_LOCAL4,
726 "local5": LOG_LOCAL5,
727 "local6": LOG_LOCAL6,
728 "local7": LOG_LOCAL7,
729 }
730
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000731 #The map below appears to be trivially lowercasing the key. However,
732 #there's more to it than meets the eye - in some locales, lowercasing
733 #gives unexpected results. See SF #1524081: in the Turkish locale,
734 #"INFO".lower() != "info"
735 priority_map = {
736 "DEBUG" : "debug",
737 "INFO" : "info",
738 "WARNING" : "warning",
739 "ERROR" : "error",
740 "CRITICAL" : "critical"
741 }
742
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000743 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
744 facility=LOG_USER, socktype=socket.SOCK_DGRAM):
Guido van Rossum57102f82002-11-13 16:15:58 +0000745 """
746 Initialize a handler.
747
Guido van Rossume7ba4952007-06-06 23:52:48 +0000748 If address is specified as a string, a UNIX socket is used. To log to a
749 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000750 If facility is not specified, LOG_USER is used.
751 """
752 logging.Handler.__init__(self)
753
754 self.address = address
755 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000756 self.socktype = socktype
757
Guido van Rossum13257902007-06-07 23:15:56 +0000758 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100759 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000760 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000761 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100762 self.unixsocket = False
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000763 self.socket = socket.socket(socket.AF_INET, socktype)
764 if socktype == socket.SOCK_STREAM:
765 self.socket.connect(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000766 self.formatter = None
767
Vinay Sajipa1974c12005-01-13 08:23:56 +0000768 def _connect_unixsocket(self, address):
769 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
770 # syslog may require either DGRAM or STREAM sockets
771 try:
772 self.socket.connect(address)
773 except socket.error:
774 self.socket.close()
775 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Vinay Sajip8b6b53f2005-11-09 13:55:13 +0000776 self.socket.connect(address)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000777
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000778 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000779 """
780 Encode the facility and priority. You can pass in strings or
781 integers - if strings are passed, the facility_names and
782 priority_names mapping dictionaries are used to convert them to
783 integers.
784 """
Guido van Rossum13257902007-06-07 23:15:56 +0000785 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000786 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000787 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000788 priority = self.priority_names[priority]
789 return (facility << 3) | priority
790
791 def close (self):
792 """
793 Closes the socket.
794 """
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000795 with self.lock:
Vinay Sajip57c22372012-02-23 20:03:49 +0000796 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000797 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000798
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000799 def mapPriority(self, levelName):
800 """
801 Map a logging level name to a key in the priority_names map.
802 This is useful in two scenarios: when custom levels are being
803 used, and in the case where you can't do a straightforward
804 mapping by lowercasing the logging level name because of locale-
805 specific issues (see SF #1524081).
806 """
807 return self.priority_map.get(levelName, "warning")
808
Vinay Sajip2353e352011-06-27 15:40:06 +0100809 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100810 append_nul = True # some old syslog daemons expect a NUL terminator
811
Guido van Rossum57102f82002-11-13 16:15:58 +0000812 def emit(self, record):
813 """
814 Emit a record.
815
816 The record is formatted, and then sent to the syslog server. If
817 exception information is present, it is NOT sent to the server.
818 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100819 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100820 if self.ident:
821 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100822 if self.append_nul:
823 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000824 """
825 We need to convert record level to lowercase, maybe this will
826 change in the future.
827 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000828 prio = '<%d>' % self.encodePriority(self.facility,
829 self.mapPriority(record.levelname))
830 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000831 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000832 msg = msg.encode('utf-8')
Florent Xicluna5252f9f2011-11-07 19:43:05 +0100833 msg = prio + BOM_UTF8 + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000834 try:
835 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000836 try:
837 self.socket.send(msg)
838 except socket.error:
839 self._connect_unixsocket(self.address)
840 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000841 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000842 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000843 else:
844 self.socket.sendall(msg)
Vinay Sajip985ef872011-04-26 19:34:04 +0100845 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip85c19092005-10-31 13:14:19 +0000846 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000847 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000848 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000849
850class SMTPHandler(logging.Handler):
851 """
852 A handler class which sends an SMTP email for each logging event.
853 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000854 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip25fcd222009-12-06 18:05:04 +0000855 credentials=None, secure=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000856 """
857 Initialize the handler.
858
859 Initialize the instance with the from and to addresses and subject
860 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000861 (host, port) tuple format for the mailhost argument. To specify
862 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000863 for the credentials argument. To specify the use of a secure
864 protocol (TLS), pass in a tuple for the secure argument. This will
865 only be used when authentication credentials are supplied. The tuple
866 will be either an empty tuple, or a single-value tuple with the name
867 of a keyfile, or a 2-value tuple with the names of the keyfile and
868 certificate file. (This tuple is passed to the `starttls` method).
Guido van Rossum57102f82002-11-13 16:15:58 +0000869 """
870 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000871 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000872 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000873 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000874 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000875 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000876 self.username, self.password = credentials
877 else:
878 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000879 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000880 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000881 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000882 self.toaddrs = toaddrs
883 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000884 self.secure = secure
Guido van Rossum57102f82002-11-13 16:15:58 +0000885
886 def getSubject(self, record):
887 """
888 Determine the subject for the email.
889
890 If you want to specify a subject line which is record-dependent,
891 override this method.
892 """
893 return self.subject
894
895 def emit(self, record):
896 """
897 Emit a record.
898
899 Format the record and send it to the specified addressees.
900 """
901 try:
902 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000903 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000904 port = self.mailport
905 if not port:
906 port = smtplib.SMTP_PORT
907 smtp = smtplib.SMTP(self.mailhost, port)
908 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000909 msg = "From: %s\r\nTo: %s\r\nSubject: %s\r\nDate: %s\r\n\r\n%s" % (
Guido van Rossum57102f82002-11-13 16:15:58 +0000910 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000911 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000912 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000913 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000914 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000915 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000916 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000917 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000918 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000919 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000920 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
921 smtp.quit()
Vinay Sajip985ef872011-04-26 19:34:04 +0100922 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000923 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000924 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000925 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000926
927class NTEventLogHandler(logging.Handler):
928 """
929 A handler class which sends events to the NT Event Log. Adds a
930 registry entry for the specified application name. If no dllname is
931 provided, win32service.pyd (which contains some basic message
932 placeholders) is used. Note that use of these placeholders will make
933 your event logs big, as the entire message source is held in the log.
934 If you want slimmer logs, you have to pass in the name of your own DLL
935 which contains the message definitions you want to use in the event log.
936 """
937 def __init__(self, appname, dllname=None, logtype="Application"):
938 logging.Handler.__init__(self)
939 try:
940 import win32evtlogutil, win32evtlog
941 self.appname = appname
942 self._welu = win32evtlogutil
943 if not dllname:
944 dllname = os.path.split(self._welu.__file__)
945 dllname = os.path.split(dllname[0])
946 dllname = os.path.join(dllname[0], r'win32service.pyd')
947 self.dllname = dllname
948 self.logtype = logtype
949 self._welu.AddSourceToRegistry(appname, dllname, logtype)
950 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
951 self.typemap = {
952 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
953 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000954 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000955 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
956 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
957 }
958 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000959 print("The Python Win32 extensions for NT (service, event "\
960 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000961 self._welu = None
962
963 def getMessageID(self, record):
964 """
965 Return the message ID for the event record. If you are using your
966 own messages, you could do this by having the msg passed to the
967 logger being an ID rather than a formatting string. Then, in here,
968 you could use a dictionary lookup to get the message ID. This
969 version returns 1, which is the base message ID in win32service.pyd.
970 """
971 return 1
972
973 def getEventCategory(self, record):
974 """
975 Return the event category for the record.
976
977 Override this if you want to specify your own categories. This version
978 returns 0.
979 """
980 return 0
981
982 def getEventType(self, record):
983 """
984 Return the event type for the record.
985
986 Override this if you want to specify your own types. This version does
987 a mapping using the handler's typemap attribute, which is set up in
988 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000989 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000990 either need to override this method or place a suitable dictionary in
991 the handler's typemap attribute.
992 """
993 return self.typemap.get(record.levelno, self.deftype)
994
995 def emit(self, record):
996 """
997 Emit a record.
998
999 Determine the message ID, event category and event type. Then
1000 log the message in the NT event log.
1001 """
1002 if self._welu:
1003 try:
1004 id = self.getMessageID(record)
1005 cat = self.getEventCategory(record)
1006 type = self.getEventType(record)
1007 msg = self.format(record)
1008 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip985ef872011-04-26 19:34:04 +01001009 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001010 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001011 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001012 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001013
1014 def close(self):
1015 """
1016 Clean up this handler.
1017
1018 You can remove the application name from the registry as a
1019 source of event log entries. However, if you do this, you will
1020 not be able to see the events as you intended in the Event Log
1021 Viewer - it needs to be able to access the registry to get the
1022 DLL name.
1023 """
1024 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001025 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001026
1027class HTTPHandler(logging.Handler):
1028 """
1029 A class which sends records to a Web server, using either GET or
1030 POST semantics.
1031 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001032 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001033 """
1034 Initialize the instance with the host, the request URL, and the method
1035 ("GET" or "POST")
1036 """
1037 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001038 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001039 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001040 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001041 self.host = host
1042 self.url = url
1043 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001044 self.secure = secure
1045 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001046
Neal Norwitzf297bd12003-04-23 03:49:43 +00001047 def mapLogRecord(self, record):
1048 """
1049 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001050 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001051 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001052 """
1053 return record.__dict__
1054
Guido van Rossum57102f82002-11-13 16:15:58 +00001055 def emit(self, record):
1056 """
1057 Emit a record.
1058
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001059 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001060 """
1061 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001062 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001063 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001064 if self.secure:
1065 h = http.client.HTTPSConnection(host)
1066 else:
1067 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001068 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001069 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001070 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001071 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001072 sep = '&'
1073 else:
1074 sep = '?'
1075 url = url + "%c%s" % (sep, data)
1076 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001077 # support multiple hosts on one IP address...
1078 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001079 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001080 if i >= 0:
1081 host = host[:i]
1082 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001083 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001084 h.putheader("Content-type",
1085 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001086 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001087 if self.credentials:
1088 import base64
1089 s = ('u%s:%s' % self.credentials).encode('utf-8')
1090 s = 'Basic ' + base64.b64encode(s).strip()
1091 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001092 h.endheaders()
1093 if self.method == "POST":
1094 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001095 h.getresponse() #can't do anything with the result
Vinay Sajip985ef872011-04-26 19:34:04 +01001096 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001097 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001098 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001099 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001100
1101class BufferingHandler(logging.Handler):
1102 """
1103 A handler class which buffers logging records in memory. Whenever each
1104 record is added to the buffer, a check is made to see if the buffer should
1105 be flushed. If it should, then flush() is expected to do what's needed.
1106 """
1107 def __init__(self, capacity):
1108 """
1109 Initialize the handler with the buffer size.
1110 """
1111 logging.Handler.__init__(self)
1112 self.capacity = capacity
1113 self.buffer = []
1114
1115 def shouldFlush(self, record):
1116 """
1117 Should the handler flush its buffer?
1118
1119 Returns true if the buffer is up to capacity. This method can be
1120 overridden to implement custom flushing strategies.
1121 """
1122 return (len(self.buffer) >= self.capacity)
1123
1124 def emit(self, record):
1125 """
1126 Emit a record.
1127
1128 Append the record. If shouldFlush() tells us to, call flush() to process
1129 the buffer.
1130 """
1131 self.buffer.append(record)
1132 if self.shouldFlush(record):
1133 self.flush()
1134
1135 def flush(self):
1136 """
1137 Override to implement custom flushing behaviour.
1138
1139 This version just zaps the buffer to empty.
1140 """
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001141 with self.lock:
1142 self.buffer = []
Guido van Rossum57102f82002-11-13 16:15:58 +00001143
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001144 def close(self):
1145 """
1146 Close the handler.
1147
1148 This version just flushes and chains to the parent class' close().
1149 """
1150 self.flush()
1151 logging.Handler.close(self)
1152
Guido van Rossum57102f82002-11-13 16:15:58 +00001153class MemoryHandler(BufferingHandler):
1154 """
1155 A handler class which buffers logging records in memory, periodically
1156 flushing them to a target handler. Flushing occurs whenever the buffer
1157 is full, or when an event of a certain severity or greater is seen.
1158 """
1159 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1160 """
1161 Initialize the handler with the buffer size, the level at which
1162 flushing should occur and an optional target.
1163
1164 Note that without a target being set either here or via setTarget(),
1165 a MemoryHandler is no use to anyone!
1166 """
1167 BufferingHandler.__init__(self, capacity)
1168 self.flushLevel = flushLevel
1169 self.target = target
1170
1171 def shouldFlush(self, record):
1172 """
1173 Check for buffer full or a record at the flushLevel or higher.
1174 """
1175 return (len(self.buffer) >= self.capacity) or \
1176 (record.levelno >= self.flushLevel)
1177
1178 def setTarget(self, target):
1179 """
1180 Set the target handler for this handler.
1181 """
1182 self.target = target
1183
1184 def flush(self):
1185 """
1186 For a MemoryHandler, flushing means just sending the buffered
1187 records to the target, if there is one. Override if you want
1188 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001189
1190 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001191 """
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001192 with self.lock:
1193 if self.target:
1194 for record in self.buffer:
1195 self.target.handle(record)
1196 self.buffer = []
Guido van Rossum57102f82002-11-13 16:15:58 +00001197
1198 def close(self):
1199 """
1200 Flush, set the target to None and lose the buffer.
1201 """
1202 self.flush()
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001203 with self.lock:
1204 self.target = None
1205 BufferingHandler.close(self)
Vinay Sajip121a1c42010-09-08 10:46:15 +00001206
1207
1208class QueueHandler(logging.Handler):
1209 """
1210 This handler sends events to a queue. Typically, it would be used together
1211 with a multiprocessing Queue to centralise logging to file in one process
1212 (in a multi-process application), so as to avoid file write contention
1213 between processes.
1214
1215 This code is new in Python 3.2, but this class can be copy pasted into
1216 user code for use with earlier Python versions.
1217 """
1218
1219 def __init__(self, queue):
1220 """
1221 Initialise an instance, using the passed queue.
1222 """
1223 logging.Handler.__init__(self)
1224 self.queue = queue
1225
1226 def enqueue(self, record):
1227 """
1228 Enqueue a record.
1229
1230 The base implementation uses put_nowait. You may want to override
1231 this method if you want to use blocking, timeouts or custom queue
1232 implementations.
1233 """
1234 self.queue.put_nowait(record)
1235
Vinay Sajip0258ce82010-09-22 20:34:53 +00001236 def prepare(self, record):
1237 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001238 Prepares a record for queuing. The object returned by this method is
1239 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001240
1241 The base implementation formats the record to merge the message
1242 and arguments, and removes unpickleable items from the record
1243 in-place.
1244
1245 You might want to override this method if you want to convert
1246 the record to a dict or JSON string, or send a modified copy
1247 of the record while leaving the original intact.
1248 """
1249 # The format operation gets traceback text into record.exc_text
1250 # (if there's exception data), and also puts the message into
1251 # record.message. We can then use this to replace the original
1252 # msg + args, as these might be unpickleable. We also zap the
1253 # exc_info attribute, as it's no longer needed and, if not None,
1254 # will typically not be pickleable.
1255 self.format(record)
1256 record.msg = record.message
1257 record.args = None
1258 record.exc_info = None
1259 return record
1260
Vinay Sajip121a1c42010-09-08 10:46:15 +00001261 def emit(self, record):
1262 """
1263 Emit a record.
1264
Vinay Sajip0637d492010-09-23 08:15:54 +00001265 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001266 """
1267 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001268 self.enqueue(self.prepare(record))
Vinay Sajip985ef872011-04-26 19:34:04 +01001269 except (KeyboardInterrupt, SystemExit): #pragma: no cover
Vinay Sajip121a1c42010-09-08 10:46:15 +00001270 raise
1271 except:
1272 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001273
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001274if threading:
1275 class QueueListener(object):
1276 """
1277 This class implements an internal threaded listener which watches for
1278 LogRecords being added to a queue, removes them and passes them to a
1279 list of handlers for processing.
1280 """
1281 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001282
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001283 def __init__(self, queue, *handlers):
1284 """
1285 Initialise an instance with the specified queue and
1286 handlers.
1287 """
1288 self.queue = queue
1289 self.handlers = handlers
1290 self._stop = threading.Event()
1291 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001292
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001293 def dequeue(self, block):
1294 """
1295 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001296
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001297 The base implementation uses get. You may want to override this method
1298 if you want to use timeouts or work with custom queue implementations.
1299 """
1300 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001301
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001302 def start(self):
1303 """
1304 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001305
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001306 This starts up a background thread to monitor the queue for
1307 LogRecords to process.
1308 """
1309 self._thread = t = threading.Thread(target=self._monitor)
1310 t.setDaemon(True)
1311 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001312
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001313 def prepare(self , record):
1314 """
1315 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001316
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001317 This method just returns the passed-in record. You may want to
1318 override this method if you need to do any custom marshalling or
1319 manipulation of the record before passing it to the handlers.
1320 """
1321 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001322
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001323 def handle(self, record):
1324 """
1325 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001326
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001327 This just loops through the handlers offering them the record
1328 to handle.
1329 """
1330 record = self.prepare(record)
1331 for handler in self.handlers:
1332 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001333
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001334 def _monitor(self):
1335 """
1336 Monitor the queue for records, and ask the handler
1337 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001338
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001339 This method runs on a separate, internal thread.
1340 The thread will terminate if it sees a sentinel object in the queue.
1341 """
1342 q = self.queue
1343 has_task_done = hasattr(q, 'task_done')
1344 while not self._stop.isSet():
1345 try:
1346 record = self.dequeue(True)
1347 if record is self._sentinel:
1348 break
1349 self.handle(record)
1350 if has_task_done:
1351 q.task_done()
1352 except queue.Empty:
1353 pass
1354 # There might still be records in the queue.
1355 while True:
1356 try:
1357 record = self.dequeue(False)
1358 if record is self._sentinel:
1359 break
1360 self.handle(record)
1361 if has_task_done:
1362 q.task_done()
1363 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001364 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001365
Victor Stinner59bec362011-05-02 16:14:16 +02001366 def enqueue_sentinel(self):
1367 """
1368 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001369
Victor Stinner59bec362011-05-02 16:14:16 +02001370 The base implementation uses put_nowait. You may want to override this
1371 method if you want to use timeouts or work with custom queue
1372 implementations.
1373 """
1374 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001375
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001376 def stop(self):
1377 """
1378 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001379
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001380 This asks the thread to terminate, and then waits for it to do so.
1381 Note that if you don't call this before your application exits, there
1382 may be some records still left on the queue, which won't be processed.
1383 """
1384 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001385 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001386 self._thread.join()
1387 self._thread = None