관리-도구
편집 파일: juju.cpython-39.opt-1.pyc
a -�_g�! � @ sb d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dd�Zdd� Z G d d � d e�Z dS ) � N)�Cluster)�sos_parse_version��sos_get_command_outputc C s | sg S dd� | � d�D �S )z Parse comma separated string.c S s g | ] }|� � �qS � )�strip)�.0�stringr r �?/usr/lib/python3.9/site-packages/sos/collector/clusters/juju.py� <listcomp> � z(_parse_option_string.<locals>.<listcomp>�,)�split)�stringsr r r �_parse_option_string s r c C s G dd� d�}|| �S )aR Helper function to get Index. The reason why we need Index defined in function is because currently the collector.__init__ will load all the classes in this module and also Index. This will cause bug because it think Index is Cluster type. Also We don't want to provide a customized filter to remove Index class. c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) z_get_index.<locals>.Indexz�Index structure to help parse juju status output. Attributes apps, units and machines are dict which key is the app/unit/machine name and the value is list of targets which format are {model_name}:{machine_id}. c S s( || _ i | _i | _i | _t�d�| _d S )NZsos)� model_name�apps�units�machines�loggingZ getLogger�ui_log)�selfr r r r �__init__. s z"_get_index.<locals>.Index.__init__c S s� |d � � D ]n\}}g }|�di �}|� � D ]B\}}|d }| j� d|� �} | g| j|< | g| j|< |�| � q,|| j|< q dS )z Adds principal units to index.� applicationsr �machine�:N)�items�getr r r �appendr ) r � juju_status�app�app_info�nodesr Zunit� unit_infor �noder r r �add_principals5 s z(_get_index.<locals>.Index.add_principalsc S s� |d � � D ]�\}}|�dg �}|D ]�}| j�|�sJ| j�d|� d�� q$| j| �| j| � d|d | vr�| j�d|� d�� q$|d | d }|� � D ]P\}}| j� d|d � �} |�d i �� � D ]"\} }| �|d �r�| g| j| < q�q�q$q d S ) z�Add subordinates to index. Since subordinates does not have units they need to be manually added. r zsubordinate-tozPrincipal charm z is missingr z is missing unitsr r Z subordinates�/N) r r r r Zwarning�extendr � startswithr ) r r r r! Zsubordinate_to�parentr �_r# r$ Zsub_keyr r r �add_subordinatesC s, � ��z*_get_index.<locals>.Index.add_subordinatesc S s2 |d � � D ] }| j� d|� �}|g| j|< q dS )z�Add machines to index. If model does not have any applications it needs to be manually added. r r N)�keysr r )r r r r$ r r r � add_machinesd s z&_get_index.<locals>.Index.add_machinesN)�__name__� __module__� __qualname__�__doc__r r% r+ r- r r r r �Index% s !r2 r )r r2 r r r � _get_index s Ir3 c @ s` e Zd ZdZd ZdZg d�Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� ZdS )�jujua! The juju cluster profile is intended to be used on juju managed clouds. It"s assumed that `juju` is installed on the machine where `sos` is called, and that the juju user has superuser privilege to the current controller. By default, the sos reports will be collected from all the applications in the current model. If necessary, you can filter the nodes by models / applications / units / machines with cluster options. Example: sos collect --cluster-type juju -c "juju.models=sos" -c "juju.apps=a,b,c" zJuju Managed Clouds))r � z1Filter node list by apps (comma separated regex).)r r5 z3Filter node list by units (comma separated string).)�modelsr5 z4Filter node list by models (comma separated string).)r r5 z6Filter node list by machines (comma separated string).c C s t �dd|dt j�S )z#Remove leading characters before {.z (^[^{]*)(.*)z\2r )�re�sub� MULTILINE)r �outputr r r �_cleanup_juju_output� s zjuju._cleanup_juju_outputc C s6 | � |�}t|d�}|�|� |�|� |�|� |S )z�Parse juju status output and return target dict. Here are couple helper functions to parse the juju principals units, subordinate units and machines. )r )�_execute_juju_statusr3 r% r+ r- )r r r �indexr r r �_get_model_info� s zjuju._get_model_infoc C s t d�}|d S )zGrab the version of jujuz juju versionr: r )r �resr r r �_get_juju_version� s zjuju._get_juju_versionc C s� |rd|� �nd}d}| � � }t|�td�kr6|d7 }| j� d|� d|� �}| �|�}|d d ksztd |� d |d � ���| �|d �}d }t�|�}|S ) Nz-m r5 z --format json�3z --no-colorz status � �statusr �'z' returned error: r: )r@ r �cmdZexec_primary_cmd� Exceptionr; �json�loads) r r Z model_optionZ format_optionZ juju_versionZ status_cmdr? Zjuju_json_outputr r r r r< � s zjuju._execute_juju_statusc C sF t � }|D ]6}t||��� D ]"\}}t�||�r |�|p:g � q q |S )zFilter with regex match.)�set�getattrr r7 �match�update�r �key�patterns� model_infor" �patternZparam�valuer r r �_filter_by_pattern� s zjuju._filter_by_patternc C sB t � }|D ]2}t||��� D ] \}}||kr |�|p6g � q q |S )zFilter with fixed match.)rI rJ r rL rM r r r �_filter_by_fixed� s zjuju._filter_by_fixedc C s dS )z'Dynamically change transport to 'juju'.r4 r )r r r r �set_transport_type� s zjuju.set_transport_typec C s� t | �d��}t | �d��}t | �d��}t | �d��}|||d�}t|�� �sX|sXg S |sbdg}t� }|D ]P}| �|�}|�� D ]8\} } | dkr�| �| | |�} n| �| | |�} |� | � q�qlt |�S )z+Get the machine numbers from `juju status`.r6 r r r )r r r r5 ) r Z get_option�any�valuesrI r> r rS rT rL �list) r r6 r r r �filtersr" ZmodelrP rN �resourceZ_nodesr r r � get_nodes� s$ zjuju.get_nodesN)r. r/ r0 r1 rE Z cluster_nameZ option_listr; r> r@ r< rS rT rU r[ r r r r r4 q s r4 )N) r rG r7 Zsos.collector.clustersr Z sos.utilitiesr r r r3 r4 r r r r �<module> s V