o

i��d�
�@s@dd
lZddl
mZ
ddlmZ
ddlmZ
Gdd�de�Zd
S)�N)
�JujuNotInstalledException)
�RemoteTransport)
�sos_get_command_outputc@sZeZ
dZd
ZdZdZdd�Zdd�Zdd	d
�
Zdd�Z	e
d
d��
Ze
dd��
Zdd�Z
dS)�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
Cs<d
}
ztj
|
tjdd�
WdStjy


|�d�

t�
w)Nzjuju versionT��stderr�shellzFailed to check `juju` version)�
subprocess�check_output�STDOUT�CalledProcessError�	log_errorr)�self�cmd�r�?/usr/lib/python3/dist-packages/sos/collector/transports/juju.py�_check_juju_installed's


�


�zJujuSSH._check_juju_installedcCsN|j�d
|
��}zt
j|t
jdd�
WdSt
jy&


|�d|
�d��

�w)Nz sudo chmod o+r TrzFailed to make z world-readable)�remote_execrrr
rr)r�fnamerrrr�_chmod0s


�

�zJujuSSH._chmod�cCs|��|_
|j
S�
N)r�
_connected)r�passwordrrr�_connect9s


zJujuSSH._connectc


Csd
S)NTr�
rrrr�_disconnect=s
zJujuSSH._disconnectc


Cs|jSr)
rrrrr�	connected@szJujuSSH.connectedc
Cs:|j�
d
�
\}
}|
rd|
��n
d}|�d|��}d|��S)N�
:�-m r�
 z	juju ssh )�address�split)r�model�
target_option�model_option�optionrrrrDs



zJujuSSH.remote_execcCs\|�|
�

|j
�d
�
\}}|rd|��n
d}d|�d|�d
|
�d|��}t|�
}|ddkS)	Nr r!rz	juju scp z -- -r r"�statusr
)rr#r$r)rr�destr%�unitr'r�resrrr�_retrieve_fileKs






zJujuSSH._retrieve_fileN)
r)�__name__�
__module__�__qualname__�__doc__�name�default_userrrrr�propertyrrr-rrrrrs

	
	



r)r�sos.collector.exceptionsr�sos.collector.transportsr�
sos.utilitiesrrrrrr�<module>
s