File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/thorium/__pycache__/__init__.cpython-310.pyc
o
�N�g^ � @ sr d Z ddlZddlZddlZddlZddlZddlZddlZddl Zddl
m Z e� e
�ZG dd� dejj�ZdS )zE
The thorium system allows for advanced event tracking and reactions
� N)�SaltRenderErrorc @ sF e Zd ZdZ ddd�Zdd� Zdd � Zdd
d �Zd d
� Zdd� Z dS )� ThorStatezH
Compile the thorium state and manage it in the thorium runtime
FNc
C s || _ || _|| _|| _|d |d< |d |d< |d |d< d|d< || _|�d �r1tj�|�| _tj j
j | | jd
d � tj �
| ji �| _
| j�d d �| _i }| jd urxz | j
| j� d
� � }W n tyw } z t�|� W Y d }~nd }~ww d|i| j _tjj�| j| jd �| _d S )NZ
thorium_rootsZ
file_rootsZ
thoriumenvZsaltenvZ thorium_topZ state_top�localZ file_client�minion_data_cacheZthorium)�loaderZregister_returnerz .load_regZ__reg__Zsock_dir)�grains�
grain_keys�pillar� pillar_keys�opts�get�salt�cache�factory�state� HighState�__init__r � returners�reg_ret� Exception�log�error�inject_globals�utils�eventZget_master_event)�selfr r r r r
Zregdata�exc� r �I/opt/saltstack/salt/lib/python3.10/site-packages/salt/thorium/__init__.pyr s0
�� zThorState.__init__c C s i i d�}| j s | jr�| j�d�r�| j�d�}|s|S |D ]j}| j�d|� �d�}d|v rR| jrI| jD ]}||d v rG|d | |d | |< q3n|d |d |< ni |d |< d|v r�| jrx| jD ]}||d v rv|d | |d | |< qbq |d |d |< q i |d |< q |S )zF
Gather the specified data from the minion data cache
)r r r �minionszminions/�datar r ) r r r r r �listZfetchr
r )r r r Zminion�total�keyr r r � gather_cache< s6
��
��zThorState.gather_cachec C sB z| � � W n ty tjddd� t�| jd � Y nw q)z#
Start the system!
TzException in Thorium: )�exc_info�thorium_interval)� call_runtimer r r �time�sleepr )r r r r �
start_runtime^ s ��zThorState.start_runtimec
C s& i }g }z| � � }W n' ty } z|W Y d}~S d}~w ty1 t�� }|�|� | Y S w || �|�7 }| �|�}|sFd}t|��| �||�}| � |�\} }
|rst
|t �r_|� d�}d| v rk| d �
|� n|| d< ||
7 }| j�| �\} } || 7 }|| j�| �7 }|r�t|��| j�| �S )zn
Compile the top file and return the lowstate for the thorium runtime
to iterate over
NzNo Top file found!�,Z __exclude__)Zget_topr r � traceback�
format_exc�appendZ verify_topsZ top_matchesZmatches_whitelistZrender_highstate�
isinstance�str�split�extendr Zreconcile_extendZ verify_highZcompile_high_data) r ZexcludeZ whitelist�ret�err�topZtrb�matches�msg�high�errorsZ
ext_errorsr r r �
get_chunksi s@ �
�
zThorState.get_chunksc C s. g } | j jddd�}|du r|S |�|� q)zO
iterate over the available events and return a list of events
T� )�wait�fullN)r Z get_eventr. )r r3 r r r r �
get_events� s
�zThorState.get_eventsc
C s� | � � }| �� }| jd }| j�dd�}t�� } | �� }|s%t�|� qt�� }|| jjd< | j� |� t�� | }|| } | dkrHt�| � | j�
� || |kro| � � }| �� }| j durk| j | j � d� |� t�� }q) z%
Execute the runtime
r& Zthorium_recompilei, TZ
__events__r Nz .save_reg)
r$ r: r r r( r> r) r r Z call_chunksZ
reset_run_numr r )
r r �chunks�intervalZ recompileZr_start�events�start�elapsed�leftr r r r' � s2
�zThorState.call_runtime)FNFN)NN)
�__name__�
__module__� __qualname__�__doc__r r$ r* r: r> r' r r r r r s
� "
% r )rH �logging�osr( r, Z
salt.cacher
Z salt.loaderZ salt.payloadZ
salt.stateZsalt.exceptionsr � getLoggerrE r r r r r r r r �<module> s