jupyter/spawner
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix SLURM sinfo command parser

Browse source
This commit is contained in:
Benoît Seignovert 2024-02-20 18:08:46 +01:00
parent e3fda2c317
commit a038d46231
Signed by: Benoît Seignovert
GPG key ID: F5D8895227D18A0B
2 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/glicid_spawner/slurm.py
View file

@ -140,9 +140,9 @@ def sinfo_run(username: str = None) -> str:
cmd = f'sinfo {flags} --Format={fmt}' cmd = f'sinfo {flags} --Format={fmt}'
if username: if username:
cmd = f'su - {username} -c "{cmd}"' cmd = f"su - {username} -c '{cmd}'"
return subprocess.check_output(shlex.split(cmd, posix=False)).decode('utf-8') return subprocess.check_output(shlex.split(cmd)).decode('utf-8')
def sinfo_reader(result: str) -> list: def sinfo_reader(result: str) -> list:

3
tests/test_slurm.py
View file

@ -109,13 +109,12 @@ def test_slurm_sinfo_run(monkeypatch):
) )
assert sinfo_run(username='john-doe') == ( assert sinfo_run(username='john-doe') == (
'su - john-doe -c "' 'su - john-doe -c '
'sinfo ' 'sinfo '
'--federation ' '--federation '
'--noheader ' '--noheader '
'--responding ' '--responding '
'--Format=Cluster,PartitionName,NodeHost,StateLong,CPUsState,Memory,Gres' '--Format=Cluster,PartitionName,NodeHost,StateLong,CPUsState,Memory,Gres'
'"'
) )