| Current File : //usr/lib/python3.6/site-packages/sos/collector/transports/__pycache__/oc.cpython-36.pyc |
3
\��h�% � @ sL d dl Z d dlZd dlZd dlmZ d dlmZmZmZ G dd� de�Z dS )� N)�RemoteTransport)�
is_executable�sos_get_command_output�SoSTimeoutErrorc s~ e Zd ZdZdZdZdd� Zedd� �Zdd � Z d
d � Z
� fd d
�Z d� fdd� Z dd� Z
edd� �Zdd� Zdd� Z� ZS ) � OCTransportaq
This transport leverages the execution of commands via a locally
available and configured ``oc`` binary for OCPv4 environments.
The location of the oc binary MUST be in the $PATH used by the locally
loaded SoS policy. Specifically this means that the binary cannot be in the
running user's home directory, such as ~/.local/bin.
OCPv4 clusters generally discourage the use of SSH, so this transport may
be used to remove our use of SSH in favor of the environment provided
method of connecting to nodes and executing commands via debug pods.
The debug pod created will be a privileged pod that mounts the host's
filesystem internally so that sos report collections reflect the host, and
not the container in which it runs.
This transport will execute within a temporary 'sos-collect-tmp' project
created by the OCP cluster profile. The project will be removed at the end
of execution.
In the event of failures due to a misbehaving OCP API or oc binary, it is
recommended to fallback to the control_persist transport by manually
setting the --transport option.
�oczsos-collect-tmpc K s t d| j� d|� �f|�S )z\Format and run a command with `oc` in the project defined for our
execution
zoc -n � )r �project)�self�cmd�kwargs� r
�/usr/lib/python3.6/oc.py�run_oc1 s zOCTransport.run_occ C s | j d| j� ��}|d dkS )Nz,wait --timeout=0s --for=condition=ready pod/�statusr )r �pod_name)r
�upr
r
r � connected: s zOCTransport.connectedc C s� dd| j jd�d � d�| jd�ddd d
d �d �d
dd
d �d �ddd
d �d �dddd �d �gd| jjsjdn| jjdgddd�gi ddd�d
dd�ddd�ddd�gddd �dddd �
g| jjr�d nd d | j dddd!�d"�S )#z�Based on our template for the debug container, add the node-specific
items so that we can deploy one of these on each node we're collecting
from
ZPodZv1�.r z-sos-collector)�name� namespacezsystem-cluster-critical�host�/Z Directory)�path�type)r ZhostPathZrunz/runZvarlogz/var/logz
machine-idz/etc/machine-idZFilezsos-collector-tmpz®istry.redhat.io/rhel8/support-toolsz /bin/bashZHOSTz/host)r �value)r Z mountPathT)Z
privilegedZ runAsUser)
r �imageZcommand�envZ resourcesZ volumeMountsZsecurityContext�stdinZ stdinOnceZttyZAlwaysZ IfNotPresentZNever)ZvolumesZ
containersZimagePullPolicyZ
restartPolicyZnodeNameZ hostNetworkZhostPIDZhostIPC)ZkindZ
apiVersion�metadataZpriorityClassName�spec)Zaddress�splitr Zoptsr Zforce_pull_image)r
r
r
r �get_node_pod_configA sT
z OCTransport.get_node_pod_configc C sf t d�s dS | j� }|d d | _tj| jd�\}| _t|ddd��}tj ||� W d Q R X | j
d | j� d
�� t d | j� ��}|d d
ks�d| j� d�|d kr�| j d� | j
d|d � �� dS | j
d| j� d�� y8| j
d| j� d�dd�}|d d
k�s| j d� dS W nP tk
�r0 | j d� dS tk
�r` } z| j d|� �� dS d }~X nX d S ) Nr Fr r )�dir�wzutf-8)�encodingz"Starting sos collector container '�'z
oc create -f r r zpod/z created�outputz Unable to deploy sos collect podz Debug pod deployment failed: zPod 'z=' successfully deployed, waiting for pod to enter ready statez wait --for=condition=Ready pod/z --timeout=30s�( )�timeoutz"Pod not available after 30 secondsz'Timeout while polling for pod readinessz)Error while waiting for pod to be ready: T)r r" r �tempfileZmkstempZtmpdir� pod_tmp_conf�open�json�dump� log_debugr Z log_errorr r � Exception)r
ZpasswordZpodconf�fd�cfile�outr �errr
r
r �_connect� s:
zOCTransport._connectc s0 |j d�r$d| j� d| j� d|� �S t� j|�S )Nr zoc -n z exec --request-timeout=0 z -- chroot /host )�
startswithr r �super�_format_cmd_for_exec)r
r )� __class__r
r r8 � s
z OCTransport._format_cmd_for_exec� FNc s" |r |d7 }t � j||||dd�S )N�
F)� use_shell)r7 � run_command)r
r r) Z need_rootr r< )r9 r
r r= � s zOCTransport.run_commandc C sX t jj| j�rt j| j� | jd| j� ��}d|d krT| jd| j� d|� �� dS dS )Nz delete pod Zdeletedr' zCalling delete on pod 'z
' failed: FT)�osr �existsr+ �unlinkr r r/ )r
Zremovedr
r
r � _disconnect� s zOCTransport._disconnectc C s d| j � d| j� d�S )Nzoc -n z exec --request-timeout=0 z -- /bin/bash -c)r r )r
r
r
r � remote_exec� s zOCTransport.remote_execc C sV | j ddd�}d|d kr dnd}| j d|� d |� d | j� d
|� �d d �}|d
dkS )Nz cp --retriesT)�stderrz unknown flagr' � z --retries=5zcp r �:� )r) r r )r r )r
�fname�dest�result�flagsr r
r
r �_copy_file_to_remote� s
z OCTransport._copy_file_to_remotec C sR | j ddd�}d|d kr dnd}| j d|� d | j� d
|� d |� ��}|d d kS )
Nz cp --retriesT)rC z unknown flagr' rD z --retries=5zcp r rE r r )r r )r
rG rH rI rJ r r
r
r �_retrieve_file� s $zOCTransport._retrieve_file)r: FNF)�__name__�
__module__� __qualname__�__doc__r r r �propertyr r" r5 r8 r= rA rB rK rL �
__classcell__r
r
)r9 r r s `*
r )
r- r* r> Zsos.collector.transportsr Z
sos.utilitiesr r r r r
r
r
r �<module> s