sface/sfiprocess.py

   1
   2 import os
   3 import sys
   4 import time
   5
   6 from PyQt4.QtCore import *
   7 from sface.config import config
   8
   9 def find_executable(exec_name):
  10     """find the given executable in $PATH"""
  11     paths = os.getenv("PATH").split(':')
  12     for p in paths:
  13         exec_path = os.path.join(p, exec_name)
  14         if os.path.exists(exec_path):
  15             return exec_path
  16     return None
  17
  18
  19 class SfiProcess(QObject):
  20     def __init__(self, parent=None):
  21         QObject.__init__(self, parent)
  22
  23         self.process = QProcess()
  24         self.connect(self.process, SIGNAL("finished(int, QProcess::ExitStatus)"),
  25                      self.processFinished)
  26
  27     def __init_command(self, args):
  28         self.args = QStringList()
  29         for arg in args:
  30             self.args << QString(arg)
  31
  32         self.exe = find_executable("sfi.py")
  33         if not self.exe:
  34             print "FATAL.. Could not locate binary sfi.py - not much we can do without that"
  35
  36     def isRunning(self):
  37         return self.process.state() != QProcess.NotRunning
  38
  39     def processFinished(self):
  40         if self.process.exitStatus() == QProcess.CrashExit:
  41             print self.readOutput()
  42             print "Process exited with errors:",
  43             err = self.process.error()
  44             if err == QProcess.FailedToStart:
  45                 print "FailedToStart"
  46             elif err == QProcess.Crashed:
  47                 print "Crashed"
  48             elif err == QProcess.Timedout:
  49                 print "Timedout"
  50             elif err == QProcess.WriteError:
  51                 print "WriteError"
  52             elif err == QProcess.ReadError:
  53                 print "ReadError"
  54             elif err == QProcess.UnknownError:
  55                 print "UnknownError"
  56         self.trace_end()
  57         self.emit(SIGNAL("finished()"))
  58
  59     def __getRSpec(self, mgr):
  60         slice = config.getSlice()
  61         # Write RSpec to file for testing.
  62         filename = os.path.expanduser("~/.sfi/" + slice + ".rspec")
  63         try:
  64             os.remove(filename)
  65         except:
  66             pass
  67         args = ["-u", config.getUser(), "-a", config.getAuthority(),
  68                 "-r", config.getRegistry(), "-s", mgr, "resources",
  69                 "-o", filename, slice]
  70
  71         self.__init_command(args)
  72         self.start()
  73         return filename
  74
  75     def getRSpecFromSM(self):
  76         return self.__getRSpec(config.getSlicemgr())
  77
  78     def getRSpecFromAM(self):
  79         return self.__getRSpec(config.getAggmgr())
  80
  81     def getRecord(self, hrn):
  82         args = ["-u", config.getUser(), "-a", config.getAuthority(),
  83                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "show", hrn]
  84         self.__init_command(args)
  85         self.start()
  86
  87     def applyRSpec(self, rspec):
  88         filename = config.getSliceRSpecFile() + "_new"
  89         rspec.save(filename)
  90         args = ["-u", config.getUser(), "-a", config.getAuthority(),
  91                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "create",
  92                 config.getSlice(), filename]
  93         self.__init_command(args)
  94         self.start()
  95         return filename
  96
  97     def start(self):
  98         self.trace_command()
  99         self.process.start(self.exe, self.args)
 100
 101     def readOutput(self):
 102         if self.process.state() == QProcess.NotRunning:
 103             return self.process.readAll()
 104
 105     def trace_command (self):
 106         if config.verbose:
 107             self._trace=time.time()
 108             command = "%s %s" % (self.exe, self.args.join(" "))
 109             print time.strftime('%M:%S'),'Invoking',command
 110
 111     def trace_end (self):
 112         if config.verbose:
 113             command = "%s %s" % (self.exe, self.args.join(" "))
 114             print time.strftime('%M:%S'),"[%.3f s]"%(time.time()-self._trace),command,'Done'