o

-�_g��@sBdd
lZdd
l
Z
dd
lZddlmZmZmZ
Gdd�dee�Zd
S)�N)�Plugin�UbuntuPlugin�	PluginOptc@sXeZ
dZd
ZdZdZdZdZedde	dd	�ed
de
dd	�gZd
d�Zdd�Z
dd�ZdS)�SunbeamzSunbeam Openstack Controller�sunbeam)
�cloud)
�	openstackz/var/snap/openstack/common�sunbeam-user�ubuntuz&The user used for sunbeam installation)�default�val_type�desc�juju-allow-loginFzAllow sos to login to jujuc
Cs�
|�d
�

|�
|j�d�|j�d�|j�d�|j�d�|j�d�dg�

|jgd�
d	d
�
|�d�
}
|
dd
krNt�|
d�
}|D]}|�d|d���

qA|�d�
}zt�	|�
}Wnt
ym


|�d|�d��

YdSw|r�|j�d�}|�
|�d�|�d�|�d�g�

|�d�
r�|�
d|�d��

|j
d|d�}|j
d|dd�}d }	|dd
ks�|dd
kr�z!|d��d
}
|d��d
}d!|
vs�d!|vr�|�|�

d	}	Wn	ty�


Yn
w|	s�|�d"�

dSdSdS)#Nzsnap.openstack.*z
/hooks.logz/state/daemon.yaml�#/state/truststore/sunbeam.maas.yamlz/state/database/info.yamlz/state/database/cluster.yamlz'/var/snap/openstack/current/config.yaml)zsunbeam cluster listz"sunbeam cluster list --format yamlzsunbeam manifest listT)
�snap_cmdz#sunbeam manifest list --format yaml�statusr
�outputzsunbeam manifest show --id �
manifestidr	zUser "z�" does not exist, will not collect juju information. Use `-k sunbeam.sunbeam-user` option to define the user to use to collect data for sunbeamz/snap/openstack/commonz/*.logz/etc/*/*.logz/logs/*.logrzsu - z -c "sunbeam utils juju-login"zjuju whoami�
�runaszjuju status�)r�timeoutF�
Controllerz�juju is not logged in, will not collect juju information. Use `-k sunbeam.juju-allow-login=True` to login or use `juju login` as the sunbeam user to login)�add_service_status�
add_copy_spec�
common_dir�add_cmd_output�collect_cmd_output�yaml�	safe_load�
get_option�pwd�getpwnam�KeyError�	_log_warn�pw_dir�exec_cmd�
splitlines�_get_juju_cmd_details�
IndexError)�self�manifest_raw�	manifests�manifest�sunbeam_user�user_pwd�sb_snap_homedir�juju_whoami�juju_status�	logged_in�j_whoami�j_status�r6�</usr/lib/python3/dist-packages/sos/report/plugins/sunbeam.py�setup"s~











�
	�
�



�





��



�



�

�






�
�

��!z
Sunbeam.setupc		Cs
|jd
|
dd�
|j
d|
d�}|ddkr�t�|d�
}|d	��D]_}|jd
|��d|��d|��d|�d
�g|
dd�
|j
d
|�d
�|
d�}|ddkrt�|d�
}|dD]%}|�d|d��}|jd|��d|�d
�d|��d|�d
�g|
dd�
qYq"dSdS)N�juju controllersT)rrzjuju controllers --format jsonrrr
r�controllerszjuju models -c zjuju model-defaults -c zjuju controller-config -c z --format json�models�
:�namezjuju status -m zjuju model-config -m )rr�json�loads�keys)	r*�user�juju_controllers�juju_ctrl_json�
controller�juju_models�juju_status_json�model�
model_namer6r6r7r(qsD


�






��


�






����zSunbeam._get_juju_cmd_detailsc

Cs*|�|j
�d
��

|�d�

|�d�

dS)Nrr9zjuju controller-config)�do_file_private_subr�do_cmd_private_sub)
r*r6r6r7�postproc�s


�

zSunbeam.postprocN)�__name__�
__module__�__qualname__�
short_desc�plugin_name�profiles�packagesrr�str�bool�option_listr8r(rKr6r6r6r7rs 


�
��O"r)r>r!r�sos.report.pluginsrrrrr6r6r6r7�<module>
s