Name	Size	Mode	Actions
control_persist.cpython-310.pyc	7045	0644	editdlrm
juju.cpython-310.pyc	3124	0644	editdlrm
local.cpython-310.pyc	1873	0644	editdlrm
oc.cpython-310.pyc	6506	0644	editdlrm
saltstack.cpython-310.pyc	5697	0644	editdlrm
__init__.cpython-310.pyc	14872	0644	editdlrm

o G%@i8�@s`dd
lZdd
l
Z
dd
lZdd
lZddlmZ
ddlmZmZ
ddl m Z mZ
Gdd�de�Zd
S)�N)
�RemoteTransport)�ConnectionException�#SaltStackMasterUnsupportedException)� is_executable�sos_get_command_outputcs�eZ
dZd
ZdZdd�Z  d�f
dd � Zd d�Zdd �Ze dd��
Z ddd�
Zddd�
Zdd�Z e dd��
Zdd�Zdd�Z�ZS)�SaltStackMasterz� A transport for collect that leverages SaltStack's Master Pub/Sub functionality to send commands to minions. This transport will by default assume the use cmd.shell module to execute commands on the minions. � saltstackcCstt
�|
�
���
d
S)Nr
)�list�json�loads�values)�self�json_output�r�D/usr/lib/python3/dist-packages/sos/collector/transports/saltstack.py�_convert_output_json!s
z$SaltStackMaster._convert_output_json�FNcsVt��
|
||||�}t�t�
�
|�|d
�
|d
<Wd�
|S1s$w



Y
|S)z� Run a command on the remote host using SaltStack Master. If the output is json, convert it to a string. �outputN)�super�run_command� contextlib�suppress� Exceptionr)r �cmd�timeout� need_root�env� use_shell�ret�
� __class__rrr$s
�
��zSaltStackMaster.run_commandcCs.d
|
�d|�d|��}t|dd�}|ddkS)zO Execute cp.get_file on the remote host using SaltStack Master zsalt-cp �
��
r�statusr
)
r)r �node�fname�destr�resrrr�_salt_copy_file0s

zSaltStackMaster._salt_copy_filecCsZd
|
�d|��}t|�
}|ddkr+d|j
�d�}tj�||�d�
�}t�||�
dSd S) zK Execute cp.push on the remote host using SaltStack Master �salt z cp.push r$r
z/var/cache/salt/master/minions/z/files�
/TF)r�address�os�path�join�lstrip�shutil�move)r r%r&r'rr(�cachedir� cachedir_filerrr�_salt_retrieve_file8s






z#SaltStackMaster._salt_retrieve_filec
Cs|jd
dd�}
|
ddkS)z>Check if the remote host is responding using SaltStack Master.�echo Connected� r#r$r
)
r)r �uprrr� connectedEs
zSaltStackMaster.connectedcCs(d
}t|�
}|ddkr|ddkSt
�
)a�
Checks to see if the local system supported SaltStack Master. This check relies on feedback from the salt binary. The command being run should always generate stderr output, but depending on what that output reads we can determine if SaltStack Master is supported or not. For our purposes, a host that does not support SaltStack Master is not able to run sos collect. Returns True if SaltStack Master is supported, else raise Exception zsalt-run manage.statusr$r
)rr)r �passwordrr(rrr�_check_for_saltstackLs 



z$SaltStackMaster._check_for_saltstackcCs|td
�
s|�
d�

dSz|��
Wn ty


|�
d�

�w|�d�

d}|j|dd�}|d d kr8t|j�
�
|d dkS)aD
Connect to the remote host using SaltStack Master. This method will attempt to connect to the remote host using SaltStack Master. If the connection fails, an exception will be raised. If the connection is successful, the connection will be stored in the self._connection attribute. �saltz salt command is not executable. Fz$Transport is not locally supported. z4Transport is locally supported and service running. r6rr#r$�
r
)r� log_errorr;r�log_inforr,)r r:r�resultrrr�_connect`s





� 




zSaltStackMaster._connectc


Csd
S)NTr)
r rrr�_disconnectys
zSaltStackMaster._disconnectc
Csd
}
d|
�d|j�d�S)z7The remote execution command to use for this transport.z--out json --static --no-colorr*r!z cmd.exec_code sh )
r,)r � salt_argsrrr�remote_exec|s
zSaltStackMaster.remote_execcC�|jr|�
|j|
|�Sd
S)a
Copy a file to the remote host using SaltStack Master Parameters fname The path to the file on the master dest The path to the destination directory on the remote host Returns True if the file was copied, else False F)r9r)r,�r r&r'rrr�_copy_file_to_remote�� ���z$SaltStackMaster._copy_file_to_remotecCrE)a
Retrieve a file from the remote host using saltstack Parameters fname The path to the file on the remote host dest The path to the destination directory on the master Returns True if the file was retrieved, else False F)r9r5r,rFrrr�_retrieve_file�rHzSaltStackMaster._retrieve_file)rFNF)
N)�__name__� __module__�__qualname__�__doc__�namerrr)r5�propertyr9r;rArBrDrGrI� __classcell__rrrrrs$

�
 
r) rr r-r1�sos.collector.transportsr�sos.collector.exceptionsrr� sos.utilitiesrrrrrrr�
s