ProxyManager.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__ = "Manages diagnostic, telemetry and science proxies"
__author__   = "<perazzo@slac.stanford.edu>"
__updated__  = "$Date: 2006/03/16 23:55:34 $"
__version__  = "$Revision: 1.3 $"
__release__  = "$Name: HEAD $"
__credits__  = "SLAC"

import sys
import os
import signal

from qt import QObject, QApplication, SIGNAL, SLOT

from ProxyManagerGui import ProxyManagerGui
from LICOS.util.gOptions import Options

class ProxyParams(object):
  def __init__(self, label, button, script):
    self.label = label
    self.button = button
    self.script = script
    self.pid = None

class ProxyManager(ProxyManagerGui):
  def __init__(self, app, vscip, config):
    ProxyManagerGui.__init__(self)
    self.setCaption("Proxy Manager")
    self.invscip.setText(vscip)
    self.inconfig.setText(config)
    self.__proxies = {
      'diagnostic': ProxyParams(self.status_dia, self.start_dia,
                                'telemetry/DiagnosticProxy.py'),
      'telemetry':  ProxyParams(self.status_tlm, self.start_tlm,
                                'telemetry/TelemetryProxy.py'),
      'science':    ProxyParams(self.status_sci, self.start_sci,
                                'science/ScienceProxy.py')
      }
    self.startTimer(5000)
    self.__app = app
    
  def Exit(self):
    self.StopClicked()
    self.__app.quit()

  def StartClicked(self):
    for proxy in self.__proxies.values():
      if proxy.pid is None:
        self.__start(proxy)

  def StopClicked(self):
    for proxy in self.__proxies.values():
      if proxy.pid is not None:
        self.__stop(proxy)

  def DiaClicked(self):
    self.__clicked('diagnostic')

  def TlmClicked(self):
    self.__clicked('telemetry')

  def SciClicked(self):
    self.__clicked('science')

  def __clicked(self, which):
    proxy = self.__proxies[which]
    if proxy.pid is None:
      self.__start(proxy)
    else:
      self.__stop(proxy)

  def __start(self, proxy):
    command = os.path.expandvars('$ONLINE_PACKAGES/bin/python')
    args = (os.path.expandvars('$ONLINE_ROOT/LICOS/%s'%(proxy.script)),
            '--vscip', self.invscip.text().ascii(),
            '--config', os.path.expandvars(self.inconfig.text().ascii()),
            '--logtime')
    proxy.pid = os.spawnv(os.P_NOWAIT, command, (command,)+args)
    proxy.label.setText('<font color=blue>running</font>')
    proxy.button.setText('Stop')
    proxy.button.setDown(True)

  def __stop(self, proxy):
    os.kill(proxy.pid, signal.SIGKILL)
    try:
      os.waitpid(proxy.pid, 0)
    except:
      pass
    proxy.pid = None
    proxy.label.setText('<font color=black>stopped</font>')
    proxy.button.setText('Start')
    proxy.button.setDown(False)

  def timerEvent(self, event):
    for proxy in self.__proxies.values():
      if proxy.pid is not None:
        (pid, status) = os.waitpid(proxy.pid, os.WNOHANG)
        # signum = status & 0xff
        # exitstatus = status >> 8
        if pid != 0:
          proxy.pid = None
          proxy.label.setText('<font color=red>crashed</font>')
          proxy.button.setText('Start')
          proxy.button.setDown(False)
        

if __name__ == '__main__':
  options = Options(['vscip', 'config'], [], [])
  try:
    options.parse()
  except Exception, msg:
    options.usage(str(msg))
    sys.exit()

  app = QApplication(sys.argv)
  app.connect(app, SIGNAL("lastWindowClosed()"), app, SLOT("quit()"))
  win = ProxyManager(app, options.vscip, options.config)
  win.show()
  app.exec_loop()

---