1 files changed, 146 insertions, 0 deletions
diff --git a/htc_utils/bindings/job.py b/htc_utils/bindings/job.py
new file mode 100644
index 0000000..79771fc
--- /dev/null
+++ b/htc_utils/bindings/job.py
@@ -0,0 +1,146 @@
+# 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
+from . import recast
+from collections import OrderedDict
+
+
+class Job(object):
+    def __init__(self, filename, ext='job', * args, **kwargs):
+        ''' Example usage
+        import random
+        from condor_batch import Job, Submit
+
+        # Create job object (filename will be "my_neato_condor.job")
+        job = Job('my_neato_condor')
+
+        # Populate initial job attributes
+        job.attr('getenv', True)
+        job.attr('executable', '/bin/sleep')
+        job.attr('arguments', 1)
+        job.logging('logs', create=True)
+        job.attr('queue')
+
+        # Populate additional sub-attributes (i.e. re-execute job with
+        # a different set of arguments.
+        for fileno in range(5):
+            x = random.Random().randint(2, 5)
+            job.subattr('arguments', x)
+            job.subattr('queue')
+
+        # Write job file to disk
+        job.commit()
+
+        # Pass our job to the Submit class
+        submit = Submit(job)
+
+        # Send our job to the cluster
+        submit.execute()
+        '''
+
+
+        self.filename = os.path.abspath('.'.join([filename, ext]))
+        self.shebang = '#!/usr/bin/env condor_submit'
+
+        self.config_ext = []
+        self.config = OrderedDict()
+
+        self.default_config = [
+            self.attr('universe', 'vanilla'),
+            self.attr('getenv', True),
+            self.attr('environment', ''),
+            self.attr('executable', ''),
+            self.attr('arguments', ''),
+            self.attr('notification', 'Never'),
+            self.attr('notify_user', ''),
+            self.attr('priority', 0),
+            self.attr('rank', ''),
+            self.attr('input', ''),
+            self.attr('request_cpus', ''),
+            self.attr('request_disk', ''),
+            self.attr('request_memory', ''),
+            self.attr('requirements', ''),
+            self.attr('transfer_executable', True),
+            self.attr('transfer_input_files', []),
+            self.attr('transfer_output_files', ''),
+            self.attr('transfer_output_remaps', ''),
+            self.attr('should_transfer_files', 'IF_NEEDED'),
+            self.attr('when_to_transfer_output', 'ON_EXIT'),
+            self.attr('hold', False),
+            self.attr('initialdir', ''),
+            self.attr('remote_initialdir', ''),
+            self.attr('run_as_owner', ''),
+            self.attr('nice_user', ''),
+            self.attr('stack_size', ''),
+            ]
+
+        for key, value in kwargs.items():
+            self.attr(key, value)
+
+
+    def generate(self):
+        '''Return a
+        '''
+        if not self.config:
+            print("Warning: No attributes defined!")
+        output = ''
+
+        # Parse initial job attributes
+        for key, value in self.config.items():
+            if value and key != 'queue':
+                output += ' = '.join([key, value])
+                output += os.linesep
+            if key == 'queue':
+                output += ' '.join([key, value])
+                output += os.linesep
+
+        # Parse job sub-attributes
+        for sub in self.config_ext:
+            output += os.linesep
+            for key, value in sub.items():
+                if value and key != 'queue':
+                    output += ' = '.join([key, value])
+                    # output += os.linesep
+                if key == 'queue':
+                    output += ' '.join([key, value])
+                    output += os.linesep
+
+        return output
+
+    def logging(self, path, ext='log', create=False):
+        if create and not os.path.exists(path):
+            os.mkdir(os.path.abspath(path))
+
+        self.attr('log', os.path.normpath(os.path.join(path, '.'.join(['condor_$(Cluster)', ext]))))
+        self.attr('output', os.path.normpath(os.path.join(path, '.'.join(['stdout_$(Cluster)_$(Process)', ext]))))
+        self.attr('error', os.path.normpath(os.path.join(path, '.'.join(['stderr_$(Cluster)_$(Process)', ext]))))
+
+    def attr(self, key, *args):
+        value = ' '.join([recast(x) for x in args])
+        self.config[key] = recast(value)
+
+    def subattr(self, key, *args):
+        value = ' '.join([recast(x) for x in args])
+        self.config_ext.append(OrderedDict([(key, recast(value))]))
+
+    def commit(self, ext='job'):
+        with open(self.filename, 'w+') as submit_file:
+            submit_file.write(self.generate())
+
+    def reset(self):
+        self.config = OrderedDict()
+        self.config_ext = OrderedDict()