| Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 1 | # Copyright 2001-2007 by Vinay Sajip. All Rights Reserved. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 2 | # | 
 | 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 Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 16 |  | 
 | 17 | """ | 
| Vinay Sajip | 3f74284 | 2004-02-28 16:07:46 +0000 | [diff] [blame] | 18 | Additional handlers for the logging package for Python. The core package is | 
 | 19 | based on PEP 282 and comments thereto in comp.lang.python, and influenced by | 
 | 20 | Apache's log4j system. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 21 |  | 
| Vinay Sajip | 6268cbc | 2009-01-21 00:19:28 +0000 | [diff] [blame] | 22 | Copyright (C) 2001-2009 Vinay Sajip. All Rights Reserved. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 23 |  | 
| Vinay Sajip | 6268cbc | 2009-01-21 00:19:28 +0000 | [diff] [blame] | 24 | To use, simply 'import logging.handlers' and log away! | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 25 | """ | 
 | 26 |  | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 27 | import logging, socket, os, pickle, struct, time, re | 
| Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 28 | from stat import ST_DEV, ST_INO | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 29 |  | 
| Vinay Sajip | 4600f11 | 2005-03-13 09:56:36 +0000 | [diff] [blame] | 30 | try: | 
 | 31 |     import codecs | 
 | 32 | except ImportError: | 
 | 33 |     codecs = None | 
 | 34 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 35 | # | 
 | 36 | # Some constants... | 
 | 37 | # | 
 | 38 |  | 
 | 39 | DEFAULT_TCP_LOGGING_PORT    = 9020 | 
 | 40 | DEFAULT_UDP_LOGGING_PORT    = 9021 | 
 | 41 | DEFAULT_HTTP_LOGGING_PORT   = 9022 | 
 | 42 | DEFAULT_SOAP_LOGGING_PORT   = 9023 | 
 | 43 | SYSLOG_UDP_PORT             = 514 | 
 | 44 |  | 
| Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 45 | _MIDNIGHT = 24 * 60 * 60  # number of seconds in a day | 
 | 46 |  | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 47 | class BaseRotatingHandler(logging.FileHandler): | 
 | 48 |     """ | 
 | 49 |     Base class for handlers that rotate log files at a certain point. | 
 | 50 |     Not meant to be instantiated directly.  Instead, use RotatingFileHandler | 
 | 51 |     or TimedRotatingFileHandler. | 
 | 52 |     """ | 
| Christian Heimes | e7a15bb | 2008-01-24 16:21:45 +0000 | [diff] [blame] | 53 |     def __init__(self, filename, mode, encoding=None, delay=0): | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 54 |         """ | 
 | 55 |         Use the specified filename for streamed logging | 
 | 56 |         """ | 
| Vinay Sajip | 4600f11 | 2005-03-13 09:56:36 +0000 | [diff] [blame] | 57 |         if codecs is None: | 
 | 58 |             encoding = None | 
| Christian Heimes | e7a15bb | 2008-01-24 16:21:45 +0000 | [diff] [blame] | 59 |         logging.FileHandler.__init__(self, filename, mode, encoding, delay) | 
| Vinay Sajip | 4600f11 | 2005-03-13 09:56:36 +0000 | [diff] [blame] | 60 |         self.mode = mode | 
 | 61 |         self.encoding = encoding | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 62 |  | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 63 |     def emit(self, record): | 
 | 64 |         """ | 
 | 65 |         Emit a record. | 
 | 66 |  | 
 | 67 |         Output the record to the file, catering for rollover as described | 
 | 68 |         in doRollover(). | 
 | 69 |         """ | 
| Vinay Sajip | 3970c11 | 2004-07-08 10:24:04 +0000 | [diff] [blame] | 70 |         try: | 
 | 71 |             if self.shouldRollover(record): | 
 | 72 |                 self.doRollover() | 
 | 73 |             logging.FileHandler.emit(self, record) | 
| Vinay Sajip | 85c1909 | 2005-10-31 13:14:19 +0000 | [diff] [blame] | 74 |         except (KeyboardInterrupt, SystemExit): | 
 | 75 |             raise | 
| Vinay Sajip | 3970c11 | 2004-07-08 10:24:04 +0000 | [diff] [blame] | 76 |         except: | 
 | 77 |             self.handleError(record) | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 78 |  | 
 | 79 | class RotatingFileHandler(BaseRotatingHandler): | 
 | 80 |     """ | 
 | 81 |     Handler for logging to a set of files, which switches from one file | 
 | 82 |     to the next when the current file reaches a certain size. | 
 | 83 |     """ | 
| Christian Heimes | e7a15bb | 2008-01-24 16:21:45 +0000 | [diff] [blame] | 84 |     def __init__(self, filename, mode='a', maxBytes=0, backupCount=0, encoding=None, delay=0): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 85 |         """ | 
 | 86 |         Open the specified file and use it as the stream for logging. | 
 | 87 |  | 
 | 88 |         By default, the file grows indefinitely. You can specify particular | 
 | 89 |         values of maxBytes and backupCount to allow the file to rollover at | 
 | 90 |         a predetermined size. | 
 | 91 |  | 
 | 92 |         Rollover occurs whenever the current log file is nearly maxBytes in | 
 | 93 |         length. If backupCount is >= 1, the system will successively create | 
 | 94 |         new files with the same pathname as the base file, but with extensions | 
 | 95 |         ".1", ".2" etc. appended to it. For example, with a backupCount of 5 | 
 | 96 |         and a base file name of "app.log", you would get "app.log", | 
 | 97 |         "app.log.1", "app.log.2", ... through to "app.log.5". The file being | 
 | 98 |         written to is always "app.log" - when it gets filled up, it is closed | 
 | 99 |         and renamed to "app.log.1", and if files "app.log.1", "app.log.2" etc. | 
 | 100 |         exist, then they are renamed to "app.log.2", "app.log.3" etc. | 
 | 101 |         respectively. | 
 | 102 |  | 
 | 103 |         If maxBytes is zero, rollover never occurs. | 
 | 104 |         """ | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 105 |         if maxBytes > 0: | 
| Vinay Sajip | 4600f11 | 2005-03-13 09:56:36 +0000 | [diff] [blame] | 106 |             mode = 'a' # doesn't make sense otherwise! | 
| Christian Heimes | e7a15bb | 2008-01-24 16:21:45 +0000 | [diff] [blame] | 107 |         BaseRotatingHandler.__init__(self, filename, mode, encoding, delay) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 108 |         self.maxBytes = maxBytes | 
 | 109 |         self.backupCount = backupCount | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 110 |  | 
 | 111 |     def doRollover(self): | 
 | 112 |         """ | 
 | 113 |         Do a rollover, as described in __init__(). | 
 | 114 |         """ | 
| Vinay Sajip | 6268cbc | 2009-01-21 00:19:28 +0000 | [diff] [blame] | 115 |         if self.stream: | 
 | 116 |             self.stream.close() | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 117 |         if self.backupCount > 0: | 
 | 118 |             for i in range(self.backupCount - 1, 0, -1): | 
 | 119 |                 sfn = "%s.%d" % (self.baseFilename, i) | 
 | 120 |                 dfn = "%s.%d" % (self.baseFilename, i + 1) | 
 | 121 |                 if os.path.exists(sfn): | 
 | 122 |                     #print "%s -> %s" % (sfn, dfn) | 
 | 123 |                     if os.path.exists(dfn): | 
 | 124 |                         os.remove(dfn) | 
 | 125 |                     os.rename(sfn, dfn) | 
 | 126 |             dfn = self.baseFilename + ".1" | 
 | 127 |             if os.path.exists(dfn): | 
 | 128 |                 os.remove(dfn) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 129 |             os.rename(self.baseFilename, dfn) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 130 |             #print "%s -> %s" % (self.baseFilename, dfn) | 
| Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 131 |         self.mode = 'w' | 
 | 132 |         self.stream = self._open() | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 133 |  | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 134 |     def shouldRollover(self, record): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 135 |         """ | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 136 |         Determine if rollover should occur. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 137 |  | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 138 |         Basically, see if the supplied record would cause the file to exceed | 
 | 139 |         the size limit we have. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 140 |         """ | 
| Vinay Sajip | 6268cbc | 2009-01-21 00:19:28 +0000 | [diff] [blame] | 141 |         if self.stream is None:                 # delay was set... | 
 | 142 |             self.stream = self._open() | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 143 |         if self.maxBytes > 0:                   # are we rolling over? | 
 | 144 |             msg = "%s\n" % self.format(record) | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 145 |             self.stream.seek(0, 2)  #due to non-posix-compliant Windows feature | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 146 |             if self.stream.tell() + len(msg) >= self.maxBytes: | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 147 |                 return 1 | 
 | 148 |         return 0 | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 149 |  | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 150 | class TimedRotatingFileHandler(BaseRotatingHandler): | 
 | 151 |     """ | 
 | 152 |     Handler for logging to a file, rotating the log file at certain timed | 
 | 153 |     intervals. | 
 | 154 |  | 
 | 155 |     If backupCount is > 0, when rollover is done, no more than backupCount | 
 | 156 |     files are kept - the oldest ones are deleted. | 
 | 157 |     """ | 
| Georg Brandl | 96f05e5 | 2009-06-08 08:53:39 +0000 | [diff] [blame] | 158 |     def __init__(self, filename, when='h', interval=1, backupCount=0, encoding=None, delay=0, utc=False): | 
| Christian Heimes | e7a15bb | 2008-01-24 16:21:45 +0000 | [diff] [blame] | 159 |         BaseRotatingHandler.__init__(self, filename, 'a', encoding, delay) | 
| Neal Norwitz | 9d72bb4 | 2007-04-17 08:48:32 +0000 | [diff] [blame] | 160 |         self.when = when.upper() | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 161 |         self.backupCount = backupCount | 
| Benjamin Peterson | a37cfc6 | 2008-05-26 13:48:34 +0000 | [diff] [blame] | 162 |         self.utc = utc | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 163 |         # Calculate the real rollover interval, which is just the number of | 
 | 164 |         # seconds between rollovers.  Also set the filename suffix used when | 
 | 165 |         # a rollover occurs.  Current 'when' events supported: | 
 | 166 |         # S - Seconds | 
 | 167 |         # M - Minutes | 
 | 168 |         # H - Hours | 
 | 169 |         # D - Days | 
 | 170 |         # midnight - roll over at midnight | 
 | 171 |         # W{0-6} - roll over on a certain day; 0 - Monday | 
 | 172 |         # | 
 | 173 |         # Case of the 'when' specifier is not important; lower or upper case | 
 | 174 |         # will work. | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 175 |         if self.when == 'S': | 
 | 176 |             self.interval = 1 # one second | 
 | 177 |             self.suffix = "%Y-%m-%d_%H-%M-%S" | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 178 |             self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}$" | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 179 |         elif self.when == 'M': | 
 | 180 |             self.interval = 60 # one minute | 
 | 181 |             self.suffix = "%Y-%m-%d_%H-%M" | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 182 |             self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}$" | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 183 |         elif self.when == 'H': | 
 | 184 |             self.interval = 60 * 60 # one hour | 
 | 185 |             self.suffix = "%Y-%m-%d_%H" | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 186 |             self.extMatch = r"^\d{4}-\d{2}-\d{2}_\d{2}$" | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 187 |         elif self.when == 'D' or self.when == 'MIDNIGHT': | 
 | 188 |             self.interval = 60 * 60 * 24 # one day | 
 | 189 |             self.suffix = "%Y-%m-%d" | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 190 |             self.extMatch = r"^\d{4}-\d{2}-\d{2}$" | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 191 |         elif self.when.startswith('W'): | 
 | 192 |             self.interval = 60 * 60 * 24 * 7 # one week | 
 | 193 |             if len(self.when) != 2: | 
 | 194 |                 raise ValueError("You must specify a day for weekly rollover from 0 to 6 (0 is Monday): %s" % self.when) | 
 | 195 |             if self.when[1] < '0' or self.when[1] > '6': | 
 | 196 |                 raise ValueError("Invalid day specified for weekly rollover: %s" % self.when) | 
 | 197 |             self.dayOfWeek = int(self.when[1]) | 
 | 198 |             self.suffix = "%Y-%m-%d" | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 199 |             self.extMatch = r"^\d{4}-\d{2}-\d{2}$" | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 200 |         else: | 
 | 201 |             raise ValueError("Invalid rollover interval specified: %s" % self.when) | 
 | 202 |  | 
| Antoine Pitrou | fd03645 | 2008-08-19 17:56:33 +0000 | [diff] [blame] | 203 |         self.extMatch = re.compile(self.extMatch, re.ASCII) | 
| Vinay Sajip | e7d4066 | 2004-10-03 19:12:07 +0000 | [diff] [blame] | 204 |         self.interval = self.interval * interval # multiply by units requested | 
| Vinay Sajip | affbd87 | 2009-06-11 10:11:47 +0000 | [diff] [blame] | 205 |         self.rolloverAt = self.computeRollover(int(time.time())) | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 206 |  | 
| Vinay Sajip | affbd87 | 2009-06-11 10:11:47 +0000 | [diff] [blame] | 207 |     def computeRollover(self, currentTime): | 
 | 208 |         """ | 
 | 209 |         Work out the rollover time based on the specified time. | 
 | 210 |         """ | 
 | 211 |         result = currentTime + self.interval | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 212 |         # If we are rolling over at midnight or weekly, then the interval is already known. | 
 | 213 |         # What we need to figure out is WHEN the next interval is.  In other words, | 
 | 214 |         # if you are rolling over at midnight, then your base interval is 1 day, | 
 | 215 |         # but you want to start that one day clock at midnight, not now.  So, we | 
 | 216 |         # have to fudge the rolloverAt value in order to trigger the first rollover | 
 | 217 |         # at the right time.  After that, the regular interval will take care of | 
 | 218 |         # the rest.  Note that this code doesn't care about leap seconds. :) | 
 | 219 |         if self.when == 'MIDNIGHT' or self.when.startswith('W'): | 
 | 220 |             # This could be done with less code, but I wanted it to be clear | 
| Vinay Sajip | affbd87 | 2009-06-11 10:11:47 +0000 | [diff] [blame] | 221 |             if self.utc: | 
| Benjamin Peterson | a37cfc6 | 2008-05-26 13:48:34 +0000 | [diff] [blame] | 222 |                 t = time.gmtime(currentTime) | 
 | 223 |             else: | 
 | 224 |                 t = time.localtime(currentTime) | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 225 |             currentHour = t[3] | 
 | 226 |             currentMinute = t[4] | 
 | 227 |             currentSecond = t[5] | 
 | 228 |             # r is the number of seconds left between now and midnight | 
| Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 229 |             r = _MIDNIGHT - ((currentHour * 60 + currentMinute) * 60 + | 
 | 230 |                     currentSecond) | 
| Vinay Sajip | affbd87 | 2009-06-11 10:11:47 +0000 | [diff] [blame] | 231 |             result = currentTime + r | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 232 |             # If we are rolling over on a certain day, add in the number of days until | 
 | 233 |             # the next rollover, but offset by 1 since we just calculated the time | 
 | 234 |             # until the next day starts.  There are three cases: | 
 | 235 |             # Case 1) The day to rollover is today; in this case, do nothing | 
 | 236 |             # Case 2) The day to rollover is further in the interval (i.e., today is | 
 | 237 |             #         day 2 (Wednesday) and rollover is on day 6 (Sunday).  Days to | 
 | 238 |             #         next rollover is simply 6 - 2 - 1, or 3. | 
 | 239 |             # Case 3) The day to rollover is behind us in the interval (i.e., today | 
 | 240 |             #         is day 5 (Saturday) and rollover is on day 3 (Thursday). | 
 | 241 |             #         Days to rollover is 6 - 5 + 3, or 4.  In this case, it's the | 
 | 242 |             #         number of days left in the current week (1) plus the number | 
 | 243 |             #         of days in the next week until the rollover day (3). | 
| Georg Brandl | 86def6c | 2008-01-21 20:36:10 +0000 | [diff] [blame] | 244 |             # The calculations described in 2) and 3) above need to have a day added. | 
 | 245 |             # This is because the above time calculation takes us to midnight on this | 
 | 246 |             # day, i.e. the start of the next day. | 
| Vinay Sajip | affbd87 | 2009-06-11 10:11:47 +0000 | [diff] [blame] | 247 |             if self.when.startswith('W'): | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 248 |                 day = t[6] # 0 is Monday | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 249 |                 if day != self.dayOfWeek: | 
 | 250 |                     if day < self.dayOfWeek: | 
| Georg Brandl | 86def6c | 2008-01-21 20:36:10 +0000 | [diff] [blame] | 251 |                         daysToWait = self.dayOfWeek - day | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 252 |                     else: | 
| Georg Brandl | 86def6c | 2008-01-21 20:36:10 +0000 | [diff] [blame] | 253 |                         daysToWait = 6 - day + self.dayOfWeek + 1 | 
| Vinay Sajip | affbd87 | 2009-06-11 10:11:47 +0000 | [diff] [blame] | 254 |                     newRolloverAt = result + (daysToWait * (60 * 60 * 24)) | 
 | 255 |                     if not self.utc: | 
| Benjamin Peterson | a37cfc6 | 2008-05-26 13:48:34 +0000 | [diff] [blame] | 256 |                         dstNow = t[-1] | 
 | 257 |                         dstAtRollover = time.localtime(newRolloverAt)[-1] | 
 | 258 |                         if dstNow != dstAtRollover: | 
 | 259 |                             if not dstNow:  # DST kicks in before next rollover, so we need to deduct an hour | 
 | 260 |                                 newRolloverAt = newRolloverAt - 3600 | 
 | 261 |                             else:           # DST bows out before next rollover, so we need to add an hour | 
 | 262 |                                 newRolloverAt = newRolloverAt + 3600 | 
| Vinay Sajip | affbd87 | 2009-06-11 10:11:47 +0000 | [diff] [blame] | 263 |                     result = newRolloverAt | 
 | 264 |         return result | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 265 |  | 
 | 266 |     def shouldRollover(self, record): | 
 | 267 |         """ | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 268 |         Determine if rollover should occur. | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 269 |  | 
 | 270 |         record is not used, as we are just comparing times, but it is needed so | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 271 |         the method signatures are the same | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 272 |         """ | 
 | 273 |         t = int(time.time()) | 
 | 274 |         if t >= self.rolloverAt: | 
 | 275 |             return 1 | 
| Vinay Sajip | 5e9e9e1 | 2004-07-12 09:21:41 +0000 | [diff] [blame] | 276 |         #print "No need to rollover: %d, %d" % (t, self.rolloverAt) | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 277 |         return 0 | 
 | 278 |  | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 279 |     def getFilesToDelete(self): | 
 | 280 |         """ | 
 | 281 |         Determine the files to delete when rolling over. | 
 | 282 |  | 
 | 283 |         More specific than the earlier method, which just used glob.glob(). | 
 | 284 |         """ | 
 | 285 |         dirName, baseName = os.path.split(self.baseFilename) | 
 | 286 |         fileNames = os.listdir(dirName) | 
 | 287 |         result = [] | 
 | 288 |         prefix = baseName + "." | 
 | 289 |         plen = len(prefix) | 
 | 290 |         for fileName in fileNames: | 
 | 291 |             if fileName[:plen] == prefix: | 
 | 292 |                 suffix = fileName[plen:] | 
 | 293 |                 if self.extMatch.match(suffix): | 
| Benjamin Peterson | a37cfc6 | 2008-05-26 13:48:34 +0000 | [diff] [blame] | 294 |                     result.append(os.path.join(dirName, fileName)) | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 295 |         result.sort() | 
 | 296 |         if len(result) < self.backupCount: | 
 | 297 |             result = [] | 
 | 298 |         else: | 
 | 299 |             result = result[:len(result) - self.backupCount] | 
 | 300 |         return result | 
 | 301 |  | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 302 |     def doRollover(self): | 
 | 303 |         """ | 
 | 304 |         do a rollover; in this case, a date/time stamp is appended to the filename | 
 | 305 |         when the rollover happens.  However, you want the file to be named for the | 
 | 306 |         start of the interval, not the current time.  If there is a backup count, | 
 | 307 |         then we have to get a list of matching filenames, sort them and remove | 
 | 308 |         the one with the oldest suffix. | 
 | 309 |         """ | 
| Vinay Sajip | 6268cbc | 2009-01-21 00:19:28 +0000 | [diff] [blame] | 310 |         if self.stream: | 
 | 311 |             self.stream.close() | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 312 |         # get the time that this sequence started at and make it a TimeTuple | 
 | 313 |         t = self.rolloverAt - self.interval | 
| Benjamin Peterson | a37cfc6 | 2008-05-26 13:48:34 +0000 | [diff] [blame] | 314 |         if self.utc: | 
 | 315 |             timeTuple = time.gmtime(t) | 
 | 316 |         else: | 
 | 317 |             timeTuple = time.localtime(t) | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 318 |         dfn = self.baseFilename + "." + time.strftime(self.suffix, timeTuple) | 
 | 319 |         if os.path.exists(dfn): | 
 | 320 |             os.remove(dfn) | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 321 |         os.rename(self.baseFilename, dfn) | 
| Vinay Sajip | 17c52d8 | 2004-07-03 11:48:34 +0000 | [diff] [blame] | 322 |         if self.backupCount > 0: | 
 | 323 |             # find the oldest log file and delete it | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 324 |             #s = glob.glob(self.baseFilename + ".20*") | 
 | 325 |             #if len(s) > self.backupCount: | 
 | 326 |             #    s.sort() | 
 | 327 |             #    os.remove(s[0]) | 
 | 328 |             for s in self.getFilesToDelete(): | 
 | 329 |                 os.remove(s) | 
| Vinay Sajip | 5e9e9e1 | 2004-07-12 09:21:41 +0000 | [diff] [blame] | 330 |         #print "%s -> %s" % (self.baseFilename, dfn) | 
| Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 331 |         self.mode = 'w' | 
 | 332 |         self.stream = self._open() | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 333 |         currentTime = int(time.time()) | 
| Vinay Sajip | affbd87 | 2009-06-11 10:11:47 +0000 | [diff] [blame] | 334 |         newRolloverAt = self.computeRollover(currentTime) | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 335 |         while newRolloverAt <= currentTime: | 
 | 336 |             newRolloverAt = newRolloverAt + self.interval | 
 | 337 |         #If DST changes and midnight or weekly rollover, adjust for this. | 
| Benjamin Peterson | a37cfc6 | 2008-05-26 13:48:34 +0000 | [diff] [blame] | 338 |         if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc: | 
| Benjamin Peterson | ad9d48d | 2008-04-02 21:49:44 +0000 | [diff] [blame] | 339 |             dstNow = time.localtime(currentTime)[-1] | 
 | 340 |             dstAtRollover = time.localtime(newRolloverAt)[-1] | 
 | 341 |             if dstNow != dstAtRollover: | 
 | 342 |                 if not dstNow:  # DST kicks in before next rollover, so we need to deduct an hour | 
 | 343 |                     newRolloverAt = newRolloverAt - 3600 | 
 | 344 |                 else:           # DST bows out before next rollover, so we need to add an hour | 
 | 345 |                     newRolloverAt = newRolloverAt + 3600 | 
 | 346 |         self.rolloverAt = newRolloverAt | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 347 |  | 
| Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 348 | class WatchedFileHandler(logging.FileHandler): | 
 | 349 |     """ | 
 | 350 |     A handler for logging to a file, which watches the file | 
 | 351 |     to see if it has changed while in use. This can happen because of | 
 | 352 |     usage of programs such as newsyslog and logrotate which perform | 
 | 353 |     log file rotation. This handler, intended for use under Unix, | 
 | 354 |     watches the file to see if it has changed since the last emit. | 
 | 355 |     (A file has changed if its device or inode have changed.) | 
 | 356 |     If it has changed, the old file stream is closed, and the file | 
 | 357 |     opened to get a new stream. | 
 | 358 |  | 
 | 359 |     This handler is not appropriate for use under Windows, because | 
 | 360 |     under Windows open files cannot be moved or renamed - logging | 
 | 361 |     opens the files with exclusive locks - and so there is no need | 
 | 362 |     for such a handler. Furthermore, ST_INO is not supported under | 
 | 363 |     Windows; stat always returns zero for this value. | 
 | 364 |  | 
 | 365 |     This handler is based on a suggestion and patch by Chad J. | 
 | 366 |     Schroeder. | 
 | 367 |     """ | 
| Christian Heimes | e7a15bb | 2008-01-24 16:21:45 +0000 | [diff] [blame] | 368 |     def __init__(self, filename, mode='a', encoding=None, delay=0): | 
 | 369 |         logging.FileHandler.__init__(self, filename, mode, encoding, delay) | 
 | 370 |         if not os.path.exists(self.baseFilename): | 
 | 371 |             self.dev, self.ino = -1, -1 | 
 | 372 |         else: | 
 | 373 |             stat = os.stat(self.baseFilename) | 
 | 374 |             self.dev, self.ino = stat[ST_DEV], stat[ST_INO] | 
| Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 375 |  | 
 | 376 |     def emit(self, record): | 
 | 377 |         """ | 
 | 378 |         Emit a record. | 
 | 379 |  | 
 | 380 |         First check if the underlying file has changed, and if it | 
 | 381 |         has, close the old stream and reopen the file to get the | 
 | 382 |         current stream. | 
 | 383 |         """ | 
 | 384 |         if not os.path.exists(self.baseFilename): | 
 | 385 |             stat = None | 
 | 386 |             changed = 1 | 
 | 387 |         else: | 
 | 388 |             stat = os.stat(self.baseFilename) | 
 | 389 |             changed = (stat[ST_DEV] != self.dev) or (stat[ST_INO] != self.ino) | 
| Christian Heimes | e7a15bb | 2008-01-24 16:21:45 +0000 | [diff] [blame] | 390 |         if changed and self.stream is not None: | 
| Thomas Wouters | fc7bb8c | 2007-01-15 15:49:28 +0000 | [diff] [blame] | 391 |             self.stream.flush() | 
 | 392 |             self.stream.close() | 
 | 393 |             self.stream = self._open() | 
 | 394 |             if stat is None: | 
 | 395 |                 stat = os.stat(self.baseFilename) | 
 | 396 |             self.dev, self.ino = stat[ST_DEV], stat[ST_INO] | 
 | 397 |         logging.FileHandler.emit(self, record) | 
 | 398 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 399 | class SocketHandler(logging.Handler): | 
 | 400 |     """ | 
 | 401 |     A handler class which writes logging records, in pickle format, to | 
 | 402 |     a streaming socket. The socket is kept open across logging calls. | 
 | 403 |     If the peer resets it, an attempt is made to reconnect on the next call. | 
| Raymond Hettinger | 6f3eaa6 | 2003-06-27 21:43:39 +0000 | [diff] [blame] | 404 |     The pickle which is sent is that of the LogRecord's attribute dictionary | 
 | 405 |     (__dict__), so that the receiver does not need to have the logging module | 
 | 406 |     installed in order to process the logging event. | 
 | 407 |  | 
 | 408 |     To unpickle the record at the receiving end into a LogRecord, use the | 
 | 409 |     makeLogRecord function. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 410 |     """ | 
 | 411 |  | 
 | 412 |     def __init__(self, host, port): | 
 | 413 |         """ | 
 | 414 |         Initializes the handler with a specific host address and port. | 
 | 415 |  | 
 | 416 |         The attribute 'closeOnError' is set to 1 - which means that if | 
 | 417 |         a socket error occurs, the socket is silently closed and then | 
 | 418 |         reopened on the next logging call. | 
 | 419 |         """ | 
 | 420 |         logging.Handler.__init__(self) | 
 | 421 |         self.host = host | 
 | 422 |         self.port = port | 
 | 423 |         self.sock = None | 
 | 424 |         self.closeOnError = 0 | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 425 |         self.retryTime = None | 
 | 426 |         # | 
 | 427 |         # Exponential backoff parameters. | 
 | 428 |         # | 
 | 429 |         self.retryStart = 1.0 | 
 | 430 |         self.retryMax = 30.0 | 
 | 431 |         self.retryFactor = 2.0 | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 432 |  | 
| Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 433 |     def makeSocket(self, timeout=1): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 434 |         """ | 
 | 435 |         A factory method which allows subclasses to define the precise | 
 | 436 |         type of socket they want. | 
 | 437 |         """ | 
 | 438 |         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | 
| Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 439 |         if hasattr(s, 'settimeout'): | 
 | 440 |             s.settimeout(timeout) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 441 |         s.connect((self.host, self.port)) | 
 | 442 |         return s | 
 | 443 |  | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 444 |     def createSocket(self): | 
 | 445 |         """ | 
 | 446 |         Try to create a socket, using an exponential backoff with | 
 | 447 |         a max retry time. Thanks to Robert Olson for the original patch | 
 | 448 |         (SF #815911) which has been slightly refactored. | 
 | 449 |         """ | 
 | 450 |         now = time.time() | 
 | 451 |         # Either retryTime is None, in which case this | 
 | 452 |         # is the first time back after a disconnect, or | 
 | 453 |         # we've waited long enough. | 
 | 454 |         if self.retryTime is None: | 
| Tim Peters | 4e0e1b6 | 2004-07-07 20:54:48 +0000 | [diff] [blame] | 455 |             attempt = 1 | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 456 |         else: | 
| Tim Peters | 4e0e1b6 | 2004-07-07 20:54:48 +0000 | [diff] [blame] | 457 |             attempt = (now >= self.retryTime) | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 458 |         if attempt: | 
 | 459 |             try: | 
 | 460 |                 self.sock = self.makeSocket() | 
 | 461 |                 self.retryTime = None # next time, no delay before trying | 
| Thomas Wouters | 902d6eb | 2007-01-09 23:18:33 +0000 | [diff] [blame] | 462 |             except socket.error: | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 463 |                 #Creation failed, so set the retry time and return. | 
 | 464 |                 if self.retryTime is None: | 
 | 465 |                     self.retryPeriod = self.retryStart | 
 | 466 |                 else: | 
 | 467 |                     self.retryPeriod = self.retryPeriod * self.retryFactor | 
 | 468 |                     if self.retryPeriod > self.retryMax: | 
 | 469 |                         self.retryPeriod = self.retryMax | 
 | 470 |                 self.retryTime = now + self.retryPeriod | 
 | 471 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 472 |     def send(self, s): | 
 | 473 |         """ | 
 | 474 |         Send a pickled string to the socket. | 
 | 475 |  | 
 | 476 |         This function allows for partial sends which can happen when the | 
 | 477 |         network is busy. | 
 | 478 |         """ | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 479 |         if self.sock is None: | 
 | 480 |             self.createSocket() | 
 | 481 |         #self.sock can be None either because we haven't reached the retry | 
 | 482 |         #time yet, or because we have reached the retry time and retried, | 
 | 483 |         #but are still unable to connect. | 
 | 484 |         if self.sock: | 
 | 485 |             try: | 
 | 486 |                 if hasattr(self.sock, "sendall"): | 
 | 487 |                     self.sock.sendall(s) | 
 | 488 |                 else: | 
 | 489 |                     sentsofar = 0 | 
 | 490 |                     left = len(s) | 
 | 491 |                     while left > 0: | 
 | 492 |                         sent = self.sock.send(s[sentsofar:]) | 
 | 493 |                         sentsofar = sentsofar + sent | 
 | 494 |                         left = left - sent | 
 | 495 |             except socket.error: | 
 | 496 |                 self.sock.close() | 
 | 497 |                 self.sock = None  # so we can call createSocket next time | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 498 |  | 
 | 499 |     def makePickle(self, record): | 
 | 500 |         """ | 
 | 501 |         Pickles the record in binary format with a length prefix, and | 
 | 502 |         returns it ready for transmission across the socket. | 
 | 503 |         """ | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 504 |         ei = record.exc_info | 
 | 505 |         if ei: | 
| Tim Peters | 4e0e1b6 | 2004-07-07 20:54:48 +0000 | [diff] [blame] | 506 |             dummy = self.format(record) # just to get traceback text into record.exc_text | 
 | 507 |             record.exc_info = None  # to avoid Unpickleable error | 
| Guido van Rossum | ba205d6 | 2006-08-17 08:57:26 +0000 | [diff] [blame] | 508 |         s = pickle.dumps(record.__dict__, 1) | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 509 |         if ei: | 
| Tim Peters | 4e0e1b6 | 2004-07-07 20:54:48 +0000 | [diff] [blame] | 510 |             record.exc_info = ei  # for next handler | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 511 |         slen = struct.pack(">L", len(s)) | 
 | 512 |         return slen + s | 
 | 513 |  | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 514 |     def handleError(self, record): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 515 |         """ | 
 | 516 |         Handle an error during logging. | 
 | 517 |  | 
 | 518 |         An error has occurred during logging. Most likely cause - | 
 | 519 |         connection lost. Close the socket so that we can retry on the | 
 | 520 |         next event. | 
 | 521 |         """ | 
 | 522 |         if self.closeOnError and self.sock: | 
 | 523 |             self.sock.close() | 
 | 524 |             self.sock = None        #try to reconnect next time | 
 | 525 |         else: | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 526 |             logging.Handler.handleError(self, record) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 527 |  | 
 | 528 |     def emit(self, record): | 
 | 529 |         """ | 
 | 530 |         Emit a record. | 
 | 531 |  | 
 | 532 |         Pickles the record and writes it to the socket in binary format. | 
 | 533 |         If there is an error with the socket, silently drop the packet. | 
 | 534 |         If there was a problem with the socket, re-establishes the | 
 | 535 |         socket. | 
 | 536 |         """ | 
 | 537 |         try: | 
 | 538 |             s = self.makePickle(record) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 539 |             self.send(s) | 
| Vinay Sajip | 85c1909 | 2005-10-31 13:14:19 +0000 | [diff] [blame] | 540 |         except (KeyboardInterrupt, SystemExit): | 
 | 541 |             raise | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 542 |         except: | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 543 |             self.handleError(record) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 544 |  | 
 | 545 |     def close(self): | 
 | 546 |         """ | 
 | 547 |         Closes the socket. | 
 | 548 |         """ | 
 | 549 |         if self.sock: | 
 | 550 |             self.sock.close() | 
 | 551 |             self.sock = None | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 552 |         logging.Handler.close(self) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 553 |  | 
 | 554 | class DatagramHandler(SocketHandler): | 
 | 555 |     """ | 
 | 556 |     A handler class which writes logging records, in pickle format, to | 
| Raymond Hettinger | 6f3eaa6 | 2003-06-27 21:43:39 +0000 | [diff] [blame] | 557 |     a datagram socket.  The pickle which is sent is that of the LogRecord's | 
 | 558 |     attribute dictionary (__dict__), so that the receiver does not need to | 
 | 559 |     have the logging module installed in order to process the logging event. | 
 | 560 |  | 
 | 561 |     To unpickle the record at the receiving end into a LogRecord, use the | 
 | 562 |     makeLogRecord function. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 563 |  | 
 | 564 |     """ | 
 | 565 |     def __init__(self, host, port): | 
 | 566 |         """ | 
 | 567 |         Initializes the handler with a specific host address and port. | 
 | 568 |         """ | 
 | 569 |         SocketHandler.__init__(self, host, port) | 
 | 570 |         self.closeOnError = 0 | 
 | 571 |  | 
 | 572 |     def makeSocket(self): | 
 | 573 |         """ | 
 | 574 |         The factory method of SocketHandler is here overridden to create | 
 | 575 |         a UDP socket (SOCK_DGRAM). | 
 | 576 |         """ | 
 | 577 |         s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) | 
 | 578 |         return s | 
 | 579 |  | 
 | 580 |     def send(self, s): | 
 | 581 |         """ | 
 | 582 |         Send a pickled string to a socket. | 
 | 583 |  | 
 | 584 |         This function no longer allows for partial sends which can happen | 
 | 585 |         when the network is busy - UDP does not guarantee delivery and | 
 | 586 |         can deliver packets out of sequence. | 
 | 587 |         """ | 
| Vinay Sajip | fb15417 | 2004-08-24 09:36:23 +0000 | [diff] [blame] | 588 |         if self.sock is None: | 
 | 589 |             self.createSocket() | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 590 |         self.sock.sendto(s, (self.host, self.port)) | 
 | 591 |  | 
 | 592 | class SysLogHandler(logging.Handler): | 
 | 593 |     """ | 
 | 594 |     A handler class which sends formatted logging records to a syslog | 
 | 595 |     server. Based on Sam Rushing's syslog module: | 
 | 596 |     http://www.nightmare.com/squirl/python-ext/misc/syslog.py | 
 | 597 |     Contributed by Nicolas Untz (after which minor refactoring changes | 
 | 598 |     have been made). | 
 | 599 |     """ | 
 | 600 |  | 
 | 601 |     # from <linux/sys/syslog.h>: | 
 | 602 |     # ====================================================================== | 
 | 603 |     # priorities/facilities are encoded into a single 32-bit quantity, where | 
 | 604 |     # the bottom 3 bits are the priority (0-7) and the top 28 bits are the | 
 | 605 |     # facility (0-big number). Both the priorities and the facilities map | 
 | 606 |     # roughly one-to-one to strings in the syslogd(8) source code.  This | 
 | 607 |     # mapping is included in this file. | 
 | 608 |     # | 
 | 609 |     # priorities (these are ordered) | 
 | 610 |  | 
 | 611 |     LOG_EMERG     = 0       #  system is unusable | 
 | 612 |     LOG_ALERT     = 1       #  action must be taken immediately | 
 | 613 |     LOG_CRIT      = 2       #  critical conditions | 
 | 614 |     LOG_ERR       = 3       #  error conditions | 
 | 615 |     LOG_WARNING   = 4       #  warning conditions | 
 | 616 |     LOG_NOTICE    = 5       #  normal but significant condition | 
 | 617 |     LOG_INFO      = 6       #  informational | 
 | 618 |     LOG_DEBUG     = 7       #  debug-level messages | 
 | 619 |  | 
 | 620 |     #  facility codes | 
 | 621 |     LOG_KERN      = 0       #  kernel messages | 
 | 622 |     LOG_USER      = 1       #  random user-level messages | 
 | 623 |     LOG_MAIL      = 2       #  mail system | 
 | 624 |     LOG_DAEMON    = 3       #  system daemons | 
 | 625 |     LOG_AUTH      = 4       #  security/authorization messages | 
 | 626 |     LOG_SYSLOG    = 5       #  messages generated internally by syslogd | 
 | 627 |     LOG_LPR       = 6       #  line printer subsystem | 
 | 628 |     LOG_NEWS      = 7       #  network news subsystem | 
 | 629 |     LOG_UUCP      = 8       #  UUCP subsystem | 
 | 630 |     LOG_CRON      = 9       #  clock daemon | 
 | 631 |     LOG_AUTHPRIV  = 10  #  security/authorization messages (private) | 
 | 632 |  | 
 | 633 |     #  other codes through 15 reserved for system use | 
 | 634 |     LOG_LOCAL0    = 16      #  reserved for local use | 
 | 635 |     LOG_LOCAL1    = 17      #  reserved for local use | 
 | 636 |     LOG_LOCAL2    = 18      #  reserved for local use | 
 | 637 |     LOG_LOCAL3    = 19      #  reserved for local use | 
 | 638 |     LOG_LOCAL4    = 20      #  reserved for local use | 
 | 639 |     LOG_LOCAL5    = 21      #  reserved for local use | 
 | 640 |     LOG_LOCAL6    = 22      #  reserved for local use | 
 | 641 |     LOG_LOCAL7    = 23      #  reserved for local use | 
 | 642 |  | 
 | 643 |     priority_names = { | 
 | 644 |         "alert":    LOG_ALERT, | 
 | 645 |         "crit":     LOG_CRIT, | 
 | 646 |         "critical": LOG_CRIT, | 
 | 647 |         "debug":    LOG_DEBUG, | 
 | 648 |         "emerg":    LOG_EMERG, | 
 | 649 |         "err":      LOG_ERR, | 
 | 650 |         "error":    LOG_ERR,        #  DEPRECATED | 
 | 651 |         "info":     LOG_INFO, | 
 | 652 |         "notice":   LOG_NOTICE, | 
 | 653 |         "panic":    LOG_EMERG,      #  DEPRECATED | 
 | 654 |         "warn":     LOG_WARNING,    #  DEPRECATED | 
 | 655 |         "warning":  LOG_WARNING, | 
 | 656 |         } | 
 | 657 |  | 
 | 658 |     facility_names = { | 
 | 659 |         "auth":     LOG_AUTH, | 
 | 660 |         "authpriv": LOG_AUTHPRIV, | 
 | 661 |         "cron":     LOG_CRON, | 
 | 662 |         "daemon":   LOG_DAEMON, | 
 | 663 |         "kern":     LOG_KERN, | 
 | 664 |         "lpr":      LOG_LPR, | 
 | 665 |         "mail":     LOG_MAIL, | 
 | 666 |         "news":     LOG_NEWS, | 
 | 667 |         "security": LOG_AUTH,       #  DEPRECATED | 
 | 668 |         "syslog":   LOG_SYSLOG, | 
 | 669 |         "user":     LOG_USER, | 
 | 670 |         "uucp":     LOG_UUCP, | 
 | 671 |         "local0":   LOG_LOCAL0, | 
 | 672 |         "local1":   LOG_LOCAL1, | 
 | 673 |         "local2":   LOG_LOCAL2, | 
 | 674 |         "local3":   LOG_LOCAL3, | 
 | 675 |         "local4":   LOG_LOCAL4, | 
 | 676 |         "local5":   LOG_LOCAL5, | 
 | 677 |         "local6":   LOG_LOCAL6, | 
 | 678 |         "local7":   LOG_LOCAL7, | 
 | 679 |         } | 
 | 680 |  | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 681 |     #The map below appears to be trivially lowercasing the key. However, | 
 | 682 |     #there's more to it than meets the eye - in some locales, lowercasing | 
 | 683 |     #gives unexpected results. See SF #1524081: in the Turkish locale, | 
 | 684 |     #"INFO".lower() != "info" | 
 | 685 |     priority_map = { | 
 | 686 |         "DEBUG" : "debug", | 
 | 687 |         "INFO" : "info", | 
 | 688 |         "WARNING" : "warning", | 
 | 689 |         "ERROR" : "error", | 
 | 690 |         "CRITICAL" : "critical" | 
 | 691 |     } | 
 | 692 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 693 |     def __init__(self, address=('localhost', SYSLOG_UDP_PORT), facility=LOG_USER): | 
 | 694 |         """ | 
 | 695 |         Initialize a handler. | 
 | 696 |  | 
| Guido van Rossum | e7ba495 | 2007-06-06 23:52:48 +0000 | [diff] [blame] | 697 |         If address is specified as a string, a UNIX socket is used. To log to a | 
 | 698 |         local syslogd, "SysLogHandler(address="/dev/log")" can be used. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 699 |         If facility is not specified, LOG_USER is used. | 
 | 700 |         """ | 
 | 701 |         logging.Handler.__init__(self) | 
 | 702 |  | 
 | 703 |         self.address = address | 
 | 704 |         self.facility = facility | 
| Guido van Rossum | 1325790 | 2007-06-07 23:15:56 +0000 | [diff] [blame] | 705 |         if isinstance(address, str): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 706 |             self.unixsocket = 1 | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 707 |             self._connect_unixsocket(address) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 708 |         else: | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 709 |             self.unixsocket = 0 | 
| Thomas Wouters | 89f507f | 2006-12-13 04:49:30 +0000 | [diff] [blame] | 710 |             self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 711 |  | 
 | 712 |         self.formatter = None | 
 | 713 |  | 
| Vinay Sajip | a1974c1 | 2005-01-13 08:23:56 +0000 | [diff] [blame] | 714 |     def _connect_unixsocket(self, address): | 
 | 715 |         self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM) | 
 | 716 |         # syslog may require either DGRAM or STREAM sockets | 
 | 717 |         try: | 
 | 718 |             self.socket.connect(address) | 
 | 719 |         except socket.error: | 
 | 720 |             self.socket.close() | 
 | 721 |             self.socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) | 
| Vinay Sajip | 8b6b53f | 2005-11-09 13:55:13 +0000 | [diff] [blame] | 722 |             self.socket.connect(address) | 
| Vinay Sajip | a1974c1 | 2005-01-13 08:23:56 +0000 | [diff] [blame] | 723 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 724 |     # curious: when talking to the unix-domain '/dev/log' socket, a | 
 | 725 |     #   zero-terminator seems to be required.  this string is placed | 
 | 726 |     #   into a class variable so that it can be overridden if | 
 | 727 |     #   necessary. | 
 | 728 |     log_format_string = '<%d>%s\000' | 
 | 729 |  | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 730 |     def encodePriority(self, facility, priority): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 731 |         """ | 
 | 732 |         Encode the facility and priority. You can pass in strings or | 
 | 733 |         integers - if strings are passed, the facility_names and | 
 | 734 |         priority_names mapping dictionaries are used to convert them to | 
 | 735 |         integers. | 
 | 736 |         """ | 
| Guido van Rossum | 1325790 | 2007-06-07 23:15:56 +0000 | [diff] [blame] | 737 |         if isinstance(facility, str): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 738 |             facility = self.facility_names[facility] | 
| Guido van Rossum | 1325790 | 2007-06-07 23:15:56 +0000 | [diff] [blame] | 739 |         if isinstance(priority, str): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 740 |             priority = self.priority_names[priority] | 
 | 741 |         return (facility << 3) | priority | 
 | 742 |  | 
 | 743 |     def close (self): | 
 | 744 |         """ | 
 | 745 |         Closes the socket. | 
 | 746 |         """ | 
 | 747 |         if self.unixsocket: | 
 | 748 |             self.socket.close() | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 749 |         logging.Handler.close(self) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 750 |  | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 751 |     def mapPriority(self, levelName): | 
 | 752 |         """ | 
 | 753 |         Map a logging level name to a key in the priority_names map. | 
 | 754 |         This is useful in two scenarios: when custom levels are being | 
 | 755 |         used, and in the case where you can't do a straightforward | 
 | 756 |         mapping by lowercasing the logging level name because of locale- | 
 | 757 |         specific issues (see SF #1524081). | 
 | 758 |         """ | 
 | 759 |         return self.priority_map.get(levelName, "warning") | 
 | 760 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 761 |     def emit(self, record): | 
 | 762 |         """ | 
 | 763 |         Emit a record. | 
 | 764 |  | 
 | 765 |         The record is formatted, and then sent to the syslog server. If | 
 | 766 |         exception information is present, it is NOT sent to the server. | 
 | 767 |         """ | 
 | 768 |         msg = self.format(record) | 
 | 769 |         """ | 
 | 770 |         We need to convert record level to lowercase, maybe this will | 
 | 771 |         change in the future. | 
 | 772 |         """ | 
 | 773 |         msg = self.log_format_string % ( | 
 | 774 |             self.encodePriority(self.facility, | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 775 |                                 self.mapPriority(record.levelname)), | 
 | 776 |                                 msg) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 777 |         try: | 
 | 778 |             if self.unixsocket: | 
| Vinay Sajip | a1974c1 | 2005-01-13 08:23:56 +0000 | [diff] [blame] | 779 |                 try: | 
 | 780 |                     self.socket.send(msg) | 
 | 781 |                 except socket.error: | 
 | 782 |                     self._connect_unixsocket(self.address) | 
 | 783 |                     self.socket.send(msg) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 784 |             else: | 
 | 785 |                 self.socket.sendto(msg, self.address) | 
| Vinay Sajip | 85c1909 | 2005-10-31 13:14:19 +0000 | [diff] [blame] | 786 |         except (KeyboardInterrupt, SystemExit): | 
 | 787 |             raise | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 788 |         except: | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 789 |             self.handleError(record) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 790 |  | 
 | 791 | class SMTPHandler(logging.Handler): | 
 | 792 |     """ | 
 | 793 |     A handler class which sends an SMTP email for each logging event. | 
 | 794 |     """ | 
| Guido van Rossum | 360e4b8 | 2007-05-14 22:51:27 +0000 | [diff] [blame] | 795 |     def __init__(self, mailhost, fromaddr, toaddrs, subject, credentials=None): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 796 |         """ | 
 | 797 |         Initialize the handler. | 
 | 798 |  | 
 | 799 |         Initialize the instance with the from and to addresses and subject | 
 | 800 |         line of the email. To specify a non-standard SMTP port, use the | 
| Guido van Rossum | 360e4b8 | 2007-05-14 22:51:27 +0000 | [diff] [blame] | 801 |         (host, port) tuple format for the mailhost argument. To specify | 
 | 802 |         authentication credentials, supply a (username, password) tuple | 
 | 803 |         for the credentials argument. | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 804 |         """ | 
 | 805 |         logging.Handler.__init__(self) | 
| Guido van Rossum | 1325790 | 2007-06-07 23:15:56 +0000 | [diff] [blame] | 806 |         if isinstance(mailhost, tuple): | 
| Guido van Rossum | 360e4b8 | 2007-05-14 22:51:27 +0000 | [diff] [blame] | 807 |             self.mailhost, self.mailport = mailhost | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 808 |         else: | 
| Guido van Rossum | 360e4b8 | 2007-05-14 22:51:27 +0000 | [diff] [blame] | 809 |             self.mailhost, self.mailport = mailhost, None | 
| Guido van Rossum | 1325790 | 2007-06-07 23:15:56 +0000 | [diff] [blame] | 810 |         if isinstance(credentials, tuple): | 
| Guido van Rossum | 360e4b8 | 2007-05-14 22:51:27 +0000 | [diff] [blame] | 811 |             self.username, self.password = credentials | 
 | 812 |         else: | 
 | 813 |             self.username = None | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 814 |         self.fromaddr = fromaddr | 
| Guido van Rossum | 1325790 | 2007-06-07 23:15:56 +0000 | [diff] [blame] | 815 |         if isinstance(toaddrs, str): | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 816 |             toaddrs = [toaddrs] | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 817 |         self.toaddrs = toaddrs | 
 | 818 |         self.subject = subject | 
 | 819 |  | 
 | 820 |     def getSubject(self, record): | 
 | 821 |         """ | 
 | 822 |         Determine the subject for the email. | 
 | 823 |  | 
 | 824 |         If you want to specify a subject line which is record-dependent, | 
 | 825 |         override this method. | 
 | 826 |         """ | 
 | 827 |         return self.subject | 
 | 828 |  | 
| Vinay Sajip | e7d4066 | 2004-10-03 19:12:07 +0000 | [diff] [blame] | 829 |     weekdayname = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun'] | 
 | 830 |  | 
 | 831 |     monthname = [None, | 
 | 832 |                  'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', | 
 | 833 |                  'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'] | 
 | 834 |  | 
 | 835 |     def date_time(self): | 
 | 836 |         """ | 
 | 837 |         Return the current date and time formatted for a MIME header. | 
 | 838 |         Needed for Python 1.5.2 (no email package available) | 
 | 839 |         """ | 
 | 840 |         year, month, day, hh, mm, ss, wd, y, z = time.gmtime(time.time()) | 
 | 841 |         s = "%s, %02d %3s %4d %02d:%02d:%02d GMT" % ( | 
 | 842 |                 self.weekdayname[wd], | 
 | 843 |                 day, self.monthname[month], year, | 
 | 844 |                 hh, mm, ss) | 
 | 845 |         return s | 
 | 846 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 847 |     def emit(self, record): | 
 | 848 |         """ | 
 | 849 |         Emit a record. | 
 | 850 |  | 
 | 851 |         Format the record and send it to the specified addressees. | 
 | 852 |         """ | 
 | 853 |         try: | 
 | 854 |             import smtplib | 
| Vinay Sajip | e7d4066 | 2004-10-03 19:12:07 +0000 | [diff] [blame] | 855 |             try: | 
| Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 856 |                 from email.utils import formatdate | 
| Thomas Wouters | 902d6eb | 2007-01-09 23:18:33 +0000 | [diff] [blame] | 857 |             except ImportError: | 
| Vinay Sajip | e7d4066 | 2004-10-03 19:12:07 +0000 | [diff] [blame] | 858 |                 formatdate = self.date_time | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 859 |             port = self.mailport | 
 | 860 |             if not port: | 
 | 861 |                 port = smtplib.SMTP_PORT | 
 | 862 |             smtp = smtplib.SMTP(self.mailhost, port) | 
 | 863 |             msg = self.format(record) | 
| Neal Norwitz | f297bd1 | 2003-04-23 03:49:43 +0000 | [diff] [blame] | 864 |             msg = "From: %s\r\nTo: %s\r\nSubject: %s\r\nDate: %s\r\n\r\n%s" % ( | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 865 |                             self.fromaddr, | 
| Neal Norwitz | 9d72bb4 | 2007-04-17 08:48:32 +0000 | [diff] [blame] | 866 |                             ",".join(self.toaddrs), | 
| Neal Norwitz | f297bd1 | 2003-04-23 03:49:43 +0000 | [diff] [blame] | 867 |                             self.getSubject(record), | 
| Martin v. Löwis | 318a12e | 2004-08-18 12:27:40 +0000 | [diff] [blame] | 868 |                             formatdate(), msg) | 
| Guido van Rossum | 360e4b8 | 2007-05-14 22:51:27 +0000 | [diff] [blame] | 869 |             if self.username: | 
 | 870 |                 smtp.login(self.username, self.password) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 871 |             smtp.sendmail(self.fromaddr, self.toaddrs, msg) | 
 | 872 |             smtp.quit() | 
| Vinay Sajip | 245a5ab | 2005-10-31 14:27:01 +0000 | [diff] [blame] | 873 |         except (KeyboardInterrupt, SystemExit): | 
 | 874 |             raise | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 875 |         except: | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 876 |             self.handleError(record) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 877 |  | 
 | 878 | class NTEventLogHandler(logging.Handler): | 
 | 879 |     """ | 
 | 880 |     A handler class which sends events to the NT Event Log. Adds a | 
 | 881 |     registry entry for the specified application name. If no dllname is | 
 | 882 |     provided, win32service.pyd (which contains some basic message | 
 | 883 |     placeholders) is used. Note that use of these placeholders will make | 
 | 884 |     your event logs big, as the entire message source is held in the log. | 
 | 885 |     If you want slimmer logs, you have to pass in the name of your own DLL | 
 | 886 |     which contains the message definitions you want to use in the event log. | 
 | 887 |     """ | 
 | 888 |     def __init__(self, appname, dllname=None, logtype="Application"): | 
 | 889 |         logging.Handler.__init__(self) | 
 | 890 |         try: | 
 | 891 |             import win32evtlogutil, win32evtlog | 
 | 892 |             self.appname = appname | 
 | 893 |             self._welu = win32evtlogutil | 
 | 894 |             if not dllname: | 
 | 895 |                 dllname = os.path.split(self._welu.__file__) | 
 | 896 |                 dllname = os.path.split(dllname[0]) | 
 | 897 |                 dllname = os.path.join(dllname[0], r'win32service.pyd') | 
 | 898 |             self.dllname = dllname | 
 | 899 |             self.logtype = logtype | 
 | 900 |             self._welu.AddSourceToRegistry(appname, dllname, logtype) | 
 | 901 |             self.deftype = win32evtlog.EVENTLOG_ERROR_TYPE | 
 | 902 |             self.typemap = { | 
 | 903 |                 logging.DEBUG   : win32evtlog.EVENTLOG_INFORMATION_TYPE, | 
 | 904 |                 logging.INFO    : win32evtlog.EVENTLOG_INFORMATION_TYPE, | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 905 |                 logging.WARNING : win32evtlog.EVENTLOG_WARNING_TYPE, | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 906 |                 logging.ERROR   : win32evtlog.EVENTLOG_ERROR_TYPE, | 
 | 907 |                 logging.CRITICAL: win32evtlog.EVENTLOG_ERROR_TYPE, | 
 | 908 |          } | 
 | 909 |         except ImportError: | 
| Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 910 |             print("The Python Win32 extensions for NT (service, event "\ | 
 | 911 |                         "logging) appear not to be available.") | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 912 |             self._welu = None | 
 | 913 |  | 
 | 914 |     def getMessageID(self, record): | 
 | 915 |         """ | 
 | 916 |         Return the message ID for the event record. If you are using your | 
 | 917 |         own messages, you could do this by having the msg passed to the | 
 | 918 |         logger being an ID rather than a formatting string. Then, in here, | 
 | 919 |         you could use a dictionary lookup to get the message ID. This | 
 | 920 |         version returns 1, which is the base message ID in win32service.pyd. | 
 | 921 |         """ | 
 | 922 |         return 1 | 
 | 923 |  | 
 | 924 |     def getEventCategory(self, record): | 
 | 925 |         """ | 
 | 926 |         Return the event category for the record. | 
 | 927 |  | 
 | 928 |         Override this if you want to specify your own categories. This version | 
 | 929 |         returns 0. | 
 | 930 |         """ | 
 | 931 |         return 0 | 
 | 932 |  | 
 | 933 |     def getEventType(self, record): | 
 | 934 |         """ | 
 | 935 |         Return the event type for the record. | 
 | 936 |  | 
 | 937 |         Override this if you want to specify your own types. This version does | 
 | 938 |         a mapping using the handler's typemap attribute, which is set up in | 
 | 939 |         __init__() to a dictionary which contains mappings for DEBUG, INFO, | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 940 |         WARNING, ERROR and CRITICAL. If you are using your own levels you will | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 941 |         either need to override this method or place a suitable dictionary in | 
 | 942 |         the handler's typemap attribute. | 
 | 943 |         """ | 
 | 944 |         return self.typemap.get(record.levelno, self.deftype) | 
 | 945 |  | 
 | 946 |     def emit(self, record): | 
 | 947 |         """ | 
 | 948 |         Emit a record. | 
 | 949 |  | 
 | 950 |         Determine the message ID, event category and event type. Then | 
 | 951 |         log the message in the NT event log. | 
 | 952 |         """ | 
 | 953 |         if self._welu: | 
 | 954 |             try: | 
 | 955 |                 id = self.getMessageID(record) | 
 | 956 |                 cat = self.getEventCategory(record) | 
 | 957 |                 type = self.getEventType(record) | 
 | 958 |                 msg = self.format(record) | 
 | 959 |                 self._welu.ReportEvent(self.appname, id, cat, type, [msg]) | 
| Vinay Sajip | 245a5ab | 2005-10-31 14:27:01 +0000 | [diff] [blame] | 960 |             except (KeyboardInterrupt, SystemExit): | 
 | 961 |                 raise | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 962 |             except: | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 963 |                 self.handleError(record) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 964 |  | 
 | 965 |     def close(self): | 
 | 966 |         """ | 
 | 967 |         Clean up this handler. | 
 | 968 |  | 
 | 969 |         You can remove the application name from the registry as a | 
 | 970 |         source of event log entries. However, if you do this, you will | 
 | 971 |         not be able to see the events as you intended in the Event Log | 
 | 972 |         Viewer - it needs to be able to access the registry to get the | 
 | 973 |         DLL name. | 
 | 974 |         """ | 
 | 975 |         #self._welu.RemoveSourceFromRegistry(self.appname, self.logtype) | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 976 |         logging.Handler.close(self) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 977 |  | 
 | 978 | class HTTPHandler(logging.Handler): | 
 | 979 |     """ | 
 | 980 |     A class which sends records to a Web server, using either GET or | 
 | 981 |     POST semantics. | 
 | 982 |     """ | 
 | 983 |     def __init__(self, host, url, method="GET"): | 
 | 984 |         """ | 
 | 985 |         Initialize the instance with the host, the request URL, and the method | 
 | 986 |         ("GET" or "POST") | 
 | 987 |         """ | 
 | 988 |         logging.Handler.__init__(self) | 
| Neal Norwitz | 9d72bb4 | 2007-04-17 08:48:32 +0000 | [diff] [blame] | 989 |         method = method.upper() | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 990 |         if method not in ["GET", "POST"]: | 
| Collin Winter | ce36ad8 | 2007-08-30 01:19:48 +0000 | [diff] [blame] | 991 |             raise ValueError("method must be GET or POST") | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 992 |         self.host = host | 
 | 993 |         self.url = url | 
 | 994 |         self.method = method | 
 | 995 |  | 
| Neal Norwitz | f297bd1 | 2003-04-23 03:49:43 +0000 | [diff] [blame] | 996 |     def mapLogRecord(self, record): | 
 | 997 |         """ | 
 | 998 |         Default implementation of mapping the log record into a dict | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 999 |         that is sent as the CGI data. Overwrite in your class. | 
| Neal Norwitz | f297bd1 | 2003-04-23 03:49:43 +0000 | [diff] [blame] | 1000 |         Contributed by Franz  Glasner. | 
 | 1001 |         """ | 
 | 1002 |         return record.__dict__ | 
 | 1003 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1004 |     def emit(self, record): | 
 | 1005 |         """ | 
 | 1006 |         Emit a record. | 
 | 1007 |  | 
 | 1008 |         Send the record to the Web server as an URL-encoded dictionary | 
 | 1009 |         """ | 
 | 1010 |         try: | 
| Georg Brandl | 029986a | 2008-06-23 11:44:14 +0000 | [diff] [blame] | 1011 |             import http.client, urllib.parse | 
| Vinay Sajip | b793506 | 2005-10-11 13:15:31 +0000 | [diff] [blame] | 1012 |             host = self.host | 
| Georg Brandl | 2442015 | 2008-05-26 16:32:26 +0000 | [diff] [blame] | 1013 |             h = http.client.HTTP(host) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1014 |             url = self.url | 
| Georg Brandl | 029986a | 2008-06-23 11:44:14 +0000 | [diff] [blame] | 1015 |             data = urllib.parse.urlencode(self.mapLogRecord(record)) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1016 |             if self.method == "GET": | 
| Neal Norwitz | 9d72bb4 | 2007-04-17 08:48:32 +0000 | [diff] [blame] | 1017 |                 if (url.find('?') >= 0): | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1018 |                     sep = '&' | 
 | 1019 |                 else: | 
 | 1020 |                     sep = '?' | 
 | 1021 |                 url = url + "%c%s" % (sep, data) | 
 | 1022 |             h.putrequest(self.method, url) | 
| Vinay Sajip | b793506 | 2005-10-11 13:15:31 +0000 | [diff] [blame] | 1023 |             # support multiple hosts on one IP address... | 
 | 1024 |             # need to strip optional :port from host, if present | 
| Neal Norwitz | 9d72bb4 | 2007-04-17 08:48:32 +0000 | [diff] [blame] | 1025 |             i = host.find(":") | 
| Vinay Sajip | b793506 | 2005-10-11 13:15:31 +0000 | [diff] [blame] | 1026 |             if i >= 0: | 
 | 1027 |                 host = host[:i] | 
 | 1028 |             h.putheader("Host", host) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1029 |             if self.method == "POST": | 
| Vinay Sajip | b793506 | 2005-10-11 13:15:31 +0000 | [diff] [blame] | 1030 |                 h.putheader("Content-type", | 
 | 1031 |                             "application/x-www-form-urlencoded") | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1032 |                 h.putheader("Content-length", str(len(data))) | 
| Benjamin Peterson | 458ad47 | 2009-01-18 00:08:45 +0000 | [diff] [blame] | 1033 |             h.endheaders(data if self.method == "POST" else None) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1034 |             h.getreply()    #can't do anything with the result | 
| Vinay Sajip | 245a5ab | 2005-10-31 14:27:01 +0000 | [diff] [blame] | 1035 |         except (KeyboardInterrupt, SystemExit): | 
 | 1036 |             raise | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1037 |         except: | 
| Neal Norwitz | 6fa635d | 2003-02-18 14:20:07 +0000 | [diff] [blame] | 1038 |             self.handleError(record) | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1039 |  | 
 | 1040 | class BufferingHandler(logging.Handler): | 
 | 1041 |     """ | 
 | 1042 |   A handler class which buffers logging records in memory. Whenever each | 
 | 1043 |   record is added to the buffer, a check is made to see if the buffer should | 
 | 1044 |   be flushed. If it should, then flush() is expected to do what's needed. | 
 | 1045 |     """ | 
 | 1046 |     def __init__(self, capacity): | 
 | 1047 |         """ | 
 | 1048 |         Initialize the handler with the buffer size. | 
 | 1049 |         """ | 
 | 1050 |         logging.Handler.__init__(self) | 
 | 1051 |         self.capacity = capacity | 
 | 1052 |         self.buffer = [] | 
 | 1053 |  | 
 | 1054 |     def shouldFlush(self, record): | 
 | 1055 |         """ | 
 | 1056 |         Should the handler flush its buffer? | 
 | 1057 |  | 
 | 1058 |         Returns true if the buffer is up to capacity. This method can be | 
 | 1059 |         overridden to implement custom flushing strategies. | 
 | 1060 |         """ | 
 | 1061 |         return (len(self.buffer) >= self.capacity) | 
 | 1062 |  | 
 | 1063 |     def emit(self, record): | 
 | 1064 |         """ | 
 | 1065 |         Emit a record. | 
 | 1066 |  | 
 | 1067 |         Append the record. If shouldFlush() tells us to, call flush() to process | 
 | 1068 |         the buffer. | 
 | 1069 |         """ | 
 | 1070 |         self.buffer.append(record) | 
 | 1071 |         if self.shouldFlush(record): | 
 | 1072 |             self.flush() | 
 | 1073 |  | 
 | 1074 |     def flush(self): | 
 | 1075 |         """ | 
 | 1076 |         Override to implement custom flushing behaviour. | 
 | 1077 |  | 
 | 1078 |         This version just zaps the buffer to empty. | 
 | 1079 |         """ | 
 | 1080 |         self.buffer = [] | 
 | 1081 |  | 
| Vinay Sajip | f42d95e | 2004-02-21 22:14:34 +0000 | [diff] [blame] | 1082 |     def close(self): | 
 | 1083 |         """ | 
 | 1084 |         Close the handler. | 
 | 1085 |  | 
 | 1086 |         This version just flushes and chains to the parent class' close(). | 
 | 1087 |         """ | 
 | 1088 |         self.flush() | 
 | 1089 |         logging.Handler.close(self) | 
 | 1090 |  | 
| Guido van Rossum | 57102f8 | 2002-11-13 16:15:58 +0000 | [diff] [blame] | 1091 | class MemoryHandler(BufferingHandler): | 
 | 1092 |     """ | 
 | 1093 |     A handler class which buffers logging records in memory, periodically | 
 | 1094 |     flushing them to a target handler. Flushing occurs whenever the buffer | 
 | 1095 |     is full, or when an event of a certain severity or greater is seen. | 
 | 1096 |     """ | 
 | 1097 |     def __init__(self, capacity, flushLevel=logging.ERROR, target=None): | 
 | 1098 |         """ | 
 | 1099 |         Initialize the handler with the buffer size, the level at which | 
 | 1100 |         flushing should occur and an optional target. | 
 | 1101 |  | 
 | 1102 |         Note that without a target being set either here or via setTarget(), | 
 | 1103 |         a MemoryHandler is no use to anyone! | 
 | 1104 |         """ | 
 | 1105 |         BufferingHandler.__init__(self, capacity) | 
 | 1106 |         self.flushLevel = flushLevel | 
 | 1107 |         self.target = target | 
 | 1108 |  | 
 | 1109 |     def shouldFlush(self, record): | 
 | 1110 |         """ | 
 | 1111 |         Check for buffer full or a record at the flushLevel or higher. | 
 | 1112 |         """ | 
 | 1113 |         return (len(self.buffer) >= self.capacity) or \ | 
 | 1114 |                 (record.levelno >= self.flushLevel) | 
 | 1115 |  | 
 | 1116 |     def setTarget(self, target): | 
 | 1117 |         """ | 
 | 1118 |         Set the target handler for this handler. | 
 | 1119 |         """ | 
 | 1120 |         self.target = target | 
 | 1121 |  | 
 | 1122 |     def flush(self): | 
 | 1123 |         """ | 
 | 1124 |         For a MemoryHandler, flushing means just sending the buffered | 
 | 1125 |         records to the target, if there is one. Override if you want | 
 | 1126 |         different behaviour. | 
 | 1127 |         """ | 
 | 1128 |         if self.target: | 
 | 1129 |             for record in self.buffer: | 
 | 1130 |                 self.target.handle(record) | 
 | 1131 |             self.buffer = [] | 
 | 1132 |  | 
 | 1133 |     def close(self): | 
 | 1134 |         """ | 
 | 1135 |         Flush, set the target to None and lose the buffer. | 
 | 1136 |         """ | 
 | 1137 |         self.flush() | 
 | 1138 |         self.target = None | 
| Vinay Sajip | 48cfe38 | 2004-02-20 13:17:27 +0000 | [diff] [blame] | 1139 |         BufferingHandler.close(self) |