HskReceiver.py

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

__facility__ = "Online"
__abstract__ = "GLAST LAT Housekeeping server infrastructure"
__author__   = "Jim Panetta <panetta@slac.stanford.edu> SLAC - GLAST I&T"
__date__     = "07/28/04"
__version__  = "$Revision: 1.9 $"
__release__  = "$Name: R04-12-00 $"
__credits__  = "SLAC"

import LATTE.copyright_SLAC

from LATTE.monitoring.Receiver       import Receiver
from LATTE.client.gOptions           import Options

import logging
from linecache import *
import time

second  = 1
minute  = 60 * second
hour    = minute * 60
day     = hour * 24
era     = 10**9

timeFmt = '%Y/%m/%d  %H:%M:%S'
dataFile     = None
timeDepth    = 2*hour / 5

class HskSimpleTextReceiver(Receiver):
  """
  """
  def __init__(self, flatFile, startTime):
    Receiver.__init__(self)
    self.log     = logging.getLogger('HskSimpleTextReceiver')
    self.__flatFile   = flatFile
    self.__keyList    = []
    self.__lineNumber = 1
    self.__startTime  = startTime

  def shutdown(self):
    self.setReceiving(0)

  def receive(self):
    if self.__lineNumber == 1:
      checkcache()
      keyList = getline(self.__flatFile, self.__lineNumber)
      if keyList == '': time.sleep(0.1); return
      # create a list of keys from whitespace separated text blobs
      self.__keyList = keyList.split()
      self.__lineNumber += 1
    while True:
      line = getline(self.__flatFile, self.__lineNumber)
      # print line
      if line == '': 
        checkcache()
        break
      # if self.__lineNumber % 100 == 3: 
        # from HskGuiReceiver import memget
        # print  memget()
      self.__lineNumber += 1
      trendMap = {}
      values = line.split()
      # first key is 'TimeStamp', so this is always safe:
      if float(values[0]) < self.__startTime:
        continue
      for key in self.__keyList:
        v = values[self.__keyList.index(key)]
        if v == 'None':    # No value in table.
          trendMap[key] = None
        else:
          trendMap[key] = float(values[self.__keyList.index(key)])
      self.dispatch(trendMap)
    time.sleep(10)

  def dispatch(self, trendMap):
    if trendMap == {}: return
    print time.strftime(timeFmt, time.localtime(trendMap['TimeStamp'])), '    ' \
          'DAQ TEMP:', trendMap['LHKAEMDAQBT']

class HskCfgTextReceiver(HskSimpleTextReceiver):
  def __init__(self, flatFile, startTime, trendList):
    HskSimpleTextReceiver.__init__(self, flatFile, startTime)
    self.__trendList = trendList
    
  def dispatch(self, trendMap):
    if trendMap == {}: return
    dispatchMap = {'TimeStamp':trendMap['TimeStamp']}
    for item in trendMap:
      if self.trendItem(item):
        dispatchMap[item] = trendMap[item]
    self.process(dispatchMap)
    
  def process(self, dispatchMap):
    line = ''
    for item in dispatchMap:
      line += item + ":" + str(dispatchMap[item]) + "    "
    # print line
    
  def trendItem(self, item):
    if item in self.__trendList:
      return True
    else:
      return False

def parseKeyFile(keyFile):
  keyList = {}
  f       = open(keyFile, 'r')
  for line in f.readlines():
    line.strip()
    if line[0] != '#' and line != '':
      (key,title)  = line.split(':', 1)
      keyList[key.strip()] = title.strip()
  f.close()
  return keyList

def startup():
  logging.basicConfig()
  logging.root.setLevel(logging.DEBUG)
  
def usage():
  pass

if __name__ == '__main__':
  startup()

  # Options, first list mandatory, second list optional 
  options = Options(['dataFile'], ['timeDepth', 'keyFile'])
  try:
    options.parse()
  except Exception, msg:
    options.usage(usage())
    raise Exception, msg
  
  dataFile = options.dataFile
  if options.timeDepth is not None:
    timeDepth = int(options.timeDepth)
  else:
    timeDepth = 2*hour
  startTime = time.time() - timeDepth

  if options.keyFile is not None:
    keyFile = options.keyFile
    keyList = parseKeyFile(keyFile)
  else:  # really really basic set of histos.
    keyList = {'LHKAEMDAQB33V':'AEM DAQ board 3.3 V', 
               'LHKAEMFR11T'  :'AEM Free board 11 Temp',
               'LHKAEMDAQBT'  :'AEM DAQ board Temp', 
               'LHKAEMFR11V1' :'AEM Free 11 HV 1',
               'LHKAEMFR11V2' :'AEM Free 11 HV 2',
               'LHKAEMFR11VD' :'AEM Free 11 VDD',
               }
  
  # r = HskSimpleTextReceiver(dataFile, startTime)
  r = HskCfgTextReceiver(dataFile, startTime, keyList.keys())
  
  
  r.setReceiving(1)
  
  try:
    while 1:
      time.sleep(1)
  except Exception, e:
    r.shutdown()

---