File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/__pycache__/salt.cpython-310.pyc
o
�N�g�B � @ s| d dl Z d dlZdejd< d dlZd dlZd dlZd dlmZm Z m
Z
m Z m Z m
Z
mZ d dlmZ G dd� dejjj�ZdS )� NZ
pkg_resources)�AuthenticationError�AuthorizationError�EauthAuthenticationError� LoaderError�SaltClientError�SaltInvocationError�SaltSystemExit)� yamlify_argc @ sr e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
d dd�Z dd� Z dd� Z
dd� Zdd� ZdS ) �SaltCMDz6
The execution of a salt command happens here
c sj ddl � | �� z| jjdk}� jj| �� |dd�| _W n ty8 } z| � d|� d�� W Y d}~dS d}~ww | jj
sA| jj rG| � � dS | jj
r\| �� }| �|| j�dd �� dS | jjdkrj| jjd
| j_| jd | jd | jd
| jj| jj| jjd�}d| jv r�ddl� z(� jj�tj� | jd d�d��}|� � |d< W d � n1 s�w Y W n t y� | jd |d< Y nw | jj |d< | j r�| j |d< nd|d< | jj!dk�r t"| �� �| jj!k�r � jj#�$d� | jj%| j_
z | � � W | j�&� dS | j�&� w t'| jd��rt'| jd�|d< t'| jd��r't'| jd�|d < t'| jd ��r8t(t'| jd ��|d < t'| jd ��rIt(t'| jd ��|d < t'| jd ��rZt(t'| jd ��|d < t'| jd!��rkt(t'| jd!��|d!< d|v�r�d|v�r�| jj�r�ddl)� � j*�+| j�}|�,| jj�}| jj-�r�|�r�|�.| jj|�}|�r�|�dd�|d< |�s�t/j0�1d"� t/� d� |�2|� | jj|d#< | jd$ �r�| jj3d1i |��} � jj#�$d%| � �� dS | j�s�dS g }
g } �zH�z| jj4�r�| jj5} | jj4|d&< d|d'< n| jj6} | jj7�rMd|d(< d| jd(< i }
| d1i |��D ])}d(}z| �8||� W n t9�y5 } zt:|��d}~ww d)|v�r@|
�2|� �q| �;|� | �<|
� nq| jd d*k�r|i }
d}| jj6d1i |��D ]}| �=|�\}}}|
�2|� �qb| j|
||d+� nB| jj>�r�d|d,< i }
| d1i |��D ]/}z | �=|�\}}}|
�?|� | j|||d+� |
�2|� W �q� t@�y� | �?|� Y �q�w | jd- du �r�| jd d*k�r�| jjAdu �r�| �<|
� | �B| � tC� fd.d/�|
D ���s�t/j0�1d0� t/� � jDjEjF� W n$ tGtHtItJtf�y } ztK|�}
| j|
ddd+� W Y d}~n
d}~ww W | j�&� dS W | j�&� dS | j�&� w )2z/
Execute the salt command line
r N� T)�skip_perm_errorsZauto_reconnect� �
�output�nested�timeout�tgt�fun�arg)r r r r � show_timeout�show_jid�tokenZcachedirz .root_key�r�key� delimiterZtgt_type�glob� zA
NOTICE: Too many minions targeted, switching to batch execution.�return�retZ
return_configZ
ret_configZ
return_kwargsZ
ret_kwargsZmodule_executorsZ
executor_opts�metadata� ERROR: Authentication failed
�eauth�asyncz Executed command with job ID: �subset�cli�progressZ return_count�sys.doc)�retcode�verboseZ cli_summaryc 3 s � | ]
}|� j jjkV qd S )N)�defaults� exitcodes�EX_OK)�.0Z exit_code��salt� �A/opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/salt.py� <genexpr>� s �
�z SaltCMD.run.<locals>.<genexpr>z0ERROR: Minions returned with non-zero exit code
r/ )LZ salt.client�
parse_args�optionsr! �clientZget_local_clientZget_config_file_path� local_clientr �exit�batch�static�
_run_batchZpreview_target�_preview_target� _output_ret�config�getr �optsr r Zsalt.utils.files�utils�filesZfopen�os�path�join�readline�OSErrorr �selected_target_optionZbatch_safe_limit�len� stringutils� print_cliZbatch_safe_sizeZdestroy�getattrr � salt.auth�auth�Resolverr$ �mktoken� token_cli�sys�stderr�write�updateZ cmd_asyncr# Z
cmd_subsetZcmd_clir% �
_progress_retr r �
_progress_end�_print_returns_summary� _format_retr( �append�KeyErrorr �_print_errors_summary�allr) r* Z
EX_GENERICr r r r �str)�selfr �excZ minion_list�kwargsZfp_�resolver�res�tokZjidZretcodes�errorsZcmd_funcr r% �out�full_retZret_r' r/ r- r0 �run sF ��� �
� �� � �
�
�
�
��
�
�
� �� ���
�z SaltCMD.runc C s | j �| jd | jp d�S )z>
Return a list of minions from a given target
r r )r5 Zgather_minionsr< rF )r] r/ r/ r0 r: � s �zSaltCMD._preview_targetc C s� dd l }i }d| jv r| jd |d< d|vr\| jjr\dd l}|j�| j�}|�| jj�}| jjrD|rD|� | jj|�}|rD|�
dd�|d< |sQt j �
d� t �d� |�|� | jj|d< | jjr�| jjsid| jd< z
|jjj| j|d d
�}W n ty� t �d� Y nw i }|�� D ] \}}|�|� q�| �|d� d S z| jj| jd< |jjj| j|| jd �}W n ty� t �d � Y nw d} |�� D ]
\}}
|
| kr�|
} q�t �| � d S )
Nr r r r r
r! z100%r7 T)r! �quiet)r! Z_parserr )Zsalt.cli.batchr< r3 r! rK rL rM r$ rN rO r= rP rQ rR r6 rS r8 r7 ZBatchr rf r; ) r] r. r! r` ra rb r7 r �_r' Z job_retcoder/ r/ r0 r9 � sV
�
� ��zSaltCMD._run_batchc C sb |r-t jj�d� t jj�d� t jj�d� t jj�d� |D ]}t jj�| �|�� q d S d S )Nr z---------------------------ZErrors)r. r? rH rI �
_format_error)r] rc �errorr/ r/ r0 rZ ) s �z SaltCMD._print_errors_summaryc
C s� d}d}g }g }g }g }|D ]M}|| } t | t�r$d| v r$|| �d�} t | t�rK| �d�rKd| v r8|�|� n d| v rA|�|� |d7 }|�|� q|d7 }| �|| �r[|�|� qtjj �
d� tjj �
d� tjj �
d � tjj �
d� tjj �
d
|| � �� tjj �
d |� �� tjj �
d |� �� tjj �
d
t |�� �� | j j
r�|r�tjj �
d�d�|��� |r�tjj �
d�d�|��� |r�tjj �
d�d�|��� tjj �
d� dS )z)
Display returns summary
r r �Minion did not returnz
Not connectedz No responser r z+-------------------------------------------ZSummaryz# of minions targeted: z# of minions returned: z"# of minions that did not return: z# of minions with errors: zMinions not connected: {}� zMinions not responding: {}zMinions with failures: {}N)�
isinstance�dictr= r\ �
startswithrX � _get_retcoder. r? rH rI rG r3 r( �formatrC )
r] r Zreturn_counterZnot_return_counterZnot_return_minionsZnot_response_minionsZnot_connected_minionsZfailed_minionsZ each_minionZ
minion_retr/ r/ r0 rV 2 sf �
� �� ����z SaltCMD._print_returns_summaryc C s dd l }|j�| j� d S )Nr )� salt.outputr Z progress_end� progress_bar)r] rd r. r/ r/ r0 rU j s zSaltCMD._progress_endc C s\ ddl }t| d�s!z |j�| j||�| _W n ty td��w |j�| j|| j|� dS )z'
Print progress events
r Nrs zp
WARNING: Install the `progressbar` python package. Requested job was still run but output cannot be displayed.
) rr �hasattrr Z get_progressr< rs � Exceptionr Zupdate_progress)r] r% rd r. r/ r/ r0 rT o s
��zSaltCMD._progress_retr c C sd ddl }| jd dkrt|t�s| �|� n |jj||| j|d� |s0tj� d� t�
d� dS dS )zG
Print the output from a single return to the terminal
r Nr r& )rd r> Z_retcodezERROR: No return received
r
) rr r< rm ru � _print_docsr �display_outputrP rQ rR r6 )r] r rd r' r. r/ r/ r0 r; � s �zSaltCMD._output_retc C sZ i }d}d}|� � D ] \}}|d ||< d|v r |d }| �|�}||kr'|}q
|||fS )zJ
Take the full return data and format it to simple output
r r r rd )�itemsrp )r] re r rd r' r �dataZ ret_retcoder/ r/ r0 rW � s
�
zSaltCMD._format_retc C sf d}t |t�r(|�dd�dkr(t |�dd�t�r$t|�dddi��� �S |d S t |t�r1|s1dS |S )z8
Determine a retcode for a given return
r r' r )rm rn r= �max�values�bool)r] r r' r/ r/ r0 rp � s zSaltCMD._get_retcodec C s& |� � D ] \}}d�||d �}q|S )Nz&Minion [{}] encountered exception '{}'�message)rx rq )r] Z minion_errorZminionZ error_docrj r/ r/ r0 ri � s
�zSaltCMD._format_errorc C s ddl }i }|s| �dd� t|t�r | �d|� d�� |D ]0}t|| t�r5|| �d�s4|| dkr5q || D ]}||vrM|| | rM|| | ||< q9q | jjrit|�D ]}|j�||| id| j � qWdS t|�D ] }|j
j � |� d �� |j
j � || � |j
j � d
� qmdS ) zR
Print out the docstrings for all of the functions on the minions
r Nr
z%No minions found to gather docs from
r rk Z
VALUE_TRIMMEDr �:r )
rr r6 rm r\ ro r3 r �sortedrw r< r? rH rI )r] r r. Zdocs�hostr r/ r/ r0 rv � s4
� �� � �zSaltCMD._print_docsN)r )�__name__�
__module__� __qualname__�__doc__rf r: r9 rZ rV rU rT r; rW rp ri rv r/ r/ r/ r0 r
s K; 8
r
)rA rP �modulesZsalt.defaults.exitcodesr. Zsalt.utils.parsersZsalt.utils.stringutilsZsalt.exceptionsr r r r r r r Zsalt.utils.argsr r? �parsersZSaltCMDOptionParserr
r/ r/ r/ r0 �<module> s
$