Current File : //usr/lib/python2.7/site-packages/salt/cli/batch.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l m Z m Z d d l
Z d d l Z d d l
Z d d l Z d d l m Z d d l m Z d d l Z e j e � Z d e f d � � YZ d S( u
Execute batch runs
i ( t absolute_importt print_functiont unicode_literalsN( t datetimet timedelta( t six( t ranget Batchc B@ sD e Z d Z d e d d � Z d � Z d � Z d � Z d � Z RS( u,
Manage the execution of batch runs
c C@ s� | | _ | r | n i | _ | r* | n i | _ | | _ t j j | d � | _ | j � \ | _ | _
| _ | | _ d S( Nu conf_file(
t optst eautht
pub_kwargst quiett saltt clientt get_local_clientt localt _Batch__gather_minionst minionst ping_gent down_minionst options( t selfR R R t parser( ( s2 /usr/lib/python2.7/site-packages/salt/cli/batch.pyt __init__! s c C@ s� | j d d g | j d g } | j j d d
� } | d
k rQ | j | � n | j | j j d d � � t | j d <| j j d | j d | | j � } t � } t � } x� | D]� } d
o� d | k r� x | d
D] } | j | � q� Wq� q� y t
t j | � � } Wn1 t
k
rD| j s@t j j j d � n Pn X| d
k r� | j | � q� q� Wt | � | | j | � f S( uC
Return a list of minions to use for the batch run
u tgtu test.pingu timeoutu selected_target_optionu tgt_typeu globu yield_pub_datat gather_job_timeoutu gather_job_timeoutu minionsu jidu No minions matched the target.N( R t gett Nonet appendt TrueR
R t cmd_itert sett addt nextR t iterkeyst
StopIterationR R t utilst stringutilst print_clit listt
difference( R t argst selected_target_optionR t frett nrett rett miniont m( ( s2 /usr/lib/python2.7/site-packages/salt/cli/batch.pyt __gather_minions* s8
c @ s� � f d � } y� t � j d t j � r� d � j d k r� | t � j d j d � � � } | d k r t t j | � � St | � Sn t � j d � SWn@ t k
r� � j
s� t j j
j d j � j d � � q� n Xd S( uA
Return the active number of minions to maintain
c @ s t | � d t � j � S( Ng Y@( t floatt lenR ( t x( R ( s2 /usr/lib/python2.7/site-packages/salt/cli/batch.pyt <lambda>V s u batchu %i uF Invalid batch data sent: {0}
Data must be in the form of %10, 10% or 3N( t
isinstanceR R t string_typesR0 t stript intt matht ceilt
ValueErrorR R R# R$ R% t format( R t partitiont res( ( R s2 /usr/lib/python2.7/site-packages/salt/cli/batch.pyt get_bnumR s ,"
c C@ sY t j � } d } x0 | t | � k rD | | | k rD | d 7} q W| rU | | 4n d S( Ni i ( R t nowR1 ( R t waitR? t i( ( s2 /usr/lib/python2.7/site-packages/salt/cli/batch.pyt
__update_waite s %c c@ s� g | j d | j d | j d d g } | j � } | j sC d St j | j � } g } i } g } | j j d d � } g } | j r� | j j } | j j }
n t } t }
i } | j
s� x0 | j D]" } t j
j j d j | � � q� Wn x�t | � t | j � k r�g }
| r4| r4| j | � n t | � | t | � k rz| rzx� | rv|
j | j � � qZWnv xs t | t | � t | � � D]Q } | r�| j � } t | t � r�|
j | j � d � q�|
j | � q�q�W| |
7} |
| d <|
r�| j
s8t j
j j d j t |
� � � n | j j d
| j j d t � d | j j d
d � d | d |
d | j d | | j � } | j | � i | | <|
| | d <t | | d <n
t j d � i } xe | j D]Z } | d k r�Pn t" t# j$ | � � } | | j k r�| j j | � | j | � q�q�Wx�| D]�} yOd } xBt r�t" | � } | d k r�t j d � | d 7} | d k r]Pq]q]n | j j d � r5| j% i | | d d 6� | d d | | d k r| | d j& | d d � q�t j
j j d j | d � � q]| j% | � xV | D]N } | | | d k r{| | d j& | � qIt j
j j d j | � � qIWq]WWqKt' k
r| | k rt | | d <x@ | | d D]- } | | k r�i | | <i | | d <q�q�WqqKXqKWx�t# j( | � D]�\ } } | | k rv| j& | � | rv| j t) j* � t+ d | � � qvn t } d | k r�t | d t � r�d | d k r�| d | d d <| j j d � r| d d d k rt } qn+ | j j d � r| d d k rt } n | j j d � r>| | | <| Vn | d | | <i | d | 6V| j
s�| d | | <| j d � | | <d | k r�| j d � } n d } t j, j- | | | j � n | r%t. j/ d! | � t' � q%q%Wx� | D]� } | | d r�| | k r�| j& | � x^ | | d D]K } | | k r-| j& | � | rx| j t) j* � t+ d | � � qxq-q-Wq�q�Wq� Wd S(" u'
Execute the batch run
u funu argu timeoutu listNu
batch_waiti u0 Minion {0} did not respond. No job will be sent.u
Executing run on {0}
t rawu rawR, u returnu t show_jidt verboseR u gather_job_timeoutu minionsu activeg{�G�z�?g{�G�z�?i i u datau iduE minion {0} was already deleted from tracker, probably a duplicate keyu rett secondsu retcodeu failhardu outuM Minion %s returned with non-zero exit code. Batch run stopped due to failhard(0 R R> R t copyt deepcopyR R RD RE t FalseR R R R# R$ R% R; R1 t _Batch__update_waitR t popR R4 t dictt keyst sortedR t cmd_iter_no_blockR R t timet sleepR R R R R! t updatet removeR" t iteritemsR R? R t outputt display_outputt logt error( R R( t bnumt to_runt activeR, t iterst bwaitR@ RD t show_verboset minion_trackert down_miniont next_RA t minion_idt new_itert partst ping_retR. t queuet ncntt partt idR- t datat failhardt out( ( s2 /usr/lib/python2.7/site-packages/salt/cli/batch.pyt runm s
# # '
%
#
(
&/& "
N(
t __name__t
__module__t __doc__R RI R R R> RJ Rm ( ( ( s2 /usr/lib/python2.7/site-packages/salt/cli/batch.pyR s ( ( Rp t
__future__R R R R8 RP RG R R t salt.utils.stringutilsR t salt.clientt salt.outputt salt.exceptionst salt.extR t salt.ext.six.movesR t loggingt getLoggerRn RW t objectR ( ( ( s2 /usr/lib/python2.7/site-packages/salt/cli/batch.pyt <module> s
Mini Shell