1 files changed, 80 insertions, 0 deletions
diff --git a/htc_utils/bindings/submit.py b/htc_utils/bindings/submit.py
new file mode 100644
index 0000000..fba59a5
--- /dev/null
+++ b/htc_utils/bindings/submit.py
@@ -0,0 +1,80 @@
+# This file is part of htc_utils.
+#
+# htc_utils is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# htc_utils is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with htc_utils.  If not, see <http://www.gnu.org/licenses/>.
+
+
+import os
+import subprocess
+from . import htcondor_path
+
+
+class Submit(object):
+    def __init__(self, job, **kwargs):
+        self.job = job
+        self.cluster = None
+        self.environ = os.environ
+        self._prefix = ':'.join([os.environ['PATH'], htcondor_path()])
+        self.environ['PATH'] = self._prefix
+        self.cli_args = []
+
+        print("Received job: {}".format(self.job.filename))
+
+    def sendto_name(self, schedd_name):
+        self.cli_args.append(['-name', schedd_name])
+
+    def sendto_remote(self, schedd_name):
+        self.cli_args.append(['-remote', schedd_name])
+
+    def sendto_addr(self, ip, port):
+        addr = ':'.join([ip, str(port)])
+        addr = '<' + addr + '>'
+        self.cli_args.append(['-addr', addr])
+
+    def sendto_pool(self, pool_name):
+        self.cli_args.append(['-pool', pool_name])
+
+    def toggle_verbose(self):
+        self.cli_args.append(['-verbose'])
+
+    def toggle_unused_variables(self):
+        self.cli_args.append(['-unused'])
+
+    def execute(self):
+        condor_submit_args = ' '.join([str(arg) for record in self.cli_args for arg in record])
+        condor_submit = ' '.join(['condor_submit', condor_submit_args, self.job.filename])
+        proc = subprocess.Popen(condor_submit.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=self.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('.')
+
+    def monitor(self):
+        if self.cluster is None:
+            print('No cluster data for job.')
+            return False
+        print('Monitoring cluster {}'.format(self.cluster))
+        os.environ['PATH'] = self._prefix
+        '''
+        proc = subprocess.Popen('condor_q -analyze:summary'.split(), env=self.environ)
+        proc.communicate()
+        proc.wait()
+        '''
+        print('NOT IMPLEMENTED')
+        return True