File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/__pycache__/caller.cpython-310.pyc
o
�N�gq+ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Zddl
Zddl Zddl Zddl
ZddlZddlZddlZddlZddlmZ ddlmZmZmZmZ e�e�ZG dd� d�Z G dd� d�Z G d d
� d
e �Z dS ) z]
The caller module is used as a front-end to manage direct calls to the salt
minion modules.
� N)�
LOG_LEVELS)�CommandExecutionError�CommandNotFoundError�SaltClientError�SaltInvocationErrorc @ s e Zd ZdZedd� �ZdS )�CallerzK
Factory class to create salt-call callers for different transport
c K s t | fi |��S )N)� ZeroMQCaller)�opts�kwargs� r �C/opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/caller.py�factory( s zCaller.factoryN)�__name__�
__module__� __qualname__�__doc__� staticmethodr
r r r r r # s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�
BaseCallerz*
Base class for caller transports
c
C sj || _ d| j d< z| j �d�rtj�|�| _W dS tj�|�| _W dS ty4 } ztt|���d}~ww )z+
Pass in command line opts
TZcallerZproxyidN) r �get�salt�minionZ SProxyMinionZSMinionr �
SystemExit�str)�selfr �excr r r �__init__2 s
��zBaseCaller.__init__c C sv i }| j j�� D ]\}}||vr|jr|j||< qt|�D ]}|�| j�dd��r8tj j
� |� d|| � d�� q dS )zT
Pick up the documentation for all of the modules and print it out.
�fun� z:
�
N) r � functions�itemsr �sorted�
startswithr r r �utils� stringutils� print_cli)r �docs�name�funcr r r �
print_docsC s
� ��zBaseCaller.print_docsc C s6 | j j�d�p
tj�| j�}tj�d|id| j� dS )z&
Print out the grains
�grains�localN)r r r r �loaderr* �output�display_output)r r* r r r � print_grainsP s zBaseCaller.print_grainsc C s* | j �dd�}z}tjj�|�}z| �� }W tjjj|| j �dd�dd� ntjjj|| j �dd�dd� w |�dd�}| j d rE|}d}n|�d
i �}tjj d |i|| j |�d d
�d� | j �dd�rmt
� |d � W dS |�d �tj j
jkr�t
� tj j
j� W dS W dS ty� } zt|��d}~ww )z-
Execute the salt call logic
�profiling_enabledFZprofiling_pathz
/tmp/statsT)Z
stats_path�stop�out�nestedZprint_metadata�returnr+ �retcoder )r2 r Z_retcodeZretcode_passthroughN)r r r r# ZprofileZactivate_profile�callZoutput_profiler- r. �sys�exit�defaults� exitcodes�EX_OK�
EX_GENERICr r )r r0 Zpr�retr2 Z print_ret�errr r r �runW sD
� �
����zBaseCaller.runc
sx i � | j d }tjj�| j �� d< tj�tj� | j d �� d �}|| jj
vrx| jj
d |� d��}|rG| jj
� |�||< d� d< |� d< � S t j
�| jj
� |�� |�d �d
}|| jjv rmt j
�d �| jj| �� nt j
�d � t �d
� | j �d�}|dur�tjj�|�}�zx|t�� � d dd�}|dur�||d< tjj| jj
| tjjj| j d | j �dg �d�|d�\}}z#tjj�|d��} | �tj� |�� W d � n1 s�w Y W n t y� Y n t y� t j
�d�|�� Y nw | jj
| }
||d�} | � |� t | jdg ��p tjj�| j �dd��} | j �dd��r2tjj�| j d �| d< t!| t"��r;| g} z4| D ]/}
|
� d �}|| jj#v�rTt$d |
� d ���| jj#| | j | |
||�� d< � d du�rl n�q>W nI t%�y� } z<t j
�d |� d �� tjj&�'|
j(� t)�| j d! �*� t+j,�}|t+j-k�r�t.�/� }t j
�|� t �tj0j1j2� W Y d}~nd}~ww z
| jj#j3d" �d#d
�}W n t4�y� tj0j1j2}Y nw |d
k�r�z
t5� fd$d%�d&D ��}W n t6�y� d'}Y nw |�s�tj0j1j2}|� d#< W ng t7�yC } z4d(}t)�| j d! �*� t+j,�}|t+j-k�r't j
�t.�/� � t j
�|�||�� t �tj0j1j2� W Y d}~n+d}~w t8�yi } zd)}t j
�|�||�� t �tj0j1j2� W Y d}~nd}~ww zt�9|� W n
t �y{ Y nw t:| jj
| d*��r�| jj
| j;}t!|t"��r�|� d< | j d+ �p�| j �d,d-�d+k�p�| j �d.�d/k}| j �dd0��d1�}|�r�|�r�| j d2 � d2< |� d< | j d � d3< |du�r�|� d< |D ]$}|�s�q�zd'� d4< | jj<|� d5� � � W �q� t6�y Y �q�w |�s(| j �d6d-��s(z� �=� }d7|d< | �>|� W � S t6�y' Y � S w | j d8 �r:tjj�?| j � d � � � S )9z!
Call the module
r �jidZcachedirzsys.doc�*r3 r2 r4 �.r z Possible reasons: {}
r ����metadataNz salt-call)r �pidr@ Ztgt�arg�no_parse)rG )�datazw+bzLCannot write to process directory. Do you have permissions to write to {} ?
)rF r Zmodule_executorsz
[direct_call]Z
executor_optsz.executez
Executor 'z' is not availablez
Passed invalid arguments: z
.
Usage:
Z log_level� __context__r5 c 3 s � | ] }� d � |d�V qdS )r4 TN)r )�.0�x�r= r r � <genexpr>� s �
�z"BaseCaller.call.<locals>.<genexpr>)�result�successTzError running '{0}': {1}
z*Command required for '{0}' not found: {1}
�
__outputter__r+ Z file_clientFZ master_type�disabler �,�idZfun_argsrO z .returnerZno_return_eventZreq�
cache_jobs)@r r r# r@ Zgen_jid�os�path�joinr Z get_proc_dirr Zmissing_fun_stringr7 �stderr�write�splitZfunction_errors�formatr8 r �argsZ yamlify_arg�getpidZload_args_and_kwargsZ parse_input�filesZfopen�payload�dumps� NameError�OSError�update�getattr�
isinstancer � executorsr � TypeErrorr$ r% r r �lower�logging�ERROR�DEBUG� traceback�
format_excr9 r: r< �pack�AttributeError�all� Exceptionr r �remove�hasattrrP � returners�copy�
return_pubrT )r r Zproc_fnr& �mod_namerD Zsdatar\ r
Zfp_r( rH rf r' �fnamer Z active_levelZtracer5 Z func_result�msgZoputZis_localrt ZreturnerZmretr rL r r6 z sN
�
��
�
�
� �� ��
� ���
���� ��
�
��
� ��� ��
� �� zBaseCaller.callN) r r r r r r) r/ r? r6 r r r r r - s
#r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r zT
Object to wrap the calling of local salt modules for the salt-call command
c s t � �|� dS )z2
Pass in the command line options
N)�superr )r r �� __class__r r r ' s zZeroMQCaller.__init__c C sp t jjjj| jdd��#}d| jd d�}|�� D ]\}}|||< q|�|� W d � dS 1 s1w Y dS )z2
Return the data up to the master
Z salt_call)�usageZ_returnrS )�cmdrS N)r �channel�clientZ
ReqChannelr
r r �send)r r= r �load�key�valuer r r rv - s
�
"�zZeroMQCaller.return_pub)r r r r r rv �
__classcell__r r r{ r r " s r ) r ri rU r7 rl r Zsalt.channel.clientZsalt.defaults.exitcodesZ salt.loaderZ salt.minionZ salt.outputZ salt.payloadZsalt.utils.argsZsalt.utils.filesZsalt.utils.jidZsalt.utils.minionZsalt.utils.profileZsalt.utils.stringutilsZ
salt._loggingr Zsalt.exceptionsr r r r � getLoggerr �logr r r r r r r �<module> s2
v