blob: d4df5c6d61d5930367d6ecf06ae89dd0b24395f5 [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.mode = 'w'
141 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000142
Vinay Sajip17c52d82004-07-03 11:48:34 +0000143 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000144 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000145 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000146
Vinay Sajip17c52d82004-07-03 11:48:34 +0000147 Basically, see if the supplied record would cause the file to exceed
148 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000149 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000150 if self.stream is None: # delay was set...
151 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000152 if self.maxBytes > 0: # are we rolling over?
153 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000154 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000155 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000156 return 1
157 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000158
Vinay Sajip17c52d82004-07-03 11:48:34 +0000159class TimedRotatingFileHandler(BaseRotatingHandler):
160 """
161 Handler for logging to a file, rotating the log file at certain timed
162 intervals.
163
164 If backupCount is > 0, when rollover is done, no more than backupCount
165 files are kept - the oldest ones are deleted.
166 """
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000167 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Christian Heimese7a15bb2008-01-24 16:21:45 +0000168 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000169 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000170 self.backupCount = backupCount
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000171 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000172 # Calculate the real rollover interval, which is just the number of
173 # seconds between rollovers. Also set the filename suffix used when
174 # a rollover occurs. Current 'when' events supported:
175 # S - Seconds
176 # M - Minutes
177 # H - Hours
178 # D - Days
179 # midnight - roll over at midnight
180 # W{0-6} - roll over on a certain day; 0 - Monday
181 #
182 # Case of the 'when' specifier is not important; lower or upper case
183 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000184 if self.when == 'S':
185 self.interval = 1 # one second
186 self.suffix = "%Y-%m-%d_%H-%M-%S"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000187 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000188 elif self.when == 'M':
189 self.interval = 60 # one minute
190 self.suffix = "%Y-%m-%d_%H-%M"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000191 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000192 elif self.when == 'H':
193 self.interval = 60 * 60 # one hour
194 self.suffix = "%Y-%m-%d_%H"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000195 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000196 elif self.when == 'D' or self.when == 'MIDNIGHT':
197 self.interval = 60 * 60 * 24 # one day
198 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000199 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000200 elif self.when.startswith('W'):
201 self.interval = 60 * 60 * 24 * 7 # one week
202 if len(self.when) != 2:
203 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
204 if self.when[1] < '0' or self.when[1] > '6':
205 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
206 self.dayOfWeek = int(self.when[1])
207 self.suffix = "%Y-%m-%d"
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000208 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000209 else:
210 raise ValueError("Invalid rollover interval specified: %s" % self.when)
211
Antoine Pitroufd036452008-08-19 17:56:33 +0000212 self.extMatch = re.compile(self.extMatch, re.ASCII)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000213 self.interval = self.interval * interval # multiply by units requested
Benjamin Peterson9451a1c2010-03-13 22:30:34 +0000214 if os.path.exists(filename):
215 t = os.stat(filename)[ST_MTIME]
216 else:
217 t = int(time.time())
218 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000219
Vinay Sajipaffbd872009-06-11 10:11:47 +0000220 def computeRollover(self, currentTime):
221 """
222 Work out the rollover time based on the specified time.
223 """
224 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000225 # If we are rolling over at midnight or weekly, then the interval is already known.
226 # What we need to figure out is WHEN the next interval is. In other words,
227 # if you are rolling over at midnight, then your base interval is 1 day,
228 # but you want to start that one day clock at midnight, not now. So, we
229 # have to fudge the rolloverAt value in order to trigger the first rollover
230 # at the right time. After that, the regular interval will take care of
231 # the rest. Note that this code doesn't care about leap seconds. :)
232 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
233 # This could be done with less code, but I wanted it to be clear
Vinay Sajipaffbd872009-06-11 10:11:47 +0000234 if self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000235 t = time.gmtime(currentTime)
236 else:
237 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000238 currentHour = t[3]
239 currentMinute = t[4]
240 currentSecond = t[5]
241 # r is the number of seconds left between now and midnight
Thomas Wouters477c8d52006-05-27 19:21:47 +0000242 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
243 currentSecond)
Vinay Sajipaffbd872009-06-11 10:11:47 +0000244 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000245 # If we are rolling over on a certain day, add in the number of days until
246 # the next rollover, but offset by 1 since we just calculated the time
247 # until the next day starts. There are three cases:
248 # Case 1) The day to rollover is today; in this case, do nothing
249 # Case 2) The day to rollover is further in the interval (i.e., today is
250 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
251 # next rollover is simply 6 - 2 - 1, or 3.
252 # Case 3) The day to rollover is behind us in the interval (i.e., today
253 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
254 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
255 # number of days left in the current week (1) plus the number
256 # of days in the next week until the rollover day (3).
Georg Brandl86def6c2008-01-21 20:36:10 +0000257 # The calculations described in 2) and 3) above need to have a day added.
258 # This is because the above time calculation takes us to midnight on this
259 # day, i.e. the start of the next day.
Vinay Sajipaffbd872009-06-11 10:11:47 +0000260 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000261 day = t[6] # 0 is Monday
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000262 if day != self.dayOfWeek:
263 if day < self.dayOfWeek:
Georg Brandl86def6c2008-01-21 20:36:10 +0000264 daysToWait = self.dayOfWeek - day
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000265 else:
Georg Brandl86def6c2008-01-21 20:36:10 +0000266 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajipaffbd872009-06-11 10:11:47 +0000267 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
268 if not self.utc:
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000269 dstNow = t[-1]
270 dstAtRollover = time.localtime(newRolloverAt)[-1]
271 if dstNow != dstAtRollover:
272 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000273 addend = -3600
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000274 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000275 addend = 3600
276 newRolloverAt += addend
Vinay Sajipaffbd872009-06-11 10:11:47 +0000277 result = newRolloverAt
278 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000279
280 def shouldRollover(self, record):
281 """
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000282 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000283
284 record is not used, as we are just comparing times, but it is needed so
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000285 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000286 """
287 t = int(time.time())
288 if t >= self.rolloverAt:
289 return 1
Vinay Sajip17c52d82004-07-03 11:48:34 +0000290 return 0
291
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000292 def getFilesToDelete(self):
293 """
294 Determine the files to delete when rolling over.
295
296 More specific than the earlier method, which just used glob.glob().
297 """
298 dirName, baseName = os.path.split(self.baseFilename)
299 fileNames = os.listdir(dirName)
300 result = []
301 prefix = baseName + "."
302 plen = len(prefix)
303 for fileName in fileNames:
304 if fileName[:plen] == prefix:
305 suffix = fileName[plen:]
306 if self.extMatch.match(suffix):
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000307 result.append(os.path.join(dirName, fileName))
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000308 result.sort()
309 if len(result) < self.backupCount:
310 result = []
311 else:
312 result = result[:len(result) - self.backupCount]
313 return result
314
Vinay Sajip17c52d82004-07-03 11:48:34 +0000315 def doRollover(self):
316 """
317 do a rollover; in this case, a date/time stamp is appended to the filename
318 when the rollover happens. However, you want the file to be named for the
319 start of the interval, not the current time. If there is a backup count,
320 then we have to get a list of matching filenames, sort them and remove
321 the one with the oldest suffix.
322 """
Vinay Sajip6268cbc2009-01-21 00:19:28 +0000323 if self.stream:
324 self.stream.close()
Vinay Sajip01241d62011-01-21 23:35:57 +0000325 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000326 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip27f48972012-03-13 12:10:33 +0000327 currentTime = int(time.time())
328 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000329 t = self.rolloverAt - self.interval
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000330 if self.utc:
331 timeTuple = time.gmtime(t)
332 else:
333 timeTuple = time.localtime(t)
Vinay Sajip27f48972012-03-13 12:10:33 +0000334 dstThen = timeTuple[-1]
335 if dstNow != dstThen:
336 if dstNow:
337 addend = 3600
338 else:
339 addend = -3600
340 timeTuple = time.localtime(t + addend)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000341 dfn = self.baseFilename + "." + time.strftime(self.suffix, timeTuple)
342 if os.path.exists(dfn):
343 os.remove(dfn)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000344 os.rename(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000345 if self.backupCount > 0:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000346 for s in self.getFilesToDelete():
347 os.remove(s)
Thomas Woutersb2137042007-02-01 18:02:27 +0000348 self.mode = 'w'
349 self.stream = self._open()
Vinay Sajipaffbd872009-06-11 10:11:47 +0000350 newRolloverAt = self.computeRollover(currentTime)
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000351 while newRolloverAt <= currentTime:
352 newRolloverAt = newRolloverAt + self.interval
353 #If DST changes and midnight or weekly rollover, adjust for this.
Benjamin Petersona37cfc62008-05-26 13:48:34 +0000354 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000355 dstAtRollover = time.localtime(newRolloverAt)[-1]
356 if dstNow != dstAtRollover:
357 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000358 addend = -3600
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000359 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip27f48972012-03-13 12:10:33 +0000360 addend = 3600
361 newRolloverAt += addend
Benjamin Petersonad9d48d2008-04-02 21:49:44 +0000362 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000363
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000364class WatchedFileHandler(logging.FileHandler):
365 """
366 A handler for logging to a file, which watches the file
367 to see if it has changed while in use. This can happen because of
368 usage of programs such as newsyslog and logrotate which perform
369 log file rotation. This handler, intended for use under Unix,
370 watches the file to see if it has changed since the last emit.
371 (A file has changed if its device or inode have changed.)
372 If it has changed, the old file stream is closed, and the file
373 opened to get a new stream.
374
375 This handler is not appropriate for use under Windows, because
376 under Windows open files cannot be moved or renamed - logging
377 opens the files with exclusive locks - and so there is no need
378 for such a handler. Furthermore, ST_INO is not supported under
379 Windows; stat always returns zero for this value.
380
381 This handler is based on a suggestion and patch by Chad J.
382 Schroeder.
383 """
Christian Heimese7a15bb2008-01-24 16:21:45 +0000384 def __init__(self, filename, mode='a', encoding=None, delay=0):
385 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
Vinay Sajip66b8b082012-04-24 23:25:30 +0100386 self.dev, self.ino = -1, -1
387 self._statstream()
388
389 def _statstream(self):
390 if self.stream:
391 sres = os.fstat(self.stream.fileno())
392 self.dev, self.ino = sres[ST_DEV], sres[ST_INO]
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000393
394 def emit(self, record):
395 """
396 Emit a record.
397
398 First check if the underlying file has changed, and if it
399 has, close the old stream and reopen the file to get the
400 current stream.
401 """
Vinay Sajip66b8b082012-04-24 23:25:30 +0100402 # Reduce the chance of race conditions by stat'ing by path only
403 # once and then fstat'ing our new fd if we opened a new log stream.
404 # See issue #14632: Thanks to John Mulligan for the problem report
405 # and patch.
406 try:
407 # stat the file by path, checking for existence
408 sres = os.stat(self.baseFilename)
409 except OSError as err:
410 if err.errno == errno.ENOENT:
411 sres = None
412 else:
413 raise
414 # compare file system stat with that of our stream file handle
415 if not sres or sres[ST_DEV] != self.dev or sres[ST_INO] != self.ino:
416 if self.stream is not None:
417 # we have an open file handle, clean it up
418 self.stream.flush()
419 self.stream.close()
420 # open a new file handle and get new stat info from that fd
421 self.stream = self._open()
422 self._statstream()
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000423 logging.FileHandler.emit(self, record)
424
Vinay Sajip66b8b082012-04-24 23:25:30 +0100425
Guido van Rossum57102f82002-11-13 16:15:58 +0000426class SocketHandler(logging.Handler):
427 """
428 A handler class which writes logging records, in pickle format, to
429 a streaming socket. The socket is kept open across logging calls.
430 If the peer resets it, an attempt is made to reconnect on the next call.
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000431 The pickle which is sent is that of the LogRecord's attribute dictionary
432 (__dict__), so that the receiver does not need to have the logging module
433 installed in order to process the logging event.
434
435 To unpickle the record at the receiving end into a LogRecord, use the
436 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000437 """
438
439 def __init__(self, host, port):
440 """
441 Initializes the handler with a specific host address and port.
442
443 The attribute 'closeOnError' is set to 1 - which means that if
444 a socket error occurs, the socket is silently closed and then
445 reopened on the next logging call.
446 """
447 logging.Handler.__init__(self)
448 self.host = host
449 self.port = port
450 self.sock = None
451 self.closeOnError = 0
Vinay Sajip48cfe382004-02-20 13:17:27 +0000452 self.retryTime = None
453 #
454 # Exponential backoff parameters.
455 #
456 self.retryStart = 1.0
457 self.retryMax = 30.0
458 self.retryFactor = 2.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000459
Guido van Rossumd8faa362007-04-27 19:54:29 +0000460 def makeSocket(self, timeout=1):
Guido van Rossum57102f82002-11-13 16:15:58 +0000461 """
462 A factory method which allows subclasses to define the precise
463 type of socket they want.
464 """
465 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000466 if hasattr(s, 'settimeout'):
467 s.settimeout(timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000468 s.connect((self.host, self.port))
469 return s
470
Vinay Sajip48cfe382004-02-20 13:17:27 +0000471 def createSocket(self):
472 """
473 Try to create a socket, using an exponential backoff with
474 a max retry time. Thanks to Robert Olson for the original patch
475 (SF #815911) which has been slightly refactored.
476 """
477 now = time.time()
478 # Either retryTime is None, in which case this
479 # is the first time back after a disconnect, or
480 # we've waited long enough.
481 if self.retryTime is None:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000482 attempt = 1
Vinay Sajip48cfe382004-02-20 13:17:27 +0000483 else:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000484 attempt = (now >= self.retryTime)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000485 if attempt:
486 try:
487 self.sock = self.makeSocket()
488 self.retryTime = None # next time, no delay before trying
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000489 except socket.error:
Vinay Sajip48cfe382004-02-20 13:17:27 +0000490 #Creation failed, so set the retry time and return.
491 if self.retryTime is None:
492 self.retryPeriod = self.retryStart
493 else:
494 self.retryPeriod = self.retryPeriod * self.retryFactor
495 if self.retryPeriod > self.retryMax:
496 self.retryPeriod = self.retryMax
497 self.retryTime = now + self.retryPeriod
498
Guido van Rossum57102f82002-11-13 16:15:58 +0000499 def send(self, s):
500 """
501 Send a pickled string to the socket.
502
503 This function allows for partial sends which can happen when the
504 network is busy.
505 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000506 if self.sock is None:
507 self.createSocket()
508 #self.sock can be None either because we haven't reached the retry
509 #time yet, or because we have reached the retry time and retried,
510 #but are still unable to connect.
511 if self.sock:
512 try:
513 if hasattr(self.sock, "sendall"):
514 self.sock.sendall(s)
515 else:
516 sentsofar = 0
517 left = len(s)
518 while left > 0:
519 sent = self.sock.send(s[sentsofar:])
520 sentsofar = sentsofar + sent
521 left = left - sent
522 except socket.error:
523 self.sock.close()
524 self.sock = None # so we can call createSocket next time
Guido van Rossum57102f82002-11-13 16:15:58 +0000525
526 def makePickle(self, record):
527 """
528 Pickles the record in binary format with a length prefix, and
529 returns it ready for transmission across the socket.
530 """
Vinay Sajip48cfe382004-02-20 13:17:27 +0000531 ei = record.exc_info
532 if ei:
Vinay Sajip6f5e54e2012-03-29 20:17:18 +0100533 # just to get traceback text into record.exc_text ...
534 dummy = self.format(record)
535 # See issue #14436: If msg or args are objects, they may not be
536 # available on the receiving end. So we convert the msg % args
537 # to a string, save it as msg and zap the args.
538 d = dict(record.__dict__)
539 d['msg'] = record.getMessage()
540 d['args'] = None
541 d['exc_info'] = None
542 s = pickle.dumps(d, 1)
Guido van Rossum57102f82002-11-13 16:15:58 +0000543 slen = struct.pack(">L", len(s))
544 return slen + s
545
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000546 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000547 """
548 Handle an error during logging.
549
550 An error has occurred during logging. Most likely cause -
551 connection lost. Close the socket so that we can retry on the
552 next event.
553 """
554 if self.closeOnError and self.sock:
555 self.sock.close()
556 self.sock = None #try to reconnect next time
557 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000558 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000559
560 def emit(self, record):
561 """
562 Emit a record.
563
564 Pickles the record and writes it to the socket in binary format.
565 If there is an error with the socket, silently drop the packet.
566 If there was a problem with the socket, re-establishes the
567 socket.
568 """
569 try:
570 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000571 self.send(s)
Vinay Sajip85c19092005-10-31 13:14:19 +0000572 except (KeyboardInterrupt, SystemExit):
573 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000574 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000575 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000576
577 def close(self):
578 """
579 Closes the socket.
580 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000581 self.acquire()
582 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000583 if self.sock:
584 self.sock.close()
585 self.sock = None
586 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000587 finally:
588 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000589
590class DatagramHandler(SocketHandler):
591 """
592 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000593 a datagram socket. The pickle which is sent is that of the LogRecord's
594 attribute dictionary (__dict__), so that the receiver does not need to
595 have the logging module installed in order to process the logging event.
596
597 To unpickle the record at the receiving end into a LogRecord, use the
598 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000599
600 """
601 def __init__(self, host, port):
602 """
603 Initializes the handler with a specific host address and port.
604 """
605 SocketHandler.__init__(self, host, port)
606 self.closeOnError = 0
607
608 def makeSocket(self):
609 """
610 The factory method of SocketHandler is here overridden to create
611 a UDP socket (SOCK_DGRAM).
612 """
613 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
614 return s
615
616 def send(self, s):
617 """
618 Send a pickled string to a socket.
619
620 This function no longer allows for partial sends which can happen
621 when the network is busy - UDP does not guarantee delivery and
622 can deliver packets out of sequence.
623 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000624 if self.sock is None:
625 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000626 self.sock.sendto(s, (self.host, self.port))
627
628class SysLogHandler(logging.Handler):
629 """
630 A handler class which sends formatted logging records to a syslog
631 server. Based on Sam Rushing's syslog module:
632 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
633 Contributed by Nicolas Untz (after which minor refactoring changes
634 have been made).
635 """
636
637 # from <linux/sys/syslog.h>:
638 # ======================================================================
639 # priorities/facilities are encoded into a single 32-bit quantity, where
640 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
641 # facility (0-big number). Both the priorities and the facilities map
642 # roughly one-to-one to strings in the syslogd(8) source code. This
643 # mapping is included in this file.
644 #
645 # priorities (these are ordered)
646
647 LOG_EMERG = 0 # system is unusable
648 LOG_ALERT = 1 # action must be taken immediately
649 LOG_CRIT = 2 # critical conditions
650 LOG_ERR = 3 # error conditions
651 LOG_WARNING = 4 # warning conditions
652 LOG_NOTICE = 5 # normal but significant condition
653 LOG_INFO = 6 # informational
654 LOG_DEBUG = 7 # debug-level messages
655
656 # facility codes
657 LOG_KERN = 0 # kernel messages
658 LOG_USER = 1 # random user-level messages
659 LOG_MAIL = 2 # mail system
660 LOG_DAEMON = 3 # system daemons
661 LOG_AUTH = 4 # security/authorization messages
662 LOG_SYSLOG = 5 # messages generated internally by syslogd
663 LOG_LPR = 6 # line printer subsystem
664 LOG_NEWS = 7 # network news subsystem
665 LOG_UUCP = 8 # UUCP subsystem
666 LOG_CRON = 9 # clock daemon
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000667 LOG_AUTHPRIV = 10 # security/authorization messages (private)
668 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000669
670 # other codes through 15 reserved for system use
671 LOG_LOCAL0 = 16 # reserved for local use
672 LOG_LOCAL1 = 17 # reserved for local use
673 LOG_LOCAL2 = 18 # reserved for local use
674 LOG_LOCAL3 = 19 # reserved for local use
675 LOG_LOCAL4 = 20 # reserved for local use
676 LOG_LOCAL5 = 21 # reserved for local use
677 LOG_LOCAL6 = 22 # reserved for local use
678 LOG_LOCAL7 = 23 # reserved for local use
679
680 priority_names = {
681 "alert": LOG_ALERT,
682 "crit": LOG_CRIT,
683 "critical": LOG_CRIT,
684 "debug": LOG_DEBUG,
685 "emerg": LOG_EMERG,
686 "err": LOG_ERR,
687 "error": LOG_ERR, # DEPRECATED
688 "info": LOG_INFO,
689 "notice": LOG_NOTICE,
690 "panic": LOG_EMERG, # DEPRECATED
691 "warn": LOG_WARNING, # DEPRECATED
692 "warning": LOG_WARNING,
693 }
694
695 facility_names = {
696 "auth": LOG_AUTH,
697 "authpriv": LOG_AUTHPRIV,
698 "cron": LOG_CRON,
699 "daemon": LOG_DAEMON,
Benjamin Peterson22005fc2010-04-11 16:25:06 +0000700 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000701 "kern": LOG_KERN,
702 "lpr": LOG_LPR,
703 "mail": LOG_MAIL,
704 "news": LOG_NEWS,
705 "security": LOG_AUTH, # DEPRECATED
706 "syslog": LOG_SYSLOG,
707 "user": LOG_USER,
708 "uucp": LOG_UUCP,
709 "local0": LOG_LOCAL0,
710 "local1": LOG_LOCAL1,
711 "local2": LOG_LOCAL2,
712 "local3": LOG_LOCAL3,
713 "local4": LOG_LOCAL4,
714 "local5": LOG_LOCAL5,
715 "local6": LOG_LOCAL6,
716 "local7": LOG_LOCAL7,
717 }
718
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000719 #The map below appears to be trivially lowercasing the key. However,
720 #there's more to it than meets the eye - in some locales, lowercasing
721 #gives unexpected results. See SF #1524081: in the Turkish locale,
722 #"INFO".lower() != "info"
723 priority_map = {
724 "DEBUG" : "debug",
725 "INFO" : "info",
726 "WARNING" : "warning",
727 "ERROR" : "error",
728 "CRITICAL" : "critical"
729 }
730
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000731 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
732 facility=LOG_USER, socktype=socket.SOCK_DGRAM):
Guido van Rossum57102f82002-11-13 16:15:58 +0000733 """
734 Initialize a handler.
735
Guido van Rossume7ba4952007-06-06 23:52:48 +0000736 If address is specified as a string, a UNIX socket is used. To log to a
737 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000738 If facility is not specified, LOG_USER is used.
739 """
740 logging.Handler.__init__(self)
741
742 self.address = address
743 self.facility = facility
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000744 self.socktype = socktype
745
Guido van Rossum13257902007-06-07 23:15:56 +0000746 if isinstance(address, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000747 self.unixsocket = 1
Thomas Wouters89f507f2006-12-13 04:49:30 +0000748 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000749 else:
Guido van Rossum57102f82002-11-13 16:15:58 +0000750 self.unixsocket = 0
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000751 self.socket = socket.socket(socket.AF_INET, socktype)
752 if socktype == socket.SOCK_STREAM:
753 self.socket.connect(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000754 self.formatter = None
755
Vinay Sajipa1974c12005-01-13 08:23:56 +0000756 def _connect_unixsocket(self, address):
757 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
758 # syslog may require either DGRAM or STREAM sockets
759 try:
760 self.socket.connect(address)
761 except socket.error:
762 self.socket.close()
763 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Vinay Sajip3e6d71d2012-06-25 23:18:45 +0100764 try:
765 self.socket.connect(address)
766 except socket.error:
767 self.socket.close()
768 raise
Vinay Sajipa1974c12005-01-13 08:23:56 +0000769
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000770 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000771 """
772 Encode the facility and priority. You can pass in strings or
773 integers - if strings are passed, the facility_names and
774 priority_names mapping dictionaries are used to convert them to
775 integers.
776 """
Guido van Rossum13257902007-06-07 23:15:56 +0000777 if isinstance(facility, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000778 facility = self.facility_names[facility]
Guido van Rossum13257902007-06-07 23:15:56 +0000779 if isinstance(priority, str):
Guido van Rossum57102f82002-11-13 16:15:58 +0000780 priority = self.priority_names[priority]
781 return (facility << 3) | priority
782
783 def close (self):
784 """
785 Closes the socket.
786 """
Vinay Sajipf0509032012-02-23 20:49:08 +0000787 self.acquire()
788 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +0000789 if self.unixsocket:
790 self.socket.close()
791 logging.Handler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +0000792 finally:
793 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +0000794
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000795 def mapPriority(self, levelName):
796 """
797 Map a logging level name to a key in the priority_names map.
798 This is useful in two scenarios: when custom levels are being
799 used, and in the case where you can't do a straightforward
800 mapping by lowercasing the logging level name because of locale-
801 specific issues (see SF #1524081).
802 """
803 return self.priority_map.get(levelName, "warning")
804
Vinay Sajip8168d102011-06-09 16:50:49 +0100805 append_nul = True # some old syslog daemons expect a NUL terminator
806
Guido van Rossum57102f82002-11-13 16:15:58 +0000807 def emit(self, record):
808 """
809 Emit a record.
810
811 The record is formatted, and then sent to the syslog server. If
812 exception information is present, it is NOT sent to the server.
813 """
Vinay Sajip8168d102011-06-09 16:50:49 +0100814 msg = self.format(record)
815 if self.append_nul:
816 msg += '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000817 """
818 We need to convert record level to lowercase, maybe this will
819 change in the future.
820 """
Vinay Sajip467d12f2010-08-19 22:20:22 +0000821 prio = '<%d>' % self.encodePriority(self.facility,
822 self.mapPriority(record.levelname))
823 prio = prio.encode('utf-8')
Vinay Sajip609364a2010-08-30 18:31:13 +0000824 # Message is a string. Convert to bytes as required by RFC 5424
Vinay Sajip42ead482009-10-21 20:22:14 +0000825 msg = msg.encode('utf-8')
Vinay Sajip467d12f2010-08-19 22:20:22 +0000826 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000827 try:
828 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000829 try:
830 self.socket.send(msg)
831 except socket.error:
832 self._connect_unixsocket(self.address)
833 self.socket.send(msg)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000834 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000835 self.socket.sendto(msg, self.address)
Vinay Sajipcbabd7e2009-10-10 20:32:36 +0000836 else:
837 self.socket.sendall(msg)
Vinay Sajip85c19092005-10-31 13:14:19 +0000838 except (KeyboardInterrupt, SystemExit):
839 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000840 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000841 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000842
843class SMTPHandler(logging.Handler):
844 """
845 A handler class which sends an SMTP email for each logging event.
846 """
Vinay Sajip540f2152009-12-06 17:57:11 +0000847 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajip25fcd222009-12-06 18:05:04 +0000848 credentials=None, secure=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000849 """
850 Initialize the handler.
851
852 Initialize the instance with the from and to addresses and subject
853 line of the email. To specify a non-standard SMTP port, use the
Guido van Rossum360e4b82007-05-14 22:51:27 +0000854 (host, port) tuple format for the mailhost argument. To specify
855 authentication credentials, supply a (username, password) tuple
Vinay Sajip25fcd222009-12-06 18:05:04 +0000856 for the credentials argument. To specify the use of a secure
857 protocol (TLS), pass in a tuple for the secure argument. This will
858 only be used when authentication credentials are supplied. The tuple
859 will be either an empty tuple, or a single-value tuple with the name
860 of a keyfile, or a 2-value tuple with the names of the keyfile and
861 certificate file. (This tuple is passed to the `starttls` method).
Guido van Rossum57102f82002-11-13 16:15:58 +0000862 """
863 logging.Handler.__init__(self)
Guido van Rossum13257902007-06-07 23:15:56 +0000864 if isinstance(mailhost, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000865 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000866 else:
Guido van Rossum360e4b82007-05-14 22:51:27 +0000867 self.mailhost, self.mailport = mailhost, None
Guido van Rossum13257902007-06-07 23:15:56 +0000868 if isinstance(credentials, tuple):
Guido van Rossum360e4b82007-05-14 22:51:27 +0000869 self.username, self.password = credentials
870 else:
871 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000872 self.fromaddr = fromaddr
Guido van Rossum13257902007-06-07 23:15:56 +0000873 if isinstance(toaddrs, str):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000874 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000875 self.toaddrs = toaddrs
876 self.subject = subject
Vinay Sajip540f2152009-12-06 17:57:11 +0000877 self.secure = secure
Vinay Sajipee64b282012-03-23 14:36:22 +0000878 self._timeout = 5.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000879
880 def getSubject(self, record):
881 """
882 Determine the subject for the email.
883
884 If you want to specify a subject line which is record-dependent,
885 override this method.
886 """
887 return self.subject
888
889 def emit(self, record):
890 """
891 Emit a record.
892
893 Format the record and send it to the specified addressees.
894 """
895 try:
896 import smtplib
Vinay Sajipdb81c4c2010-02-25 23:13:06 +0000897 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000898 port = self.mailport
899 if not port:
900 port = smtplib.SMTP_PORT
Vinay Sajipee64b282012-03-23 14:36:22 +0000901 smtp = smtplib.SMTP(self.mailhost, port, timeout=self._timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000902 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000903 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 +0000904 self.fromaddr,
Neal Norwitz9d72bb42007-04-17 08:48:32 +0000905 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000906 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000907 formatdate(), msg)
Guido van Rossum360e4b82007-05-14 22:51:27 +0000908 if self.username:
Vinay Sajip25fcd222009-12-06 18:05:04 +0000909 if self.secure is not None:
Vinay Sajip540f2152009-12-06 17:57:11 +0000910 smtp.ehlo()
Vinay Sajip25fcd222009-12-06 18:05:04 +0000911 smtp.starttls(*self.secure)
Vinay Sajip540f2152009-12-06 17:57:11 +0000912 smtp.ehlo()
Guido van Rossum360e4b82007-05-14 22:51:27 +0000913 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000914 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
915 smtp.quit()
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000916 except (KeyboardInterrupt, SystemExit):
917 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000918 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000919 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000920
921class NTEventLogHandler(logging.Handler):
922 """
923 A handler class which sends events to the NT Event Log. Adds a
924 registry entry for the specified application name. If no dllname is
925 provided, win32service.pyd (which contains some basic message
926 placeholders) is used. Note that use of these placeholders will make
927 your event logs big, as the entire message source is held in the log.
928 If you want slimmer logs, you have to pass in the name of your own DLL
929 which contains the message definitions you want to use in the event log.
930 """
931 def __init__(self, appname, dllname=None, logtype="Application"):
932 logging.Handler.__init__(self)
933 try:
934 import win32evtlogutil, win32evtlog
935 self.appname = appname
936 self._welu = win32evtlogutil
937 if not dllname:
938 dllname = os.path.split(self._welu.__file__)
939 dllname = os.path.split(dllname[0])
940 dllname = os.path.join(dllname[0], r'win32service.pyd')
941 self.dllname = dllname
942 self.logtype = logtype
943 self._welu.AddSourceToRegistry(appname, dllname, logtype)
944 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
945 self.typemap = {
946 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
947 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000948 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000949 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
950 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
951 }
952 except ImportError:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000953 print("The Python Win32 extensions for NT (service, event "\
954 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000955 self._welu = None
956
957 def getMessageID(self, record):
958 """
959 Return the message ID for the event record. If you are using your
960 own messages, you could do this by having the msg passed to the
961 logger being an ID rather than a formatting string. Then, in here,
962 you could use a dictionary lookup to get the message ID. This
963 version returns 1, which is the base message ID in win32service.pyd.
964 """
965 return 1
966
967 def getEventCategory(self, record):
968 """
969 Return the event category for the record.
970
971 Override this if you want to specify your own categories. This version
972 returns 0.
973 """
974 return 0
975
976 def getEventType(self, record):
977 """
978 Return the event type for the record.
979
980 Override this if you want to specify your own types. This version does
981 a mapping using the handler's typemap attribute, which is set up in
982 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000983 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000984 either need to override this method or place a suitable dictionary in
985 the handler's typemap attribute.
986 """
987 return self.typemap.get(record.levelno, self.deftype)
988
989 def emit(self, record):
990 """
991 Emit a record.
992
993 Determine the message ID, event category and event type. Then
994 log the message in the NT event log.
995 """
996 if self._welu:
997 try:
998 id = self.getMessageID(record)
999 cat = self.getEventCategory(record)
1000 type = self.getEventType(record)
1001 msg = self.format(record)
1002 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001003 except (KeyboardInterrupt, SystemExit):
1004 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001005 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001006 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001007
1008 def close(self):
1009 """
1010 Clean up this handler.
1011
1012 You can remove the application name from the registry as a
1013 source of event log entries. However, if you do this, you will
1014 not be able to see the events as you intended in the Event Log
1015 Viewer - it needs to be able to access the registry to get the
1016 DLL name.
1017 """
1018 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001019 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001020
1021class HTTPHandler(logging.Handler):
1022 """
1023 A class which sends records to a Web server, using either GET or
1024 POST semantics.
1025 """
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001026 def __init__(self, host, url, method="GET", secure=False, credentials=None):
Guido van Rossum57102f82002-11-13 16:15:58 +00001027 """
1028 Initialize the instance with the host, the request URL, and the method
1029 ("GET" or "POST")
1030 """
1031 logging.Handler.__init__(self)
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001032 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001033 if method not in ["GET", "POST"]:
Collin Winterce36ad82007-08-30 01:19:48 +00001034 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001035 self.host = host
1036 self.url = url
1037 self.method = method
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001038 self.secure = secure
1039 self.credentials = credentials
Guido van Rossum57102f82002-11-13 16:15:58 +00001040
Neal Norwitzf297bd12003-04-23 03:49:43 +00001041 def mapLogRecord(self, record):
1042 """
1043 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001044 that is sent as the CGI data. Overwrite in your class.
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001045 Contributed by Franz Glasner.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001046 """
1047 return record.__dict__
1048
Guido van Rossum57102f82002-11-13 16:15:58 +00001049 def emit(self, record):
1050 """
1051 Emit a record.
1052
Senthil Kumaran30e86a42010-08-09 20:01:35 +00001053 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001054 """
1055 try:
Georg Brandl029986a2008-06-23 11:44:14 +00001056 import http.client, urllib.parse
Vinay Sajipb7935062005-10-11 13:15:31 +00001057 host = self.host
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001058 if self.secure:
1059 h = http.client.HTTPSConnection(host)
1060 else:
1061 h = http.client.HTTPConnection(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001062 url = self.url
Georg Brandl029986a2008-06-23 11:44:14 +00001063 data = urllib.parse.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001064 if self.method == "GET":
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001065 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001066 sep = '&'
1067 else:
1068 sep = '?'
1069 url = url + "%c%s" % (sep, data)
1070 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001071 # support multiple hosts on one IP address...
1072 # need to strip optional :port from host, if present
Neal Norwitz9d72bb42007-04-17 08:48:32 +00001073 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001074 if i >= 0:
1075 host = host[:i]
1076 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001077 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001078 h.putheader("Content-type",
1079 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001080 h.putheader("Content-length", str(len(data)))
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001081 if self.credentials:
1082 import base64
1083 s = ('u%s:%s' % self.credentials).encode('utf-8')
1084 s = 'Basic ' + base64.b64encode(s).strip()
1085 h.putheader('Authorization', s)
Benjamin Peterson458ad472009-01-18 00:08:45 +00001086 h.endheaders(data if self.method == "POST" else None)
Vinay Sajipaf9d10a2010-09-13 20:40:30 +00001087 h.getresponse() #can't do anything with the result
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001088 except (KeyboardInterrupt, SystemExit):
1089 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001090 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001091 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001092
1093class BufferingHandler(logging.Handler):
1094 """
1095 A handler class which buffers logging records in memory. Whenever each
1096 record is added to the buffer, a check is made to see if the buffer should
1097 be flushed. If it should, then flush() is expected to do what's needed.
1098 """
1099 def __init__(self, capacity):
1100 """
1101 Initialize the handler with the buffer size.
1102 """
1103 logging.Handler.__init__(self)
1104 self.capacity = capacity
1105 self.buffer = []
1106
1107 def shouldFlush(self, record):
1108 """
1109 Should the handler flush its buffer?
1110
1111 Returns true if the buffer is up to capacity. This method can be
1112 overridden to implement custom flushing strategies.
1113 """
1114 return (len(self.buffer) >= self.capacity)
1115
1116 def emit(self, record):
1117 """
1118 Emit a record.
1119
1120 Append the record. If shouldFlush() tells us to, call flush() to process
1121 the buffer.
1122 """
1123 self.buffer.append(record)
1124 if self.shouldFlush(record):
1125 self.flush()
1126
1127 def flush(self):
1128 """
1129 Override to implement custom flushing behaviour.
1130
1131 This version just zaps the buffer to empty.
1132 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001133 self.acquire()
1134 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001135 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001136 finally:
1137 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001138
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001139 def close(self):
1140 """
1141 Close the handler.
1142
1143 This version just flushes and chains to the parent class' close().
1144 """
1145 self.flush()
1146 logging.Handler.close(self)
1147
Guido van Rossum57102f82002-11-13 16:15:58 +00001148class MemoryHandler(BufferingHandler):
1149 """
1150 A handler class which buffers logging records in memory, periodically
1151 flushing them to a target handler. Flushing occurs whenever the buffer
1152 is full, or when an event of a certain severity or greater is seen.
1153 """
1154 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1155 """
1156 Initialize the handler with the buffer size, the level at which
1157 flushing should occur and an optional target.
1158
1159 Note that without a target being set either here or via setTarget(),
1160 a MemoryHandler is no use to anyone!
1161 """
1162 BufferingHandler.__init__(self, capacity)
1163 self.flushLevel = flushLevel
1164 self.target = target
1165
1166 def shouldFlush(self, record):
1167 """
1168 Check for buffer full or a record at the flushLevel or higher.
1169 """
1170 return (len(self.buffer) >= self.capacity) or \
1171 (record.levelno >= self.flushLevel)
1172
1173 def setTarget(self, target):
1174 """
1175 Set the target handler for this handler.
1176 """
1177 self.target = target
1178
1179 def flush(self):
1180 """
1181 For a MemoryHandler, flushing means just sending the buffered
1182 records to the target, if there is one. Override if you want
1183 different behaviour.
Vinay Sajipc84f0162010-09-21 11:25:39 +00001184
1185 The record buffer is also cleared by this operation.
Guido van Rossum57102f82002-11-13 16:15:58 +00001186 """
Vinay Sajipf0509032012-02-23 20:49:08 +00001187 self.acquire()
1188 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001189 if self.target:
1190 for record in self.buffer:
1191 self.target.handle(record)
1192 self.buffer = []
Vinay Sajipf0509032012-02-23 20:49:08 +00001193 finally:
1194 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001195
1196 def close(self):
1197 """
1198 Flush, set the target to None and lose the buffer.
1199 """
1200 self.flush()
Vinay Sajipf0509032012-02-23 20:49:08 +00001201 self.acquire()
1202 try:
Vinay Sajip0abf61d2012-02-23 19:45:52 +00001203 self.target = None
1204 BufferingHandler.close(self)
Vinay Sajipf0509032012-02-23 20:49:08 +00001205 finally:
1206 self.release()
Vinay Sajip121a1c42010-09-08 10:46:15 +00001207
1208
1209class QueueHandler(logging.Handler):
1210 """
1211 This handler sends events to a queue. Typically, it would be used together
1212 with a multiprocessing Queue to centralise logging to file in one process
1213 (in a multi-process application), so as to avoid file write contention
1214 between processes.
1215
1216 This code is new in Python 3.2, but this class can be copy pasted into
1217 user code for use with earlier Python versions.
1218 """
1219
1220 def __init__(self, queue):
1221 """
1222 Initialise an instance, using the passed queue.
1223 """
1224 logging.Handler.__init__(self)
1225 self.queue = queue
1226
1227 def enqueue(self, record):
1228 """
1229 Enqueue a record.
1230
1231 The base implementation uses put_nowait. You may want to override
1232 this method if you want to use blocking, timeouts or custom queue
1233 implementations.
1234 """
1235 self.queue.put_nowait(record)
1236
Vinay Sajip0258ce82010-09-22 20:34:53 +00001237 def prepare(self, record):
1238 """
Vinay Sajip0637d492010-09-23 08:15:54 +00001239 Prepares a record for queuing. The object returned by this method is
1240 enqueued.
Vinay Sajip0258ce82010-09-22 20:34:53 +00001241
1242 The base implementation formats the record to merge the message
1243 and arguments, and removes unpickleable items from the record
1244 in-place.
1245
1246 You might want to override this method if you want to convert
1247 the record to a dict or JSON string, or send a modified copy
1248 of the record while leaving the original intact.
1249 """
1250 # The format operation gets traceback text into record.exc_text
1251 # (if there's exception data), and also puts the message into
1252 # record.message. We can then use this to replace the original
1253 # msg + args, as these might be unpickleable. We also zap the
1254 # exc_info attribute, as it's no longer needed and, if not None,
1255 # will typically not be pickleable.
1256 self.format(record)
1257 record.msg = record.message
1258 record.args = None
1259 record.exc_info = None
1260 return record
1261
Vinay Sajip121a1c42010-09-08 10:46:15 +00001262 def emit(self, record):
1263 """
1264 Emit a record.
1265
Vinay Sajip0637d492010-09-23 08:15:54 +00001266 Writes the LogRecord to the queue, preparing it for pickling first.
Vinay Sajip121a1c42010-09-08 10:46:15 +00001267 """
1268 try:
Vinay Sajip0258ce82010-09-22 20:34:53 +00001269 self.enqueue(self.prepare(record))
Vinay Sajip121a1c42010-09-08 10:46:15 +00001270 except (KeyboardInterrupt, SystemExit):
1271 raise
1272 except:
1273 self.handleError(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001274
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001275if threading:
1276 class QueueListener(object):
1277 """
1278 This class implements an internal threaded listener which watches for
1279 LogRecords being added to a queue, removes them and passes them to a
1280 list of handlers for processing.
1281 """
1282 _sentinel = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001283
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001284 def __init__(self, queue, *handlers):
1285 """
1286 Initialise an instance with the specified queue and
1287 handlers.
1288 """
1289 self.queue = queue
1290 self.handlers = handlers
1291 self._stop = threading.Event()
1292 self._thread = None
Vinay Sajip0637d492010-09-23 08:15:54 +00001293
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001294 def dequeue(self, block):
1295 """
1296 Dequeue a record and return it, optionally blocking.
Vinay Sajip0637d492010-09-23 08:15:54 +00001297
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001298 The base implementation uses get. You may want to override this method
1299 if you want to use timeouts or work with custom queue implementations.
1300 """
1301 return self.queue.get(block)
Vinay Sajip0637d492010-09-23 08:15:54 +00001302
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001303 def start(self):
1304 """
1305 Start the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001306
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001307 This starts up a background thread to monitor the queue for
1308 LogRecords to process.
1309 """
1310 self._thread = t = threading.Thread(target=self._monitor)
1311 t.setDaemon(True)
1312 t.start()
Vinay Sajip0637d492010-09-23 08:15:54 +00001313
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001314 def prepare(self , record):
1315 """
1316 Prepare a record for handling.
Vinay Sajip0637d492010-09-23 08:15:54 +00001317
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001318 This method just returns the passed-in record. You may want to
1319 override this method if you need to do any custom marshalling or
1320 manipulation of the record before passing it to the handlers.
1321 """
1322 return record
Vinay Sajip0637d492010-09-23 08:15:54 +00001323
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001324 def handle(self, record):
1325 """
1326 Handle a record.
Vinay Sajip0637d492010-09-23 08:15:54 +00001327
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001328 This just loops through the handlers offering them the record
1329 to handle.
1330 """
1331 record = self.prepare(record)
1332 for handler in self.handlers:
1333 handler.handle(record)
Vinay Sajip0637d492010-09-23 08:15:54 +00001334
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001335 def _monitor(self):
1336 """
1337 Monitor the queue for records, and ask the handler
1338 to deal with them.
Vinay Sajip0637d492010-09-23 08:15:54 +00001339
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001340 This method runs on a separate, internal thread.
1341 The thread will terminate if it sees a sentinel object in the queue.
1342 """
1343 q = self.queue
1344 has_task_done = hasattr(q, 'task_done')
1345 while not self._stop.isSet():
1346 try:
1347 record = self.dequeue(True)
1348 if record is self._sentinel:
1349 break
1350 self.handle(record)
1351 if has_task_done:
1352 q.task_done()
1353 except queue.Empty:
1354 pass
1355 # There might still be records in the queue.
1356 while True:
1357 try:
1358 record = self.dequeue(False)
1359 if record is self._sentinel:
1360 break
1361 self.handle(record)
1362 if has_task_done:
1363 q.task_done()
1364 except queue.Empty:
Vinay Sajip0637d492010-09-23 08:15:54 +00001365 break
Vinay Sajip0637d492010-09-23 08:15:54 +00001366
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001367 def stop(self):
1368 """
1369 Stop the listener.
Vinay Sajip0637d492010-09-23 08:15:54 +00001370
Victor Stinnercafa2ef2011-05-02 16:11:28 +02001371 This asks the thread to terminate, and then waits for it to do so.
1372 Note that if you don't call this before your application exits, there
1373 may be some records still left on the queue, which won't be processed.
1374 """
1375 self._stop.set()
1376 self.queue.put_nowait(self._sentinel)
1377 self._thread.join()
1378 self._thread = None