File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/beacons/__pycache__/__init__.cpython-310.pyc
o
�N�g�H � @ sP d Z ddlZddlZddlZddlZddlZddlZe�e �Z
G dd� d�Z dS )zF
This package contains the loader modules for the salt streams system
� Nc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z d,dd�Z d,dd�Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+S )-�BeaconzI
This class is used to evaluate and execute on the beacon system
c C s0 dd l }|| _|| _|j�||�| _t� | _d S )Nr )Z salt.loader�opts� functions�loader�beacons�dict� interval_map)�selfr r �salt� r �I/opt/saltstack/salt/lib/python3.10/site-packages/salt/beacons/__init__.py�__init__ s
zBeacon.__init__c C sb g }t �|�}d|v r|d sdS |D �]�}|dkrqd}t|| t�r1i }tt|j|| �� n t|| t�r<|| }d|v r`|d sKt�d|� qt|| t�rX|| d= n| � || d� t�d|� d}| �
|d�rs|d }n|}|� d�}|| j v r�| j | || �\} }
| s�t� d||
� qnt�
d|� t|| t�s�t� d � q|| �d
|i� |� d �} | | j v �r�| �
|d �} | �
|d
�}
|
r�| �||d
�}| �||
�s�t�d|� q| �
|d��r1t�d|� | �||d�}d}tjj�| j�}|D ]
}t�d|d ��r d}q�|�r1|� d�}|| j v �r*t�d|� | j | || � nt�d|� q|| j | jd< d}z | j | || �}W n% t�� d � }t� d||� d� | jd |�}|�||i |d �� Y |�s�|D ]-}d� | jd |�}d |v �r�||� d �7 }d |v�r�| jd |d < |�|||d �� �qq| �r�| � |� qt�
d |� q|S )!a
Process the configured beacons
The config must be a list and looks like this in yaml
.. code_block:: yaml
beacons:
inotify:
- files:
- /etc/fstab: {}
- /var/cache/foo: {}
�enabledNzBeacon %s disabledzBeacon processing: %s�
beacon_module� .validatez0Beacon %s configuration invalid, not running.
%szCNo validate function found for %s, running basic beacon validation.z(Configuration for beacon must be a list.Z _beacon_name�.beaconZrun_once�intervalz)Skipping beacon %s. Interval not reached.Zdisable_during_state_runz<Evaluting if beacon %s should be skipped due to a state run.Fzstate.*ZfunTz.closez)Closing beacon %s. State run in progress.z*Skipping beacon %s. State run in progress.Z
__grains__� z Unable to start %s beacon, %szsalt/beacon/{}/{}/�id)�tag�error�data� beacon_namer )r r r zUnable to process beacon %s) �copy�deepcopy�
isinstance�list�map�updater �log�trace�_remove_list_item�_determine_beacon_configr r �warning�append� _trim_config�_process_intervalr
�utils�minionZrunningr �re�match�info� __globals__�sys�exc_info�format�pop�disable_beacon)r ZconfigZgrains�ret�b_config�mod�current_beacon_configr � validate_str�valid�vcommentZfun_strZrunoncer Z
is_runningZ running_jobsZjobZ close_strr �rawr r r r r �process s�
���
�� ����
��
�
�zBeacon.processc C sB t || t�r| �|| |� |S t || t�r || �|� |S )zM
Take a beacon configuration and strip out the interval bits
)r r r! r r0 )r r3 r4 �keyr r r r% � s �zBeacon._trim_configc C s d}t |t�r
|�|d�}|S )zJ
Process a beacon configuration to determine its interval
F)r r �get)r r5 r; r r r r r" � s
z Beacon._determine_beacon_configc C s� t �d||� | jd }|| jv r9t �d� | j| }t �d|� || |kr.d| j|< dS | j| d7 < dS t �d|� d| j|< dS ) zk
Process beacons with intervals
Return True if a beacon should be run on this loop
z(Processing interval %s for beacon mod %s�
loop_intervalzProcessing interval in mapzInterval counter: %sr Tz"Interval process inserting mod: %sF)r r r r )r r4 r r= Zcounterr r r r& � s
�
zBeacon._process_intervalc s& � fdd�t |�D �}|sdS |d S )zn
Return the index of a labeled config item in the beacon config, -1 if the index is not found
c s g | ]
\}}� |v r|�qS r r )�.0�index�item��labelr r �
<listcomp>� s z%Beacon._get_index.<locals>.<listcomp>���r )� enumerate)r �
beacon_configrB Zindexesr rA r �
_get_index� s zBeacon._get_indexc C s | � ||�}||= dS )z:
Remove an item from a beacon config list
N)rG )r rF rB r? r r r r! � s
zBeacon._remove_list_itemc C s~ t | jd | t�r|| jd | d< dS | �| jd | d�}|dkr1|| jd | | d< dS | jd | �d|i� dS )z@
Update whether an individual beacon is enabled
r r r N)r r r rG r$ )r �nameZ
enabled_valueZ
enabled_indexr r r �_update_enabled� s zBeacon._update_enabledTc C sl i }|r | j �di ��di �}t|t�std��|�|� |r4| j �di �}t|t�s/td��|�|� |S )z3
Return the beacons data structure
Zpillarr z Beacons must be of type dict.)r r< r r �
ValueErrorr )r � include_opts�include_pillarr Zpillar_beaconsZ opts_beaconsr r r � _get_beacons� s
zBeacon._get_beaconsc C s\ | j ||d�}tjjjd| jd��}|jd|d�dd� W d � dS 1 s'w Y dS ) a
List the beacon items
include_pillar: Whether to include beacons that are
configured in pillar, default is True.
include_opts: Whether to include beacons that are
configured in opts, default is True.
)rL rK r( �r T��completer z)/salt/minion/minion_beacons_list_complete�r N)rM r
r' �event� get_eventr �
fire_event)r rL rK r �evtr r r � list_beacons s
��
��zBeacon.list_beaconsc C s^ dd� | j D �}tjjjd| jd��}|jd|d�dd� W d � dS 1 s(w Y dS )
z,
List the available beacons
c S s& g | ]}d |v rd� |�d d���qS )r z{}� )r/ �replace)r> Z_beaconr r r rC # s
�z1Beacon.list_available_beacons.<locals>.<listcomp>r( rN TrO z3/salt/minion/minion_beacons_list_available_completerQ N)r r
r' rR rS r rT )r Z_beaconsrU r r r �list_available_beacons s ��
��z Beacon.list_available_beaconsc s� t � fdd�|D ��}|� d�}|| jv r&d|v r |d= | j| |�\}}nd�|�}d}tjjjd| jd��}|jd||d �d
d � W d � dS 1 sNw Y dS )
z3
Return available beacon functions
c 3 s � | ] }|� d � �V qdS )r N)r< )r> r@ �rH r r � <genexpr>6 s � z)Beacon.validate_beacon.<locals>.<genexpr>r r zABeacon {} does not have a validate function, skipping validation.Tr( rN )rP r8 r7 z./salt/minion/minion_beacon_validation_completerQ N) �nextr r/ r
r' rR rS r rT )r rH � beacon_datar r6 r7 r8 rU r rZ r �validate_beacon2 s&
�
�
��zBeacon.validate_beaconc C s� i }|||< || j dd�v rd�|�}d}n || jd v r#d|� �}nd|� �}d}| jd �|� tjjjd| jd ��}|j||| jd d
�d d � W d
� dS 1 sVw Y dS )z#
Add a beacon item
F�rK zACannot update beacon item {}, because it is configured in pillar.r �#Updating settings for beacon item: zAdded new beacon item: Tr( rN �rP �commentr z'/salt/minion/minion_beacon_add_completerQ N) rM r/ r r r
r' rR rS rT �r rH r] r rb rP rU r r r �
add_beaconO s0 �
��
�
�
zBeacon.add_beaconc C s� i }|||< || j dd�v rd|� d�}d}nd|� �}d}| jd �|� tjjjd| jd ��}|j||| jd d
�d d � W d
� dS 1 sJw Y dS )z&
Modify a beacon item
Fr_ zCannot modify beacon item � , it is configured in pillar.r` Tr r( rN ra z*/salt/minion/minion_beacon_modify_completerQ N)rM r r r
r' rR rS rT rc r r r �
modify_beaconr s(
��
� � zBeacon.modify_beaconc C s� || j dd�v rd|� d�}d}n|| jd v r$| jd |= d|� �}nd|� d�}d }tjjjd
| jd ��}|j||| jd d �d
d� W d � d S 1 sPw Y d S )z&
Delete a beacon item
Fr_ zCannot delete beacon item re r zDeleting beacon item: z Beacon item z not found.Tr( rN ra z*/salt/minion/minion_beacon_delete_completerQ N)rM r r
r' rR rS rT �r rH rb rP rU r r r �
delete_beacon� s( ��
�
�
zBeacon.delete_beaconc C sb d| j d d< tjjjd| j d��}|jd| j d d�dd� W d � dS 1 s*w Y dS )
�
Enable beacons
Tr r r( rN rO z,/salt/minion/minion_beacons_enabled_completerQ N�r r
r' rR rS rT �r rU r r r �enable_beacons� � �
��zBeacon.enable_beaconsc C sb d| j d d< tjjjd| j d��}|jd| j d d�dd � W d
� dS 1 s*w Y dS ) ri Fr r r( rN TrO z-/salt/minion/minion_beacons_disabled_completerQ Nrj rk r r r �disable_beacons� rm zBeacon.disable_beaconsc C s� || j dd�v rd|� d�}d}n
| �|d� d|� �}d}tjjjd| jd��}|j||| jd d
�d d � W d
� dS 1 sBw Y dS )z!
Enable a beacon
Fr_ zCannot enable beacon item re TzEnabling beacon item r( rN r ra z+/salt/minion/minion_beacon_enabled_completerQ N)rM rI r
r' rR rS r rT rg r r r �
enable_beacon� s$
��
�
�
zBeacon.enable_beaconc C s� || j dd�v rd�|�}d}n
| �|d� d|� �}d}tjjjd| jd��}|j||| jd d �d
d � W d � dS 1 sAw Y dS )
z"
Disable a beacon
Fr_ z:Cannot disable beacon item {}, it is configured in pillar.zDisabling beacon item Tr( rN r ra z,/salt/minion/minion_beacon_disabled_completerQ N) rM r/ rI r
r' rR rS r rT rg r r r r1 � s* ��
��
�
�
zBeacon.disable_beaconc C s` i | j d< tjjjd| j d��}|jdd| j d d�dd� W d � dS 1 s)w Y dS )
z/
Reset the beacons to defaults
r r( rN TzBeacons have been resetra z)/salt/minion/minion_beacon_reset_completerQ Nrj rk r r r �reset s
��
� � z Beacon.resetN)TT)�__name__�
__module__� __qualname__�__doc__r
r: r% r" r& rG r! rI rM rV rY r^ rd rf rh rl rn ro r1 rp r r r r r s.
# r ) rt r �loggingr) r- Zsalt.utils.eventr
Zsalt.utils.minion� getLoggerrq r r r r r r �<module> s