Add first draft for flux

pysqa/queueadapter.py

@@ -308,7 +308,7 @@ def set_queue_adapter(config, directory, execute_command=execute_command):
         config (dict): configuration for one cluster
         directory (str): directory which contains the queue configurations
     """
-    if config["queue_type"] in ["SGE", "TORQUE", "SLURM", "LSF", "MOAB"]:
+    if config["queue_type"] in ["SGE", "TORQUE", "SLURM", "LSF", "MOAB", "FLUX"]:
         return BasisQueueAdapter(
             config=config, directory=directory, execute_command=execute_command
         )

pysqa/utils/basic.py

@@ -66,6 +66,9 @@ def __init__(self, config, directory="~/.queues", execute_command=execute_comman
         elif self._config["queue_type"] == "REMOTE":
             class_name = None
             module_name = None
+        elif self._config["queue_type"] == "FLUX":
+            class_name = "FluxCommands"
+            module_name = "pysqa.wrapper.flux"
         else:
             raise ValueError()
         if self._config["queue_type"] != "REMOTE":

pysqa/wrapper/flux.py

@@ -0,0 +1,30 @@
+# coding: utf-8
+from pysqa.wrapper.generic import SchedulerCommands
+import subprocess
+
+
+class FluxCommands(SchedulerCommands):
+    @property
+    def submit_job_command(self):
+        return ["flux", "batch"]
+
+    @property
+    def delete_job_command(self):
+        return ["flux", "cancel"]
+
+    @property
+    def get_queue_status_command(self):
+        return ["flux", "jobs", "-a", "--no-header"]
+
+    @staticmethod
+    def get_job_id_from_output(queue_submit_output):
+        result = subprocess.run(
+            [
+                "flux",
+                "job",
+                "id",
+                queue_submit_output.splitlines()[-1].rstrip().lstrip().split()[-1],
+            ],
+            stdout=subprocess.PIPE,
+        )
+        return int(result.stdout.strip())

tests/config/flux/flux.sh

@@ -0,0 +1,3 @@
+#!/bin/bash
+#flux: -n{{cores}} --job-name={{job_name}} --env=CORES={{cores}} --output=time.out --error=error.out
+{{command}}

tests/config/flux/queue.yaml

@@ -0,0 +1,4 @@
+queue_type: FLUX
+queue_primary: flux
+queues:
+  flux: {cores_max: 64, cores_min: 1, run_time_max: 172800, script: flux.sh}