o

-�_g�@s8dd
lm
Z
mZmZ
ddlmZ
Gdd�de
ee�ZdS)�)�Plugin�UbuntuPlugin�RedHatPlugin)
�
is_executablec@s0eZ
dZd
ZdZdZdZdZdd�Zdd	�Z	d
S)�SlurmzSlurm Workload Manager�slurm)
�hpc)z	slurm-wlm�slurmd�slurmdbd�	slurmctldrzslurm-slurmctldzslurm-slurmdzslurm-slurmdbd)r	r
rc
	
Csx
|�d
dg�

t
d�
r|�ddg�

t
d�
r|�dg
�

gd�
}
t
d	�
r.|�d
d�|
D�
�

d}|�|�
s7d
}d}d}zFt|ddd��5}|����D]&}|sOqJ|�d�
}|d��dkrb|d��}|d��dkrp|d��}qJWd�
n1s{w



Y
Wnt	y�
}
z|�
d|�d|���

WYd}~nd}~w
w|�d�
s�|�||g�

dS|�|�d�|�d�g�

dS)z  Slurm Workload Manager
        z/etc/slurm/*.confz/var/run/slurm/conf/*.conf�sinfoz!sinfo --all --list-reasons --longzsinfo --all --long�squeuezsqueue --all --long)�aliases�	assoc_mgr�bbstat�burstBuffer�config�daemons�dwstat�
federation�frontend�job�licenses�node�	partition�reservationr	�step�topology�scontrolc
Ssg|]}
d|
���qS)
zscontrol show �)�.0�
irr�:/usr/lib/python3/dist-packages/sos/report/plugins/slurm.py�
<listcomp>OszSlurm.setup.<locals>.<listcomp>z/etc/slurm/slurm.confz/var/run/slurm/conf/slurm.confz/var/log/slurmd.logz/var/log/slurmctld.log�
rzUTF-8)
�encoding�
=r
�
SlurmdLogFile�
�SlurmctldLogFileNzCould not open conf file z: �all_logs�
*)�
add_copy_specr�add_cmd_output�path_exists�open�read�
splitlines�split�strip�IOError�
_log_error�
get_option)	�self�
scontrol_cmds�config_file�slurmd_log_file�slurmctld_log_file�cfile�line�words�errorrrr"�setup%sh

�


�

�

�












����	

���




�


�zSlurm.setupc
Cs`d
dg}
ddg}dd�|�
�d�}d}d	}|
D]}|�
|�d
�||�
|�
|�d�||�
qdS)Nz
/etc/slurmz/var/run/slurm/conf�AccountingStoragePass�JobCompPassz(^\s*(�
|z
)\s*=\s*)(.*)z(^\s*(StoragePass)\s*=\s*)(.*)z
\1********z/slurm.confz/slurmdbd.conf)�join�do_file_sub)r7�
conf_paths�
slurm_keys�slurm_keys_regex�slurmdbd_key_regex�sub�	conf_pathrrr"�postprocss&
�
�



�

��zSlurm.postprocN)
�__name__�
__module__�__qualname__�
short_desc�plugin_name�profiles�packages�servicesr@rLrrrr"r
s

NrN)�sos.report.pluginsrrr�
sos.utilitiesrrrrrr"�<module>
s