Setting tag sface-0.1-14

[sface.git] / sface / sfiprocess.py

diff --git a/sface/sfiprocess.py b/sface/sfiprocess.py
index 9fd56a9..bef0902 100644 (file)
--- a/sface/sfiprocess.py
+++ b/sface/sfiprocess.py
@@ -1,7 +1,11 @@
 
 import os
+import sys
+import time
+
 from PyQt4.QtCore import *
-from sface.sficonfig import config
+from sface.config import config
+from sface.xmlrpcwindow import XmlrpcTracker
 
 def find_executable(exec_name):
     """find the given executable in $PATH"""
@@ -17,17 +21,65 @@ class SfiProcess(QObject):
     def __init__(self, parent=None):
         QObject.__init__(self, parent)
 
+        env = QProcess.systemEnvironment()
+        env << "PYTHONPATH=%s" % ":".join(sys.path)
+        self.process = QProcess()
+        self.process.setEnvironment(env)
+        self.connect(self.process, SIGNAL("finished(int, QProcess::ExitStatus)"),
+                     self.processFinished)
+        
+        self.xmlrpctracker = XmlrpcTracker()
+        # in case self.output is read by the XmlrpcTracker before any
+        # readyReadStandardOutput signal
+        self.output = ''
+
+        self.connect(self.process, SIGNAL("readyReadStandardOutput()"),
+                     self.processStandardOutput)
+        self.connect(self.process, SIGNAL("readyReadStandardError()"),
+                     self.processStandardError)
+
     def __init_command(self, args):
         self.args = QStringList()
+        if config.debug:
+            # this shows xmlrpc conversation, see sfi.py docs.
+            self.args << QString('-D')
         for arg in args:
             self.args << QString(arg)
 
         self.exe = find_executable("sfi.py")
-        self.process = QProcess()
+        if not self.exe:
+            print "FATAL.. Could not locate binary sfi.py - not much we can do without that"
 
-        self.finished = pyqtSignal()
-        self.connect(self.process, SIGNAL("finished(int, QProcess::ExitStatus)"),
-                     self.finished)
+    def isRunning(self):
+        return self.process.state() != QProcess.NotRunning
+
+    def processStandardOutput(self):
+        output = self.process.readAllStandardOutput()
+        if config.debug:
+            print output
+
+    def processStandardError(self):
+        print self.process.readAllStandardError()
+
+    def processFinished(self):
+        if self.process.exitStatus() == QProcess.CrashExit:
+            print self.readOutput()
+            print "Process exited with errors:",
+            err = self.process.error()
+            if err == QProcess.FailedToStart:
+                print "FailedToStart"
+            elif err == QProcess.Crashed:
+                print "Crashed"
+            elif err == QProcess.Timedout:
+                print "Timedout"
+            elif err == QProcess.WriteError:
+                print "WriteError"
+            elif err == QProcess.ReadError:
+                print "ReadError"
+            elif err == QProcess.UnknownError:
+                print "UnknownError"
+        self.trace_end()
+        self.emit(SIGNAL("finished()"))
 
     def __getRSpec(self, mgr):
         slice = config.getSlice()
@@ -38,7 +90,7 @@ class SfiProcess(QObject):
         except:
             pass
         args = ["-u", config.getUser(), "-a", config.getAuthority(), 
-                "-r", config.registry(), "-s", mgr, "resources", 
+                "-r", config.getRegistry(), "-s", mgr, "resources", 
                 "-o", filename, slice]
 
         self.__init_command(args)
@@ -46,21 +98,61 @@ class SfiProcess(QObject):
         return filename
 
     def getRSpecFromSM(self):
-        return self.__getRSpec(config.slicemgr())
+        return self.__getRSpec(config.getSlicemgr())
 
-    def getRSpecFromAM(self):
-        return self.__getRSpec(config.aggmgr())
+#    def getRSpecFromAM(self):
+#        return self.__getRSpec(config.getAggmgr())
 
-    def getRecord(self, hrn):
-        args = ["-u", config.getUser(), "-a", config.getAuthority(), 
-                "-r", config.registry(), "-s", config.slicemgr(), "show", hrn]
+    def getRecord(self, hrn, filename=None):
+        args = ["-u", config.getUser(), "-a", config.getAuthority(),
+                "-r", config.getRegistry(), "-s", config.getSlicemgr(), "show", hrn]
+        if filename:
+            args.append("-o")
+            args.append(filename)
+        self.__init_command(args)
+        self.start()
+
+    def applyRSpec(self, rspec):
+        filename = config.getSliceRSpecFile() + "_new"
+        rspec.save(filename)
+        args = ["-u", config.getUser(), "-a", config.getAuthority(),
+                "-r", config.getRegistry(), "-s", config.getSlicemgr(), "create",
+                config.getSlice(), filename]
+        self.__init_command(args)
+        self.start()
+        return filename
+
+    def updateRecord(self, filename):
+        args = ["-u", config.getUser(), "-a", config.getAuthority(),
+                "-r", config.getRegistry(), "-s", config.getSlicemgr(), "update", filename]
+        self.__init_command(args)
+        self.start()
+
+    def renewSlivers(self, expiration):
+        args = ["-u", config.getUser(), "-a", config.getAuthority(),
+                "-r", config.getRegistry(), "-s", config.getSlicemgr(), "renew",
+                config.getSlice(), expiration]
         self.__init_command(args)
         self.start()
 
     def start(self):
+        self.trace_command()
         self.process.start(self.exe, self.args)
 
     def readOutput(self):
         if self.process.state() == QProcess.NotRunning:
             return self.process.readAll()
 
+    def trace_command (self):
+        if config.verbose:
+            self._trace=time.time()
+            command = "%s %s" % (self.exe, self.args.join(" "))
+            print time.strftime('%H:%M:%S'),'Invoking',command
+
+    def trace_end (self):
+        if config.verbose:
+#            command = "%s %s" % (self.exe, self.args.join(" "))
+            print time.strftime('%H:%M:%S'),"Done [%.3f s]"%(time.time()-self._trace)
+        if config.debug:
+            self.xmlrpctracker.getAndPrint(self.output)
+