File: //lib/python3/dist-packages/sos/collector/__pycache__/sosnode.cpython-38.pyc
U
i��ds� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d dlmZ d d lmZ d d
lmZmZmZmZ d d lmZ eeeee d �Z G d
d� d�Z dS )� N)�quote)�load)�
InitSystem)�JujuSSH)�SSHControlPersist)�LocalTransport)� OCTransport)�SaltStackMaster)�CommandTimeoutException�ConnectionException�UnsupportedHostException�InvalidTransportException)�
parse_version)�localZcontrol_persistZocZ saltstackZjujuc @ s� e Zd Zdgdd�Zedd� �Zdd� Zd d
� Zd d � Zed
d� �Z dd� Z
dd� Z dd� Z dd� Z
dhdd�Zedd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zdid.d/�Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z djd9d:�Z d;d<� Z d=d>� Z d?d@� Z!dAdB� Z"dCdD� Z#dEdF� Z$dGdH� Z%dIdJ� Z&dKdL� Z'dMdN� Z(dOdP� Z)dQdR� Z*dSdT� Z+dUdV� Z,dWdX� Z-dYdZ� Z.d[d\� Z/d]d^� Z0d_d`� Z1dadb� Z2dcdd� Z3dedf� Z4dS )k�SosNodeNTc
C s� |� � | _|| _|d | _| �� |d | _|d | _|d | _|d | _d| _ d | _
d | _ d | _ i | _
i | _|px| jj| _| jjs�| jjs�| j| j_|r�|| j_d | _d| _d| _g | _d g g g g |d d�| _d | _t�d
�| _t�d �| _ | � |�| _ | j j �r| j j d k| _z| j � | j� W n4 t!k
�rZ } z| �"d
| � � W 5 d }~X Y nX | �#� | _
| j j | _ | j �r�| jj$�r�d}| j%�r�|�r�| j
�s�| j �&� d S | j �r�| �'� �r�d| j
_(| j
j(�r�| �)� | �*� d S )NZ cmdlineopts�tmpdir�hostlen� need_sudo� sos_optionsF�sos_cmd)�version�enabled�disabled�options�presetsr � sosreportZsosZsos_ui�rootz!Unable to open remote session: %s)+�strip�address�commons�opts�_assign_config_optsr r r r r �host�cluster�hostname� sos_env_vars� _env_vars�password� _passwordZ
nopasswd_sudoZsudo_pw�sos_path� retrievedZhash_retrieved� file_list�sos_info�sos_bin�loggingZ getLogger�soslog�ui_log�_load_remote_transport�
_transportZ default_userZconnect� Exception� log_error�determine_host_policyZno_local� connected�
disconnect�check_in_container�
containerized�create_sos_container�_load_sos_info)�selfr r r' Z
local_sudoZ
load_facts�err� r> �7/usr/lib/python3/dist-packages/sos/collector/sosnode.py�__init__* sn
�
zSosNode.__init__c C s | j r| j jS dS )NT)r2 r6 �r< r> r> r? r6 l s zSosNode.connectedc C s | j �� dS )zDWrapper to close the remote session via our transport agent
N)r2 r7 rA r> r> r? r7 s s zSosNode.disconnectc C sv | j dkr d| _t| j |�S | jjt�� kr@t| jj | j |�S | jjdkrj| �d| jj � t| jj��t | j |�S )z�Determine the type of remote transport to load for this node, then
return an instantiated instance of that transport
)� localhostz 127.0.0.1T�autoz6Connection failed: unknown or unsupported transport %s)
r r r r � transport�
TRANSPORTS�keysr4 r
r )r< r r> r> r? r1 x s
�� z SosNode._load_remote_transportc C s d� | j| jd |�S )Nz {:<{}} : {}� )�format� _hostnamer �r< �msgr> r> r? �_fmt_msg� s zSosNode._fmt_msgc C sl | j sf| jrtj�� | _ nL| �d�}|d dkrf|d �d�D ]&}|sHq>|�d�}|d | j |d < q>| j S )Nz
env --null�statusr �output� �=rG )r& r �os�environ�copy� run_command�split)r< �retZln�_valr> r> r? �env_vars� s
zSosNode.env_varsc C s` || _ | j �d| j� | j �d| jj� | j �d| jd � | j �dd� | j �d| jj� dS ) z>Set the manifest section that this node will write to
r$ �policyZ sos_versionr �final_sos_command� rD N)�manifest� add_fieldrI r" �distror, r2 �name)r< r\ r> r> r? �set_node_manifest� s zSosNode.set_node_manifestc C s< t j�d�r| �d� dS t j�d�dk r8| �d� dS dS )zV
Tries to identify if we are currently running in a container or not.
z/run/.containerenvz/Found /run/.containerenv. Running in container.TZ containerNz/Found env var 'container'. Running in containerF)rQ �path�exists� log_debugrR �getrA r> r> r? r8 � s
zSosNode.check_in_containerc C s� | j jr�| j j| jj| �� | jjd�}| j|dd�}|d dkr�|d dkr�d|d krh| �d � t �nd
|d kr�| �d � t �| j| j �
� dd�}|d d kr�| � d
| j j � dS | �d|d � t �n| �d|d � t �dS )zJIf the host is containerized, create the container we'll be using
)�imageZauthZ
force_pullT�� need_rootrM )r �} rh z unable to retrieve auth tokenrN zHCould not pull image. Provide either a username and password or authfilezunknown: Not foundz%Specified image not found on registryr z Temporary container %s createdz*Could not start container after create: %sz&Could not create container on host: %sN)
r" r9 r: r re �get_container_authZforce_pull_imagerT r4 r3 Zrestart_sos_container�log_infoZsos_container_name)r< �cmd�resrV r> r> r? r: � sB � �
� ���z SosNode.create_sos_containerc C sF | j jr$| jjd �| j j| j j�S | jjd �| j jp>| jj�S dS )zrDetermine what the auth string should be to pull the image used to
deploy our temporary container
�defaultN) r Z
registry_userr" ZruntimesZfmt_registry_credentialsZregistry_passwordZfmt_registry_authfileZregistry_authfileZcontainer_authfilerA r> r> r? ri � s � �zSosNode.get_container_authFc C s< z | j d| |d�}|d dkW S tk
r6 Y dS X dS )z3Checks for the presence of fname on the remote nodezstat %srf rM r FN)rT r3 )r< �fnamerg rl r> r> r? � file_exists� s
zSosNode.file_existsc C s | j rd| j kr| j S | jS )NrB )r$ r rA r> r> r? rI � s zSosNode._hostnamec C s d}t �|d|�S )zUAttempts to obfuscate sensitive information in log messages such as
passwordsz>(?P<var>(pass|key|secret|PASS|KEY|SECRET).*?=)(?P<value>.*?\s)z\g<var>****** )�re�sub)r< rK Zregr> r> r? �_sanitize_log_msg� s zSosNode._sanitize_log_msgc C s | j �| �|�� dS )z:Format a ui message that includes host name and formattingN)r0 �inforL rJ r> r> r? �ui_msg� s zSosNode.ui_msgc C s0 t �� d d }d| j||f }| j�|� dS )z#Used to print and log info messagesrG � �
[%s:%s] %sN)�inspect�stackrI r/ rs �r< rK �callerZlmsgr> r> r? rj � s zSosNode.log_infoc C s0 t �� d d }d| j||f }| j�|� dS )z$Used to print and log error messagesrG ru rv N)rw rx rI r/ �errorry r> r> r? r4 s zSosNode.log_errorc C s: | � |�}t�� d d }d| j||f }| j�|� dS )z$Used to print and log debug messagesrG ru rv N)rr rw rx rI r/ �debug)r< rK rz r> r> r? rc s
zSosNode.log_debugc C s& | j jrdt|� S | jr"d| S |S )z�If we need to provide a sudo or root password to a command, then
here we prefix the command with the correct bits
zsu -c %sz
sudo -S %s)r � become_rootr r )r< rk r> r> r? � _format_cmd s
zSosNode._format_cmdc
C s� d}d}| j jdkrN| j j�| j j�}|dk r�d�|d �}|d r�|d }n8| j| j jddd�}|d dkr�|d �� �d
�\}}}|r�t |�d��d kr�|d 7 }z|d
|�d�d 7 }W n0 t
k
r� } z| � d| � W 5 d}~X Y nX || j d< | j d �r | �
d| j d � n$| j| jjk�s6| �d� d| _dS | �d��rRd| _d| j }| j|ddd�}|d dk�r�| �|d � | �d��r�| �� dS )zLQueries the node for information about the installed version of sos
N�.r �releaseT��
use_containerrg rM r rN �-� �.0�-%sz%Unable to fully parse sos release: %szsos version is %sz!sos is not installed on this nodeFz4.0z
sos reportz%s -l�3.6)r" Zcontainer_version_command�package_managerZ pkg_versionZ sos_pkg_name�joinrT r rU �lenr3 rc r, rj r r Zprimaryr4 r6 �check_sos_versionr- �_load_sos_plugins�_load_sos_presets) r< �verZrel�pkgZpkgs�_r= rk �sosinfor> r> r? r; sH
�
zSosNode._load_sos_infoc C sl d| j }| j|ddd�}|d dkrh|d �� D ]4}|�� �d�r2|�d�d �� }| jd �|� q2d S )
Nz%s --list-presetsTr� rM r rN zname:rG r )r- rT �
splitlinesr �
startswithrU r, �append)r< rk rl �lineZpnamer> r> r? r� O s
zSosNode._load_sos_presetsc C s� d}d}d}d}d}|d | }|d | }|d | } |d d }
| � ||�| jd< | � ||�| jd < | � | |�| jd
< | � |
|d �| jd < d S )
Nz,The following plugins are currently enabled:z-The following plugins are currently disabled:z4The following options are available for ALL plugins:z+The following plugin options are available:z Profiles:z(.*?)z
r r r TZprofiles)�_regex_sos_helpr, ) r< r� ZENABLEDZDISABLEDZ ALL_OPTIONSZOPTIONSZPROFILESZ enableregZdisregZoptregZproregr> r> r? r� X s zSosNode._load_sos_pluginsc
C s| g }t �||t j�D ]b}|�� D ]T}|sVz|�|�� d � W qt tk
rR Y qtX q |�d�}|�dd� |D �� q q|S )Nr �,c s s | ]}|� � r|� � V qd S �N�r )�.0�pr> r> r? � <genexpr>t s z*SosNode._regex_sos_help.<locals>.<genexpr>)rp �findall�Sr� r� rU r3 �extend)r< Zregexr� Zis_listrl �resultr� �rr> r> r? r� i s
zSosNode._regex_sos_helpc
C s\ z | � d| � | j�|�W S tk
rV } z | �d||f � W Y �dS d}~X Y nX dS )z1Reads the specified file and returns the contentszReading file %sz Exception while reading %s: %sr[ N)rj r2 � read_filer3 r4 )r< Zto_readr= r> r> r? r� w s zSosNode.read_filec C st | j r&| �d| jd j � | jd S ti | jjt� d| jj | �
d�d�}|rb| �d|j � |S | � d� t �dS ) z[Attempts to identify the host installation against supported
distributions
zusing local policy %srY Tz/etc/os-release)�cache�sysrootZinitZ
probe_runtimeZ remote_execZ remote_checkzloaded policy %s for hostz4Unable to determine host installation. Ignoring nodeN)
r rj r r^ r r r� r r2 rT r� r4 r )r< r" r> r> r? r5 � s
�
�
z SosNode.determine_host_policyc
sp � fdd�}||�}z t � jd �}t |�}||kW S tk
rj } z� �d| � W Y �dS d}~X Y nX dS )a� Checks to see if the sos installation on the node is AT LEAST the
given ver. This means that if the installed version is greater than
ver, this will still return True
:param ver: Version number we are trying to verify is installed
:type ver: ``str``
:returns: True if installed version is at least ``ver``, else False
:rtype: ``bool``
c
s� zZ| � d�d }d}d| kr8d| � d�d � d�d }t|� d��dkrR|d7 }|| W S tk
r� } z � �d| |f � | W Y �S d }~X Y nX d S ) Nr� r r[ ���r r� r� zUnable to format '%s': %s)rU r� r3 rc )r� Z_fverZ_relr= rA r> r? �_format_version� s
z2SosNode.check_sos_version.<locals>._format_versionr z Error checking sos version: %sFN)r r, r3 r4 )r< r� r� Z_verZ _node_verZ _test_verr= r> rA r? r� � s
zSosNode.check_sos_versionc C s | j s
dS | j j�|�dk S )z2Checks if a given package is installed on the nodeFN)r" r� Z pkg_by_name)r< r� r> r> r? � is_installed� s zSosNode.is_installed� c
C s� | j sn| jsn| �d� z$| j�| j�}|s8| �d� t�W n2 tk
rl } z| �d| � � W 5 d}~X Y nX |r�| jj r�| j�
|�}|r�| � |�}d|kr�d}|r�| j } | �
|�}| j�|||||�S )a� Runs a given cmd, either via the SSH session or locally
Arguments:
cmd - the full command to be run
timeout - time in seconds to wait for the command to complete
get_pty - If a shell is absolutely needed to run a command, set
this to True
need_root - if a command requires root privileges, setting this to
True tells sos-collector to format the command with
sudo or su - as appropriate and to input the password
use_container - Run this command in a container *IF* the host is
containerized
z-Node is disconnected, attempting to reconnectzFailed to reconnect to nodez#Error while trying to reconnect: %sNZatomicT)r6 r rc r2 Z reconnectr( r r3 r" r9 Zformat_container_commandr~ rX �updaterT )
r< rk �timeout�get_ptyrg r� �envZ reconnectedr= Z_cmd_envr> r> r? rT � s,
�zSosNode.run_commandc C sX z6| � � }|r| �|� n
| �d� | jr4| �� | _W n tk
rJ Y nX | �� dS )z.Run an sos report on the node, then collect itz'Unable to determine path of sos archiveN)�execute_sos_command�finalize_sos_pathr4 r) �retrieve_sosreportr* r3 �cleanup)r< ra r> r> r? r � s
zSosNode.sosreportc C s || j d kS )z/Verifies if the given preset exists on the noder �r, )r< �presetr> r> r? �_preset_exists� s zSosNode._preset_existsc s& t � fdd�| jd | jd fD ��S )z/Verifies if the given plugin exists on the nodec 3 s | ]}� |kV qd S r� r> )r� �s��pluginr> r? r� � s z)SosNode._plugin_exists.<locals>.<genexpr>r r )�anyr, �r< r� r> r� r? �_plugin_exists� s �zSosNode._plugin_existsc C s || j d kS )z6Checks to see if the plugin is default enabled on noder r� r� r> r> r? �_check_enabled� s zSosNode._check_enabledc C s || j d kS )z7Checks to see if the plugin is default disabled on noder r� r� r> r> r? �_check_disabled� s zSosNode._check_disabledc C sR |� d�d }| �|�s dS | �|�r6|| jjkr6dS | �|�rN|| jd kS dS )z�Attempts to verify that the given option is available on the node.
Note that we only get available options for enabled plugins, so if a
plugin has been force-enabled we cannot validate if the plugin option
is correct or notr r Fr T)rU r� r� r �enable_pluginsr� r, )r< �opt�plugr> r> r? �_plugin_option_exists s
�
z SosNode._plugin_option_existsc s d� � fdd�|D ��S )z^Returns a comma delimited list for sos plugins that are confirmed
to exist on the noder� c 3 s | ]}� � |�r|V qd S r� �r� �r� �orA r> r? r� s
z,SosNode._fmt_sos_opt_list.<locals>.<genexpr>)r� )r< r r> rA r? �_fmt_sos_opt_list s zSosNode._fmt_sos_opt_listc C s
|| _ dS )zVExpose the node to the cluster profile determined for the
environment
N)r# )r< r# r> r> r? � set_cluster s zSosNode.set_clusterc s, | j jr,| js| j j| _n| �d| j j � | j jrX| j jD ]}|| jkr<| j�|� q<| j jr�| j jD ] � � | jkrh| j j� | j� < qh| j jr�| j jD ]<� t � fdd�| j
D ��s�d� | j j� f }| j
�|� q�| j � | ��r| j j � | j �
| � W 5 Q R X n | j j � | j �| � W 5 Q R X dS )a+ This is used to modify the sos report command run on the nodes.
By default, sos report is run without any options, using this will
allow the profile to specify what plugins to run or not and what
options to use.
This will NOT override user supplied options.
zYCluster specified preset %s but user has also defined a preset. Using user specification.c 3 s | ]}� |kV qd S r� r> r� �r� r> r? r� : s z2SosNode.update_cmd_from_cluster.<locals>.<genexpr>�%s=%sN)r# Z
sos_presetr� rj Z sos_pluginsr� r� r Zsos_plugin_optionsr� �plugoptsZcheck_node_is_primary�lockZset_primary_optionsZset_node_options)r< r� Zoptionr> r� r? �update_cmd_from_cluster s4 �
�
z SosNode.update_cmd_from_clusterc C sJ t | jj�| _t | jj�| _t | jj�| _t | jj�| _t | jj�| _dS )z�From the global opts configuration, assign those values locally
to this node so that they may be acted on individually.
N)�listr � only_plugins� skip_pluginsr� r� r� rA r> r> r? r! G s
zSosNode._assign_config_optsc s� � j d }� �� }|r&d|t|�f }g }� �d�rZ� jjdkrZ|�dtt� jj�� � � �d�r�� jjr�|�dtt� jj�� � � �d�r҈ jj r�|�d � � jj
r�|�d
� � jj r�|�d t� jj � � � �d ��r*� jj �r|�d
td�
� jj �� � � jj�r*|�dtd�
� jj�� � � �d��rZ� jj�rZ|�dtt� jj�� � � �d��sr� �d��r�� jjdk�r�|�d� jj � � jj�r�|�d� jj � � �d��r� jj�r�|�d� jj� �� � jj�r�|�d� � �� |�dtj�
� jj� j��}� jD ]0}� j| }|�d|� d |�r:|nd � �� �q� j �r�� fd d �� j D �}|�r�|�d td�
d!d"� |D ��� � � j �rȈ � � j ��r�|�d#t� j � � n� � d$� j � � j �rx� fd%d �� j D �}t!|�t!� j �k�r t"t#� j �t#|� �}� � d&| � � �$� j �} | �r@|�d't| � � d(|d �
|�f � _%� �&d)� j% � � j'�(d*� j%� d+S � j)�r� fd,d �� j)D �}
t!|
�t!� j)�k�r�t"t#� j)�t#|
� �} � � d-| � � �$|
�} | �r�|�d.t| � � � j*�rd� fd/d �� j*D �}t!|�t!� j*�k�rBt"t#� j*�t#|� �}
� � d0|
� � �$|�}|�rd|�d1t|� � d(|d �
|�f � _%� �&d)� j% � � j'�(d*� j%� d+S )2zZUse host facts and compare to the cluster type to modify the sos
command if neededr z%s %s r� � z --threads=%sz3.7z--plugin-timeout=%sz3.8z--allow-system-changesz
--no-env-varsz
--since=%sz4.1z--skip-commands=%sr� z--skip-files=%sz4.2z--cmd-timeout=%sz4.3z4.2-13rC z--container-runtime=%sz--namespaces=%sz4.5.2z--journal-size=z--low-priorityr z--� r[ c s8 g | ]0}� � |�d �d �r� �|�d�d �r|�qS )r r rP )r� rU r� r� rA r> r? �
<listcomp>� s �z,SosNode.finalize_sos_cmd.<locals>.<listcomp>z-k %sc s s | ]
}|V qd S r� r> r� r> r> r? r� � s z+SosNode.finalize_sos_cmd.<locals>.<genexpr>z --preset=%sz?Requested to enable preset %s but preset does not exist on nodec s g | ]}� � |�r|�qS r> r� r� rA r> r? r� � s
zBRequested plugins %s were requested to be enabled but do not existz--only-plugins=%sz%s %szFinal sos command set to %srZ Nc s g | ]}� � |�r|�qS r> )r� r� rA r> r? r� � s
zARequested to skip plugins %s, but plugins are already not enabledz--skip-plugins=%sc s. g | ]&}|� j kr� �|�r� �|�r|�qS r> )r� r� r� r� rA r> r? r� � s
�zORequested to enable plugins %s, but plugins are already enabled or do not existz--enable-plugins=%s)+r, �determine_sos_labelr r� r Zthreadsr� �strZplugin_timeoutZallow_system_changesZ no_env_varsZsinceZ
skip_commandsr� Z
skip_filesZ cmd_timeoutZcontainer_runtimeZ
namespacesZ journal_sizeZ low_priorityr� �replacerQ ra r" Z sos_bin_pathr- r r� r� r� rc r� r� r� �setr� r rj r\ r] r� r� )r< r �labelZsos_optsr� rW r ZplugsZnot_onlyZonly�skipZnot_skipZskiplnZnot_on�enabler> rA r? �finalize_sos_cmdR s�
�
��
�
�
�
�
�
$"�� �
�
zSosNode.finalize_sos_cmdc C s� d}|| j �| �7 }| jjr<||s.d| jj n
d| jj 7 }|sDdS | �d| � | �d�rbd}n d}d | j�d
�d |f }d ||f S )
z?Determine what, if any, label should be added to the sos reportr[ z%sr� Nz Label for sos report set to %sr� z--labelz--namez%s-%sr r r� )r# Zget_node_labelr r� rc r� r rU )r< r� Zlcmdr> r> r? r� � s
�
zSosNode.determine_sos_labelc C s\ | j j}|r|�|d�}|�� d }| �d| � || _|�d�d | _| j�d| j� dS )z\Use host facts to determine if we need to change the sos path
we are retrieving fromr[ r zFinal sos path: %s�/r� Zcollected_archiveN) r" Zsos_path_stripr� rU rj r) �archiver\ r] )r< ra Zpstripr> r> r? r� � s zSosNode.finalize_sos_pathc C sV |dkr dS |dkr d|kr dS |dkr,dS t |�dkrJ|�d �dd� S d
| S d S ) Nr� z+sos report process received SIGKILL on noderG Zsudozsudo attempt failed� z4sos report terminated unexpectedly. Check disk spacer �
zsos exited with code %s)r� rU )r< Zrc�stdoutr> r> r? �determine_sos_error� s zSosNode.determine_sos_errorc
C s� | � d� �z0d}d}| j| j| jjddd| jd�}|d dk�r|d �� D ]H}t�|d�rf|�� }|� d �r�|�
d
�d }qN|� d �rN|�
� d
}qN|r�| j � d|� t
|�dkr�| j � dd� n*t
|�dkr�| j � dd� n| j � dd� n| j � dd� n6| �|d |d �}| �d|d |d f � t|��|W S tk
�r^ | �d� � Y nF tk
�r� } z&| �d|� �� | � d|� �� � W 5 d}~X Y nX dS )z2Run sos report and capture the resulting file pathzGenerating sos report...FT)r� r� rg r� r� rM r rN z*sosreport-*tar*)z sha256 z md5 � rG zThe checksum is: ru �checksum� Z
checksum_typeZmd5�@ Zsha256�unknownz*Error running sos report. rc = %s msg = %szTimeout exceededz'Exception during sos report execution: zError running sos report: N)rt rT r r r� r% r� �fnmatchr r� rU r\ r] r� r� rc r3 r
r4 rj )r< ra r� rl r� r= r> r> r? r� sP
�
�
zSosNode.execute_sos_commandc
C s� | j d }tj�||�d�d �}zD| �|�rP| �d||f � | j�||�W S | � d| � W dS W n: t
k
r� } z | � d||f � W Y �dS d}~X Y nX dS )z7Copies the specified file from the host to our temp dirr� r� z Copying remote %s to local %szFAttempting to copy remote file %s, but it does not exist on filesystemFzFailed to retrieve %s: %sN) r rQ ra r� rU ro rj r2 �
retrieve_filerc r3 )r< ra Zdestdir�destr= r> r> r? r� 0 s
��
zSosNode.retrieve_filec
C s� d� |�� �}zzt|�d��dkr6| �d| � W dS | �|�rr| �d| � d| }| j|dd �}|d
d kW S | �d | � W dS W n: tk
r� } z | �d
||f � W Y �dS d}~X Y nX dS )z|Removes the spciefied file from the host. This should only be used
after we have retrieved the file already
r[ r� r� zJRefusing to remove path %s: appears to be incorrect and possibly dangerousFzRemoving file %szrm -f %sTrf rM r zHAttempting to remove remote file %s, but it does not exist on filesystemzFailed to remove %s: %sN)r� rU r� rc ro rj rT r3 )r< ra rk rl �er> r> r? � remove_fileA s& �
�
zSosNode.remove_filec
C s8 | j r�| js| jjrFz| �| j � W n tk
rD | �d� Y dS X | �d| j � | � d� z| �
| j �}W n2 tk
r� } z| �|� W Y �dS d}~X Y nX |r�| � d� | j � | j �
d�d � d S | � d
� dS nV| j�� r�| j�� }nd d � | j�� D �d }| j�d
| j|f � | �d| � dS dS )z+Collect the sosreport archive from the nodez Failed to make archive readableFz Retrieving sos report from %szRetrieving sos report...Nz!Successfully collected sos reportr� r� Tz Failed to retrieve sos reportc S s g | ]}|j r|� � �qS r> r� )r� �xr> r> r? r� t s z.SosNode.retrieve_sosreport.<locals>.<listcomp>z"Failed to run sos report on %s: %sz Failed to run sos report. %s)r) r r r} �make_archive_readabler3 r4 rj r rt r� r+ r� rU �stderr�readr� � readlinesr/ r{ )r< rV r= r� r> r> r? r� X s8
�zSosNode.retrieve_sosreportc C sP | j dks| jrdS d| j kr2| �d| j � dS | �| j �}|sL| �d� dS )zuRemove the sosreport archive from the node, since we have
collected it and it would be wasted space otherwiseNr zFNode sos report path %s looks incorrect. Not attempting to remove pathzFailed to remove sos report)r) r rc r� r4 )r< Zremovedr> r> r? �remove_sos_archivez s
� zSosNode.remove_sos_archivec C sL | � � | jr,dD ]}| �| j| �r q,q| j�� }|rH| j|dd� dS )z<Remove the sos archive from the node once we have it locally)z.sha256z.md5Trf N)r� r) r� r" Zset_cleanup_cmdrT )r< Zextr� r> r> r? r� � s
zSosNode.cleanupc C s� |D ]�}z�| j s| jjrtz| �|� W nL tk
rr } z.| �d| � | �d||f � W Y �W qW 5 d}~X Y nX | �|�}|r�| j� |�
d�d � | � |� n| �d| � W q tk
r� } zd| }| �|� W 5 d}~X Y qX qdS )z4Collect the file created by a cluster outside of soszUnable to retrieve file %sz#Failed to make file %s readable: %sNr� r� z1Error collecting additional data from primary: %s) r r r} r� r3 r4 rc r� r+ r� rU r� )r< � filenames�filenamer= rV r� rK r> r> r? �collect_extra_cmd� s& �
zSosNode.collect_extra_cmdc C sJ d| }| j |ddd�}|d dkr(dS d}| �|||d f � t�dS ) z�Used to make the given archive world-readable, which is slightly
better than changing the ownership outright.
This is only used when we're not connecting as root.
z chmod o+r %s�
T)r� rg rM r z6Exception while making %s readable. Return code was %sN)rT r4 r3 )r< �filepathrk rl rK r> r> r? r� � s z SosNode.make_archive_readable)NNT)F)F)r� FFFN)5�__name__�
__module__� __qualname__r@ �propertyr6 r7 r1 rL rX r` r8 r: ri ro rI rr rt rj r4 rc r~ r; r� r� r� r� r5 r� r� rT r r� r� r� r� r� r� r� r� r! r� r� r� r� r� r� r� r� r� r� r� r� r> r> r> r? r ( st �
B
(
3
$ �
')
," r ) r� rw r. rQ rp Zpipesr Z sos.policiesr Zsos.policies.init_systemsr Z sos.collector.transports.jujur Z(sos.collector.transports.control_persistr Z sos.collector.transports.localr Zsos.collector.transports.ocr Z"sos.collector.transports.saltstackr Zsos.collector.exceptionsr
r r r
Z
sos.utilitiesr rE r r> r> r> r? �<module> s* �