6 from PyQt4.QtCore import *
7 from sface.config import config
8 from sface.xmlrpcwindow import XmlrpcTracker, XmlrpcReader
10 def find_executable(exec_name):
11 """find the given executable in $PATH"""
12 paths = os.getenv("PATH").split(':')
14 exec_path = os.path.join(p, exec_name)
15 if os.path.exists(exec_path):
20 class SfiProcess(QObject):
21 def __init__(self, parent=None):
22 QObject.__init__(self, parent)
24 env = QProcess.systemEnvironment()
25 env << "PYTHONPATH=%s" % ":".join(sys.path)
26 self.process = QProcess()
27 self.process.setEnvironment(env)
28 self.connect(self.process, SIGNAL("finished(int, QProcess::ExitStatus)"),
31 self.xmlrpcreader = XmlrpcReader() # this one is for parsing XMLRPC responses
32 self.xmlrpctracker = XmlrpcTracker() # this one is for the debug window
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
69 def processStandardError(self):
70 print self.process.readAllStandardError()
72 def processFinished(self):
73 if self.process.exitStatus() == QProcess.CrashExit:
74 print self.readOutput()
75 print "Process exited with errors:",
76 err = self.process.error()
77 if err == QProcess.FailedToStart:
79 elif err == QProcess.Crashed:
81 elif err == QProcess.Timedout:
83 elif err == QProcess.WriteError:
85 elif err == QProcess.ReadError:
87 elif err == QProcess.UnknownError:
90 # extract any faults from the XMLRPC response(s)
91 self.xmlrpcreader.responses = []
92 self.xmlrpcreader.store(self.output)
93 self.xmlrpcreader.extractXml()
94 self.responses = self.xmlrpcreader.responses
95 self.faults = [x for x in self.responses if (x["kind"]=="fault")]
98 self.emit(SIGNAL("finished()"))
100 def __getRSpec(self, mgr):
101 slice = config.getSlice()
102 # Write RSpec to file for testing.
103 filename = os.path.expanduser("~/.sfi/" + slice + ".rspec")
108 args = ["-u", config.getUser(), "-a", config.getAuthority(),
109 "-r", config.getRegistry(), "-s", mgr, "resources",
110 "-o", filename, slice]
112 self.__init_command(args)
116 def getRSpecFromSM(self):
117 return self.__getRSpec(config.getSlicemgr())
119 # def getRSpecFromAM(self):
120 # return self.__getRSpec(config.getAggmgr())
122 def listRecords(self, hrn, rectype=None, filename=None):
123 args = ["-u", config.getUser(), "-a", config.getAuthority(),
124 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "list", hrn]
127 # we can't tell whether SFI will create one file or many, so delete
128 # leftovers from last time, then we'll know what we got, after we get it.
129 if os.path.exists(filename):
131 if os.path.exists(filename + ".1"):
132 os.remove(filename + ".1")
134 args.append(filename)
140 self.__init_command(args)
143 def getRecord(self, hrn, filename=None):
144 args = ["-u", config.getUser(), "-a", config.getAuthority(),
145 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "show", hrn]
148 args.append(filename)
149 self.__init_command(args)
152 def getSliceRecord(self):
153 self.getRecord(config.getSlice(), config.getSliceRecordFile())
155 def getAuthorityRecord(self):
156 self.getRecord(config.getAuthority(), config.getAuthorityRecordFile())
158 def applyRSpec(self, rspec):
159 filename = config.getSliceRSpecFile() + "_new"
161 args = ["-u", config.getUser(), "-a", config.getAuthority(),
162 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "create",
163 config.getSlice(), filename]
164 self.__init_command(args)
168 def updateRecord(self, filename):
169 args = ["-u", config.getUser(), "-a", config.getAuthority(),
170 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "update", filename]
171 self.__init_command(args)
174 def addRecord(self, filename):
175 args = ["-u", config.getUser(), "-a", config.getAuthority(),
176 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "add", filename]
177 self.__init_command(args)
180 def renewSlivers(self, expiration):
181 args = ["-u", config.getUser(), "-a", config.getAuthority(),
182 "-r", config.getRegistry(), "-s", config.getSlicemgr(), "renew",
183 config.getSlice(), expiration]
184 self.__init_command(args)
192 self.process.start(self.exe, self.args)
194 def readOutput(self):
195 if self.process.state() == QProcess.NotRunning:
196 return self.process.readAll()
198 def trace_command (self):
200 self._trace=time.time()
201 command = "%s %s" % (self.exe, self.args.join(" "))
202 print time.strftime('%H:%M:%S'),'Invoking',command
204 def trace_end (self):
206 # command = "%s %s" % (self.exe, self.args.join(" "))
207 print time.strftime('%H:%M:%S'),"Done [%.3f s]"%(time.time()-self._trace)
209 self.xmlrpctracker.getAndPrint(self.output)