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         env = QProcess.systemEnvironment()
  24         env << "PYTHONPATH=%s" % ":".join(sys.path)
  25         self.process = QProcess()
  26         self.process.setEnvironment(env)
  27         self.connect(self.process, SIGNAL("finished(int, QProcess::ExitStatus)"),
  28                      self.processFinished)
  29         self.connect(self.process, SIGNAL("readyReadStandardOutput()"),
  30                      self.processStandardOutput)
  31         self.connect(self.process, SIGNAL("readyReadStandardError()"),
  32                      self.processStandardError)
  33
  34     def __init_command(self, args):
  35         self.args = QStringList()
  36         for arg in args:
  37             self.args << QString(arg)
  38
  39         self.exe = find_executable("sfi.py")
  40         if not self.exe:
  41             print "FATAL.. Could not locate binary sfi.py - not much we can do without that"
  42
  43     def isRunning(self):
  44         return self.process.state() != QProcess.NotRunning
  45
  46     def processStandardOutput(self):
  47         output = self.process.readAllStandardOutput()
  48         if config.debug:
  49             print output
  50
  51     def processStandardError(self):
  52         print self.process.readAllStandardError()
  53
  54     def processFinished(self):
  55         if self.process.exitStatus() == QProcess.CrashExit:
  56             print self.readOutput()
  57             print "Process exited with errors:",
  58             err = self.process.error()
  59             if err == QProcess.FailedToStart:
  60                 print "FailedToStart"
  61             elif err == QProcess.Crashed:
  62                 print "Crashed"
  63             elif err == QProcess.Timedout:
  64                 print "Timedout"
  65             elif err == QProcess.WriteError:
  66                 print "WriteError"
  67             elif err == QProcess.ReadError:
  68                 print "ReadError"
  69             elif err == QProcess.UnknownError:
  70                 print "UnknownError"
  71         self.trace_end()
  72         self.emit(SIGNAL("finished()"))
  73
  74     def __getRSpec(self, mgr):
  75         slice = config.getSlice()
  76         # Write RSpec to file for testing.
  77         filename = os.path.expanduser("~/.sfi/" + slice + ".rspec")
  78         try:
  79             os.remove(filename)
  80         except:
  81             pass
  82         args = ["-u", config.getUser(), "-a", config.getAuthority(),
  83                 "-r", config.getRegistry(), "-s", mgr, "resources",
  84                 "-o", filename, slice]
  85
  86         self.__init_command(args)
  87         self.start()
  88         return filename
  89
  90     def getRSpecFromSM(self):
  91         return self.__getRSpec(config.getSlicemgr())
  92
  93     def getRSpecFromAM(self):
  94         return self.__getRSpec(config.getAggmgr())
  95
  96     def getRecord(self, hrn):
  97         args = ["-u", config.getUser(), "-a", config.getAuthority(),
  98                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "show", hrn]
  99         self.__init_command(args)
 100         self.start()
 101
 102     def applyRSpec(self, rspec):
 103         filename = config.getSliceRSpecFile() + "_new"
 104         rspec.save(filename)
 105         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 106                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "create",
 107                 config.getSlice(), filename]
 108         self.__init_command(args)
 109         self.start()
 110         return filename
 111
 112     def start(self):
 113         self.trace_command()
 114         self.process.start(self.exe, self.args)
 115
 116     def readOutput(self):
 117         if self.process.state() == QProcess.NotRunning:
 118             return self.process.readAll()
 119
 120     def trace_command (self):
 121         if config.verbose:
 122             self._trace=time.time()
 123             command = "%s %s" % (self.exe, self.args.join(" "))
 124             print time.strftime('%M:%S'),'Invoking',command
 125
 126     def trace_end (self):
 127         if config.verbose:
 128             command = "%s %s" % (self.exe, self.args.join(" "))
 129             print time.strftime('%M:%S'),"[%.3f s]"%(time.time()-self._trace),command,'Done'