blob: 4cbc320d354ee6461b055ebf4171f74e26c90909 [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
Vinay Sajip66b8b082012-04-24 23:25:30 +010026import errno, 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 Sajip8cf4eb12012-10-09 08:06:13 +010075 except Exception:
Vinay Sajip3970c112004-07-08 10:24:04 +000076 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000077
Vinay Sajip23b94d02012-01-04 12:02:26 +000078 def rotation_filename(self, default_name):
79 """
80 Modify the filename of a log file when rotating.
81
82 This is provided so that a custom filename can be provided.
83
84 The default implementation calls the 'namer' attribute of the
85 handler, if it's callable, passing the default name to
86 it. If the attribute isn't callable (the default is None), the name
87 is returned unchanged.
88
89 :param default_name: The default name for the log file.
90 """
91 if not callable(self.namer):
92 result = default_name
93 else:
94 result = self.namer(default_name)
95 return result
96
97 def rotate(self, source, dest):
98 """
99 When rotating, rotate the current log.
100
101 The default implementation calls the 'rotator' attribute of the
102 handler, if it's callable, passing the source and dest arguments to
103 it. If the attribute isn't callable (the default is None), the source
104 is simply renamed to the destination.
105
106 :param source: The source filename. This is normally the base
107 filename, e.g. 'test.log'
108 :param dest: The destination filename. This is normally
109 what the source is rotated to, e.g. 'test.log.1'.
110 """
111 if not callable(self.rotator):
112 os.rename(source, dest)
113 else:
114 self.rotator(source, dest)
115
Vinay Sajip17c52d82004-07-03 11:48:34 +0000116class RotatingFileHandler(BaseRotatingHandler):
117 """
118 Handler for logging to a set of files, which switches from one file
119 to the next when the current file reaches a certain size.
120 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000121 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=False):
Guido van Rossum57102f82002-11-13 16:15:58 +0000122 """
123 Open the specified file and use it as the stream for logging.
124
125 By default, the file grows indefinitely. You can specify particular
126 values of maxBytes and backupCount to allow the file to rollover at
127 a predetermined size.
128
129 Rollover occurs whenever the current log file is nearly maxBytes in
130 length. If backupCount is >= 1, the system will successively create
131 new files with the same pathname as the base file, but with extensions
132 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
133 and a base file name of "app.log", you would get "app.log",
134 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
135 written to is always "app.log" - when it gets filled up, it is closed
136 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
137 exist, then they are renamed to "app.log.2", "app.log.3" etc.
138 respectively.
139
140 If maxBytes is zero, rollover never occurs.
141 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000142 # If rotation/rollover is wanted, it doesn't make sense to use another
143 # mode. If for example 'w' were specified, then if there were multiple
144 # runs of the calling application, the logs from previous runs would be
145 # lost if the 'w' is respected, because the log file would be truncated
146 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000147 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000148 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000149 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000150 self.maxBytes = maxBytes
151 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000152
153 def doRollover(self):
154 """
155 Do a rollover, as described in __init__().
156 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000157 if self.stream:
158 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000159 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000160 if self.backupCount > 0:
161 for i in range(self.backupCount - 1, 0, -1):
Vinay Sajip23b94d02012-01-04 12:02:26 +0000162 sfn = self.rotation_filename("%s.%d" % (self.baseFilename, i))
163 dfn = self.rotation_filename("%s.%d" % (self.baseFilename,
164 i + 1))
Guido van Rossum57102f82002-11-13 16:15:58 +0000165 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000166 if os.path.exists(dfn):
167 os.remove(dfn)
168 os.rename(sfn, dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000169 dfn = self.rotation_filename(self.baseFilename + ".1")
Guido van Rossum57102f82002-11-13 16:15:58 +0000170 if os.path.exists(dfn):
171 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000172 self.rotate(self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000173 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000174
Vinay Sajip17c52d82004-07-03 11:48:34 +0000175 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000176 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000177 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000178
Vinay Sajip17c52d82004-07-03 11:48:34 +0000179 Basically, see if the supplied record would cause the file to exceed
180 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000181 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000182 if self.stream is None: # delay was set...
183 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000184 if self.maxBytes > 0: # are we rolling over?
185 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000186 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000187 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000188 return 1
189 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000190
Vinay Sajip17c52d82004-07-03 11:48:34 +0000191class TimedRotatingFileHandler(BaseRotatingHandler):
192 """
193 Handler for logging to a file, rotating the log file at certain timed
194 intervals.
195
196 If backupCount is > 0, when rollover is done, no more than backupCount
197 files are kept - the oldest ones are deleted.
198 """
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000199 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000200 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000201 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000202 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000203 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000204 # Calculate the real rollover interval, which is just the number of
205 # seconds between rollovers. Also set the filename suffix used when
206 # a rollover occurs. Current 'when' events supported:
207 # S - Seconds
208 # M - Minutes
209 # H - Hours
210 # D - Days
211 # midnight - roll over at midnight
212 # W{0-6} - roll over on a certain day; 0 - Monday
213 #
214 # Case of the 'when' specifier is not important; lower or upper case
215 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000216 if self.when == 'S':
217 self.interval = 1 # one second
218 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000219 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000220 elif self.when == 'M':
221 self.interval = 60 # one minute
222 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000223 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000224 elif self.when == 'H':
225 self.interval = 60 * 60 # one hour
226 self.suffix = "%Y-%m-%d_%H"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000227 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000228 elif self.when == 'D' or self.when == 'MIDNIGHT':
229 self.interval = 60 * 60 * 24 # one day
230 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000231 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000232 elif self.when.startswith('W'):
233 self.interval = 60 * 60 * 24 * 7 # one week
234 if len(self.when) != 2:
235 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
236 if self.when[1] < '0' or self.when[1] > '6':
237 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
238 self.dayOfWeek = int(self.when[1])
239 self.suffix = "%Y-%m-%d"
Vinay Sajip23b94d02012-01-04 12:02:26 +0000240 self.extMatch = r"^\d{4}-\d{2}-\d{2}(\.\w+)?$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000241 else:
242 raise ValueError("Invalid rollover interval specified: %s" % self.when)
243
Antoine Pitroufd036452008-08-19 17:56:33 +0000244 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000245 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000246 if os.path.exists(filename):
247 t = os.stat(filename)[ST_MTIME]
248 else:
249 t = int(time.time())
250 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000251
Vinay Sajipaffbd872009-06-11 10:11:47 +0000252 def computeRollover(self, currentTime):
253 """
254 Work out the rollover time based on the specified time.
255 """
256 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000257 # If we are rolling over at midnight or weekly, then the interval is already known.
258 # What we need to figure out is WHEN the next interval is. In other words,
259 # if you are rolling over at midnight, then your base interval is 1 day,
260 # but you want to start that one day clock at midnight, not now. So, we
261 # have to fudge the rolloverAt value in order to trigger the first rollover
262 # at the right time. After that, the regular interval will take care of
263 # the rest. Note that this code doesn't care about leap seconds. :)
264 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
265 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000266 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000267 t = time.gmtime(currentTime)
268 else:
269 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000270 currentHour = t[3]
271 currentMinute = t[4]
272 currentSecond = t[5]
273 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000274 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
275 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000276 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000277 # If we are rolling over on a certain day, add in the number of days until
278 # the next rollover, but offset by 1 since we just calculated the time
279 # until the next day starts. There are three cases:
280 # Case 1) The day to rollover is today; in this case, do nothing
281 # Case 2) The day to rollover is further in the interval (i.e., today is
282 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
283 # next rollover is simply 6 - 2 - 1, or 3.
284 # Case 3) The day to rollover is behind us in the interval (i.e., today
285 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
286 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
287 # number of days left in the current week (1) plus the number
288 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000289 # The calculations described in 2) and 3) above need to have a day added.
290 # This is because the above time calculation takes us to midnight on this
291 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000292 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000293 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000294 if day != self.dayOfWeek:
295 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000296 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000297 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000298 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000299 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
300 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000301 dstNow = t[-1]
302 dstAtRollover = time.localtime(newRolloverAt)[-1]
303 if dstNow != dstAtRollover:
304 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000305 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000306 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000307 addend = 3600
308 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000309 result = newRolloverAt
310 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000311
312 def shouldRollover(self, record):
313 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000314 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000315
316 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000317 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000318 """
319 t = int(time.time())
320 if t >= self.rolloverAt:
321 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000322 return 0
323
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000324 def getFilesToDelete(self):
325 """
326 Determine the files to delete when rolling over.
327
328 More specific than the earlier method, which just used glob.glob().
329 """
330 dirName, baseName = os.path.split(self.baseFilename)
331 fileNames = os.listdir(dirName)
332 result = []
333 prefix = baseName + "."
334 plen = len(prefix)
335 for fileName in fileNames:
336 if fileName[:plen] == prefix:
337 suffix = fileName[plen:]
338 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000339 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000340 result.sort()
341 if len(result) < self.backupCount:
342 result = []
343 else:
344 result = result[:len(result) - self.backupCount]
345 return result
346
Vinay Sajip17c52d82004-07-03 11:48:34 +0000347 def doRollover(self):
348 """
349 do a rollover; in this case, a date/time stamp is appended to the filename
350 when the rollover happens. However, you want the file to be named for the
351 start of the interval, not the current time. If there is a backup count,
352 then we have to get a list of matching filenames, sort them and remove
353 the one with the oldest suffix.
354 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000355 if self.stream:
356 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000357 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000358 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000359 currentTime = int(time.time())
360 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000361 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 Sajip27f48972012-03-13 12:10:33 +0000366 dstThen = timeTuple[-1]
367 if dstNow != dstThen:
368 if dstNow:
369 addend = 3600
370 else:
371 addend = -3600
372 timeTuple = time.localtime(t + addend)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000373 dfn = self.rotation_filename(self.baseFilename + "." +
374 time.strftime(self.suffix, timeTuple))
Vinay Sajip17c52d82004-07-03 11:48:34 +0000375 if os.path.exists(dfn):
376 os.remove(dfn)
Vinay Sajip23b94d02012-01-04 12:02:26 +0000377 self.rotate(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000378 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000379 for s in self.getFilesToDelete():
380 os.remove(s)
Thomas Woutersb2137042007-02-01 18:02:27 +0000381 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000382 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000383 while newRolloverAt <= currentTime:
384 newRolloverAt = newRolloverAt + self.interval
385 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000386 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000387 dstAtRollover = time.localtime(newRolloverAt)[-1]
388 if dstNow != dstAtRollover:
389 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000390 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000391 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000392 addend = 3600
393 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000394 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000395
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000396class WatchedFileHandler(logging.FileHandler):
397 """
398 A handler for logging to a file, which watches the file
399 to see if it has changed while in use. This can happen because of
400 usage of programs such as newsyslog and logrotate which perform
401 log file rotation. This handler, intended for use under Unix,
402 watches the file to see if it has changed since the last emit.
403 (A file has changed if its device or inode have changed.)
404 If it has changed, the old file stream is closed, and the file
405 opened to get a new stream.
406
407 This handler is not appropriate for use under Windows, because
408 under Windows open files cannot be moved or renamed - logging
409 opens the files with exclusive locks - and so there is no need
410 for such a handler. Furthermore, ST_INO is not supported under
411 Windows; stat always returns zero for this value.
412
413 This handler is based on a suggestion and patch by Chad J.
414 Schroeder.
415 """
Vinay Sajip23b94d02012-01-04 12:02:26 +0000416 def __init__(self, filename, mode='a', encoding=None, delay=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000417 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100418 self.dev, self.ino = -1, -1
419 self._statstream()
420
421 def _statstream(self):
422 if self.stream:
423 sres = os.fstat(self.stream.fileno())
424 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000425
426 def emit(self, record):
427 """
428 Emit a record.
429
430 First check if the underlying file has changed, and if it
431 has, close the old stream and reopen the file to get the
432 current stream.
433 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100434 # Reduce the chance of race conditions by stat'ing by path only
435 # once and then fstat'ing our new fd if we opened a new log stream.
436 # See issue #14632: Thanks to John Mulligan for the problem report
437 # and patch.
438 try:
439 # stat the file by path, checking for existence
440 sres = os.stat(self.baseFilename)
441 except OSError as err:
442 if err.errno == errno.ENOENT:
443 sres = None
444 else:
445 raise
446 # compare file system stat with that of our stream file handle
447 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
448 if self.stream is not None:
449 # we have an open file handle, clean it up
450 self.stream.flush()
451 self.stream.close()
452 # open a new file handle and get new stat info from that fd
453 self.stream = self._open()
454 self._statstream()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000455 logging.FileHandler.emit(self, record)
456
Vinay Sajip66b8b082012-04-24 23:25:30 +0100457
Guido van Rossum57102f82002-11-13 16:15:58 +0000458class SocketHandler(logging.Handler):
459 """
460 A handler class which writes logging records, in pickle format, to
461 a streaming socket. The socket is kept open across logging calls.
462 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000463 The pickle which is sent is that of the LogRecord's attribute dictionary
464 (__dict__), so that the receiver does not need to have the logging module
465 installed in order to process the logging event.
466
467 To unpickle the record at the receiving end into a LogRecord, use the
468 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000469 """
470
471 def __init__(self, host, port):
472 """
473 Initializes the handler with a specific host address and port.
474
Vinay Sajipde19e082011-04-30 21:52:26 +0100475 When the attribute *closeOnError* is set to True - if a socket error
476 occurs, the socket is silently closed and then reopened on the next
477 logging call.
Guido van Rossum57102f82002-11-13 16:15:58 +0000478 """
479 logging.Handler.__init__(self)
480 self.host = host
481 self.port = port
482 self.sock = None
Vinay Sajipde19e082011-04-30 21:52:26 +0100483 self.closeOnError = False
Vinay Sajip48cfe382004-02-20 13:17:27 +0000484 self.retryTime = None
485 #
486 # Exponential backoff parameters.
487 #
488 self.retryStart = 1.0
489 self.retryMax = 30.0
490 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000491
Guido van Rossumd8faa362007-04-27 19:54:29 +0000492 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000493 """
494 A factory method which allows subclasses to define the precise
495 type of socket they want.
496 """
Giampaolo Rodola'35eeb302012-10-15 16:57:37 +0200497 return socket.create_connection((self.host, self.port), timeout=timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000498
Vinay Sajip48cfe382004-02-20 13:17:27 +0000499 def createSocket(self):
500 """
501 Try to create a socket, using an exponential backoff with
502 a max retry time. Thanks to Robert Olson for the original patch
503 (SF #815911) which has been slightly refactored.
504 """
505 now = time.time()
506 # Either retryTime is None, in which case this
507 # is the first time back after a disconnect, or
508 # we've waited long enough.
509 if self.retryTime is None:
Vinay Sajipde19e082011-04-30 21:52:26 +0100510 attempt = True
Vinay Sajip48cfe382004-02-20 13:17:27 +0000511 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000512 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000513 if attempt:
514 try:
515 self.sock = self.makeSocket()
516 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000517 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000518 #Creation failed, so set the retry time and return.
519 if self.retryTime is None:
520 self.retryPeriod = self.retryStart
521 else:
522 self.retryPeriod = self.retryPeriod * self.retryFactor
523 if self.retryPeriod > self.retryMax:
524 self.retryPeriod = self.retryMax
525 self.retryTime = now + self.retryPeriod
526
Guido van Rossum57102f82002-11-13 16:15:58 +0000527 def send(self, s):
528 """
529 Send a pickled string to the socket.
530
531 This function allows for partial sends which can happen when the
532 network is busy.
533 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000534 if self.sock is None:
535 self.createSocket()
536 #self.sock can be None either because we haven't reached the retry
537 #time yet, or because we have reached the retry time and retried,
538 #but are still unable to connect.
539 if self.sock:
540 try:
Giampaolo Rodola'f9de90a2012-10-26 18:32:24 +0200541 self.sock.sendall(s)
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:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100553 # just to get traceback text into record.exc_text ...
554 dummy = self.format(record)
555 # See issue #14436: If msg or args are objects, they may not be
556 # available on the receiving end. So we convert the msg % args
557 # to a string, save it as msg and zap the args.
558 d = dict(record.__dict__)
559 d['msg'] = record.getMessage()
560 d['args'] = None
561 d['exc_info'] = None
562 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000563 slen = struct.pack(">L", len(s))
564 return slen + s
565
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000566 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000567 """
568 Handle an error during logging.
569
570 An error has occurred during logging. Most likely cause -
571 connection lost. Close the socket so that we can retry on the
572 next event.
573 """
574 if self.closeOnError and self.sock:
575 self.sock.close()
576 self.sock = None #try to reconnect next time
577 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000578 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000579
580 def emit(self, record):
581 """
582 Emit a record.
583
584 Pickles the record and writes it to the socket in binary format.
585 If there is an error with the socket, silently drop the packet.
586 If there was a problem with the socket, re-establishes the
587 socket.
588 """
589 try:
590 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000591 self.send(s)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100592 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000593 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000594
595 def close(self):
596 """
597 Closes the socket.
598 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000599 self.acquire()
600 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000601 if self.sock:
602 self.sock.close()
603 self.sock = None
604 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000605 finally:
606 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000607
608class DatagramHandler(SocketHandler):
609 """
610 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000611 a datagram socket. The pickle which is sent is that of the LogRecord's
612 attribute dictionary (__dict__), so that the receiver does not need to
613 have the logging module installed in order to process the logging event.
614
615 To unpickle the record at the receiving end into a LogRecord, use the
616 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000617
618 """
619 def __init__(self, host, port):
620 """
621 Initializes the handler with a specific host address and port.
622 """
623 SocketHandler.__init__(self, host, port)
Vinay Sajipde19e082011-04-30 21:52:26 +0100624 self.closeOnError = False
Guido van Rossum57102f82002-11-13 16:15:58 +0000625
626 def makeSocket(self):
627 """
628 The factory method of SocketHandler is here overridden to create
629 a UDP socket (SOCK_DGRAM).
630 """
631 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
632 return s
633
634 def send(self, s):
635 """
636 Send a pickled string to a socket.
637
638 This function no longer allows for partial sends which can happen
639 when the network is busy - UDP does not guarantee delivery and
640 can deliver packets out of sequence.
641 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000642 if self.sock is None:
643 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000644 self.sock.sendto(s, (self.host, self.port))
645
646class SysLogHandler(logging.Handler):
647 """
648 A handler class which sends formatted logging records to a syslog
649 server. Based on Sam Rushing's syslog module:
650 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
651 Contributed by Nicolas Untz (after which minor refactoring changes
652 have been made).
653 """
654
655 # from <linux/sys/syslog.h>:
656 # ======================================================================
657 # priorities/facilities are encoded into a single 32-bit quantity, where
658 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
659 # facility (0-big number). Both the priorities and the facilities map
660 # roughly one-to-one to strings in the syslogd(8) source code. This
661 # mapping is included in this file.
662 #
663 # priorities (these are ordered)
664
665 LOG_EMERG = 0 # system is unusable
666 LOG_ALERT = 1 # action must be taken immediately
667 LOG_CRIT = 2 # critical conditions
668 LOG_ERR = 3 # error conditions
669 LOG_WARNING = 4 # warning conditions
670 LOG_NOTICE = 5 # normal but significant condition
671 LOG_INFO = 6 # informational
672 LOG_DEBUG = 7 # debug-level messages
673
674 # facility codes
675 LOG_KERN = 0 # kernel messages
676 LOG_USER = 1 # random user-level messages
677 LOG_MAIL = 2 # mail system
678 LOG_DAEMON = 3 # system daemons
679 LOG_AUTH = 4 # security/authorization messages
680 LOG_SYSLOG = 5 # messages generated internally by syslogd
681 LOG_LPR = 6 # line printer subsystem
682 LOG_NEWS = 7 # network news subsystem
683 LOG_UUCP = 8 # UUCP subsystem
684 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000685 LOG_AUTHPRIV = 10 # security/authorization messages (private)
686 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000687
688 # other codes through 15 reserved for system use
689 LOG_LOCAL0 = 16 # reserved for local use
690 LOG_LOCAL1 = 17 # reserved for local use
691 LOG_LOCAL2 = 18 # reserved for local use
692 LOG_LOCAL3 = 19 # reserved for local use
693 LOG_LOCAL4 = 20 # reserved for local use
694 LOG_LOCAL5 = 21 # reserved for local use
695 LOG_LOCAL6 = 22 # reserved for local use
696 LOG_LOCAL7 = 23 # reserved for local use
697
698 priority_names = {
699 "alert": LOG_ALERT,
700 "crit": LOG_CRIT,
701 "critical": LOG_CRIT,
702 "debug": LOG_DEBUG,
703 "emerg": LOG_EMERG,
704 "err": LOG_ERR,
705 "error": LOG_ERR, # DEPRECATED
706 "info": LOG_INFO,
707 "notice": LOG_NOTICE,
708 "panic": LOG_EMERG, # DEPRECATED
709 "warn": LOG_WARNING, # DEPRECATED
710 "warning": LOG_WARNING,
711 }
712
713 facility_names = {
714 "auth": LOG_AUTH,
715 "authpriv": LOG_AUTHPRIV,
716 "cron": LOG_CRON,
717 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000718 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000719 "kern": LOG_KERN,
720 "lpr": LOG_LPR,
721 "mail": LOG_MAIL,
722 "news": LOG_NEWS,
723 "security": LOG_AUTH, # DEPRECATED
724 "syslog": LOG_SYSLOG,
725 "user": LOG_USER,
726 "uucp": LOG_UUCP,
727 "local0": LOG_LOCAL0,
728 "local1": LOG_LOCAL1,
729 "local2": LOG_LOCAL2,
730 "local3": LOG_LOCAL3,
731 "local4": LOG_LOCAL4,
732 "local5": LOG_LOCAL5,
733 "local6": LOG_LOCAL6,
734 "local7": LOG_LOCAL7,
735 }
736
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000737 #The map below appears to be trivially lowercasing the key. However,
738 #there's more to it than meets the eye - in some locales, lowercasing
739 #gives unexpected results. See SF #1524081: in the Turkish locale,
740 #"INFO".lower() != "info"
741 priority_map = {
742 "DEBUG" : "debug",
743 "INFO" : "info",
744 "WARNING" : "warning",
745 "ERROR" : "error",
746 "CRITICAL" : "critical"
747 }
748
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000749 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
750 facility=LOG_USER, socktype=socket.SOCK_DGRAM):
Guido van Rossum57102f82002-11-13 16:15:58 +0000751 """
752 Initialize a handler.
753
Guido van Rossume7ba4952007-06-06 23:52:48 +0000754 If address is specified as a string, a UNIX socket is used. To log to a
755 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000756 If facility is not specified, LOG_USER is used.
757 """
758 logging.Handler.__init__(self)
759
760 self.address = address
761 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000762 self.socktype = socktype
763
Guido van Rossum13257902007-06-07 23:15:56 +0000764 if isinstance(address, str):
Vinay Sajip5a35b062011-04-27 11:31:14 +0100765 self.unixsocket = True
Thomas Wouters89f507f2006-12-13 04:49:30 +0000766 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000767 else:
Vinay Sajip5a35b062011-04-27 11:31:14 +0100768 self.unixsocket = False
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000769 self.socket = socket.socket(socket.AF_INET, socktype)
770 if socktype == socket.SOCK_STREAM:
771 self.socket.connect(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000772 self.formatter = None
773
Vinay Sajipa1974c12005-01-13 08:23:56 +0000774 def _connect_unixsocket(self, address):
Vinay Sajipb5ca9322012-10-09 08:19:25 +0100775 self.socket = socket.socket(socket.AF_UNIX, self.socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000776 try:
777 self.socket.connect(address)
778 except socket.error:
779 self.socket.close()
Vinay Sajipb5ca9322012-10-09 08:19:25 +0100780 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000781
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000782 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000783 """
784 Encode the facility and priority. You can pass in strings or
785 integers - if strings are passed, the facility_names and
786 priority_names mapping dictionaries are used to convert them to
787 integers.
788 """
Guido van Rossum13257902007-06-07 23:15:56 +0000789 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000790 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000791 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000792 priority = self.priority_names[priority]
793 return (facility << 3) | priority
794
795 def close (self):
796 """
797 Closes the socket.
798 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000799 self.acquire()
800 try:
Vinay Sajip57c22372012-02-23 20:03:49 +0000801 self.socket.close()
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000802 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000803 finally:
804 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000805
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000806 def mapPriority(self, levelName):
807 """
808 Map a logging level name to a key in the priority_names map.
809 This is useful in two scenarios: when custom levels are being
810 used, and in the case where you can't do a straightforward
811 mapping by lowercasing the logging level name because of locale-
812 specific issues (see SF #1524081).
813 """
814 return self.priority_map.get(levelName, "warning")
815
Vinay Sajip2353e352011-06-27 15:40:06 +0100816 ident = '' # prepended to all messages
Vinay Sajip8168d102011-06-09 16:50:49 +0100817 append_nul = True # some old syslog daemons expect a NUL terminator
818
Guido van Rossum57102f82002-11-13 16:15:58 +0000819 def emit(self, record):
820 """
821 Emit a record.
822
823 The record is formatted, and then sent to the syslog server. If
824 exception information is present, it is NOT sent to the server.
825 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100826 msg = self.format(record)
Vinay Sajip2353e352011-06-27 15:40:06 +0100827 if self.ident:
828 msg = self.ident + msg
Vinay Sajip8168d102011-06-09 16:50:49 +0100829 if self.append_nul:
830 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000831 """
832 We need to convert record level to lowercase, maybe this will
833 change in the future.
834 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000835 prio = '<%d>' % self.encodePriority(self.facility,
836 self.mapPriority(record.levelname))
837 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000838 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000839 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000840 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000841 try:
842 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000843 try:
844 self.socket.send(msg)
845 except socket.error:
846 self._connect_unixsocket(self.address)
847 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000848 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000849 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000850 else:
851 self.socket.sendall(msg)
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100852 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000853 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000854
855class SMTPHandler(logging.Handler):
856 """
857 A handler class which sends an SMTP email for each logging event.
858 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000859 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip26308da2012-04-22 18:16:14 +0100860 credentials=None, secure=None, timeout=5.0):
Guido van Rossum57102f82002-11-13 16:15:58 +0000861 """
862 Initialize the handler.
863
864 Initialize the instance with the from and to addresses and subject
865 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000866 (host, port) tuple format for the mailhost argument. To specify
867 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000868 for the credentials argument. To specify the use of a secure
869 protocol (TLS), pass in a tuple for the secure argument. This will
870 only be used when authentication credentials are supplied. The tuple
871 will be either an empty tuple, or a single-value tuple with the name
872 of a keyfile, or a 2-value tuple with the names of the keyfile and
873 certificate file. (This tuple is passed to the `starttls` method).
Vinay Sajip17160fd2012-03-15 12:02:08 +0000874 A timeout in seconds can be specified for the SMTP connection (the
875 default is one second).
Guido van Rossum57102f82002-11-13 16:15:58 +0000876 """
877 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000878 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000879 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000880 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000881 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000882 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000883 self.username, self.password = credentials
884 else:
885 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000886 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000887 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000888 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000889 self.toaddrs = toaddrs
890 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000891 self.secure = secure
Vinay Sajip17160fd2012-03-15 12:02:08 +0000892 self.timeout = timeout
Guido van Rossum57102f82002-11-13 16:15:58 +0000893
894 def getSubject(self, record):
895 """
896 Determine the subject for the email.
897
898 If you want to specify a subject line which is record-dependent,
899 override this method.
900 """
901 return self.subject
902
903 def emit(self, record):
904 """
905 Emit a record.
906
907 Format the record and send it to the specified addressees.
908 """
909 try:
910 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000911 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000912 port = self.mailport
913 if not port:
914 port = smtplib.SMTP_PORT
Vinay Sajip17160fd2012-03-15 12:02:08 +0000915 smtp = smtplib.SMTP(self.mailhost, port, timeout=self.timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000916 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000917 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 +0000918 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000919 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000920 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000921 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000922 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000923 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000924 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000925 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000926 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000927 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000928 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
929 smtp.quit()
Vinay Sajip8cf4eb12012-10-09 08:06:13 +0100930 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000931 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000932
933class NTEventLogHandler(logging.Handler):
934 """
935 A handler class which sends events to the NT Event Log. Adds a
936 registry entry for the specified application name. If no dllname is
937 provided, win32service.pyd (which contains some basic message
938 placeholders) is used. Note that use of these placeholders will make
939 your event logs big, as the entire message source is held in the log.
940 If you want slimmer logs, you have to pass in the name of your own DLL
941 which contains the message definitions you want to use in the event log.
942 """
943 def __init__(self, appname, dllname=None, logtype="Application"):
944 logging.Handler.__init__(self)
945 try:
946 import win32evtlogutil, win32evtlog
947 self.appname = appname
948 self._welu = win32evtlogutil
949 if not dllname:
950 dllname = os.path.split(self._welu.__file__)
951 dllname = os.path.split(dllname[0])
952 dllname = os.path.join(dllname[0], r'win32service.pyd')
953 self.dllname = dllname
954 self.logtype = logtype
955 self._welu.AddSourceToRegistry(appname, dllname, logtype)
956 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
957 self.typemap = {
958 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
959 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000960 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000961 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
962 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
963 }
964 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000965 print("The Python Win32 extensions for NT (service, event "\
966 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000967 self._welu = None
968
969 def getMessageID(self, record):
970 """
971 Return the message ID for the event record. If you are using your
972 own messages, you could do this by having the msg passed to the
973 logger being an ID rather than a formatting string. Then, in here,
974 you could use a dictionary lookup to get the message ID. This
975 version returns 1, which is the base message ID in win32service.pyd.
976 """
977 return 1
978
979 def getEventCategory(self, record):
980 """
981 Return the event category for the record.
982
983 Override this if you want to specify your own categories. This version
984 returns 0.
985 """
986 return 0
987
988 def getEventType(self, record):
989 """
990 Return the event type for the record.
991
992 Override this if you want to specify your own types. This version does
993 a mapping using the handler's typemap attribute, which is set up in
994 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000995 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000996 either need to override this method or place a suitable dictionary in
997 the handler's typemap attribute.
998 """
999 return self.typemap.get(record.levelno, self.deftype)
1000
1001 def emit(self, record):
1002 """
1003 Emit a record.
1004
1005 Determine the message ID, event category and event type. Then
1006 log the message in the NT event log.
1007 """
1008 if self._welu:
1009 try:
1010 id = self.getMessageID(record)
1011 cat = self.getEventCategory(record)
1012 type = self.getEventType(record)
1013 msg = self.format(record)
1014 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001015 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001016 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001017
1018 def close(self):
1019 """
1020 Clean up this handler.
1021
1022 You can remove the application name from the registry as a
1023 source of event log entries. However, if you do this, you will
1024 not be able to see the events as you intended in the Event Log
1025 Viewer - it needs to be able to access the registry to get the
1026 DLL name.
1027 """
1028 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001029 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001030
1031class HTTPHandler(logging.Handler):
1032 """
1033 A class which sends records to a Web server, using either GET or
1034 POST semantics.
1035 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001036 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001037 """
1038 Initialize the instance with the host, the request URL, and the method
1039 ("GET" or "POST")
1040 """
1041 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001042 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001043 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001044 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001045 self.host = host
1046 self.url = url
1047 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001048 self.secure = secure
1049 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001050
Neal Norwitzf297bd12003-04-23 03:49:43 +00001051 def mapLogRecord(self, record):
1052 """
1053 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001054 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001055 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001056 """
1057 return record.__dict__
1058
Guido van Rossum57102f82002-11-13 16:15:58 +00001059 def emit(self, record):
1060 """
1061 Emit a record.
1062
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001063 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001064 """
1065 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001066 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001067 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001068 if self.secure:
1069 h = http.client.HTTPSConnection(host)
1070 else:
1071 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001072 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001073 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001074 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001075 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001076 sep = '&'
1077 else:
1078 sep = '?'
1079 url = url + "%c%s" % (sep, data)
1080 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001081 # support multiple hosts on one IP address...
1082 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001083 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001084 if i >= 0:
1085 host = host[:i]
1086 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001087 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001088 h.putheader("Content-type",
1089 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001090 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001091 if self.credentials:
1092 import base64
1093 s = ('u%s:%s' % self.credentials).encode('utf-8')
1094 s = 'Basic ' + base64.b64encode(s).strip()
1095 h.putheader('Authorization', s)
Vinay Sajip0372e102011-05-05 12:59:14 +01001096 h.endheaders()
1097 if self.method == "POST":
1098 h.send(data.encode('utf-8'))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001099 h.getresponse() #can't do anything with the result
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001100 except Exception:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001101 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001102
1103class BufferingHandler(logging.Handler):
1104 """
1105 A handler class which buffers logging records in memory. Whenever each
1106 record is added to the buffer, a check is made to see if the buffer should
1107 be flushed. If it should, then flush() is expected to do what's needed.
1108 """
1109 def __init__(self, capacity):
1110 """
1111 Initialize the handler with the buffer size.
1112 """
1113 logging.Handler.__init__(self)
1114 self.capacity = capacity
1115 self.buffer = []
1116
1117 def shouldFlush(self, record):
1118 """
1119 Should the handler flush its buffer?
1120
1121 Returns true if the buffer is up to capacity. This method can be
1122 overridden to implement custom flushing strategies.
1123 """
1124 return (len(self.buffer) >= self.capacity)
1125
1126 def emit(self, record):
1127 """
1128 Emit a record.
1129
1130 Append the record. If shouldFlush() tells us to, call flush() to process
1131 the buffer.
1132 """
1133 self.buffer.append(record)
1134 if self.shouldFlush(record):
1135 self.flush()
1136
1137 def flush(self):
1138 """
1139 Override to implement custom flushing behaviour.
1140
1141 This version just zaps the buffer to empty.
1142 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001143 self.acquire()
1144 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001145 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001146 finally:
1147 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001148
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001149 def close(self):
1150 """
1151 Close the handler.
1152
1153 This version just flushes and chains to the parent class' close().
1154 """
1155 self.flush()
1156 logging.Handler.close(self)
1157
Guido van Rossum57102f82002-11-13 16:15:58 +00001158class MemoryHandler(BufferingHandler):
1159 """
1160 A handler class which buffers logging records in memory, periodically
1161 flushing them to a target handler. Flushing occurs whenever the buffer
1162 is full, or when an event of a certain severity or greater is seen.
1163 """
1164 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1165 """
1166 Initialize the handler with the buffer size, the level at which
1167 flushing should occur and an optional target.
1168
1169 Note that without a target being set either here or via setTarget(),
1170 a MemoryHandler is no use to anyone!
1171 """
1172 BufferingHandler.__init__(self, capacity)
1173 self.flushLevel = flushLevel
1174 self.target = target
1175
1176 def shouldFlush(self, record):
1177 """
1178 Check for buffer full or a record at the flushLevel or higher.
1179 """
1180 return (len(self.buffer) >= self.capacity) or \
1181 (record.levelno >= self.flushLevel)
1182
1183 def setTarget(self, target):
1184 """
1185 Set the target handler for this handler.
1186 """
1187 self.target = target
1188
1189 def flush(self):
1190 """
1191 For a MemoryHandler, flushing means just sending the buffered
1192 records to the target, if there is one. Override if you want
1193 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001194
1195 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001196 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001197 self.acquire()
1198 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001199 if self.target:
1200 for record in self.buffer:
1201 self.target.handle(record)
1202 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001203 finally:
1204 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001205
1206 def close(self):
1207 """
1208 Flush, set the target to None and lose the buffer.
1209 """
1210 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001211 self.acquire()
1212 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001213 self.target = None
1214 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001215 finally:
1216 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001217
1218
1219class QueueHandler(logging.Handler):
1220 """
1221 This handler sends events to a queue. Typically, it would be used together
1222 with a multiprocessing Queue to centralise logging to file in one process
1223 (in a multi-process application), so as to avoid file write contention
1224 between processes.
1225
1226 This code is new in Python 3.2, but this class can be copy pasted into
1227 user code for use with earlier Python versions.
1228 """
1229
1230 def __init__(self, queue):
1231 """
1232 Initialise an instance, using the passed queue.
1233 """
1234 logging.Handler.__init__(self)
1235 self.queue = queue
1236
1237 def enqueue(self, record):
1238 """
1239 Enqueue a record.
1240
1241 The base implementation uses put_nowait. You may want to override
1242 this method if you want to use blocking, timeouts or custom queue
1243 implementations.
1244 """
1245 self.queue.put_nowait(record)
1246
Vinay Sajip0258ce82010-09-22 20:34:53 +00001247 def prepare(self, record):
1248 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001249 Prepares a record for queuing. The object returned by this method is
1250 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001251
1252 The base implementation formats the record to merge the message
1253 and arguments, and removes unpickleable items from the record
1254 in-place.
1255
1256 You might want to override this method if you want to convert
1257 the record to a dict or JSON string, or send a modified copy
1258 of the record while leaving the original intact.
1259 """
1260 # The format operation gets traceback text into record.exc_text
1261 # (if there's exception data), and also puts the message into
1262 # record.message. We can then use this to replace the original
1263 # msg + args, as these might be unpickleable. We also zap the
1264 # exc_info attribute, as it's no longer needed and, if not None,
1265 # will typically not be pickleable.
1266 self.format(record)
1267 record.msg = record.message
1268 record.args = None
1269 record.exc_info = None
1270 return record
1271
Vinay Sajip121a1c42010-09-08 10:46:15 +00001272 def emit(self, record):
1273 """
1274 Emit a record.
1275
Vinay Sajip0637d492010-09-23 08:15:54 +00001276 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001277 """
1278 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001279 self.enqueue(self.prepare(record))
Vinay Sajip8cf4eb12012-10-09 08:06:13 +01001280 except Exception:
Vinay Sajip121a1c42010-09-08 10:46:15 +00001281 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001282
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001283if threading:
1284 class QueueListener(object):
1285 """
1286 This class implements an internal threaded listener which watches for
1287 LogRecords being added to a queue, removes them and passes them to a
1288 list of handlers for processing.
1289 """
1290 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001291
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001292 def __init__(self, queue, *handlers):
1293 """
1294 Initialise an instance with the specified queue and
1295 handlers.
1296 """
1297 self.queue = queue
1298 self.handlers = handlers
1299 self._stop = threading.Event()
1300 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001301
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001302 def dequeue(self, block):
1303 """
1304 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001305
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001306 The base implementation uses get. You may want to override this method
1307 if you want to use timeouts or work with custom queue implementations.
1308 """
1309 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001310
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001311 def start(self):
1312 """
1313 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001314
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001315 This starts up a background thread to monitor the queue for
1316 LogRecords to process.
1317 """
1318 self._thread = t = threading.Thread(target=self._monitor)
1319 t.setDaemon(True)
1320 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001321
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001322 def prepare(self , record):
1323 """
1324 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001325
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001326 This method just returns the passed-in record. You may want to
1327 override this method if you need to do any custom marshalling or
1328 manipulation of the record before passing it to the handlers.
1329 """
1330 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001331
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001332 def handle(self, record):
1333 """
1334 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001335
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001336 This just loops through the handlers offering them the record
1337 to handle.
1338 """
1339 record = self.prepare(record)
1340 for handler in self.handlers:
1341 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001342
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001343 def _monitor(self):
1344 """
1345 Monitor the queue for records, and ask the handler
1346 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001347
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001348 This method runs on a separate, internal thread.
1349 The thread will terminate if it sees a sentinel object in the queue.
1350 """
1351 q = self.queue
1352 has_task_done = hasattr(q, 'task_done')
1353 while not self._stop.isSet():
1354 try:
1355 record = self.dequeue(True)
1356 if record is self._sentinel:
1357 break
1358 self.handle(record)
1359 if has_task_done:
1360 q.task_done()
1361 except queue.Empty:
1362 pass
1363 # There might still be records in the queue.
1364 while True:
1365 try:
1366 record = self.dequeue(False)
1367 if record is self._sentinel:
1368 break
1369 self.handle(record)
1370 if has_task_done:
1371 q.task_done()
1372 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001373 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001374
Victor Stinner59bec362011-05-02 16:14:16 +02001375 def enqueue_sentinel(self):
1376 """
1377 This is used to enqueue the sentinel record.
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001378
Victor Stinner59bec362011-05-02 16:14:16 +02001379 The base implementation uses put_nowait. You may want to override this
1380 method if you want to use timeouts or work with custom queue
1381 implementations.
1382 """
1383 self.queue.put_nowait(self._sentinel)
Vinay Sajipaa7c1792011-02-25 15:56:55 +00001384
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001385 def stop(self):
1386 """
1387 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001388
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001389 This asks the thread to terminate, and then waits for it to do so.
1390 Note that if you don't call this before your application exits, there
1391 may be some records still left on the queue, which won't be processed.
1392 """
1393 self._stop.set()
Victor Stinner59bec362011-05-02 16:14:16 +02001394 self.enqueue_sentinel()
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001395 self._thread.join()
1396 self._thread = None