File: //usr/lib/python3/dist-packages/sos/collector/transports/__pycache__/juju.cpython-310.pyc
o
-�_g � @ s@ d dl Z d dlmZ d dlmZ d dlmZ G dd� de�ZdS )� N)�JujuNotInstalledException)�RemoteTransport)�sos_get_command_outputc @ sb e Zd ZdZdZdZdd� Zdd� Zdd d
�Zd d � Z e
d
d� �Z e
dd� �Z dd� Z
dd� ZdS )�JujuSSHa
A "transport" that leverages `juju ssh` to perform commands on the remote
hosts.
This transport is expected to be used in juju managed environment, and the
user should have the necessary credential for accessing the controller.
When using this transport, the --nodes option will be expected to be a
comma separated machine IDs, **not** IP addr, since `juju ssh` identifies
the ssh target by machine ID.
Examples:
sos collect --nodes 0,1,2 --no-local --transport juju --batch
�juju_ssh�ubuntuc
C sH d}z t j|t jdd� W dS t jy# } z| �d� t|�d }~ww )Nz juju versionT��stderr�shellz Failed to check `juju` version)�
subprocess� check_output�STDOUT�CalledProcessError� log_errorr )�self�cmd�err� r �?/usr/lib/python3/dist-packages/sos/collector/transports/juju.py�_check_juju_installed' s �
��z JujuSSH._check_juju_installedc C sN | j � d|� �}z tj|tjdd� W dS tjy& | �d|� d�� � w )Nz sudo chmod o+r Tr zFailed to make z world-readable)� remote_execr r r
r r )r �fnamer r r r �_chmod0 s ��zJujuSSH._chmod� c C s | � � | _| jS �N)r �
_connected)r �passwordr r r �_connect9 s
zJujuSSH._connectc C s dS )NTr �r r r r � _disconnect= s zJujuSSH._disconnectc C s | j S r )r r r r r � connected@ s zJujuSSH.connectedc C s: | j �d�\}}|rd|� �nd}|� d|� �}d|� �S )N�:�-m r � z juju ssh )�address�split)r �model�
target_option� model_option�optionr r r r D s
zJujuSSH.remote_execc C sV | j �d�\}}|rd|� �nd}d|� d|� d|� d|� �}t|dd�}|d d
kS ) Nr! r" r � juju scp z -- r# � )�timeout�statusr )r$ r% r �r r �destr& �unitr( r �resr r r �_copy_file_to_remoteK s
z JujuSSH._copy_file_to_remotec C s\ | � |� | j�d�\}}|rd|� �nd}d|� d|� d|� d|� �}t|�}|d dkS ) Nr! r" r r* z -- -r r# r- r )r r$ r% r r. r r r �_retrieve_fileR s
zJujuSSH._retrieve_fileN)r )�__name__�
__module__� __qualname__�__doc__�name� default_userr r r r �propertyr r r2 r3 r r r r r s
r )r �sos.collector.exceptionsr �sos.collector.transportsr �
sos.utilitiesr r r r r r �<module> s