socketLogging.py

Go to the documentation of this file.

#!/usr/local/bin/python
#
#                               Copyright 2005
#                                     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__     = "2005/07/23 00:08:27"
__updated__  = "$Date: 2005/09/22 00:29:30 $"
__version__  = "$Revision: 1.4 $"
__release__  = "$Name: HEAD $"
__credits__  = "SLAC"

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

class MyLogSocketHandler(logging.handlers.SocketHandler):
  """!\brief Socket log handler for alerts

  """
  def __init__(self, *args):
    """!\brief MyLogSocketHandler constructor

    \param *args Arguments to MyLogSocketHandler constructor
    """
    apply(logging.handlers.SocketHandler.__init__, (self,) + args)

  def handleError(self, record):
    """!\brief Error handler

    \param record Log 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):
    """!\brief createSocket

    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):
  """!\brief Setup logging for alerts.

  \param logHost  Log host
  \param logLevel Log level
  \param port     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')

---