3
ܓi @ s8 d dl mZmZmZ d dlmZ G dd deeeZdS ) )Plugin UbuntuPlugin RedHatPlugin)
is_executablec @ s0 e Zd ZdZdZdZdZdZd d Zd
d Z dS )SlurmzSlurm Workload Managerslurmhpc slurm-wlmslurmdslurmdbd slurmctldslurm-slurmctld slurm-slurmdslurm-slurmdbdc C s | j ddg tdr$| jddg tdr8| jdg dd d
d d d
dddddddddddg}tdrz| jdd |D d }| j|sd }d }d }y|t|d d!d"d}x\|j j D ]L}|sq|jd#}|d$ j d%kr|d& j }|d$ j d'kr|d& j }qW W d(Q R X W n: t k
rJ } z | j
d)| d*| W Y d(d(}~X nX | j d+sh| j ||g n| j | d,| d,g d(S )-z Slurm Workload Manager
z/etc/slurm/*.confz/var/run/slurm/conf/*.confZsinfoz!sinfo --all --list-reasons --longzsinfo --all --longZsqueuezsqueue --all --longaliasesZ assoc_mgrZbbstatZ burstBufferconfigZdaemonsZdwstatZ
federationZfrontendZjobZlicensesZnode partitionZ reservationr
stepZtopologyZscontrolc S s g | ]}d | qS )zscontrol show ).0ir r /usr/lib/python3.6/slurm.py
O s z Slurm.setup..z/etc/slurm/slurm.confz /var/run/slurm/conf/slurm.confz/var/log/slurmd.logz/var/log/slurmctld.logrzUTF-8)encoding=r Z
SlurmdLogFile ZSlurmctldLogFileNzCould not open conf file z: Zall_logs*) Z
add_copy_specr Zadd_cmd_outputZ path_existsopenread
splitlinessplitstripIOErrorZ
_log_errorZ
get_option) selfZ
scontrol_cmdsZ config_fileZslurmd_log_fileZslurmctld_log_filecfilelineZwordserrorr r r setup% sl
(
z Slurm.setupc C sd ddg}ddg}ddj | d}d}d }x4|D ],}| j| d
|| | j| d || q0W d S ) Nz
/etc/slurmz/var/run/slurm/confZAccountingStoragePassZ JobCompPassz(^\s*(|z
)\s*=\s*)(.*)z (^\s*(StoragePass)\s*=\s*)(.*)z
\1********z /slurm.confz/slurmdbd.conf)joinZ do_file_sub)r$ Z
conf_pathsZ
slurm_keysZslurm_keys_regexZslurmdbd_key_regexsubZ conf_pathr r r postprocs s
zSlurm.postprocN)r )r r
r r r r
r r )r
r r )
__name__
__module__ __qualname__Z
short_descZ plugin_nameZprofilesZpackagesZservicesr( r, r r r r r
s Nr N)Zsos.report.pluginsr r r Z
sos.utilitiesr r r r r r s