Initial commit

1 files changed, 51 insertions, 0 deletions

diff --git a/htc_utils/bindings/wait.py b/htc_utils/bindings/wait.py
new file mode 100644
index 0000000..768707f
--- /dev/null
+++ b/htc_utils/bindings/wait.py
@@ -0,0 +1,51 @@
+import os
+import subprocess
+
+
+class Wait(object):
+    '''
+    -debug            Show extra debugging info
+    -status           Show job start and terminate info
+    -echo             Echo log events relevant to [job-number]
+    -num <number>     Wait for this many jobs to end
+                       (default is all jobs)
+    -wait <seconds>   Wait no more than this time
+                       (default is unlimited)
+    '''
+    def __init__(self, submit, **kwargs):
+        self.submit = submit
+        self.cli_args = []
+    
+    def toggle_debug(self):
+        self.cli_args.append(['-debug'])
+        
+    def toggle_status(self):
+        self.cli_args.append(['-status'])    
+
+    def toggle_echo(self):
+        self.cli_args.append(['-echo'])
+    
+    def num(self, num_jobs):
+        self.cli_args.append(['-num', num_jobs])
+        
+    def timeout(self, seconds):
+        self.cli_args.append(['-wait', seconds])
+        
+    def execute(self):
+        logfile = self.submit.job.logs['log'].replace('$(Cluster)', self.submit.cluster)
+        print "logfile: ", logfile
+        condor_wait_args = ' '.join([str(arg) for record in self.cli_args for arg in record])
+        condor_wait = ' '.join(['condor_wait', condor_wait_args, logfile])
+        proc = subprocess.Popen(condor_wait.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=self.submit.environ)
+        stdout, stderr = proc.communicate()
+        print(stdout)
+        print(stderr)
+        proc.wait()
+        
+        '''
+        if not stderr:
+            if 'cluster' in stdout:
+                for line in stdout.split(os.linesep):
+                    if 'cluster' in line:
+                        self.cluster = line.split()[-1].strip('.')
+        '''