Add SLURM resources to form template

render/__main__.py

@@ -3,12 +3,16 @@
 Usage: `python -m render`
 
 """
+import sys
+from argparse import ArgumentParser
+from pathlib import Path
 from traceback import format_exc
 
 from flask import Flask, render_template, request
 from glicid_spawner.form import TEMPLATES, options_from_form
 from glicid_spawner.micromamba import MicromambaEnv
-from glicid_spawner.resources import CPU, GPU, RAM
+from glicid_spawner.resources import CPU, MEMORY, gpu_max_duration
+from glicid_spawner.slurm import gres, sinfo_from_file
 from livereload import Server
 
 # Dummy username and python environments
@@ -18,15 +22,30 @@ ENVS = [
     MicromambaEnv('USER', 'bar', f'/{USERNAME}/envs/bar'),
     MicromambaEnv('GLOBAL', 'baz', '/global/envs/baz'),
 ]
+
+# Dummy SLURM config
+DATA = Path(__file__).parent / '..' / 'tests' / 'data'
+SINFO = sinfo_from_file(
+    DATA / 'sinfo.txt', with_states=('idle', 'mixed', 'allocated', 'completing', 'planned')
+)
+
+# Single vs. multi-cluster implementation
+SLURM_SINGLE_CLUSTER = {'N/A': SINFO.pop('N/A')}
+SLURM_MULTI_CLUSTER = SINFO
+GPU_SINGLE_CLUSTER = gpu_max_duration(gres(SLURM_SINGLE_CLUSTER))
+GPU_MULTI_CLUSTER = gpu_max_duration(gres(SLURM_MULTI_CLUSTER))
+
+# Format dummy options
 OPTIONS = {
     'username': USERNAME,
     'envs': ENVS,
     'cpus': CPU,
-    'rams': RAM,
-    'gpus': GPU,
+    'mems': MEMORY,
+    # Multi-cluster by default. See `--single-cluster` flag in CLI for single cluster config.
+    'gpus': GPU_MULTI_CLUSTER,
+    'sinfo': SLURM_MULTI_CLUSTER,
 }
 
-
 # Flask app
 app = Flask(__name__)
 app.debug = True
@@ -60,11 +79,35 @@ def submit():
         return render_template('options.html', formdata=formdata, err=format_exc())
 
 
+@app.route('/favicon.ico')
+def favicon():
+    """Dummy favicon."""
+    return ''
+
+
 def server_autoreload():
     """Start auto-reload server."""
     server = Server(app.wsgi_app)
     server.serve()
 
 
-if __name__ == '__main__':
+def cli(argv=None):
+    """Command line interface."""
+    parser = ArgumentParser('Spawner form render')
+    parser.add_argument(
+        '--single-cluster',
+        action='store_true',
+        help='Toggle SLURM config to single cluster configuration.',
+    )
+
+    args, _ = parser.parse_known_args(argv)
+
+    if args.single_cluster:
+        OPTIONS['gpus'] = GPU_SINGLE_CLUSTER
+        OPTIONS['sinfo'] = SLURM_SINGLE_CLUSTER
+
     server_autoreload()
+
+
+if __name__ == '__main__':
+    cli(sys.argv)