Added basic msub support

Author: mgermain

scripts/smart_dispatch.py

@@ -20,6 +20,7 @@
 LOGS_FOLDERNAME = "SMART_DISPATCH_LOGS"
 CLUSTER_NAME = utils.detect_cluster()
 AVAILABLE_QUEUES = get_available_queues(CLUSTER_NAME)
+LAUNCHER = utils.get_launcher(CLUSTER_NAME)
 
 
 def main():
@@ -85,7 +86,7 @@ def main():
     # Launch the jobs with QSUB
     if not args.doNotLaunch:
         for pbs_filename in pbs_filenames:
-            qsub_output = check_output('qsub ' + pbs_filename, shell=True)
+            qsub_output = check_output('{launcher} {pbs_filename}'.format(launcher=LAUNCHER, pbs_filename=pbs_filename), shell=True)
             print qsub_output,
 
 

smartdispatch/utils.py

@@ -53,6 +53,7 @@ def unhexify(match):
 
     return re.sub(r"\\x..", unhexify, text)
 
+
 @contextmanager
 def open_with_lock(*args, **kwargs):
     """ Context manager for opening file with an exclusive lock. """
@@ -95,3 +96,10 @@ def detect_cluster():
     elif server_name.split('.')[-1] == 'helios':
         cluster_name = "helios"
     return cluster_name
+
+
+def get_launcher(cluster_name):
+    if cluster_name == "helios":
+        return "msub"
+    else:
+        return "qsub"