Current File : //lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyc
�
���^c @@ s� d d l m Z m Z m Z d d l Z d d l Z d d l m Z d d l Z d d l
Z e j e � Z
d e f d � � YZ d S( i ( t absolute_importt print_functiont unicode_literalsN( t sixt SaltInfoc B@ s_ e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z RS(
uP
Class to handle processing and publishing of "real time" Salt upates.
c C@ s | | _ i | _ i | _ d S( ub
handler is expected to be the server side end of a websocket
connection.
N( t handlert jobst minions( t selfR ( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt __init__ s c C@ s� t j d � i } t j d � x^ t j | j � D]J \ } } t j | � i } | j | � | j i | d 6� | | | <q3 Wt j d � i | d 6} | j j t j j
j | � t d � � d S( u7
Publishes minions as a list of dicts.
u in publish minionsu
starting loopu idu
ended loopu minionsu
N(
t logt debugR t iteritemsR t updateR t
write_messaget saltt utilst jsont dumpst str( R R t miniont minion_infot curr_miniont ret( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt publish_minions s
c C@ s@ i | | 6} t j j j | � t d � } | j j | � d S( u9
Publishes the data to the event stream.
u
N( R R R R R R R ( R t keyt datat publish_datat pub( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt publish4 s
c C@ s� | d } | d } | j d � d } | j j | d � sL i | j | <n | j | } | j i | d d 6� t j d | j � | j � d S( uO
Associate grains data with a minion and publish minion update
u tagu datau /i����u returnu grainsu/ In process minion grains update with minions=%sN( t splitR t gett NoneR
R
R R ( R t
event_datat tagt
event_infot midR ( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt process_minion_update< s
c
C@ s
| d } | d } | j d � \ } } } } } | j j | i � } | j d i � j | i � } | j i | d d 6� | j i | d d 6� | j i | d d 6� t g t j | d � D] \ } } | d ^ q� � } | r� d | d <n | j d
| j � d S( u�
Process a /ret event returned by Salt for a particular minion.
These events contain the returned results from a particular execution.
u tagu datau /u minionsu returnu retcodeu successu completeu stateu jobsN( R R t
setdefaultR
t allR R R (
R R! R" R# t _t jidR$ t jobR t job_complete( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt process_ret_job_eventN s
0
c C@ s� d } | d } | d } i } x# | d D] } i t d 6| | <q+ Wi | d d 6| d d 6| d 6| d d 6| d d 6| d
d
6d d 6} | | j | d <| j d
| j � d S( u�
Creates a new job with properties from the event data
like jid, function, args, timestamp.
Also sets the initial state to started.
Minions that are participating in this job are also noted.
u tagu datau minionsu successu jidu _stampu
start_timeu funu tgtu tgt_typeu runningu stateu jobsN( R t FalseR R ( R R! R* R" R# R R$ ( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt process_new_job_eventf s
c C@ sv | d } | d } | d d k r> | j j | d d � n* | d d k rh | j j | d i � n | j � d S( u�
Tag: salt/key
Data:
{'_stamp': '2014-05-20T22:45:04.345583',
'act': 'delete',
'id': 'compute.home',
'result': True}
u tagu datau actu deleteu idu acceptN( R t popR R& R ( R R! R" R# ( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt process_key_event� s
c C@ s t j d � t } t | d j d g � � rN t | d j d g � � } n) t | j � t | d j d g � � } x7 | D]/ } t } t j d | � | j j | d � q~ Wt | d j d g � � r� t j d � t | d j d g � � } t } nn t | d j d g � � t | j � rgt j d � t | d j d g � � t | j � } t } n g } d j | � } x( | D] } t j d
� i | j | <q�W| rt } t
j j | � }
|
j
i d d 6| d
6d d 6d d 6d d 6d d 6| d 6� n | r| j � n d S( ut
Check if any minions have connected or dropped.
Send a message to the client if they have.
u In presenceu datau lostu presentu
Popping %su newu got new minionsu detected new minionsu ,u
Adding minionu grains.itemsu funu tgtu listu expr_typeu clientu modeu localu local_asyncu asynchronousu tokenN( R
R R- t setR R t TrueR/ R t joinR t netapit NetapiClientt runR ( R t salt_datat tokent optst changedt dropped_minionsR t new_minionst tgtR$ t client( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt process_presence_events� sH
)
)
)
c C@ sR t j d t j � � t j | d � t j | � | d j d � } t | � d k r] d S| d d k r� t j d � | d d
k r� t j d � | j | � qN| d d k rNt j d
� | j | � | d d d k r� | j | � q� qNnS | d d k r(t j d � | j | � n& | d d k rN| j
| | | � n d S( u1
Process events and publish data
u
In process %su tagu /i Ni u jobu
In job part 1i u newu
In new jobu retu In retu datau funu grains.itemsu keyu In keyu presence( R
R t threadingt current_threadR t lenR. R, R% R0 R? ( R R7 R8 R9 t parts( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt process� s*
( t __name__t
__module__t __doc__R R R R% R, R. R0 R? RD ( ( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyR
s 5( t
__future__R R R t loggingR@ t salt.extR t salt.netapiR t salt.utils.jsont getLoggerRE R
t objectR ( ( ( sL /usr/lib/python2.7/site-packages/salt/netapi/rest_tornado/event_processor.pyt <module> s
Mini Shell