File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/__pycache__/batch.cpython-310.pyc
o
�N�g=5 � @ sp d Z ddlZddlZddlZddlZddlmZmZ ddlZddl Zddl
Zddl Ze� e
�ZG dd� d�ZdS )z
Execute batch runs
� N)�datetime� timedeltac @ s: e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� ZdS )�Batchz-
Manage the execution of batch runs
NFc C sH || _ |r|ni | _|r|ni | _|| _|| _tjj|d dd�| _dS )a
:param dict opts: A config options dictionary.
:param dict eauth: An eauth config to use.
The default is an empty dict.
:param bool quiet: Suppress printing to stdout
The default is False.
Z conf_fileT)�listenN) �opts�eauth�
pub_kwargs�quiet�options�salt�clientZget_local_client�local)�selfr r r Z_parser� r �B/opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/batch.py�__init__ s zBatch.__init__c C s< | j d dg | j d g}| j �dd�}|dur |�|� n
|�| j �dd�� d| jd < | jj|d
| j d
i| j��}t� }t� }|D ]P}d |v rV|d D ]}|�|� qMqCz
tt |�
� ��}W n t yt | j sqt
jj�d
� Y n w |dur�d|| v r�|| d du r�t�d|� qC|�|� qCt|�||�|�fS )zC
Return a list of minions to use for the batch run
Ztgtz test.ping�timeout�selected_target_optionNZtgt_type�globTZyield_pub_data�gather_job_timeout�minionsZjidz No minions matched the target.�failedz@minion '%s' failed test.ping - will be returned as a down minion)r �get�appendr r
Zcmd_iter�set�add�next�iter�keys�
StopIterationr r �utils� stringutils� print_cli�log�debug�list�
difference) r �argsr �ping_genZfretZnret�ret�minion�mr r r �gather_minions. sT �
��� �� �
�zBatch.gather_minionsc s� � fdd�}z4t � jd t�r3d� jd v r3|t� jd �d���}|dk r.tt�|��W S t|�W S t� jd �W S tyW � j sTt
j j �
d�� jd �� Y dS Y dS w )zA
Return the active number of minions to maintain
c s t | �d t� j� S )Ng Y@)�float�lenr )�x�r r r � partitione s z!Batch.get_bnum.<locals>.partitionZbatch�%� zEInvalid batch data sent: {}
Data must be in the form of %10, 10% or 3N)�
isinstancer �strr- �strip�int�math�ceil�
ValueErrorr r r r! r" �format)r r1 �resr r0 r �get_bnum` s
���zBatch.get_bnumc C sZ t �� }d}|t|�k r"|| |kr"|d7 }|t|�k r"|| |ks|r+|d |�= d S d S )Nr r3 )r �nowr. )r �waitr> �ir r r Z
__update_waitx s ��zBatch.__update_waitc c s: � | � � \| _| _| _g | jd | jd | jd dg}| �� }| js$dS t�| j�}g }i }g }| j�dd�}g }| j rE| j j
} | j j }
nd} d}
i } | j s^| jD ] } t
jj�d �| �� qQt|�t| j�k �rg }
|rs|rs| �|� t|�|t|� kr�|s�|r�|
�|�� � |s�n)t|t|� t|� �D ] }|r�|�� }t|t�r�|
�tt|��� q�|
�|� q�||
7 }|
|d< |
�r | j s�t
jj�d
t|
�� d �� | j�d | j�d
d��}| j j || j�dd�|| |
| jd d�| j ��}|�|� i | |< |
| | d< d| | d< nt � d� i }| jD ]#}|du �r ntt|�!� ��}|| jv�r5| j�|� |�|� �q|D ]�}z�d} t|�}|du �rYt � d� |d7 }|dk�rWng�q>| j�d��r�|�"|d d |i� |d d | | d v �r�| | d �#|d d � n6t
jj�d�|d �� n)|�"|� |D ]!}|| | d v �r�| | d �#|� �q�t
jj�d�|�� �q��q?W �q9 t$�y� || v �r�d| | d< | | d D ]}||v�r�i ||< i || d
< �q�Y �q9w |�%� D ]�\}}||v �r|�#|� |�r|�t&�'� t(|d � � d}|�d d�} | �r@t)�*d ||� | j �s2t
jj�d |� d �� | j�d!��r;d}|||< n�d} d"|v �r~t|d" t��rjz t+|d" �,� �|d"< W n t-�yi d|d"< Y nw | j�d!��rz|d" dk�rzd}|d" } | j�d��r�|||< || fV n|d
||< ||d
i| fV | j �s�|d
||< |�d
�||< d#|v �r�|�d#�} nd} t
j.�/|| | j� |�r�t)�0d$|� dS �q�| D ]7}| | d �s ||v �r |�#|� | | d D ] }||v �r |�#|� |�r |�t&�'� t(|d � � �q�q�t|�t| j�k sh| j �1� dS )%z'
Execute the batch run
Zfun�argr r% NZ
batch_waitr Fz/Minion {} did not respond. No job will be sent.z
Executing run on �
�returnr) � �rawr )rE r) �show_jid�verboser r T�activeg{�G�z�?g{�G�z�?r3 � �data�idzDminion {} was already deleted from tracker, probably a duplicate key)�secondsr z4Minion '%s' failed to respond to job sent, data '%s'zMinion 'z ' failed to respond to job sent�failhard�retcode�outzMMinion %s returned with non-zero exit code. Batch run stopped due to failhard)2r, r r( Z down_minionsr r= �copy�deepcopyr r
rF rG r r r r! r" r; r. �_Batch__update_waitr �pop�ranger4 �dictr r �sortedr
Zcmd_iter_no_blockr �time�sleepr �update�remover �itemsr r> r r# r$ �max�valuesr: �outputZdisplay_output�errorZdestroy) r r' ZbnumZto_runrH r) ZitersZbwaitr? rF Z show_verboseZminion_trackerZ down_minionZnext_r@ Z minion_idZ return_valueZnew_iter�partsZping_retr+ �queueZncnt�partrK r* rJ rM Z failed_checkrN rO r r r �run� sj ��
��
��
�� ��
�
�
�
��
��
�
�
�
��
� �� 3z Batch.run)NFN) �__name__�
__module__� __qualname__�__doc__r r, r= rR rc r r r r r s
2 r )rg rP �loggingr8 rW r r Z salt.clientr Zsalt.exceptionsZ salt.outputZsalt.utils.stringutils� getLoggerrd r# r r r r r �<module> s