7 from PyQt4.QtCore import *
8 from sface.config import config
9 from sface.xmlrpcwindow import get_tracker, XmlrpcReader
11 def find_executable(exec_name):
12 """find the given executable in $PATH"""
13 paths = os.getenv("PATH").split(':')
15 exec_path = os.path.join(p, exec_name)
16 if os.path.exists(exec_path) and os.access(exec_path,os.X_OK):
21 class SfiProcess(QObject):
22 def __init__(self, parent=None):
23 QObject.__init__(self, parent)
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)"),
32 self.xmlrpcreader = XmlrpcReader() # this one is for parsing XMLRPC responses
34 # holds aggregate output from processStandardOutput(); used by xmlrpc
38 self.connect(self.process, SIGNAL("readyReadStandardOutput()"),
39 self.processStandardOutput)
40 self.connect(self.process, SIGNAL("readyReadStandardError()"),
41 self.processStandardError)
43 def __init_command(self, args):
44 self.args = QStringList()
46 self.args << config.get_dirname()
48 # this shows xmlrpc conversation, see sfi.py docs.
49 # always do this, so we can parse the XML result for faults and show
51 self.args << QString('-D')
54 self.args << QString(arg)
56 self.exe = find_executable("sfi.py")
58 print "FATAL.. Could not locate binary sfi.py - not much we can do without that"
61 return self.process.state() != QProcess.NotRunning
63 def processStandardOutput(self):
64 output = self.process.readAllStandardOutput()
65 self.output = self.output + output
71 # XXX why is this happening??
72 print "*** caught EINTR"
77 def processStandardError(self):
78 print self.process.readAllStandardError()
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:
87 elif err == QProcess.Crashed:
89 elif err == QProcess.Timedout:
91 elif err == QProcess.WriteError:
93 elif err == QProcess.ReadError:
95 elif err == QProcess.UnknownError:
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")]
106 self.emit(SIGNAL("finished()"))
108 def getFaultString(self):
109 if self.faults == []:
112 return self.faults[0].get("faultString","") + " (" + self.faults[0].get("faultCode","") + ")"
114 def retrieveRspec(self):
115 slice = config.getSlice()
116 mgr = config.getSlicemgr()
117 # Write RSpec to file
118 filename = config.fullpath ("%s.rspec"%slice)
123 args = ["-u", config.getUser(), "-a", config.getAuthority(),
124 "-r", config.getRegistry(), "-s", mgr, "resources",
125 "-o", filename, slice]
127 self.__init_command(args)
131 def retrieveResources(self):
132 mgr = config.getSlicemgr()
133 # Write RSpec to file
134 filename = config.getResourcesRSpecFile()
139 args = ["-u", config.getUser(), "-a", config.getAuthority(),
140 "-r", config.getRegistry(), "-s", mgr, "resources",
143 self.__init_command(args)
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]
153 filename = config.getAuthorityListFile()
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):
159 if os.path.exists(filename + ".1"):
160 os.remove(filename + ".1")
162 args.append(filename)
168 self.__init_command(args)
171 def getRecord(self, hrn, filename=None):
172 args = ["-u", config.getUser(), "-a", config.getAuthority(),
173 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "show", hrn]
176 args.append(filename)
177 self.__init_command(args)
180 def getSliceRecord(self):
181 self.getRecord(config.getSlice(), config.getSliceRecordFile())
183 def getAuthorityRecord(self):
184 self.getRecord(config.getAuthority(), config.getAuthorityRecordFile())
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"]
197 args = args + ["-o", obtained]
200 args = args + ["-a", aggAddr, "-p", str(aggPort)]
202 args = args + [config.getSlice(), requested]
204 self.__init_command(args)
207 def deleteSlivers(self, slice=None, aggAddr=None, aggPort=None):
209 slice = config.getSlice()
211 args = ["-u", config.getUser(), "-a", config.getAuthority(),
212 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "delete"]
215 args = args + ["-a", aggAddr, "-p", str(aggPort)]
217 args = args + [slice]
219 self.__init_command(args)
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)
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)
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)
240 def renewSlivers(self, expiration, slice=None):
242 slice = config.getSlice()
244 args = ["-u", config.getUser(), "-a", config.getAuthority(),
245 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "renew",
247 self.__init_command(args)
250 def sliverStatus(self, slice=None, filename=None):
252 slice = config.getSlice()
255 filename = config.fullpath(slice+".sliverstatus")
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)
263 def getSliceMgrVersion(self, filename=None):
265 filename = config.fullpath("slicemgr.version")
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)
278 self.process.start(self.exe, self.args)
280 def readOutput(self):
281 if self.process.state() == QProcess.NotRunning:
282 return self.process.readAll()
284 def trace_command (self):
286 self._trace=time.time()
287 command = "%s %s" % (self.exe, self.args.join(" "))
288 print time.strftime('%H:%M:%S'),'Invoking',command
290 def trace_end (self):
292 # command = "%s %s" % (self.exe, self.args.join(" "))
293 print time.strftime('%H:%M:%S'),"Done [%.3f s]"%(time.time()-self._trace)
295 get_tracker().getAndPrint(self.output)