src/python/osrf/log.py

   1 # -----------------------------------------------------------------------
   2 # Copyright (C) 2007  Georgia Public Library Service
   3 # Bill Erickson <billserickson@gmail.com>
   4 #
   5 # This program is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU General Public License
   7 # as published by the Free Software Foundation; either version 2
   8 # of the License, or (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 # -----------------------------------------------------------------------
  15
  16 import traceback, sys, os, re
  17 from osrf.const import *
  18
  19 loglevel = 4
  20
  21 def osrfInitLog(level, facility=None, file=None):
  22     """Initialize the logging subsystem."""
  23     import syslog
  24     global loglevel
  25     if facility:
  26         osrfInitSyslog(facility, level)
  27         syslog.syslog(syslog.LOG_DEBUG, "syslog initialized")
  28     else:
  29         if file:
  30             sys.stderr.write("\n * file-based logging not implemented yet\n")
  31
  32     loglevel = level
  33
  34
  35 # -----------------------------------------------------------------------
  36 # Define wrapper functions for the log levels
  37 # -----------------------------------------------------------------------
  38 def osrfLogInternal(s): __osrfLog(OSRF_LOG_INTERNAL,s)
  39 def osrfLogDebug(s): __osrfLog(OSRF_LOG_DEBUG,s)
  40 def osrfLogInfo(s): __osrfLog(OSRF_LOG_INFO,s)
  41 def osrfLogWarn(s): __osrfLog(OSRF_LOG_WARN,s)
  42 def osrfLogErr(s): __osrfLog(OSRF_LOG_ERR,s)
  43
  44
  45 frgx = re.compile('/.*/')
  46
  47 def __osrfLog(level, msg):
  48     """Builds the log message and passes the message off to the logger."""
  49
  50     try:
  51         import syslog
  52     except:
  53         return
  54
  55     global loglevel
  56     if int(level) > int(loglevel): return
  57
  58     # find the caller info for logging the file and line number
  59     tb = traceback.extract_stack(limit=3)
  60     tb = tb[0]
  61     lvl = 'DEBG'
  62     slvl = syslog.LOG_DEBUG
  63
  64     if level == OSRF_LOG_INTERNAL: lvl = 'INT '; slvl=syslog.LOG_DEBUG
  65     if level == OSRF_LOG_INFO: lvl = 'INFO'; slvl=syslog.LOG_INFO
  66     if level == OSRF_LOG_WARN: lvl = 'WARN'; slvl=syslog.LOG_WARNING
  67     if level == OSRF_LOG_ERR:  lvl = 'ERR '; slvl=syslog.LOG_ERR
  68
  69     file = frgx.sub('',tb[0])
  70     msg = '[%s:%d:%s:%s] %s' % (lvl, os.getpid(), file, tb[1], msg)
  71     syslog.syslog(slvl, msg)
  72
  73     if level == OSRF_LOG_ERR:
  74         sys.stderr.write(msg + '\n')
  75
  76
  77 def osrfInitSyslog(facility, level):
  78     """Connect to syslog and set the logmask based on the level provided."""
  79
  80     import syslog
  81
  82     level = int(level)
  83
  84     if facility == 'local0': facility = syslog.LOG_LOCAL0
  85     if facility == 'local1': facility = syslog.LOG_LOCAL1
  86     if facility == 'local2': facility = syslog.LOG_LOCAL2
  87     if facility == 'local3': facility = syslog.LOG_LOCAL3
  88     if facility == 'local4': facility = syslog.LOG_LOCAL4
  89     if facility == 'local5': facility = syslog.LOG_LOCAL5
  90     if facility == 'local6': facility = syslog.LOG_LOCAL6
  91     # XXX add other facility maps if necessary
  92     syslog.openlog(sys.argv[0], 0, facility)
  93
  94     # this is redundant...
  95     mask = syslog.LOG_UPTO(syslog.LOG_ERR)
  96     if level >= 1: mask |= syslog.LOG_MASK(syslog.LOG_WARNING)
  97     if level >= 2: mask |= syslog.LOG_MASK(syslog.LOG_NOTICE)
  98     if level >= 3: mask |= syslog.LOG_MASK(syslog.LOG_INFO)
  99     if level >= 4: mask |= syslog.LOG_MASK(syslog.LOG_DEBUG)
 100     syslog.setlogmask(mask)
 101