File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/reactor.cpython-310.pyc
o
�N�g=Q � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Zddl
Zddl Zddl Zddl
ZddlZddlZddlZddlZddlZe�e�Zeg d��ZG dd� dejjjejj �Z G dd� d�Z dS )z0
Functions which implement running reactor jobs
� N)�__id__�__sls__�name�order�fun�key�statec sp e Zd ZdZddiZ� fdd�Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Z� ZS )�Reactorz�
Read in the reactor configuration variable and compare it to events
processed on the master.
The reactor has the capability to execute pre-programmed executions
as reactions to events
�cmd�localc sP t � jdi |�� |�� }d|d< tj�|�| _tjj�| || jj� d| _ d S )Nr Z file_clientT� )
�super�__init__�copy�salt�minionZ MasterMinionr �CompilerZrend� is_leader)�self�opts�kwargsZlocal_minion_opts�� __class__r �F/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/reactor.pyr + s
zReactor.__init__c C s� i }|� d�r| jjd |�pd}t�|�}|s t�d||� |D ]+}z| j|||d�}|D ]}||| d< q.|�|� W q! tyL t� d|� Y q!w |S )zp
Execute the render system against a single reaction file and return
the data structure
zsalt://z
cp.cache_file� z<Can not render SLS %s for tag %s. File missing or not found.)�tag�datar zFailed to render "%s": )
�
startswithr � functions�glob�log�errorZrender_template�update� Exception� exception) r Zglob_refr r ZreactZ globbed_ref�fn_�resr r r r �render_reaction3 s(
� �zReactor.render_reactionc C s. t �d|� g }t| jd t�rWz#tjj�| jd ��}tjj �
|�}W d � n1 s-w Y W n( t yE t � d| jd � Y n t
yV t � d| jd � Y nw | jd }|D ]6}t|t�sfq^t|�dkrmq^tt|�� ��}|| }t�||�r�t|t�r�|�|� q^t|t�r�|�|� q^|S )zd
Take in the tag from an event and return a list of the reactors to
process
z Gathering reactors for tag %s�reactorN� Failed to read reactor map: "%s"�)Failed to parse YAML in reactor map: "%s"� )r �debug�
isinstancer �strr �utils�files�fopen�yaml� safe_load�OSErrorr! r# �dict�len�next�iter�keys�fnmatch�append�list�extend)r r �reactors�fp_� react_mapZroptr �valr r r �
list_reactorsQ s< ��
��
�zReactor.list_reactorsc C s� t | jjd t�r^t�d| jd � z&tjj� | jd ��}tjj
� |�}W d � W |S 1 s1w Y W |S t yK t�
d| jd � Y |S ty] t�
d| jd � Y |S w t�d� | jjd }|S )z/
Return a list of the reactors
r( z Reading reactors from yaml %sNr) r* z Not reading reactors from yaml)r- r r r. r r, r r/ r0 r1 r2 r3 r4 r! r# )r r? r@ r r r �list_allr s* � � � �
��
zReactor.list_allc C sX | � � }|D ]}tt|�� ��}||krddd� S q| jjd �||i� ddd�S )z
Add a reactor
FzReactor already exists.��statusZcommentr( TzReactor added.)rC r7 r8 r9 r r r; )r r Zreactionr> r( �_tagr r r � add_reactor� s �
zReactor.add_reactorc C sT | � � }|D ] }tt|�� ��}||kr$| jjd �|� ddd� S qddd�S )z"
Delete a reactor
r( TzReactor deleted.rD FzReactor does not exists.)rC r7 r8 r9 r r �remove)r r r> r( rF r r r �delete_reactor� s �
zReactor.delete_reactorc C s> t |�D ]\}}| j�|| d �}|dur ||| d< qdS )z�
Preserve backward compatibility by rewriting the 'state' key in the low
chunks if it is using a legacy type.
r N)� enumerate�aliases�get)r �chunks�idx�_Z new_stater r r �resolve_aliases� s ��zReactor.resolve_aliasesc
C s� t �d|� i }g }z-|D ] }|�| �|||�� q
|r6| �|�}|r.t �d|||� g W S | �| �|��}W n tyN } z t � d� W Y d}~nd}~ww | �
|� |S )zN
Render a list of reactor files and returns a reaction struct
z Compiling reactions for tag %sz_Unable to render reactions for event %s due to errors (%s) in one or more of the sls files (%s)z/Exception encountered while compiling reactionsN) r r, r" r' Z verify_highr! Z order_chunksZcompile_high_datar# r$ rP ) r r r r> �highrM r% �errors�excr r r � reactions� s0
����
zReactor.reactionsc C s |D ]}| j �|� qdS )z,
Execute the reaction state
N)�wrap�run)r rM �chunkr r r �call_reactions� s �zReactor.call_reactionsc
C sd | j d r tjj�� s t�d| j d � t�| j d � tjj j
| j d | j d | j dd���}t | j �| _ |j
dd�D ]�}|d �d �| j jkrIq;d
|d v rgtjj�d | j �}|d �d
�|krgt�d� q;|d �d�rw|�d| jid� |d �d�r�|d �d�| j d kr�|d d | _|�d| jid� |d �d�r�|d }| �|d |d �}|�| �� |d�d� q;|d �d�r�|d }| �|d �}|�| �� |d�d� q;|d �d �r�|�d| �� id � q;| js�q;| �|d �}|s�q;| �|d |d |�}|�r z| �|� W q; t �y t� d � Y q;w q;W d � d S 1 �s+w Y d S ) z,
Enter into the server loop
Zreactor_nicenessz Reactor setting niceness to %iZ__roleZsock_dirT)r �listen)�fullr �userzsalt/reactors/manager �rootr zKreceived salt/reactors/manage event without matching master_key. discardingz salt/reactors/manage/is_leader�resultz!salt/reactors/manage/leader/valuez salt/reactors/manage/set_leader�id�valuezsalt/reactors/manage/add�eventr> )r> r] z!salt/reactors/manage/add-completezsalt/reactors/manage/deletez$salt/reactors/manage/delete-completezsalt/reactors/manage/listz!salt/reactors/manage/list-resultszExit ignored by reactorN) r r r/ �platformZ
is_windowsr �info�os�nicer` Z get_event� ReactWraprU Z iter_eventsrL �
event_userZmasterZget_master_keyr! �endswithZ
fire_eventr rG rC rI rB rT rX �
SystemExit�warning)r r` r Z
master_key�_datar& r> rM r r r rV � s� � �
�
� � �
����$�z Reactor.run)�__name__�
__module__� __qualname__�__doc__rK r r' rB rC rG rI rP rT rX rV �
__classcell__r r r r r s � !
r c @ sn e Zd ZdZdZdZejjej j
ej j ejj
d�Zdd� Zdd� Zd d
� Zd d � Z d
d� Z dd� Zdd� ZdS )re z?
Wrapper that executes low data for the Reactor System
Nr )r �runner�wheel�callerc C sH || _ tjd u rtjj�|d �t_tjjj| j d | j d d�| _ d S )NZreactor_refresh_intervalZreactor_worker_threadsZreactor_worker_hwm)Z
queue_size)
r re � client_cacher r/ �cacheZ CacheDict�processZ
ThreadPool�pool)r r r r r r + s
� �zReactWrap.__init__c C sr |d }|| j vr7t�d|� |dv r(| j| | j�| j |< t| j | j� dS | j| | jd �| j |< dS dS )zR
Populate the client cache with an instance of the specified type
r z%Reactor is populating %s client cache�rp rq Z conf_fileN)rs r r, �reaction_classr r6 r )r �lowZ
reaction_typer r r �populate_client_cache7 s
� ��z ReactWrap.populate_client_cachec
C s� | � |� z t| |d �}W n ty t�d|d � Y nw �ztjj�||�}|� dd�}|� di �}|d dv rH| j
|d< |� d�|d< d|v r�|� d�}d D ]}||v ret� d
|d ||| � qS|d d kr�t
|t�r�tjj�|�s�||d
< i |d< nd|d
< ||d< t
|d t�s�tjj�|d �|d< |d s�t�d|d |d |d � W dS nUi }|d dv r�d
|vs�d|vr�| j|d j� |d �} | du r�t�d|d |d |d � W dS tjjj| |td�}d
|vr�|� dd�|d
< d|v�r|� di �|d< ||i |��}
|
du �r#t�d|d |d |d � W dS W dS t�y8 t� d|d � Y dS t�yR tjd|d |d |d dd� Y dS w )zH
Execute a reaction by invoking the proper wrapper func
r z0ReactWrap is missing a wrapper function for '%s'�argsr r rw Z__user__Z __state__)�arg�kwargz�Reactor '%s' is ignoring '%s' param %s due to presence of 'args' param. Check the Reactor System documentation for the correct argument format.r rr r| r} z�Reactor '%s' failed to execute %s '%s': Incorrect argument format, check the Reactor System documentation for the correct format.r Nz>Reactor '%s' failed to execute %s '%s': function not available)Zexpected_extra_kwsFz�Reactor '%s' failed to execute %s '%s': TaskPool queue is full!Consider tuning reactor_worker_threads and/or reactor_worker_hwmz(Reactor '%s' attempted to exit. Ignored.z&Reactor '%s' failed to execute %s '%s'T)�exc_info)rz �getattr�AttributeErrorr r! r r/ r{ Z format_callrL rf �popri r- r< r Z is_dictlistr5 Zrepack_dictlistrs r �REACTOR_INTERNAL_KEYWORDSrh r# ) r ry Zl_funZ wrap_callr{ r Z reactor_args�itemZ
react_callZ react_fun�retr r r rV T s�
�
�� � �
�� ���
�� ��z
ReactWrap.runc K �B d| j vrt�d� ddi}| �|� | jj| j d j||fd�S )zZ
Wrap RunnerClient for executing :ref:`runner modules <all-salt.runners>`
rp z8reactor edge case: re-populating client_cache for runnerr �r{ �rs r r, rz rv Z
fire_asyncry �r r r ry r r r rp � �
zReactWrap.runnerc K r� )zV
Wrap Wheel to enable executing :ref:`wheel modules <all-salt.wheel>`
rq z7reactor edge case: re-populating client_cache for wheelr r� r� r� r r r rq � r� zReactWrap.wheelc K sD d| j vrt�d� ddi}| �|� | j d j||fi |�� dS )zZ
Wrap LocalClient for running :ref:`execution modules <all-salt.modules>`
r z7reactor edge case: re-populating client_cache for localr N)rs r r, rz Z cmd_async)r r Ztgtr ry r r r r � s
zReactWrap.localc K sP d| j vrt�d� ddi}| �|� | j d j|g|d �R i |d �� dS )zY
Wrap LocalCaller to execute remote exec functions locally on the Minion
rr z8reactor edge case: re-populating client_cache for callerr r| r} N)rs r r, rz r
r� r r r rr � s
*zReactWrap.caller)rk rl rm rn rs rf r �clientZ LocalClientrp Z RunnerClientrq ZWheelZCallerrx r rz rV r rr r r r r re s � { re ) rn r: r �loggingrc Z salt.clientr Zsalt.defaults.exitcodesZ salt.runnerZ
salt.stateZsalt.utils.argsZsalt.utils.cacheZsalt.utils.dataZsalt.utils.eventZsalt.utils.filesZsalt.utils.masterZsalt.utils.processZsalt.utils.yamlZ
salt.wheel� getLoggerrk r � frozensetr� r/ ru ZSignalHandlingProcessr r r re r r r r �<module> s2
� }