blob: 8349d3a7fb906f156ef53e9e8c2b0e31e69d5ef4 [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
Benjamin Peterson9451a1c2010-03-13 22:30:34 +000027from stat import ST_DEV, ST_INO, ST_MTIME
Vinay Sajipe723e962011-04-15 22:27:17 +010028import queue
Victor Stinnercafa2ef2011-05-02 16:11:28 +020029try:
30 import threading
31except ImportError:
32 threading = None
Guido van Rossum57102f82002-11-13 16:15:58 +000033
Vinay Sajip4600f112005-03-13 09:56:36 +000034try:
35 import codecs
36except ImportError:
37 codecs = None
38
Guido van Rossum57102f82002-11-13 16:15:58 +000039#
40# Some constants...
41#
42
43DEFAULT_TCP_LOGGING_PORT = 9020
44DEFAULT_UDP_LOGGING_PORT = 9021
45DEFAULT_HTTP_LOGGING_PORT = 9022
46DEFAULT_SOAP_LOGGING_PORT = 9023
47SYSLOG_UDP_PORT = 514
Vinay Sajipcbabd7e2009-10-10 20:32:36 +000048SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000049
Thomas Wouters477c8d52006-05-27 19:21:47 +000050_MIDNIGHT = 24 * 60 * 60 # number of seconds in a day
51
Vinay Sajip17c52d82004-07-03 11:48:34 +000052class BaseRotatingHandler(logging.FileHandler):
53 """
54 Base class for handlers that rotate log files at a certain point.
55 Not meant to be instantiated directly. Instead, use RotatingFileHandler
56 or TimedRotatingFileHandler.
57 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000058 def __init__(self, filename, mode, encoding=None, delay=0):
Vinay Sajip17c52d82004-07-03 11:48:34 +000059 """
60 Use the specified filename for streamed logging
61 """
Vinay Sajip4600f112005-03-13 09:56:36 +000062 if codecs is None:
63 encoding = None
Christian Heimese7a15bb2008-01-24 16:21:45 +000064 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip4600f112005-03-13 09:56:36 +000065 self.mode = mode
66 self.encoding = encoding
Guido van Rossum57102f82002-11-13 16:15:58 +000067
Vinay Sajip17c52d82004-07-03 11:48:34 +000068 def emit(self, record):
69 """
70 Emit a record.
71
72 Output the record to the file, catering for rollover as described
73 in doRollover().
74 """
Vinay Sajip3970c112004-07-08 10:24:04 +000075 try:
76 if self.shouldRollover(record):
77 self.doRollover()
78 logging.FileHandler.emit(self, record)
Vinay Sajip85c19092005-10-31 13:14:19 +000079 except (KeyboardInterrupt, SystemExit):
80 raise
Vinay Sajip3970c112004-07-08 10:24:04 +000081 except:
82 self.handleError(record)
Vinay Sajip17c52d82004-07-03 11:48:34 +000083
84class RotatingFileHandler(BaseRotatingHandler):
85 """
86 Handler for logging to a set of files, which switches from one file
87 to the next when the current file reaches a certain size.
88 """
Christian Heimese7a15bb2008-01-24 16:21:45 +000089 def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=0):
Guido van Rossum57102f82002-11-13 16:15:58 +000090 """
91 Open the specified file and use it as the stream for logging.
92
93 By default, the file grows indefinitely. You can specify particular
94 values of maxBytes and backupCount to allow the file to rollover at
95 a predetermined size.
96
97 Rollover occurs whenever the current log file is nearly maxBytes in
98 length. If backupCount is >= 1, the system will successively create
99 new files with the same pathname as the base file, but with extensions
100 ".1", ".2" etc. appended to it. For example, with a backupCount of 5
101 and a base file name of "app.log", you would get "app.log",
102 "app.log.1", "app.log.2", ... through to "app.log.5". The file being
103 written to is always "app.log" - when it gets filled up, it is closed
104 and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc.
105 exist, then they are renamed to "app.log.2", "app.log.3" etc.
106 respectively.
107
108 If maxBytes is zero, rollover never occurs.
109 """
Vinay Sajip89994b22010-08-22 18:11:02 +0000110 # If rotation/rollover is wanted, it doesn't make sense to use another
111 # mode. If for example 'w' were specified, then if there were multiple
112 # runs of the calling application, the logs from previous runs would be
113 # lost if the 'w' is respected, because the log file would be truncated
114 # on each run.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000115 if maxBytes > 0:
Vinay Sajip89994b22010-08-22 18:11:02 +0000116 mode = 'a'
Christian Heimese7a15bb2008-01-24 16:21:45 +0000117 BaseRotatingHandler.__init__(self, filename, mode, encoding, delay)
Guido van Rossum57102f82002-11-13 16:15:58 +0000118 self.maxBytes = maxBytes
119 self.backupCount = backupCount
Guido van Rossum57102f82002-11-13 16:15:58 +0000120
121 def doRollover(self):
122 """
123 Do a rollover, as described in __init__().
124 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000125 if self.stream:
126 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000127 self.stream = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000128 if self.backupCount > 0:
129 for i in range(self.backupCount - 1, 0, -1):
130 sfn = "%s.%d" % (self.baseFilename, i)
131 dfn = "%s.%d" % (self.baseFilename, i + 1)
132 if os.path.exists(sfn):
Guido van Rossum57102f82002-11-13 16:15:58 +0000133 if os.path.exists(dfn):
134 os.remove(dfn)
135 os.rename(sfn, dfn)
136 dfn = self.baseFilename + ".1"
137 if os.path.exists(dfn):
138 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000139 os.rename(self.baseFilename, dfn)
Thomas Woutersb2137042007-02-01 18:02:27 +0000140 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000141
Vinay Sajip17c52d82004-07-03 11:48:34 +0000142 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000143 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000144 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000145
Vinay Sajip17c52d82004-07-03 11:48:34 +0000146 Basically, see if the supplied record would cause the file to exceed
147 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000148 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000149 if self.stream is None: # delay was set...
150 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000151 if self.maxBytes > 0: # are we rolling over?
152 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000153 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000154 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000155 return 1
156 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000157
Vinay Sajip17c52d82004-07-03 11:48:34 +0000158class TimedRotatingFileHandler(BaseRotatingHandler):
159 """
160 Handler for logging to a file, rotating the log file at certain timed
161 intervals.
162
163 If backupCount is > 0, when rollover is done, no more than backupCount
164 files are kept - the oldest ones are deleted.
165 """
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000166 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000167 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000168 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000169 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000170 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000171 # Calculate the real rollover interval, which is just the number of
172 # seconds between rollovers. Also set the filename suffix used when
173 # a rollover occurs. Current 'when' events supported:
174 # S - Seconds
175 # M - Minutes
176 # H - Hours
177 # D - Days
178 # midnight - roll over at midnight
179 # W{0-6} - roll over on a certain day; 0 - Monday
180 #
181 # Case of the 'when' specifier is not important; lower or upper case
182 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000183 if self.when == 'S':
184 self.interval = 1 # one second
185 self.suffix = "%Y-%m-%d_%H-%M-%S"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000186 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000187 elif self.when == 'M':
188 self.interval = 60 # one minute
189 self.suffix = "%Y-%m-%d_%H-%M"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000190 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000191 elif self.when == 'H':
192 self.interval = 60 * 60 # one hour
193 self.suffix = "%Y-%m-%d_%H"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000194 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000195 elif self.when == 'D' or self.when == 'MIDNIGHT':
196 self.interval = 60 * 60 * 24 # one day
197 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000198 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000199 elif self.when.startswith('W'):
200 self.interval = 60 * 60 * 24 * 7 # one week
201 if len(self.when) != 2:
202 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
203 if self.when[1] < '0' or self.when[1] > '6':
204 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
205 self.dayOfWeek = int(self.when[1])
206 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000207 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000208 else:
209 raise ValueError("Invalid rollover interval specified: %s" % self.when)
210
Antoine Pitroufd036452008-08-19 17:56:33 +0000211 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000212 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000213 if os.path.exists(filename):
214 t = os.stat(filename)[ST_MTIME]
215 else:
216 t = int(time.time())
217 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000218
Vinay Sajipaffbd872009-06-11 10:11:47 +0000219 def computeRollover(self, currentTime):
220 """
221 Work out the rollover time based on the specified time.
222 """
223 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000224 # If we are rolling over at midnight or weekly, then the interval is already known.
225 # What we need to figure out is WHEN the next interval is. In other words,
226 # if you are rolling over at midnight, then your base interval is 1 day,
227 # but you want to start that one day clock at midnight, not now. So, we
228 # have to fudge the rolloverAt value in order to trigger the first rollover
229 # at the right time. After that, the regular interval will take care of
230 # the rest. Note that this code doesn't care about leap seconds. :)
231 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
232 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000233 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000234 t = time.gmtime(currentTime)
235 else:
236 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000237 currentHour = t[3]
238 currentMinute = t[4]
239 currentSecond = t[5]
240 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000241 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
242 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000243 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000244 # If we are rolling over on a certain day, add in the number of days until
245 # the next rollover, but offset by 1 since we just calculated the time
246 # until the next day starts. There are three cases:
247 # Case 1) The day to rollover is today; in this case, do nothing
248 # Case 2) The day to rollover is further in the interval (i.e., today is
249 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
250 # next rollover is simply 6 - 2 - 1, or 3.
251 # Case 3) The day to rollover is behind us in the interval (i.e., today
252 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
253 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
254 # number of days left in the current week (1) plus the number
255 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000256 # The calculations described in 2) and 3) above need to have a day added.
257 # This is because the above time calculation takes us to midnight on this
258 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000259 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000260 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000261 if day != self.dayOfWeek:
262 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000263 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000264 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000265 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000266 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
267 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000268 dstNow = t[-1]
269 dstAtRollover = time.localtime(newRolloverAt)[-1]
270 if dstNow != dstAtRollover:
271 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000272 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000273 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000274 addend = 3600
275 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000276 result = newRolloverAt
277 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000278
279 def shouldRollover(self, record):
280 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000281 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000282
283 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000284 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000285 """
286 t = int(time.time())
287 if t >= self.rolloverAt:
288 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000289 return 0
290
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000291 def getFilesToDelete(self):
292 """
293 Determine the files to delete when rolling over.
294
295 More specific than the earlier method, which just used glob.glob().
296 """
297 dirName, baseName = os.path.split(self.baseFilename)
298 fileNames = os.listdir(dirName)
299 result = []
300 prefix = baseName + "."
301 plen = len(prefix)
302 for fileName in fileNames:
303 if fileName[:plen] == prefix:
304 suffix = fileName[plen:]
305 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000306 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000307 result.sort()
308 if len(result) < self.backupCount:
309 result = []
310 else:
311 result = result[:len(result) - self.backupCount]
312 return result
313
Vinay Sajip17c52d82004-07-03 11:48:34 +0000314 def doRollover(self):
315 """
316 do a rollover; in this case, a date/time stamp is appended to the filename
317 when the rollover happens. However, you want the file to be named for the
318 start of the interval, not the current time. If there is a backup count,
319 then we have to get a list of matching filenames, sort them and remove
320 the one with the oldest suffix.
321 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000322 if self.stream:
323 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000324 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000325 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000326 currentTime = int(time.time())
327 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000328 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000329 if self.utc:
330 timeTuple = time.gmtime(t)
331 else:
332 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000333 dstThen = timeTuple[-1]
334 if dstNow != dstThen:
335 if dstNow:
336 addend = 3600
337 else:
338 addend = -3600
339 timeTuple = time.localtime(t + addend)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000340 dfn = self.baseFilename + "." + time.strftime(self.suffix, timeTuple)
341 if os.path.exists(dfn):
342 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000343 os.rename(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000344 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000345 for s in self.getFilesToDelete():
346 os.remove(s)
Thomas Woutersb2137042007-02-01 18:02:27 +0000347 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000348 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000349 while newRolloverAt <= currentTime:
350 newRolloverAt = newRolloverAt + self.interval
351 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000352 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000353 dstAtRollover = time.localtime(newRolloverAt)[-1]
354 if dstNow != dstAtRollover:
355 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000356 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000357 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000358 addend = 3600
359 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000360 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000361
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000362class WatchedFileHandler(logging.FileHandler):
363 """
364 A handler for logging to a file, which watches the file
365 to see if it has changed while in use. This can happen because of
366 usage of programs such as newsyslog and logrotate which perform
367 log file rotation. This handler, intended for use under Unix,
368 watches the file to see if it has changed since the last emit.
369 (A file has changed if its device or inode have changed.)
370 If it has changed, the old file stream is closed, and the file
371 opened to get a new stream.
372
373 This handler is not appropriate for use under Windows, because
374 under Windows open files cannot be moved or renamed - logging
375 opens the files with exclusive locks - and so there is no need
376 for such a handler. Furthermore, ST_INO is not supported under
377 Windows; stat always returns zero for this value.
378
379 This handler is based on a suggestion and patch by Chad J.
380 Schroeder.
381 """
Christian Heimese7a15bb2008-01-24 16:21:45 +0000382 def __init__(self, filename, mode='a', encoding=None, delay=0):
383 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100384 self.dev, self.ino = -1, -1
385 self._statstream()
386
387 def _statstream(self):
388 if self.stream:
389 sres = os.fstat(self.stream.fileno())
390 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000391
392 def emit(self, record):
393 """
394 Emit a record.
395
396 First check if the underlying file has changed, and if it
397 has, close the old stream and reopen the file to get the
398 current stream.
399 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100400 # Reduce the chance of race conditions by stat'ing by path only
401 # once and then fstat'ing our new fd if we opened a new log stream.
402 # See issue #14632: Thanks to John Mulligan for the problem report
403 # and patch.
404 try:
405 # stat the file by path, checking for existence
406 sres = os.stat(self.baseFilename)
407 except OSError as err:
408 if err.errno == errno.ENOENT:
409 sres = None
410 else:
411 raise
412 # compare file system stat with that of our stream file handle
413 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
414 if self.stream is not None:
415 # we have an open file handle, clean it up
416 self.stream.flush()
417 self.stream.close()
418 # open a new file handle and get new stat info from that fd
419 self.stream = self._open()
420 self._statstream()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000421 logging.FileHandler.emit(self, record)
422
Vinay Sajip66b8b082012-04-24 23:25:30 +0100423
Guido van Rossum57102f82002-11-13 16:15:58 +0000424class SocketHandler(logging.Handler):
425 """
426 A handler class which writes logging records, in pickle format, to
427 a streaming socket. The socket is kept open across logging calls.
428 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000429 The pickle which is sent is that of the LogRecord's attribute dictionary
430 (__dict__), so that the receiver does not need to have the logging module
431 installed in order to process the logging event.
432
433 To unpickle the record at the receiving end into a LogRecord, use the
434 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000435 """
436
437 def __init__(self, host, port):
438 """
439 Initializes the handler with a specific host address and port.
440
441 The attribute 'closeOnError' is set to 1 - which means that if
442 a socket error occurs, the socket is silently closed and then
443 reopened on the next logging call.
444 """
445 logging.Handler.__init__(self)
446 self.host = host
447 self.port = port
448 self.sock = None
449 self.closeOnError = 0
Vinay Sajip48cfe382004-02-20 13:17:27 +0000450 self.retryTime = None
451 #
452 # Exponential backoff parameters.
453 #
454 self.retryStart = 1.0
455 self.retryMax = 30.0
456 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000457
Guido van Rossumd8faa362007-04-27 19:54:29 +0000458 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000459 """
460 A factory method which allows subclasses to define the precise
461 type of socket they want.
462 """
463 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000464 if hasattr(s, 'settimeout'):
465 s.settimeout(timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000466 s.connect((self.host, self.port))
467 return s
468
Vinay Sajip48cfe382004-02-20 13:17:27 +0000469 def createSocket(self):
470 """
471 Try to create a socket, using an exponential backoff with
472 a max retry time. Thanks to Robert Olson for the original patch
473 (SF #815911) which has been slightly refactored.
474 """
475 now = time.time()
476 # Either retryTime is None, in which case this
477 # is the first time back after a disconnect, or
478 # we've waited long enough.
479 if self.retryTime is None:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000480 attempt = 1
Vinay Sajip48cfe382004-02-20 13:17:27 +0000481 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000482 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000483 if attempt:
484 try:
485 self.sock = self.makeSocket()
486 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000487 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000488 #Creation failed, so set the retry time and return.
489 if self.retryTime is None:
490 self.retryPeriod = self.retryStart
491 else:
492 self.retryPeriod = self.retryPeriod * self.retryFactor
493 if self.retryPeriod > self.retryMax:
494 self.retryPeriod = self.retryMax
495 self.retryTime = now + self.retryPeriod
496
Guido van Rossum57102f82002-11-13 16:15:58 +0000497 def send(self, s):
498 """
499 Send a pickled string to the socket.
500
501 This function allows for partial sends which can happen when the
502 network is busy.
503 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000504 if self.sock is None:
505 self.createSocket()
506 #self.sock can be None either because we haven't reached the retry
507 #time yet, or because we have reached the retry time and retried,
508 #but are still unable to connect.
509 if self.sock:
510 try:
511 if hasattr(self.sock, "sendall"):
512 self.sock.sendall(s)
513 else:
514 sentsofar = 0
515 left = len(s)
516 while left > 0:
517 sent = self.sock.send(s[sentsofar:])
518 sentsofar = sentsofar + sent
519 left = left - sent
520 except socket.error:
521 self.sock.close()
522 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000523
524 def makePickle(self, record):
525 """
526 Pickles the record in binary format with a length prefix, and
527 returns it ready for transmission across the socket.
528 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000529 ei = record.exc_info
530 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100531 # just to get traceback text into record.exc_text ...
532 dummy = self.format(record)
533 # See issue #14436: If msg or args are objects, they may not be
534 # available on the receiving end. So we convert the msg % args
535 # to a string, save it as msg and zap the args.
536 d = dict(record.__dict__)
537 d['msg'] = record.getMessage()
538 d['args'] = None
539 d['exc_info'] = None
540 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000541 slen = struct.pack(">L", len(s))
542 return slen + s
543
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000544 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000545 """
546 Handle an error during logging.
547
548 An error has occurred during logging. Most likely cause -
549 connection lost. Close the socket so that we can retry on the
550 next event.
551 """
552 if self.closeOnError and self.sock:
553 self.sock.close()
554 self.sock = None #try to reconnect next time
555 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000556 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000557
558 def emit(self, record):
559 """
560 Emit a record.
561
562 Pickles the record and writes it to the socket in binary format.
563 If there is an error with the socket, silently drop the packet.
564 If there was a problem with the socket, re-establishes the
565 socket.
566 """
567 try:
568 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000569 self.send(s)
Vinay Sajip85c19092005-10-31 13:14:19 +0000570 except (KeyboardInterrupt, SystemExit):
571 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000572 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000573 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000574
575 def close(self):
576 """
577 Closes the socket.
578 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000579 self.acquire()
580 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000581 if self.sock:
582 self.sock.close()
583 self.sock = None
584 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000585 finally:
586 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000587
588class DatagramHandler(SocketHandler):
589 """
590 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000591 a datagram socket. The pickle which is sent is that of the LogRecord's
592 attribute dictionary (__dict__), so that the receiver does not need to
593 have the logging module installed in order to process the logging event.
594
595 To unpickle the record at the receiving end into a LogRecord, use the
596 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000597
598 """
599 def __init__(self, host, port):
600 """
601 Initializes the handler with a specific host address and port.
602 """
603 SocketHandler.__init__(self, host, port)
604 self.closeOnError = 0
605
606 def makeSocket(self):
607 """
608 The factory method of SocketHandler is here overridden to create
609 a UDP socket (SOCK_DGRAM).
610 """
611 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
612 return s
613
614 def send(self, s):
615 """
616 Send a pickled string to a socket.
617
618 This function no longer allows for partial sends which can happen
619 when the network is busy - UDP does not guarantee delivery and
620 can deliver packets out of sequence.
621 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000622 if self.sock is None:
623 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000624 self.sock.sendto(s, (self.host, self.port))
625
626class SysLogHandler(logging.Handler):
627 """
628 A handler class which sends formatted logging records to a syslog
629 server. Based on Sam Rushing's syslog module:
630 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
631 Contributed by Nicolas Untz (after which minor refactoring changes
632 have been made).
633 """
634
635 # from <linux/sys/syslog.h>:
636 # ======================================================================
637 # priorities/facilities are encoded into a single 32-bit quantity, where
638 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
639 # facility (0-big number). Both the priorities and the facilities map
640 # roughly one-to-one to strings in the syslogd(8) source code. This
641 # mapping is included in this file.
642 #
643 # priorities (these are ordered)
644
645 LOG_EMERG = 0 # system is unusable
646 LOG_ALERT = 1 # action must be taken immediately
647 LOG_CRIT = 2 # critical conditions
648 LOG_ERR = 3 # error conditions
649 LOG_WARNING = 4 # warning conditions
650 LOG_NOTICE = 5 # normal but significant condition
651 LOG_INFO = 6 # informational
652 LOG_DEBUG = 7 # debug-level messages
653
654 # facility codes
655 LOG_KERN = 0 # kernel messages
656 LOG_USER = 1 # random user-level messages
657 LOG_MAIL = 2 # mail system
658 LOG_DAEMON = 3 # system daemons
659 LOG_AUTH = 4 # security/authorization messages
660 LOG_SYSLOG = 5 # messages generated internally by syslogd
661 LOG_LPR = 6 # line printer subsystem
662 LOG_NEWS = 7 # network news subsystem
663 LOG_UUCP = 8 # UUCP subsystem
664 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000665 LOG_AUTHPRIV = 10 # security/authorization messages (private)
666 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000667
668 # other codes through 15 reserved for system use
669 LOG_LOCAL0 = 16 # reserved for local use
670 LOG_LOCAL1 = 17 # reserved for local use
671 LOG_LOCAL2 = 18 # reserved for local use
672 LOG_LOCAL3 = 19 # reserved for local use
673 LOG_LOCAL4 = 20 # reserved for local use
674 LOG_LOCAL5 = 21 # reserved for local use
675 LOG_LOCAL6 = 22 # reserved for local use
676 LOG_LOCAL7 = 23 # reserved for local use
677
678 priority_names = {
679 "alert": LOG_ALERT,
680 "crit": LOG_CRIT,
681 "critical": LOG_CRIT,
682 "debug": LOG_DEBUG,
683 "emerg": LOG_EMERG,
684 "err": LOG_ERR,
685 "error": LOG_ERR, # DEPRECATED
686 "info": LOG_INFO,
687 "notice": LOG_NOTICE,
688 "panic": LOG_EMERG, # DEPRECATED
689 "warn": LOG_WARNING, # DEPRECATED
690 "warning": LOG_WARNING,
691 }
692
693 facility_names = {
694 "auth": LOG_AUTH,
695 "authpriv": LOG_AUTHPRIV,
696 "cron": LOG_CRON,
697 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000698 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000699 "kern": LOG_KERN,
700 "lpr": LOG_LPR,
701 "mail": LOG_MAIL,
702 "news": LOG_NEWS,
703 "security": LOG_AUTH, # DEPRECATED
704 "syslog": LOG_SYSLOG,
705 "user": LOG_USER,
706 "uucp": LOG_UUCP,
707 "local0": LOG_LOCAL0,
708 "local1": LOG_LOCAL1,
709 "local2": LOG_LOCAL2,
710 "local3": LOG_LOCAL3,
711 "local4": LOG_LOCAL4,
712 "local5": LOG_LOCAL5,
713 "local6": LOG_LOCAL6,
714 "local7": LOG_LOCAL7,
715 }
716
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000717 #The map below appears to be trivially lowercasing the key. However,
718 #there's more to it than meets the eye - in some locales, lowercasing
719 #gives unexpected results. See SF #1524081: in the Turkish locale,
720 #"INFO".lower() != "info"
721 priority_map = {
722 "DEBUG" : "debug",
723 "INFO" : "info",
724 "WARNING" : "warning",
725 "ERROR" : "error",
726 "CRITICAL" : "critical"
727 }
728
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000729 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
730 facility=LOG_USER, socktype=socket.SOCK_DGRAM):
Guido van Rossum57102f82002-11-13 16:15:58 +0000731 """
732 Initialize a handler.
733
Guido van Rossume7ba4952007-06-06 23:52:48 +0000734 If address is specified as a string, a UNIX socket is used. To log to a
735 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000736 If facility is not specified, LOG_USER is used.
737 """
738 logging.Handler.__init__(self)
739
740 self.address = address
741 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000742 self.socktype = socktype
743
Guido van Rossum13257902007-06-07 23:15:56 +0000744 if isinstance(address, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000745 self.unixsocket = 1
Thomas Wouters89f507f2006-12-13 04:49:30 +0000746 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000747 else:
Guido van Rossum57102f82002-11-13 16:15:58 +0000748 self.unixsocket = 0
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000749 self.socket = socket.socket(socket.AF_INET, socktype)
750 if socktype == socket.SOCK_STREAM:
751 self.socket.connect(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000752 self.formatter = None
753
Vinay Sajipa1974c12005-01-13 08:23:56 +0000754 def _connect_unixsocket(self, address):
Vinay Sajipb5ca9322012-10-09 08:19:25 +0100755 self.socket = socket.socket(socket.AF_UNIX, self.socktype)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000756 try:
757 self.socket.connect(address)
758 except socket.error:
759 self.socket.close()
Vinay Sajipb5ca9322012-10-09 08:19:25 +0100760 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000761
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000762 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000763 """
764 Encode the facility and priority. You can pass in strings or
765 integers - if strings are passed, the facility_names and
766 priority_names mapping dictionaries are used to convert them to
767 integers.
768 """
Guido van Rossum13257902007-06-07 23:15:56 +0000769 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000770 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000771 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000772 priority = self.priority_names[priority]
773 return (facility << 3) | priority
774
775 def close (self):
776 """
777 Closes the socket.
778 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000779 self.acquire()
780 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000781 if self.unixsocket:
782 self.socket.close()
783 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000784 finally:
785 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000786
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000787 def mapPriority(self, levelName):
788 """
789 Map a logging level name to a key in the priority_names map.
790 This is useful in two scenarios: when custom levels are being
791 used, and in the case where you can't do a straightforward
792 mapping by lowercasing the logging level name because of locale-
793 specific issues (see SF #1524081).
794 """
795 return self.priority_map.get(levelName, "warning")
796
Vinay Sajip8168d102011-06-09 16:50:49 +0100797 append_nul = True # some old syslog daemons expect a NUL terminator
798
Guido van Rossum57102f82002-11-13 16:15:58 +0000799 def emit(self, record):
800 """
801 Emit a record.
802
803 The record is formatted, and then sent to the syslog server. If
804 exception information is present, it is NOT sent to the server.
805 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100806 msg = self.format(record)
807 if self.append_nul:
808 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000809 """
810 We need to convert record level to lowercase, maybe this will
811 change in the future.
812 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000813 prio = '<%d>' % self.encodePriority(self.facility,
814 self.mapPriority(record.levelname))
815 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000816 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000817 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000818 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000819 try:
820 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000821 try:
822 self.socket.send(msg)
823 except socket.error:
824 self._connect_unixsocket(self.address)
825 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000826 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000827 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000828 else:
829 self.socket.sendall(msg)
Vinay Sajip85c19092005-10-31 13:14:19 +0000830 except (KeyboardInterrupt, SystemExit):
831 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000832 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000833 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000834
835class SMTPHandler(logging.Handler):
836 """
837 A handler class which sends an SMTP email for each logging event.
838 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000839 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip25fcd222009-12-06 18:05:04 +0000840 credentials=None, secure=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000841 """
842 Initialize the handler.
843
844 Initialize the instance with the from and to addresses and subject
845 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000846 (host, port) tuple format for the mailhost argument. To specify
847 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000848 for the credentials argument. To specify the use of a secure
849 protocol (TLS), pass in a tuple for the secure argument. This will
850 only be used when authentication credentials are supplied. The tuple
851 will be either an empty tuple, or a single-value tuple with the name
852 of a keyfile, or a 2-value tuple with the names of the keyfile and
853 certificate file. (This tuple is passed to the `starttls` method).
Guido van Rossum57102f82002-11-13 16:15:58 +0000854 """
855 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000856 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000857 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000858 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000859 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000860 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000861 self.username, self.password = credentials
862 else:
863 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000864 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000865 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000866 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000867 self.toaddrs = toaddrs
868 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000869 self.secure = secure
Vinay Sajipee64b282012-03-23 14:36:22 +0000870 self._timeout = 5.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000871
872 def getSubject(self, record):
873 """
874 Determine the subject for the email.
875
876 If you want to specify a subject line which is record-dependent,
877 override this method.
878 """
879 return self.subject
880
881 def emit(self, record):
882 """
883 Emit a record.
884
885 Format the record and send it to the specified addressees.
886 """
887 try:
888 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000889 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000890 port = self.mailport
891 if not port:
892 port = smtplib.SMTP_PORT
Vinay Sajipee64b282012-03-23 14:36:22 +0000893 smtp = smtplib.SMTP(self.mailhost, port, timeout=self._timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000894 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000895 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 +0000896 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000897 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000898 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000899 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000900 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000901 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000902 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000903 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000904 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000905 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000906 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
907 smtp.quit()
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000908 except (KeyboardInterrupt, SystemExit):
909 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000910 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000911 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000912
913class NTEventLogHandler(logging.Handler):
914 """
915 A handler class which sends events to the NT Event Log. Adds a
916 registry entry for the specified application name. If no dllname is
917 provided, win32service.pyd (which contains some basic message
918 placeholders) is used. Note that use of these placeholders will make
919 your event logs big, as the entire message source is held in the log.
920 If you want slimmer logs, you have to pass in the name of your own DLL
921 which contains the message definitions you want to use in the event log.
922 """
923 def __init__(self, appname, dllname=None, logtype="Application"):
924 logging.Handler.__init__(self)
925 try:
926 import win32evtlogutil, win32evtlog
927 self.appname = appname
928 self._welu = win32evtlogutil
929 if not dllname:
930 dllname = os.path.split(self._welu.__file__)
931 dllname = os.path.split(dllname[0])
932 dllname = os.path.join(dllname[0], r'win32service.pyd')
933 self.dllname = dllname
934 self.logtype = logtype
935 self._welu.AddSourceToRegistry(appname, dllname, logtype)
936 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
937 self.typemap = {
938 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
939 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000940 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000941 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
942 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
943 }
944 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000945 print("The Python Win32 extensions for NT (service, event "\
946 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000947 self._welu = None
948
949 def getMessageID(self, record):
950 """
951 Return the message ID for the event record. If you are using your
952 own messages, you could do this by having the msg passed to the
953 logger being an ID rather than a formatting string. Then, in here,
954 you could use a dictionary lookup to get the message ID. This
955 version returns 1, which is the base message ID in win32service.pyd.
956 """
957 return 1
958
959 def getEventCategory(self, record):
960 """
961 Return the event category for the record.
962
963 Override this if you want to specify your own categories. This version
964 returns 0.
965 """
966 return 0
967
968 def getEventType(self, record):
969 """
970 Return the event type for the record.
971
972 Override this if you want to specify your own types. This version does
973 a mapping using the handler's typemap attribute, which is set up in
974 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000975 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000976 either need to override this method or place a suitable dictionary in
977 the handler's typemap attribute.
978 """
979 return self.typemap.get(record.levelno, self.deftype)
980
981 def emit(self, record):
982 """
983 Emit a record.
984
985 Determine the message ID, event category and event type. Then
986 log the message in the NT event log.
987 """
988 if self._welu:
989 try:
990 id = self.getMessageID(record)
991 cat = self.getEventCategory(record)
992 type = self.getEventType(record)
993 msg = self.format(record)
994 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000995 except (KeyboardInterrupt, SystemExit):
996 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000997 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000998 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000999
1000 def close(self):
1001 """
1002 Clean up this handler.
1003
1004 You can remove the application name from the registry as a
1005 source of event log entries. However, if you do this, you will
1006 not be able to see the events as you intended in the Event Log
1007 Viewer - it needs to be able to access the registry to get the
1008 DLL name.
1009 """
1010 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001011 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001012
1013class HTTPHandler(logging.Handler):
1014 """
1015 A class which sends records to a Web server, using either GET or
1016 POST semantics.
1017 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001018 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001019 """
1020 Initialize the instance with the host, the request URL, and the method
1021 ("GET" or "POST")
1022 """
1023 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001024 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001025 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001026 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001027 self.host = host
1028 self.url = url
1029 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001030 self.secure = secure
1031 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001032
Neal Norwitzf297bd12003-04-23 03:49:43 +00001033 def mapLogRecord(self, record):
1034 """
1035 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001036 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001037 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001038 """
1039 return record.__dict__
1040
Guido van Rossum57102f82002-11-13 16:15:58 +00001041 def emit(self, record):
1042 """
1043 Emit a record.
1044
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001045 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001046 """
1047 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001048 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001049 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001050 if self.secure:
1051 h = http.client.HTTPSConnection(host)
1052 else:
1053 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001054 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001055 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001056 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001057 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001058 sep = '&'
1059 else:
1060 sep = '?'
1061 url = url + "%c%s" % (sep, data)
1062 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001063 # support multiple hosts on one IP address...
1064 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001065 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001066 if i >= 0:
1067 host = host[:i]
1068 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001069 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001070 h.putheader("Content-type",
1071 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001072 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001073 if self.credentials:
1074 import base64
1075 s = ('u%s:%s' % self.credentials).encode('utf-8')
1076 s = 'Basic ' + base64.b64encode(s).strip()
1077 h.putheader('Authorization', s)
Benjamin Peterson458ad472009-01-18 00:08:45 +00001078 h.endheaders(data if self.method == "POST" else None)
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001079 h.getresponse() #can't do anything with the result
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001080 except (KeyboardInterrupt, SystemExit):
1081 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001082 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001083 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001084
1085class BufferingHandler(logging.Handler):
1086 """
1087 A handler class which buffers logging records in memory. Whenever each
1088 record is added to the buffer, a check is made to see if the buffer should
1089 be flushed. If it should, then flush() is expected to do what's needed.
1090 """
1091 def __init__(self, capacity):
1092 """
1093 Initialize the handler with the buffer size.
1094 """
1095 logging.Handler.__init__(self)
1096 self.capacity = capacity
1097 self.buffer = []
1098
1099 def shouldFlush(self, record):
1100 """
1101 Should the handler flush its buffer?
1102
1103 Returns true if the buffer is up to capacity. This method can be
1104 overridden to implement custom flushing strategies.
1105 """
1106 return (len(self.buffer) >= self.capacity)
1107
1108 def emit(self, record):
1109 """
1110 Emit a record.
1111
1112 Append the record. If shouldFlush() tells us to, call flush() to process
1113 the buffer.
1114 """
1115 self.buffer.append(record)
1116 if self.shouldFlush(record):
1117 self.flush()
1118
1119 def flush(self):
1120 """
1121 Override to implement custom flushing behaviour.
1122
1123 This version just zaps the buffer to empty.
1124 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001125 self.acquire()
1126 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001127 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001128 finally:
1129 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001130
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001131 def close(self):
1132 """
1133 Close the handler.
1134
1135 This version just flushes and chains to the parent class' close().
1136 """
1137 self.flush()
1138 logging.Handler.close(self)
1139
Guido van Rossum57102f82002-11-13 16:15:58 +00001140class MemoryHandler(BufferingHandler):
1141 """
1142 A handler class which buffers logging records in memory, periodically
1143 flushing them to a target handler. Flushing occurs whenever the buffer
1144 is full, or when an event of a certain severity or greater is seen.
1145 """
1146 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1147 """
1148 Initialize the handler with the buffer size, the level at which
1149 flushing should occur and an optional target.
1150
1151 Note that without a target being set either here or via setTarget(),
1152 a MemoryHandler is no use to anyone!
1153 """
1154 BufferingHandler.__init__(self, capacity)
1155 self.flushLevel = flushLevel
1156 self.target = target
1157
1158 def shouldFlush(self, record):
1159 """
1160 Check for buffer full or a record at the flushLevel or higher.
1161 """
1162 return (len(self.buffer) >= self.capacity) or \
1163 (record.levelno >= self.flushLevel)
1164
1165 def setTarget(self, target):
1166 """
1167 Set the target handler for this handler.
1168 """
1169 self.target = target
1170
1171 def flush(self):
1172 """
1173 For a MemoryHandler, flushing means just sending the buffered
1174 records to the target, if there is one. Override if you want
1175 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001176
1177 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001178 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001179 self.acquire()
1180 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001181 if self.target:
1182 for record in self.buffer:
1183 self.target.handle(record)
1184 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001185 finally:
1186 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001187
1188 def close(self):
1189 """
1190 Flush, set the target to None and lose the buffer.
1191 """
1192 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001193 self.acquire()
1194 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001195 self.target = None
1196 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001197 finally:
1198 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001199
1200
1201class QueueHandler(logging.Handler):
1202 """
1203 This handler sends events to a queue. Typically, it would be used together
1204 with a multiprocessing Queue to centralise logging to file in one process
1205 (in a multi-process application), so as to avoid file write contention
1206 between processes.
1207
1208 This code is new in Python 3.2, but this class can be copy pasted into
1209 user code for use with earlier Python versions.
1210 """
1211
1212 def __init__(self, queue):
1213 """
1214 Initialise an instance, using the passed queue.
1215 """
1216 logging.Handler.__init__(self)
1217 self.queue = queue
1218
1219 def enqueue(self, record):
1220 """
1221 Enqueue a record.
1222
1223 The base implementation uses put_nowait. You may want to override
1224 this method if you want to use blocking, timeouts or custom queue
1225 implementations.
1226 """
1227 self.queue.put_nowait(record)
1228
Vinay Sajip0258ce82010-09-22 20:34:53 +00001229 def prepare(self, record):
1230 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001231 Prepares a record for queuing. The object returned by this method is
1232 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001233
1234 The base implementation formats the record to merge the message
1235 and arguments, and removes unpickleable items from the record
1236 in-place.
1237
1238 You might want to override this method if you want to convert
1239 the record to a dict or JSON string, or send a modified copy
1240 of the record while leaving the original intact.
1241 """
1242 # The format operation gets traceback text into record.exc_text
1243 # (if there's exception data), and also puts the message into
1244 # record.message. We can then use this to replace the original
1245 # msg + args, as these might be unpickleable. We also zap the
1246 # exc_info attribute, as it's no longer needed and, if not None,
1247 # will typically not be pickleable.
1248 self.format(record)
1249 record.msg = record.message
1250 record.args = None
1251 record.exc_info = None
1252 return record
1253
Vinay Sajip121a1c42010-09-08 10:46:15 +00001254 def emit(self, record):
1255 """
1256 Emit a record.
1257
Vinay Sajip0637d492010-09-23 08:15:54 +00001258 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001259 """
1260 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001261 self.enqueue(self.prepare(record))
Vinay Sajip121a1c42010-09-08 10:46:15 +00001262 except (KeyboardInterrupt, SystemExit):
1263 raise
1264 except:
1265 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001266
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001267if threading:
1268 class QueueListener(object):
1269 """
1270 This class implements an internal threaded listener which watches for
1271 LogRecords being added to a queue, removes them and passes them to a
1272 list of handlers for processing.
1273 """
1274 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001275
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001276 def __init__(self, queue, *handlers):
1277 """
1278 Initialise an instance with the specified queue and
1279 handlers.
1280 """
1281 self.queue = queue
1282 self.handlers = handlers
1283 self._stop = threading.Event()
1284 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001285
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001286 def dequeue(self, block):
1287 """
1288 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001289
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001290 The base implementation uses get. You may want to override this method
1291 if you want to use timeouts or work with custom queue implementations.
1292 """
1293 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001294
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001295 def start(self):
1296 """
1297 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001298
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001299 This starts up a background thread to monitor the queue for
1300 LogRecords to process.
1301 """
1302 self._thread = t = threading.Thread(target=self._monitor)
1303 t.setDaemon(True)
1304 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001305
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001306 def prepare(self , record):
1307 """
1308 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001309
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001310 This method just returns the passed-in record. You may want to
1311 override this method if you need to do any custom marshalling or
1312 manipulation of the record before passing it to the handlers.
1313 """
1314 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001315
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001316 def handle(self, record):
1317 """
1318 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001319
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001320 This just loops through the handlers offering them the record
1321 to handle.
1322 """
1323 record = self.prepare(record)
1324 for handler in self.handlers:
1325 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001326
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001327 def _monitor(self):
1328 """
1329 Monitor the queue for records, and ask the handler
1330 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001331
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001332 This method runs on a separate, internal thread.
1333 The thread will terminate if it sees a sentinel object in the queue.
1334 """
1335 q = self.queue
1336 has_task_done = hasattr(q, 'task_done')
1337 while not self._stop.isSet():
1338 try:
1339 record = self.dequeue(True)
1340 if record is self._sentinel:
1341 break
1342 self.handle(record)
1343 if has_task_done:
1344 q.task_done()
1345 except queue.Empty:
1346 pass
1347 # There might still be records in the queue.
1348 while True:
1349 try:
1350 record = self.dequeue(False)
1351 if record is self._sentinel:
1352 break
1353 self.handle(record)
1354 if has_task_done:
1355 q.task_done()
1356 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001357 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001358
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001359 def stop(self):
1360 """
1361 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001362
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001363 This asks the thread to terminate, and then waits for it to do so.
1364 Note that if you don't call this before your application exits, there
1365 may be some records still left on the queue, which won't be processed.
1366 """
1367 self._stop.set()
1368 self.queue.put_nowait(self._sentinel)
1369 self._thread.join()
1370 self._thread = None