upload to wxpython.wxcommunity.com instead of starship

[wxWidgets.git] / wxPython / distrib / all / taskrunner.py

diff --git a/wxPython/distrib/all/taskrunner.py b/wxPython/distrib/all/taskrunner.py
index 677bf1f1bae6ae87a92b3ab71d233af1509e8bce..5ed84826c5e4a2492c2a2bf4a9f89474f5d88f72 100644 (file)
--- a/wxPython/distrib/all/taskrunner.py
+++ b/wxPython/distrib/all/taskrunner.py
@@ -16,14 +16,38 @@ import sys
 import os
 import signal
 import select
 import os
 import signal
 import select

-import fcntl
+import time 
+

 from subprocess import Popen, PIPE, STDOUT
 
 
 from subprocess import Popen, PIPE, STDOUT
 
 

-__all__ = ["Job", "Task", "TaskRunner"]
+__all__ = ["Job", "Task", "TaskRunner", "TaskRunnerThread"]

 
 #----------------------------------------------------------------------
 
 
 #----------------------------------------------------------------------
 

+# For environment settings
+class Config:
+    def asDict(self):
+        return self.__dict__.copy()
+
+    def write(self, filename="config", outfile=None):
+        if outfile is None:
+            f = file(filename, "w")
+        else:
+            f = outfile
+        for k, v in self.__dict__.items():
+            f.write('%s="%s"\n' % (k, v))
+            
+    def read(self, filename="config"):
+        myfile = open(filename, "r")
+        for line in myfile.readlines():
+            line = line.strip()
+            if len(line) > 0 and line[0] == "#":
+                continue # it's a comment, move on
+            data = line.split("=")
+            if len(data) == 2:
+                self.__dict__[data[0].strip()] = data[1].strip()
+        myfile.close()

 
 class Job(object):
     """
 
 class Job(object):
     """


@@ -35,18 +59,28 @@ class Job(object):
 
     LOGBASE="."
     
 
     LOGBASE="."
     

-    def __init__(self, label, args):
+    def __init__(self, label, command, args=[], env=os.environ, verbose=True):

         self.label = label
         self.label = label

+        self.command = command

         self.args = args
         self.args = args

+        self.env = env

         self.proc = None
         self.proc = None

-        if self.label:
-            self.log = file("%s/%s.log" % (self.LOGBASE, label), "w", 0)
+        self.startTime = None
+        self.stopTime = None
+        self.verbose = verbose
+        self.label = label

 
     def start(self):
 
     def start(self):

-        self.proc = Popen(self.args, # the command and args to execute
-                          stdout=PIPE, stderr=STDOUT,
-                          bufsize=0, # line-buffered
+        self.proc = Popen([self.command] + self.args, # the command and args to execute
+                          stdout=PIPE, stderr=STDOUT, env=self.env,
+                          bufsize=0 # line-buffered

                           )
                           )

+        self.startTime = time.time()
+        if self.label:
+            if not os.path.exists(self.LOGBASE):
+                os.makedirs(self.LOGBASE)
+            self.log = file("%s/%s.log" % (self.LOGBASE, self.label), "w", 0)
+            

         # put the file in non-blocking mode
         #flags = fcntl.fcntl (self.proc.stdout, fcntl.F_GETFL, 0)
         #flags = flags | os.O_NONBLOCK
         # put the file in non-blocking mode
         #flags = fcntl.fcntl (self.proc.stdout, fcntl.F_GETFL, 0)
         #flags = flags | os.O_NONBLOCK


@@ -57,6 +91,7 @@ class Job(object):
         if self.proc is not None and self.proc.returncode is None:
             os.kill(self.proc.pid, signal.SIGTERM)
             self.logLines()
         if self.proc is not None and self.proc.returncode is None:
             os.kill(self.proc.pid, signal.SIGTERM)
             self.logLines()

+            self.stopTime = time.time()

 
 
     def fileno(self):
 
 
     def fileno(self):


@@ -65,7 +100,16 @@ class Job(object):
         else:
             return -1
 
         else:
             return -1
 

-        
+    def elapsedTime(self):
+        now = self.stopTime
+        if not now:
+            now = time.time()
+        elapsed_time = now-self.startTime
+        mins = elapsed_time/60
+        hours = mins/60
+        seconds = (elapsed_time - mins) % 60
+        return "%d:%d:%d" % (hours, mins, seconds)
+

     def logLines(self):
         if self.proc is not None:
             while self.linesAvailable():
     def logLines(self):
         if self.proc is not None:
             while self.linesAvailable():


@@ -74,7 +118,8 @@ class Job(object):
                 if self.label:
                     self.log.write(line)
                     line = "** %s: %s" % (self.label, line)
                 if self.label:
                     self.log.write(line)
                     line = "** %s: %s" % (self.label, line)

-                sys.stdout.write(line)
+                if self.verbose:
+                    sys.stdout.write(line)

 
 
     def linesAvailable(self):
 
 
     def linesAvailable(self):


@@ -140,16 +185,26 @@ class Task(object):
 
 class TaskRunner(object):
     """
 
 class TaskRunner(object):
     """

-    Manages the running of multiple tasks.
+    Manages the running of multiple tasks. Name can be used to identify 
+    a specific TaskRunner instance when reporting information back to the user.

     """
     """

-    def __init__(self, tasks=[]):
+    def __init__(self, tasks=[], name="TaskRunner Tasks"):

         if type(tasks) != list:
             tasks = [tasks]
         self.tasks = tasks[:]
         if type(tasks) != list:
             tasks = [tasks]
         self.tasks = tasks[:]

+        self.name = name
+        self.rc = 0

 
     def append(self, task):
         self.tasks.append(task)
 
     def append(self, task):
         self.tasks.append(task)

-
+        
+    def errorOccurred(self):
+        """
+        Only used for threaded TR instances. Once all TR tasks have completed,
+        we'll want to check to make sure there were no errors in the process.
+        """
+        return self.rc != 0
+        

     def run(self):
         # start all the active jobs
         for task in self.tasks:
     def run(self):
         # start all the active jobs
         for task in self.tasks:


@@ -175,6 +230,7 @@ class TaskRunner(object):
                         if job.returnCode() != 0:
                             rc = job.returnCode()
                             print "JOB RETURNED FAILURE CODE! (%d)" % rc
                         if job.returnCode() != 0:
                             rc = job.returnCode()
                             print "JOB RETURNED FAILURE CODE! (%d)" % rc

+                            self.rc = rc

                             self.stopAllJobs()
                             return rc
                         else:
                             self.stopAllJobs()
                             return rc
                         else:


@@ -182,6 +238,7 @@ class TaskRunner(object):
         except KeyboardInterrupt:
             print "STOPPING JOBS..."
             self.stopAllJobs()
         except KeyboardInterrupt:
             print "STOPPING JOBS..."
             self.stopAllJobs()

+            return 1

 
         except:
             print "Unknown exception..."
 
         except:
             print "Unknown exception..."


@@ -197,26 +254,60 @@ class TaskRunner(object):
             if job:
                 job.stop()
 
             if job:
                 job.stop()
 

+
+import threading
+
+class TaskRunnerThread(threading.Thread):
+    def __init__(self, taskRunner, callback=None):
+        self.taskRunner = taskRunner
+        self.startTime = None
+        self.stopTime = None
+        self.callback = callback
+        threading.Thread.__init__ ( self )
+        
+    def run(self):
+        self.startTime = time.time()
+        self.taskRunner.run()
+        self.stopTime = time.time()
+        #if self.callback:
+        #    self.callback
+
+    def elapsedTime(self):
+        now = self.stopTime
+        if not now:
+            now = time.time()
+        elapsed_time = now-self.startTime
+        mins = elapsed_time/60
+        hours = mins/60
+        seconds = (elapsed_time - mins) % 60
+        return "%d:%d:%d" % (hours, mins, seconds)
+        
+    def totalTime(self):
+        if self.stopTime:
+            return self.elapsedTime()
+        else:
+            return None
+

 #----------------------------------------------------------------------
 
 
 if __name__ == "__main__":
 
 #----------------------------------------------------------------------
 
 
 if __name__ == "__main__":
 

-    j1 = Job("label1", ["./tmp/job-1.py", "TEST-1"])
-    j2 = Job("label2", ["./tmp/job-2.sh", "TEST-2"])
+    j1 = Job("label1", "./tmp/job-1.py", ["TEST-1"])
+    j2 = Job("label2", "./tmp/job-2.sh", ["TEST-2"])

 
     t1 = Task()
     t1.append(j1)
     t1.append(j2)
 
 
     t1 = Task()
     t1.append(j1)
     t1.append(j2)
 

-    j3 = Job("task2a", ["./tmp/job-1.py", "TASK-2a"])
-    j4 = Job("task2b", ["./tmp/job-2.sh", "TASK-2b"])
+    j3 = Job("task2a", "./tmp/job-1.py", ["TASK-2a"])
+    j4 = Job("task2b", "./tmp/job-2.sh", ["TASK-2b"])

 
     t2 = Task()
     t2.append(j4)
     t2.append(j3)
 
 
     t2 = Task()
     t2.append(j4)
     t2.append(j3)
 

-    t3 = Task([Job("error", ["./tmp/job-3.sh", "TASK-3"])])
+    t3 = Task([Job("error", "./tmp/job-3.sh", ["TASK-3"])])

 
     tr = TaskRunner()
     tr.append(t1)
 
     tr = TaskRunner()
     tr.append(t1)