socketLogging.py

#!/usr/local/bin/python
#
#                               Copyright 2002
#                                     by
#                        The Board of Trustees of the
#                     Leland Stanford Junior University.
#                            All rights reserved.
#

__facility__ = "Online"
__abstract__ = "GLAST LAT Alert System GUI"
__author__   = "J. Panetta"
__date__     = "28/01/2005"
__version__  = "$Revision: 1.2 $"
__credits__  = "SLAC"

import LATTE.copyright_SLAC
import logging
import logging.handlers
import sys
ALARM_TCP_LOGGING_PORT    = 9099

class MyLogSocketHandler(logging.handlers.SocketHandler):
  def __init__(self, *args):
    apply(logging.handlers.SocketHandler.__init__, (self,) + args)

  def handleError(self, record):
    if self.closeOnError and self.sock:
      self.sock.shutdown(2)
      self.sock.close()
      self.sock = None        #try to reconnect next time
    else:
      raise IOError, sys.exc_info()[1]
      
  def createSocket(self):
    """Override base classes createSocket method.  
       This isn't allowed to fail silently
    """
    self.sock = self.makeSocket()
    
def socketLogging(logHost, logLevel, port=ALARM_TCP_LOGGING_PORT):
   
  logging.root.setLevel(logging.__dict__[logLevel])
  
  # set up the socket handler for this logger
  sockHdlr   = MyLogSocketHandler(logHost, port)

  fmt = logging.Formatter("%(asctime)s %(filename)s:%(lineno)d %(levelname)-5s - %(message)s")
  sockHdlr.setFormatter(fmt)
  logging.getLogger().addHandler(sockHdlr)
  
  try:
    logging.info('Trying to connect to the message logger')
  except Exception, e:
    logging.getLogger("").removeHandler(sockHdlr)
    logging.fatal("Can't connect to the message logger, Aborting job")
    raise
  else:
    logging.info('Successfully connected to the message logger')
  

---