blob: 3ff324d1436bc8869ef7cb9a56fbf60e55a7fea2 [file] [log] [blame]
Vinay Sajipd23845e2012-02-23 19:37:18 +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 Sajip16f6a292012-02-23 19:40:33 +000019based on PEP 282 and comments thereto in comp.lang.python.
Guido van Rossum57102f82002-11-13 16:15:58 +000020
Vinay Sajip16f6a292012-02-23 19:40:33 +000021Copyright (C) 2001-2012 Vinay Sajip. All Rights Reserved.
Guido van Rossum57102f82002-11-13 16:15:58 +000022
Vinay Sajipaecf36a2009-01-20 23:16:08 +000023To use, simply 'import logging.handlers' and log away!
Guido van Rossum57102f82002-11-13 16:15:58 +000024"""
25
Vinay Sajip1c77b7f2009-10-10 20:32:36 +000026import logging, socket, os, cPickle, struct, time, re
Vinay Sajip9098ee42010-03-12 06:01:21 +000027from stat import ST_DEV, ST_INO, ST_MTIME
Guido van Rossum57102f82002-11-13 16:15:58 +000028
Vinay Sajip4600f112005-03-13 09:56:36 +000029try:
30 import codecs
31except ImportError:
32 codecs = None
Vinay Sajip5ac65282009-10-21 20:22:14 +000033try:
34 unicode
35 _unicode = True
36except NameError:
37 _unicode = False
Vinay Sajip4600f112005-03-13 09:56:36 +000038
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 Sajip1c77b7f2009-10-10 20:32:36 +000048SYSLOG_TCP_PORT = 514
Guido van Rossum57102f82002-11-13 16:15:58 +000049
Vinay Sajip4b4a63e2006-05-02 08:35:36 +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 """
Vinay Sajip92aa2f82008-01-24 12:37:33 +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
Vinay Sajip92aa2f82008-01-24 12:37:33 +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 """
Vinay Sajip92aa2f82008-01-24 12:37:33 +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 Sajipe8b1eaf2010-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 Sajipe8b1eaf2010-08-22 18:11:02 +0000116 mode = 'a'
Vinay Sajip92aa2f82008-01-24 12:37:33 +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 Sajipaecf36a2009-01-20 23:16:08 +0000125 if self.stream:
126 self.stream.close()
Vinay Sajipcdb83882011-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):
133 #print "%s -> %s" % (sfn, dfn)
134 if os.path.exists(dfn):
135 os.remove(dfn)
136 os.rename(sfn, dfn)
137 dfn = self.baseFilename + ".1"
138 if os.path.exists(dfn):
139 os.remove(dfn)
Vinay Sajip6dd59f12006-06-27 07:34:37 +0000140 os.rename(self.baseFilename, dfn)
Guido van Rossum57102f82002-11-13 16:15:58 +0000141 #print "%s -> %s" % (self.baseFilename, dfn)
Vinay Sajipb3d8a062007-01-16 09:50:07 +0000142 self.mode = 'w'
143 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000144
Vinay Sajip17c52d82004-07-03 11:48:34 +0000145 def shouldRollover(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000146 """
Vinay Sajip17c52d82004-07-03 11:48:34 +0000147 Determine if rollover should occur.
Guido van Rossum57102f82002-11-13 16:15:58 +0000148
Vinay Sajip17c52d82004-07-03 11:48:34 +0000149 Basically, see if the supplied record would cause the file to exceed
150 the size limit we have.
Guido van Rossum57102f82002-11-13 16:15:58 +0000151 """
Vinay Sajipaecf36a2009-01-20 23:16:08 +0000152 if self.stream is None: # delay was set...
153 self.stream = self._open()
Guido van Rossum57102f82002-11-13 16:15:58 +0000154 if self.maxBytes > 0: # are we rolling over?
155 msg = "%s\n" % self.format(record)
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000156 self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
Guido van Rossum57102f82002-11-13 16:15:58 +0000157 if self.stream.tell() + len(msg) >= self.maxBytes:
Vinay Sajip17c52d82004-07-03 11:48:34 +0000158 return 1
159 return 0
Guido van Rossum57102f82002-11-13 16:15:58 +0000160
Vinay Sajip17c52d82004-07-03 11:48:34 +0000161class TimedRotatingFileHandler(BaseRotatingHandler):
162 """
163 Handler for logging to a file, rotating the log file at certain timed
164 intervals.
165
166 If backupCount is > 0, when rollover is done, no more than backupCount
167 files are kept - the oldest ones are deleted.
168 """
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000169 def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=False, utc=False):
Vinay Sajip92aa2f82008-01-24 12:37:33 +0000170 BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay)
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000171 self.when = when.upper()
Vinay Sajip17c52d82004-07-03 11:48:34 +0000172 self.backupCount = backupCount
Vinay Sajipa12da732008-05-20 15:34:36 +0000173 self.utc = utc
Vinay Sajip17c52d82004-07-03 11:48:34 +0000174 # Calculate the real rollover interval, which is just the number of
175 # seconds between rollovers. Also set the filename suffix used when
176 # a rollover occurs. Current 'when' events supported:
177 # S - Seconds
178 # M - Minutes
179 # H - Hours
180 # D - Days
181 # midnight - roll over at midnight
182 # W{0-6} - roll over on a certain day; 0 - Monday
183 #
184 # Case of the 'when' specifier is not important; lower or upper case
185 # will work.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000186 if self.when == 'S':
187 self.interval = 1 # one second
188 self.suffix = "%Y-%m-%d_%H-%M-%S"
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000189 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000190 elif self.when == 'M':
191 self.interval = 60 # one minute
192 self.suffix = "%Y-%m-%d_%H-%M"
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000193 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000194 elif self.when == 'H':
195 self.interval = 60 * 60 # one hour
196 self.suffix = "%Y-%m-%d_%H"
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000197 self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000198 elif self.when == 'D' or self.when == 'MIDNIGHT':
199 self.interval = 60 * 60 * 24 # one day
200 self.suffix = "%Y-%m-%d"
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000201 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000202 elif self.when.startswith('W'):
203 self.interval = 60 * 60 * 24 * 7 # one week
204 if len(self.when) != 2:
205 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when)
206 if self.when[1] < '0' or self.when[1] > '6':
207 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when)
208 self.dayOfWeek = int(self.when[1])
209 self.suffix = "%Y-%m-%d"
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000210 self.extMatch = r"^\d{4}-\d{2}-\d{2}$"
Vinay Sajip17c52d82004-07-03 11:48:34 +0000211 else:
212 raise ValueError("Invalid rollover interval specified: %s" % self.when)
213
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000214 self.extMatch = re.compile(self.extMatch)
Vinay Sajipe7d40662004-10-03 19:12:07 +0000215 self.interval = self.interval * interval # multiply by units requested
Vinay Sajip9098ee42010-03-12 06:01:21 +0000216 if os.path.exists(filename):
217 t = os.stat(filename)[ST_MTIME]
218 else:
219 t = int(time.time())
220 self.rolloverAt = self.computeRollover(t)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000221
Vinay Sajip91290b52009-06-11 09:53:35 +0000222 def computeRollover(self, currentTime):
223 """
224 Work out the rollover time based on the specified time.
225 """
226 result = currentTime + self.interval
Vinay Sajip17c52d82004-07-03 11:48:34 +0000227 # If we are rolling over at midnight or weekly, then the interval is already known.
228 # What we need to figure out is WHEN the next interval is. In other words,
229 # if you are rolling over at midnight, then your base interval is 1 day,
230 # but you want to start that one day clock at midnight, not now. So, we
231 # have to fudge the rolloverAt value in order to trigger the first rollover
232 # at the right time. After that, the regular interval will take care of
233 # the rest. Note that this code doesn't care about leap seconds. :)
234 if self.when == 'MIDNIGHT' or self.when.startswith('W'):
235 # This could be done with less code, but I wanted it to be clear
Vinay Sajip83da0342009-06-11 09:23:41 +0000236 if self.utc:
Vinay Sajipa12da732008-05-20 15:34:36 +0000237 t = time.gmtime(currentTime)
238 else:
239 t = time.localtime(currentTime)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000240 currentHour = t[3]
241 currentMinute = t[4]
242 currentSecond = t[5]
243 # r is the number of seconds left between now and midnight
Vinay Sajip4b4a63e2006-05-02 08:35:36 +0000244 r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 +
245 currentSecond)
Vinay Sajip91290b52009-06-11 09:53:35 +0000246 result = currentTime + r
Vinay Sajip17c52d82004-07-03 11:48:34 +0000247 # If we are rolling over on a certain day, add in the number of days until
248 # the next rollover, but offset by 1 since we just calculated the time
249 # until the next day starts. There are three cases:
250 # Case 1) The day to rollover is today; in this case, do nothing
251 # Case 2) The day to rollover is further in the interval (i.e., today is
252 # day 2 (Wednesday) and rollover is on day 6 (Sunday). Days to
253 # next rollover is simply 6 - 2 - 1, or 3.
254 # Case 3) The day to rollover is behind us in the interval (i.e., today
255 # is day 5 (Saturday) and rollover is on day 3 (Thursday).
256 # Days to rollover is 6 - 5 + 3, or 4. In this case, it's the
257 # number of days left in the current week (1) plus the number
258 # of days in the next week until the rollover day (3).
Vinay Sajipae747dc2008-01-21 17:02:26 +0000259 # The calculations described in 2) and 3) above need to have a day added.
260 # This is because the above time calculation takes us to midnight on this
261 # day, i.e. the start of the next day.
Vinay Sajip83da0342009-06-11 09:23:41 +0000262 if self.when.startswith('W'):
Vinay Sajip17c52d82004-07-03 11:48:34 +0000263 day = t[6] # 0 is Monday
Vinay Sajipbababa32007-10-24 10:47:06 +0000264 if day != self.dayOfWeek:
265 if day < self.dayOfWeek:
Vinay Sajipae747dc2008-01-21 17:02:26 +0000266 daysToWait = self.dayOfWeek - day
Vinay Sajipbababa32007-10-24 10:47:06 +0000267 else:
Vinay Sajipae747dc2008-01-21 17:02:26 +0000268 daysToWait = 6 - day + self.dayOfWeek + 1
Vinay Sajip91290b52009-06-11 09:53:35 +0000269 newRolloverAt = result + (daysToWait * (60 * 60 * 24))
Vinay Sajip83da0342009-06-11 09:23:41 +0000270 if not self.utc:
Vinay Sajipa12da732008-05-20 15:34:36 +0000271 dstNow = t[-1]
272 dstAtRollover = time.localtime(newRolloverAt)[-1]
273 if dstNow != dstAtRollover:
274 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip9790e082012-03-13 12:06:35 +0000275 addend = -3600
Vinay Sajipa12da732008-05-20 15:34:36 +0000276 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip9790e082012-03-13 12:06:35 +0000277 addend = 3600
278 newRolloverAt += addend
Vinay Sajip91290b52009-06-11 09:53:35 +0000279 result = newRolloverAt
280 return result
Vinay Sajip17c52d82004-07-03 11:48:34 +0000281
Vinay Sajip17c52d82004-07-03 11:48:34 +0000282 def shouldRollover(self, record):
283 """
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000284 Determine if rollover should occur.
Vinay Sajip17c52d82004-07-03 11:48:34 +0000285
286 record is not used, as we are just comparing times, but it is needed so
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000287 the method signatures are the same
Vinay Sajip17c52d82004-07-03 11:48:34 +0000288 """
289 t = int(time.time())
290 if t >= self.rolloverAt:
291 return 1
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000292 #print "No need to rollover: %d, %d" % (t, self.rolloverAt)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000293 return 0
294
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000295 def getFilesToDelete(self):
296 """
297 Determine the files to delete when rolling over.
298
299 More specific than the earlier method, which just used glob.glob().
300 """
301 dirName, baseName = os.path.split(self.baseFilename)
302 fileNames = os.listdir(dirName)
303 result = []
304 prefix = baseName + "."
305 plen = len(prefix)
306 for fileName in fileNames:
307 if fileName[:plen] == prefix:
308 suffix = fileName[plen:]
309 if self.extMatch.match(suffix):
Vinay Sajipa12da732008-05-20 15:34:36 +0000310 result.append(os.path.join(dirName, fileName))
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000311 result.sort()
312 if len(result) < self.backupCount:
313 result = []
314 else:
315 result = result[:len(result) - self.backupCount]
316 return result
317
Vinay Sajip17c52d82004-07-03 11:48:34 +0000318 def doRollover(self):
319 """
320 do a rollover; in this case, a date/time stamp is appended to the filename
321 when the rollover happens. However, you want the file to be named for the
322 start of the interval, not the current time. If there is a backup count,
323 then we have to get a list of matching filenames, sort them and remove
324 the one with the oldest suffix.
325 """
Vinay Sajipaecf36a2009-01-20 23:16:08 +0000326 if self.stream:
327 self.stream.close()
Vinay Sajipcdb83882011-01-21 23:35:57 +0000328 self.stream = None
Vinay Sajip17c52d82004-07-03 11:48:34 +0000329 # get the time that this sequence started at and make it a TimeTuple
Vinay Sajip9790e082012-03-13 12:06:35 +0000330 currentTime = int(time.time())
331 dstNow = time.localtime(currentTime)[-1]
Vinay Sajip17c52d82004-07-03 11:48:34 +0000332 t = self.rolloverAt - self.interval
Vinay Sajipa12da732008-05-20 15:34:36 +0000333 if self.utc:
334 timeTuple = time.gmtime(t)
335 else:
336 timeTuple = time.localtime(t)
Vinay Sajip9790e082012-03-13 12:06:35 +0000337 dstThen = timeTuple[-1]
338 if dstNow != dstThen:
339 if dstNow:
340 addend = 3600
341 else:
342 addend = -3600
343 timeTuple = time.localtime(t + addend)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000344 dfn = self.baseFilename + "." + time.strftime(self.suffix, timeTuple)
345 if os.path.exists(dfn):
346 os.remove(dfn)
Vinay Sajip6dd59f12006-06-27 07:34:37 +0000347 os.rename(self.baseFilename, dfn)
Vinay Sajip17c52d82004-07-03 11:48:34 +0000348 if self.backupCount > 0:
349 # find the oldest log file and delete it
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000350 #s = glob.glob(self.baseFilename + ".20*")
351 #if len(s) > self.backupCount:
352 # s.sort()
353 # os.remove(s[0])
354 for s in self.getFilesToDelete():
355 os.remove(s)
Vinay Sajip5e9e9e12004-07-12 09:21:41 +0000356 #print "%s -> %s" % (self.baseFilename, dfn)
Vinay Sajipb3d8a062007-01-16 09:50:07 +0000357 self.mode = 'w'
358 self.stream = self._open()
Vinay Sajip91290b52009-06-11 09:53:35 +0000359 newRolloverAt = self.computeRollover(currentTime)
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000360 while newRolloverAt <= currentTime:
361 newRolloverAt = newRolloverAt + self.interval
362 #If DST changes and midnight or weekly rollover, adjust for this.
Vinay Sajipa12da732008-05-20 15:34:36 +0000363 if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000364 dstAtRollover = time.localtime(newRolloverAt)[-1]
365 if dstNow != dstAtRollover:
366 if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
Vinay Sajip9790e082012-03-13 12:06:35 +0000367 addend = -3600
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000368 else: # DST bows out before next rollover, so we need to add an hour
Vinay Sajip9790e082012-03-13 12:06:35 +0000369 addend = 3600
370 newRolloverAt += addend
Vinay Sajipe5aefa02008-04-02 21:09:27 +0000371 self.rolloverAt = newRolloverAt
Guido van Rossum57102f82002-11-13 16:15:58 +0000372
Vinay Sajip73306b02007-01-14 21:49:59 +0000373class WatchedFileHandler(logging.FileHandler):
374 """
375 A handler for logging to a file, which watches the file
376 to see if it has changed while in use. This can happen because of
377 usage of programs such as newsyslog and logrotate which perform
378 log file rotation. This handler, intended for use under Unix,
379 watches the file to see if it has changed since the last emit.
380 (A file has changed if its device or inode have changed.)
381 If it has changed, the old file stream is closed, and the file
382 opened to get a new stream.
383
384 This handler is not appropriate for use under Windows, because
385 under Windows open files cannot be moved or renamed - logging
386 opens the files with exclusive locks - and so there is no need
387 for such a handler. Furthermore, ST_INO is not supported under
388 Windows; stat always returns zero for this value.
389
390 This handler is based on a suggestion and patch by Chad J.
391 Schroeder.
392 """
Vinay Sajip92aa2f82008-01-24 12:37:33 +0000393 def __init__(self, filename, mode='a', encoding=None, delay=0):
394 logging.FileHandler.__init__(self, filename, mode, encoding, delay)
395 if not os.path.exists(self.baseFilename):
396 self.dev, self.ino = -1, -1
397 else:
398 stat = os.stat(self.baseFilename)
399 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
Vinay Sajip73306b02007-01-14 21:49:59 +0000400
401 def emit(self, record):
402 """
403 Emit a record.
404
405 First check if the underlying file has changed, and if it
406 has, close the old stream and reopen the file to get the
407 current stream.
408 """
409 if not os.path.exists(self.baseFilename):
410 stat = None
411 changed = 1
412 else:
413 stat = os.stat(self.baseFilename)
414 changed = (stat[ST_DEV] != self.dev) or (stat[ST_INO] != self.ino)
Vinay Sajip92aa2f82008-01-24 12:37:33 +0000415 if changed and self.stream is not None:
Vinay Sajip73306b02007-01-14 21:49:59 +0000416 self.stream.flush()
417 self.stream.close()
418 self.stream = self._open()
419 if stat is None:
420 stat = os.stat(self.baseFilename)
421 self.dev, self.ino = stat[ST_DEV], stat[ST_INO]
422 logging.FileHandler.emit(self, record)
423
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
Vinay Sajipaa7b16a2007-04-09 16:16:10 +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)
Vinay Sajipaa7b16a2007-04-09 16:16:10 +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
Vinay Sajipc683a872007-01-08 18:50:32 +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:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000531 dummy = self.format(record) # just to get traceback text into record.exc_text
532 record.exc_info = None # to avoid Unpickleable error
Guido van Rossum57102f82002-11-13 16:15:58 +0000533 s = cPickle.dumps(record.__dict__, 1)
Vinay Sajip48cfe382004-02-20 13:17:27 +0000534 if ei:
Tim Peters4e0e1b62004-07-07 20:54:48 +0000535 record.exc_info = ei # for next handler
Guido van Rossum57102f82002-11-13 16:15:58 +0000536 slen = struct.pack(">L", len(s))
537 return slen + s
538
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000539 def handleError(self, record):
Guido van Rossum57102f82002-11-13 16:15:58 +0000540 """
541 Handle an error during logging.
542
543 An error has occurred during logging. Most likely cause -
544 connection lost. Close the socket so that we can retry on the
545 next event.
546 """
547 if self.closeOnError and self.sock:
548 self.sock.close()
549 self.sock = None #try to reconnect next time
550 else:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000551 logging.Handler.handleError(self, record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000552
553 def emit(self, record):
554 """
555 Emit a record.
556
557 Pickles the record and writes it to the socket in binary format.
558 If there is an error with the socket, silently drop the packet.
559 If there was a problem with the socket, re-establishes the
560 socket.
561 """
562 try:
563 s = self.makePickle(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000564 self.send(s)
Vinay Sajip85c19092005-10-31 13:14:19 +0000565 except (KeyboardInterrupt, SystemExit):
566 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000567 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000568 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000569
570 def close(self):
571 """
572 Closes the socket.
573 """
Vinay Sajip50198222012-02-23 20:45:03 +0000574 self.acquire()
575 try:
Vinay Sajipd23845e2012-02-23 19:37:18 +0000576 if self.sock:
577 self.sock.close()
578 self.sock = None
Vinay Sajip50198222012-02-23 20:45:03 +0000579 finally:
580 self.release()
Vinay Sajip48cfe382004-02-20 13:17:27 +0000581 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000582
583class DatagramHandler(SocketHandler):
584 """
585 A handler class which writes logging records, in pickle format, to
Raymond Hettinger6f3eaa62003-06-27 21:43:39 +0000586 a datagram socket. The pickle which is sent is that of the LogRecord's
587 attribute dictionary (__dict__), so that the receiver does not need to
588 have the logging module installed in order to process the logging event.
589
590 To unpickle the record at the receiving end into a LogRecord, use the
591 makeLogRecord function.
Guido van Rossum57102f82002-11-13 16:15:58 +0000592
593 """
594 def __init__(self, host, port):
595 """
596 Initializes the handler with a specific host address and port.
597 """
598 SocketHandler.__init__(self, host, port)
599 self.closeOnError = 0
600
601 def makeSocket(self):
602 """
603 The factory method of SocketHandler is here overridden to create
604 a UDP socket (SOCK_DGRAM).
605 """
606 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
607 return s
608
609 def send(self, s):
610 """
611 Send a pickled string to a socket.
612
613 This function no longer allows for partial sends which can happen
614 when the network is busy - UDP does not guarantee delivery and
615 can deliver packets out of sequence.
616 """
Vinay Sajipfb154172004-08-24 09:36:23 +0000617 if self.sock is None:
618 self.createSocket()
Guido van Rossum57102f82002-11-13 16:15:58 +0000619 self.sock.sendto(s, (self.host, self.port))
620
621class SysLogHandler(logging.Handler):
622 """
623 A handler class which sends formatted logging records to a syslog
624 server. Based on Sam Rushing's syslog module:
625 http://www.nightmare.com/squirl/python-ext/misc/syslog.py
626 Contributed by Nicolas Untz (after which minor refactoring changes
627 have been made).
628 """
629
630 # from <linux/sys/syslog.h>:
631 # ======================================================================
632 # priorities/facilities are encoded into a single 32-bit quantity, where
633 # the bottom 3 bits are the priority (0-7) and the top 28 bits are the
634 # facility (0-big number). Both the priorities and the facilities map
635 # roughly one-to-one to strings in the syslogd(8) source code. This
636 # mapping is included in this file.
637 #
638 # priorities (these are ordered)
639
640 LOG_EMERG = 0 # system is unusable
641 LOG_ALERT = 1 # action must be taken immediately
642 LOG_CRIT = 2 # critical conditions
643 LOG_ERR = 3 # error conditions
644 LOG_WARNING = 4 # warning conditions
645 LOG_NOTICE = 5 # normal but significant condition
646 LOG_INFO = 6 # informational
647 LOG_DEBUG = 7 # debug-level messages
648
649 # facility codes
650 LOG_KERN = 0 # kernel messages
651 LOG_USER = 1 # random user-level messages
652 LOG_MAIL = 2 # mail system
653 LOG_DAEMON = 3 # system daemons
654 LOG_AUTH = 4 # security/authorization messages
655 LOG_SYSLOG = 5 # messages generated internally by syslogd
656 LOG_LPR = 6 # line printer subsystem
657 LOG_NEWS = 7 # network news subsystem
658 LOG_UUCP = 8 # UUCP subsystem
659 LOG_CRON = 9 # clock daemon
Vinay Sajipb0623d62010-03-24 14:31:21 +0000660 LOG_AUTHPRIV = 10 # security/authorization messages (private)
661 LOG_FTP = 11 # FTP daemon
Guido van Rossum57102f82002-11-13 16:15:58 +0000662
663 # other codes through 15 reserved for system use
664 LOG_LOCAL0 = 16 # reserved for local use
665 LOG_LOCAL1 = 17 # reserved for local use
666 LOG_LOCAL2 = 18 # reserved for local use
667 LOG_LOCAL3 = 19 # reserved for local use
668 LOG_LOCAL4 = 20 # reserved for local use
669 LOG_LOCAL5 = 21 # reserved for local use
670 LOG_LOCAL6 = 22 # reserved for local use
671 LOG_LOCAL7 = 23 # reserved for local use
672
673 priority_names = {
674 "alert": LOG_ALERT,
675 "crit": LOG_CRIT,
676 "critical": LOG_CRIT,
677 "debug": LOG_DEBUG,
678 "emerg": LOG_EMERG,
679 "err": LOG_ERR,
680 "error": LOG_ERR, # DEPRECATED
681 "info": LOG_INFO,
682 "notice": LOG_NOTICE,
683 "panic": LOG_EMERG, # DEPRECATED
684 "warn": LOG_WARNING, # DEPRECATED
685 "warning": LOG_WARNING,
686 }
687
688 facility_names = {
689 "auth": LOG_AUTH,
690 "authpriv": LOG_AUTHPRIV,
691 "cron": LOG_CRON,
692 "daemon": LOG_DAEMON,
Vinay Sajipb0623d62010-03-24 14:31:21 +0000693 "ftp": LOG_FTP,
Guido van Rossum57102f82002-11-13 16:15:58 +0000694 "kern": LOG_KERN,
695 "lpr": LOG_LPR,
696 "mail": LOG_MAIL,
697 "news": LOG_NEWS,
698 "security": LOG_AUTH, # DEPRECATED
699 "syslog": LOG_SYSLOG,
700 "user": LOG_USER,
701 "uucp": LOG_UUCP,
702 "local0": LOG_LOCAL0,
703 "local1": LOG_LOCAL1,
704 "local2": LOG_LOCAL2,
705 "local3": LOG_LOCAL3,
706 "local4": LOG_LOCAL4,
707 "local5": LOG_LOCAL5,
708 "local6": LOG_LOCAL6,
709 "local7": LOG_LOCAL7,
710 }
711
Vinay Sajipdc579362006-07-20 23:20:12 +0000712 #The map below appears to be trivially lowercasing the key. However,
713 #there's more to it than meets the eye - in some locales, lowercasing
714 #gives unexpected results. See SF #1524081: in the Turkish locale,
715 #"INFO".lower() != "info"
716 priority_map = {
717 "DEBUG" : "debug",
718 "INFO" : "info",
719 "WARNING" : "warning",
720 "ERROR" : "error",
721 "CRITICAL" : "critical"
722 }
723
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000724 def __init__(self, address=('localhost', SYSLOG_UDP_PORT),
725 facility=LOG_USER, socktype=socket.SOCK_DGRAM):
Guido van Rossum57102f82002-11-13 16:15:58 +0000726 """
727 Initialize a handler.
728
Vinay Sajip754a5fb2007-05-25 07:05:59 +0000729 If address is specified as a string, a UNIX socket is used. To log to a
730 local syslogd, "SysLogHandler(address="/dev/log")" can be used.
Guido van Rossum57102f82002-11-13 16:15:58 +0000731 If facility is not specified, LOG_USER is used.
732 """
733 logging.Handler.__init__(self)
734
735 self.address = address
736 self.facility = facility
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000737 self.socktype = socktype
738
739 if isinstance(address, basestring):
Guido van Rossum57102f82002-11-13 16:15:58 +0000740 self.unixsocket = 1
Vinay Sajip5492e172006-12-11 14:07:16 +0000741 self._connect_unixsocket(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000742 else:
Guido van Rossum57102f82002-11-13 16:15:58 +0000743 self.unixsocket = 0
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000744 self.socket = socket.socket(socket.AF_INET, socktype)
745 if socktype == socket.SOCK_STREAM:
746 self.socket.connect(address)
Guido van Rossum57102f82002-11-13 16:15:58 +0000747 self.formatter = None
748
Vinay Sajipa1974c12005-01-13 08:23:56 +0000749 def _connect_unixsocket(self, address):
750 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
751 # syslog may require either DGRAM or STREAM sockets
752 try:
753 self.socket.connect(address)
754 except socket.error:
755 self.socket.close()
756 self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
Vinay Sajip8b6b53f2005-11-09 13:55:13 +0000757 self.socket.connect(address)
Vinay Sajipa1974c12005-01-13 08:23:56 +0000758
Guido van Rossum57102f82002-11-13 16:15:58 +0000759 # curious: when talking to the unix-domain '/dev/log' socket, a
760 # zero-terminator seems to be required. this string is placed
761 # into a class variable so that it can be overridden if
762 # necessary.
763 log_format_string = '<%d>%s\000'
764
Vinay Sajipdc579362006-07-20 23:20:12 +0000765 def encodePriority(self, facility, priority):
Guido van Rossum57102f82002-11-13 16:15:58 +0000766 """
767 Encode the facility and priority. You can pass in strings or
768 integers - if strings are passed, the facility_names and
769 priority_names mapping dictionaries are used to convert them to
770 integers.
771 """
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000772 if isinstance(facility, basestring):
Guido van Rossum57102f82002-11-13 16:15:58 +0000773 facility = self.facility_names[facility]
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000774 if isinstance(priority, basestring):
Guido van Rossum57102f82002-11-13 16:15:58 +0000775 priority = self.priority_names[priority]
776 return (facility << 3) | priority
777
778 def close (self):
779 """
780 Closes the socket.
781 """
Vinay Sajip50198222012-02-23 20:45:03 +0000782 self.acquire()
783 try:
Vinay Sajipd23845e2012-02-23 19:37:18 +0000784 if self.unixsocket:
785 self.socket.close()
Vinay Sajip50198222012-02-23 20:45:03 +0000786 finally:
787 self.release()
Vinay Sajip48cfe382004-02-20 13:17:27 +0000788 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +0000789
Vinay Sajipdc579362006-07-20 23:20:12 +0000790 def mapPriority(self, levelName):
791 """
792 Map a logging level name to a key in the priority_names map.
793 This is useful in two scenarios: when custom levels are being
794 used, and in the case where you can't do a straightforward
795 mapping by lowercasing the logging level name because of locale-
796 specific issues (see SF #1524081).
797 """
798 return self.priority_map.get(levelName, "warning")
799
Guido van Rossum57102f82002-11-13 16:15:58 +0000800 def emit(self, record):
801 """
802 Emit a record.
803
804 The record is formatted, and then sent to the syslog server. If
805 exception information is present, it is NOT sent to the server.
806 """
Vinay Sajipe900b492010-09-03 09:06:07 +0000807 msg = self.format(record) + '\000'
Guido van Rossum57102f82002-11-13 16:15:58 +0000808 """
809 We need to convert record level to lowercase, maybe this will
810 change in the future.
811 """
Vinay Sajipe900b492010-09-03 09:06:07 +0000812 prio = '<%d>' % self.encodePriority(self.facility,
813 self.mapPriority(record.levelname))
814 # Message is a string. Convert to bytes as required by RFC 5424
815 if type(msg) is unicode:
Vinay Sajip5ac65282009-10-21 20:22:14 +0000816 msg = msg.encode('utf-8')
817 if codecs:
818 msg = codecs.BOM_UTF8 + msg
Vinay Sajipe900b492010-09-03 09:06:07 +0000819 msg = prio + msg
Guido van Rossum57102f82002-11-13 16:15:58 +0000820 try:
821 if self.unixsocket:
Vinay Sajipa1974c12005-01-13 08:23:56 +0000822 try:
823 self.socket.send(msg)
824 except socket.error:
825 self._connect_unixsocket(self.address)
826 self.socket.send(msg)
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000827 elif self.socktype == socket.SOCK_DGRAM:
Guido van Rossum57102f82002-11-13 16:15:58 +0000828 self.socket.sendto(msg, self.address)
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000829 else:
830 self.socket.sendall(msg)
Vinay Sajip85c19092005-10-31 13:14:19 +0000831 except (KeyboardInterrupt, SystemExit):
832 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000833 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000834 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000835
836class SMTPHandler(logging.Handler):
837 """
838 A handler class which sends an SMTP email for each logging event.
839 """
Vinay Sajip48305662009-12-06 17:57:11 +0000840 def __init__(self, mailhost, fromaddr, toaddrs, subject,
Vinay Sajipbd1094a2009-12-06 18:05:04 +0000841 credentials=None, secure=None):
Guido van Rossum57102f82002-11-13 16:15:58 +0000842 """
843 Initialize the handler.
844
845 Initialize the instance with the from and to addresses and subject
846 line of the email. To specify a non-standard SMTP port, use the
Vinay Sajip70c8e8b2007-05-01 10:20:03 +0000847 (host, port) tuple format for the mailhost argument. To specify
848 authentication credentials, supply a (username, password) tuple
Vinay Sajip48305662009-12-06 17:57:11 +0000849 for the credentials argument. To specify the use of a secure
Vinay Sajipbd1094a2009-12-06 18:05:04 +0000850 protocol (TLS), pass in a tuple for the secure argument. This will
851 only be used when authentication credentials are supplied. The tuple
852 will be either an empty tuple, or a single-value tuple with the name
853 of a keyfile, or a 2-value tuple with the names of the keyfile and
854 certificate file. (This tuple is passed to the `starttls` method).
Guido van Rossum57102f82002-11-13 16:15:58 +0000855 """
856 logging.Handler.__init__(self)
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000857 if isinstance(mailhost, tuple):
Vinay Sajip70c8e8b2007-05-01 10:20:03 +0000858 self.mailhost, self.mailport = mailhost
Guido van Rossum57102f82002-11-13 16:15:58 +0000859 else:
Vinay Sajip70c8e8b2007-05-01 10:20:03 +0000860 self.mailhost, self.mailport = mailhost, None
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000861 if isinstance(credentials, tuple):
Vinay Sajip70c8e8b2007-05-01 10:20:03 +0000862 self.username, self.password = credentials
863 else:
864 self.username = None
Guido van Rossum57102f82002-11-13 16:15:58 +0000865 self.fromaddr = fromaddr
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000866 if isinstance(toaddrs, basestring):
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000867 toaddrs = [toaddrs]
Guido van Rossum57102f82002-11-13 16:15:58 +0000868 self.toaddrs = toaddrs
869 self.subject = subject
Vinay Sajip48305662009-12-06 17:57:11 +0000870 self.secure = secure
Vinay Sajip50d6f542012-03-23 14:33:00 +0000871 self._timeout = 5.0
Guido van Rossum57102f82002-11-13 16:15:58 +0000872
873 def getSubject(self, record):
874 """
875 Determine the subject for the email.
876
877 If you want to specify a subject line which is record-dependent,
878 override this method.
879 """
880 return self.subject
881
882 def emit(self, record):
883 """
884 Emit a record.
885
886 Format the record and send it to the specified addressees.
887 """
888 try:
889 import smtplib
Vinay Sajip7ce5c832010-02-07 13:06:51 +0000890 from email.utils import formatdate
Guido van Rossum57102f82002-11-13 16:15:58 +0000891 port = self.mailport
892 if not port:
893 port = smtplib.SMTP_PORT
Vinay Sajip50d6f542012-03-23 14:33:00 +0000894 smtp = smtplib.SMTP(self.mailhost, port, timeout=self._timeout)
Guido van Rossum57102f82002-11-13 16:15:58 +0000895 msg = self.format(record)
Neal Norwitzf297bd12003-04-23 03:49:43 +0000896 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 +0000897 self.fromaddr,
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000898 ",".join(self.toaddrs),
Neal Norwitzf297bd12003-04-23 03:49:43 +0000899 self.getSubject(record),
Martin v. Löwis318a12e2004-08-18 12:27:40 +0000900 formatdate(), msg)
Vinay Sajip70c8e8b2007-05-01 10:20:03 +0000901 if self.username:
Vinay Sajipbd1094a2009-12-06 18:05:04 +0000902 if self.secure is not None:
Vinay Sajip48305662009-12-06 17:57:11 +0000903 smtp.ehlo()
Vinay Sajipbd1094a2009-12-06 18:05:04 +0000904 smtp.starttls(*self.secure)
Vinay Sajip48305662009-12-06 17:57:11 +0000905 smtp.ehlo()
Vinay Sajip70c8e8b2007-05-01 10:20:03 +0000906 smtp.login(self.username, self.password)
Guido van Rossum57102f82002-11-13 16:15:58 +0000907 smtp.sendmail(self.fromaddr, self.toaddrs, msg)
908 smtp.quit()
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000909 except (KeyboardInterrupt, SystemExit):
910 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000911 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000912 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +0000913
914class NTEventLogHandler(logging.Handler):
915 """
916 A handler class which sends events to the NT Event Log. Adds a
917 registry entry for the specified application name. If no dllname is
918 provided, win32service.pyd (which contains some basic message
919 placeholders) is used. Note that use of these placeholders will make
920 your event logs big, as the entire message source is held in the log.
921 If you want slimmer logs, you have to pass in the name of your own DLL
922 which contains the message definitions you want to use in the event log.
923 """
924 def __init__(self, appname, dllname=None, logtype="Application"):
925 logging.Handler.__init__(self)
926 try:
927 import win32evtlogutil, win32evtlog
928 self.appname = appname
929 self._welu = win32evtlogutil
930 if not dllname:
931 dllname = os.path.split(self._welu.__file__)
932 dllname = os.path.split(dllname[0])
933 dllname = os.path.join(dllname[0], r'win32service.pyd')
934 self.dllname = dllname
935 self.logtype = logtype
936 self._welu.AddSourceToRegistry(appname, dllname, logtype)
937 self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE
938 self.typemap = {
939 logging.DEBUG : win32evtlog.EVENTLOG_INFORMATION_TYPE,
940 logging.INFO : win32evtlog.EVENTLOG_INFORMATION_TYPE,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000941 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE,
Guido van Rossum57102f82002-11-13 16:15:58 +0000942 logging.ERROR : win32evtlog.EVENTLOG_ERROR_TYPE,
943 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE,
944 }
945 except ImportError:
Vinay Sajip1c77b7f2009-10-10 20:32:36 +0000946 print("The Python Win32 extensions for NT (service, event "\
947 "logging) appear not to be available.")
Guido van Rossum57102f82002-11-13 16:15:58 +0000948 self._welu = None
949
950 def getMessageID(self, record):
951 """
952 Return the message ID for the event record. If you are using your
953 own messages, you could do this by having the msg passed to the
954 logger being an ID rather than a formatting string. Then, in here,
955 you could use a dictionary lookup to get the message ID. This
956 version returns 1, which is the base message ID in win32service.pyd.
957 """
958 return 1
959
960 def getEventCategory(self, record):
961 """
962 Return the event category for the record.
963
964 Override this if you want to specify your own categories. This version
965 returns 0.
966 """
967 return 0
968
969 def getEventType(self, record):
970 """
971 Return the event type for the record.
972
973 Override this if you want to specify your own types. This version does
974 a mapping using the handler's typemap attribute, which is set up in
975 __init__() to a dictionary which contains mappings for DEBUG, INFO,
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000976 WARNING, ERROR and CRITICAL. If you are using your own levels you will
Guido van Rossum57102f82002-11-13 16:15:58 +0000977 either need to override this method or place a suitable dictionary in
978 the handler's typemap attribute.
979 """
980 return self.typemap.get(record.levelno, self.deftype)
981
982 def emit(self, record):
983 """
984 Emit a record.
985
986 Determine the message ID, event category and event type. Then
987 log the message in the NT event log.
988 """
989 if self._welu:
990 try:
991 id = self.getMessageID(record)
992 cat = self.getEventCategory(record)
993 type = self.getEventType(record)
994 msg = self.format(record)
995 self._welu.ReportEvent(self.appname, id, cat, type, [msg])
Vinay Sajip245a5ab2005-10-31 14:27:01 +0000996 except (KeyboardInterrupt, SystemExit):
997 raise
Guido van Rossum57102f82002-11-13 16:15:58 +0000998 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +0000999 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001000
1001 def close(self):
1002 """
1003 Clean up this handler.
1004
1005 You can remove the application name from the registry as a
1006 source of event log entries. However, if you do this, you will
1007 not be able to see the events as you intended in the Event Log
1008 Viewer - it needs to be able to access the registry to get the
1009 DLL name.
1010 """
1011 #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype)
Vinay Sajip48cfe382004-02-20 13:17:27 +00001012 logging.Handler.close(self)
Guido van Rossum57102f82002-11-13 16:15:58 +00001013
1014class HTTPHandler(logging.Handler):
1015 """
1016 A class which sends records to a Web server, using either GET or
1017 POST semantics.
1018 """
1019 def __init__(self, host, url, method="GET"):
1020 """
1021 Initialize the instance with the host, the request URL, and the method
1022 ("GET" or "POST")
1023 """
1024 logging.Handler.__init__(self)
Vinay Sajip1c77b7f2009-10-10 20:32:36 +00001025 method = method.upper()
Guido van Rossum57102f82002-11-13 16:15:58 +00001026 if method not in ["GET", "POST"]:
Vinay Sajip1c77b7f2009-10-10 20:32:36 +00001027 raise ValueError("method must be GET or POST")
Guido van Rossum57102f82002-11-13 16:15:58 +00001028 self.host = host
1029 self.url = url
1030 self.method = method
1031
Neal Norwitzf297bd12003-04-23 03:49:43 +00001032 def mapLogRecord(self, record):
1033 """
1034 Default implementation of mapping the log record into a dict
Vinay Sajip48cfe382004-02-20 13:17:27 +00001035 that is sent as the CGI data. Overwrite in your class.
Neal Norwitzf297bd12003-04-23 03:49:43 +00001036 Contributed by Franz Glasner.
1037 """
1038 return record.__dict__
1039
Guido van Rossum57102f82002-11-13 16:15:58 +00001040 def emit(self, record):
1041 """
1042 Emit a record.
1043
Senthil Kumarana5ba05c2010-08-09 20:18:04 +00001044 Send the record to the Web server as a percent-encoded dictionary
Guido van Rossum57102f82002-11-13 16:15:58 +00001045 """
1046 try:
1047 import httplib, urllib
Vinay Sajipb7935062005-10-11 13:15:31 +00001048 host = self.host
1049 h = httplib.HTTP(host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001050 url = self.url
Neal Norwitzf297bd12003-04-23 03:49:43 +00001051 data = urllib.urlencode(self.mapLogRecord(record))
Guido van Rossum57102f82002-11-13 16:15:58 +00001052 if self.method == "GET":
Vinay Sajip1c77b7f2009-10-10 20:32:36 +00001053 if (url.find('?') >= 0):
Guido van Rossum57102f82002-11-13 16:15:58 +00001054 sep = '&'
1055 else:
1056 sep = '?'
1057 url = url + "%c%s" % (sep, data)
1058 h.putrequest(self.method, url)
Vinay Sajipb7935062005-10-11 13:15:31 +00001059 # support multiple hosts on one IP address...
1060 # need to strip optional :port from host, if present
Vinay Sajip1c77b7f2009-10-10 20:32:36 +00001061 i = host.find(":")
Vinay Sajipb7935062005-10-11 13:15:31 +00001062 if i >= 0:
1063 host = host[:i]
1064 h.putheader("Host", host)
Guido van Rossum57102f82002-11-13 16:15:58 +00001065 if self.method == "POST":
Vinay Sajipb7935062005-10-11 13:15:31 +00001066 h.putheader("Content-type",
1067 "application/x-www-form-urlencoded")
Guido van Rossum57102f82002-11-13 16:15:58 +00001068 h.putheader("Content-length", str(len(data)))
Kristján Valur Jónsson84040db2009-01-09 20:27:16 +00001069 h.endheaders(data if self.method == "POST" else None)
Guido van Rossum57102f82002-11-13 16:15:58 +00001070 h.getreply() #can't do anything with the result
Vinay Sajip245a5ab2005-10-31 14:27:01 +00001071 except (KeyboardInterrupt, SystemExit):
1072 raise
Guido van Rossum57102f82002-11-13 16:15:58 +00001073 except:
Neal Norwitz6fa635d2003-02-18 14:20:07 +00001074 self.handleError(record)
Guido van Rossum57102f82002-11-13 16:15:58 +00001075
1076class BufferingHandler(logging.Handler):
1077 """
1078 A handler class which buffers logging records in memory. Whenever each
1079 record is added to the buffer, a check is made to see if the buffer should
1080 be flushed. If it should, then flush() is expected to do what's needed.
1081 """
1082 def __init__(self, capacity):
1083 """
1084 Initialize the handler with the buffer size.
1085 """
1086 logging.Handler.__init__(self)
1087 self.capacity = capacity
1088 self.buffer = []
1089
1090 def shouldFlush(self, record):
1091 """
1092 Should the handler flush its buffer?
1093
1094 Returns true if the buffer is up to capacity. This method can be
1095 overridden to implement custom flushing strategies.
1096 """
1097 return (len(self.buffer) >= self.capacity)
1098
1099 def emit(self, record):
1100 """
1101 Emit a record.
1102
1103 Append the record. If shouldFlush() tells us to, call flush() to process
1104 the buffer.
1105 """
1106 self.buffer.append(record)
1107 if self.shouldFlush(record):
1108 self.flush()
1109
1110 def flush(self):
1111 """
1112 Override to implement custom flushing behaviour.
1113
1114 This version just zaps the buffer to empty.
1115 """
Vinay Sajip50198222012-02-23 20:45:03 +00001116 self.acquire()
1117 try:
Vinay Sajipd23845e2012-02-23 19:37:18 +00001118 self.buffer = []
Vinay Sajip50198222012-02-23 20:45:03 +00001119 finally:
1120 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001121
Vinay Sajipf42d95e2004-02-21 22:14:34 +00001122 def close(self):
1123 """
1124 Close the handler.
1125
1126 This version just flushes and chains to the parent class' close().
1127 """
1128 self.flush()
1129 logging.Handler.close(self)
1130
Guido van Rossum57102f82002-11-13 16:15:58 +00001131class MemoryHandler(BufferingHandler):
1132 """
1133 A handler class which buffers logging records in memory, periodically
1134 flushing them to a target handler. Flushing occurs whenever the buffer
1135 is full, or when an event of a certain severity or greater is seen.
1136 """
1137 def __init__(self, capacity, flushLevel=logging.ERROR, target=None):
1138 """
1139 Initialize the handler with the buffer size, the level at which
1140 flushing should occur and an optional target.
1141
1142 Note that without a target being set either here or via setTarget(),
1143 a MemoryHandler is no use to anyone!
1144 """
1145 BufferingHandler.__init__(self, capacity)
1146 self.flushLevel = flushLevel
1147 self.target = target
1148
1149 def shouldFlush(self, record):
1150 """
1151 Check for buffer full or a record at the flushLevel or higher.
1152 """
1153 return (len(self.buffer) >= self.capacity) or \
1154 (record.levelno >= self.flushLevel)
1155
1156 def setTarget(self, target):
1157 """
1158 Set the target handler for this handler.
1159 """
1160 self.target = target
1161
1162 def flush(self):
1163 """
1164 For a MemoryHandler, flushing means just sending the buffered
1165 records to the target, if there is one. Override if you want
1166 different behaviour.
1167 """
Vinay Sajip50198222012-02-23 20:45:03 +00001168 self.acquire()
1169 try:
Vinay Sajipd23845e2012-02-23 19:37:18 +00001170 if self.target:
1171 for record in self.buffer:
1172 self.target.handle(record)
1173 self.buffer = []
Vinay Sajip50198222012-02-23 20:45:03 +00001174 finally:
1175 self.release()
Guido van Rossum57102f82002-11-13 16:15:58 +00001176
1177 def close(self):
1178 """
1179 Flush, set the target to None and lose the buffer.
1180 """
1181 self.flush()
Vinay Sajip50198222012-02-23 20:45:03 +00001182 self.acquire()
1183 try:
Vinay Sajipd23845e2012-02-23 19:37:18 +00001184 self.target = None
1185 BufferingHandler.close(self)
Vinay Sajip50198222012-02-23 20:45:03 +00001186 finally:
1187 self.release()