sface/sfiprocess.py

   1
   2 import os
   3 import pickle
   4 import sys
   5 import time
   6
   7 from PyQt4.QtCore import *
   8 from sface.config import config
   9 from sface.xmlrpcwindow import get_tracker, XmlrpcReader
  10
  11 def find_executable(exec_name):
  12     """find the given executable in $PATH"""
  13     paths = os.getenv("PATH").split(':')
  14     for p in paths:
  15         exec_path = os.path.join(p, exec_name)
  16         if os.path.exists(exec_path):
  17             return exec_path
  18     return None
  19
  20
  21 class SfiProcess(QObject):
  22     def __init__(self, parent=None):
  23         QObject.__init__(self, parent)
  24
  25         env = QProcess.systemEnvironment()
  26         env << "PYTHONPATH=%s" % ":".join(sys.path)
  27         self.process = QProcess()
  28         self.process.setEnvironment(env)
  29         self.connect(self.process, SIGNAL("finished(int, QProcess::ExitStatus)"),
  30                      self.processFinished)
  31
  32         self.xmlrpcreader = XmlrpcReader() # this one is for parsing XMLRPC responses
  33
  34         # holds aggregate output from processStandardOutput(); used by xmlrpc
  35         # tracker.
  36         self.output = ""
  37
  38         self.connect(self.process, SIGNAL("readyReadStandardOutput()"),
  39                      self.processStandardOutput)
  40         self.connect(self.process, SIGNAL("readyReadStandardError()"),
  41                      self.processStandardError)
  42
  43     def __init_command(self, args):
  44         self.args = QStringList()
  45         self.args << "-d"
  46         self.args << config.get_dirname()
  47
  48         # this shows xmlrpc conversation, see sfi.py docs.
  49         # always do this, so we can parse the XML result for faults and show
  50         # then to the users.
  51         self.args << QString('-D')
  52
  53         for arg in args:
  54             self.args << QString(arg)
  55
  56         self.exe = find_executable("sfi.py")
  57         if not self.exe:
  58             print "FATAL.. Could not locate binary sfi.py - not much we can do without that"
  59
  60     def isRunning(self):
  61         return self.process.state() != QProcess.NotRunning
  62
  63     def processStandardOutput(self):
  64         output = self.process.readAllStandardOutput()
  65         self.output = self.output + output
  66         if config.debug:
  67             try:
  68                 print output
  69             except IOError, e:
  70                 if (e.errno == 4):
  71                     # XXX why is this happening??
  72                     print "*** caught EINTR"
  73                 else:
  74                     raise
  75
  76
  77     def processStandardError(self):
  78         print self.process.readAllStandardError()
  79
  80     def processFinished(self):
  81         if self.process.exitStatus() == QProcess.CrashExit:
  82             print self.readOutput()
  83             print "Process exited with errors:",
  84             err = self.process.error()
  85             if err == QProcess.FailedToStart:
  86                 print "FailedToStart"
  87             elif err == QProcess.Crashed:
  88                 print "Crashed"
  89             elif err == QProcess.Timedout:
  90                 print "Timedout"
  91             elif err == QProcess.WriteError:
  92                 print "WriteError"
  93             elif err == QProcess.ReadError:
  94                 print "ReadError"
  95             elif err == QProcess.UnknownError:
  96                 print "UnknownError"
  97
  98         # extract any faults from the XMLRPC response(s)
  99         self.xmlrpcreader.responses = []
 100         self.xmlrpcreader.store(self.output)
 101         self.xmlrpcreader.extractXml()
 102         self.responses = self.xmlrpcreader.responses
 103         self.faults = [x for x in self.responses if (x["kind"]=="fault")]
 104
 105         self.trace_end()
 106         self.emit(SIGNAL("finished()"))
 107
 108     def getFaultString(self):
 109         if self.faults == []:
 110             return None
 111
 112         return self.faults[0].get("faultString","") + " (" + self.faults[0].get("faultCode","") + ")"
 113
 114     def retrieveRspec(self):
 115         slice = config.getSlice()
 116         mgr = config.getSlicemgr()
 117         # Write RSpec to file
 118         filename = config.fullpath ("%s.rspec"%slice)
 119         try:
 120             os.remove(filename)
 121         except:
 122             pass
 123         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 124                 "-r", config.getRegistry(), "-s", mgr, "resources",
 125                 "-o", filename, slice]
 126
 127         self.__init_command(args)
 128         self.start()
 129         return filename
 130
 131     def retrieveResources(self):
 132         mgr = config.getSlicemgr()
 133         # Write RSpec to file
 134         filename = config.getResourcesRSpecFile()
 135         try:
 136             os.remove(filename)
 137         except:
 138             pass
 139         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 140                 "-r", config.getRegistry(), "-s", mgr, "resources",
 141                 "-o", filename]
 142
 143         self.__init_command(args)
 144         self.start()
 145         return filename
 146
 147
 148     def listRecords(self, hrn, rectype=None, filename=None):
 149         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 150                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "list", "-F", "xmllist", hrn]
 151
 152         if not filename:
 153             filename = config.getAuthorityListFile()
 154
 155         # we can't tell whether SFI will create one file or many, so delete
 156         # leftovers from last time, then we'll know what we got, after we get it.
 157         if os.path.exists(filename):
 158             os.remove(filename)
 159         if os.path.exists(filename + ".1"):
 160             os.remove(filename + ".1")
 161         args.append("-o")
 162         args.append(filename)
 163
 164         if rectype:
 165             args.append("-t")
 166             args.append(rectype)
 167
 168         self.__init_command(args)
 169         self.start()
 170
 171     def getRecord(self, hrn, filename=None):
 172         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 173                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "show", hrn]
 174         if filename:
 175             args.append("-o")
 176             args.append(filename)
 177         self.__init_command(args)
 178         self.start()
 179
 180     def getSliceRecord(self):
 181         self.getRecord(config.getSlice(), config.getSliceRecordFile())
 182
 183     def getAuthorityRecord(self):
 184         self.getRecord(config.getAuthority(), config.getAuthorityRecordFile())
 185
 186     def applyRSpec(self, rspec, aggAddr=None, aggPort=None, saveObtained=True):
 187         # that's what we pass, like in what we'd like to get
 188         requested = config.getSliceRSpecFile() + "_new"
 189         # that's what we actually receive
 190         # just overwrite the slice file as if we'd used 'resources'
 191         obtained = config.getSliceRSpecFile()
 192         rspec.save(requested)
 193         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 194                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "create"]
 195
 196         if saveObtained:
 197             args = args + ["-o", obtained]
 198
 199         if aggAddr:
 200             args = args + ["-a", aggAddr, "-p", str(aggPort)]
 201
 202         args = args + [config.getSlice(), requested]
 203
 204         self.__init_command(args)
 205         self.start()
 206
 207     def deleteSlivers(self, slice=None, aggAddr=None, aggPort=None):
 208         if not slice:
 209             slice = config.getSlice()
 210
 211         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 212                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "delete"]
 213
 214         if aggAddr:
 215             args = args + ["-a", aggAddr, "-p", str(aggPort)]
 216
 217         args = args + [slice]
 218
 219         self.__init_command(args)
 220         self.start()
 221
 222     def updateRecord(self, filename):
 223         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 224                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "update", filename]
 225         self.__init_command(args)
 226         self.start()
 227
 228     def addRecord(self, filename):
 229         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 230                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "add", filename]
 231         self.__init_command(args)
 232         self.start()
 233
 234     def removeRecord(self, hrn):
 235         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 236                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "remove", hrn]
 237         self.__init_command(args)
 238         self.start()
 239
 240     def renewSlivers(self, expiration, slice=None):
 241         if not slice:
 242             slice = config.getSlice()
 243
 244         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 245                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "renew",
 246                 slice, expiration]
 247         self.__init_command(args)
 248         self.start()
 249
 250     def sliverStatus(self, slice=None, filename=None):
 251         if not slice:
 252             slice = config.getSlice()
 253
 254         if not filename:
 255             filename = config.fullpath(slice+".sliverstatus")
 256
 257         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 258                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "status",
 259                 "-o", filename, "-F", "pickled", slice]
 260         self.__init_command(args)
 261         self.start()
 262
 263     def getSliceMgrVersion(self, filename=None):
 264         if not filename:
 265             filename = config.fullpath("slicemgr.version")
 266
 267         args = ["-u", config.getUser(), "-a", config.getAuthority(),
 268                 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "version",
 269                 "-o", filename, "-F", "pickled",]
 270         self.__init_command(args)
 271         self.start()
 272
 273     def start(self):
 274         self.respones = []
 275         self.faults = []
 276         self.output = ""
 277         self.trace_command()
 278         self.process.start(self.exe, self.args)
 279
 280     def readOutput(self):
 281         if self.process.state() == QProcess.NotRunning:
 282             return self.process.readAll()
 283
 284     def trace_command (self):
 285         if config.verbose:
 286             self._trace=time.time()
 287             command = "%s %s" % (self.exe, self.args.join(" "))
 288             print time.strftime('%H:%M:%S'),'Invoking',command
 289
 290     def trace_end (self):
 291         if config.verbose:
 292 #            command = "%s %s" % (self.exe, self.args.join(" "))
 293             print time.strftime('%H:%M:%S'),"Done [%.3f s]"%(time.time()-self._trace)
 294         if config.debug:
 295             get_tracker().getAndPrint(self.output)
 296