o

�N�g�@sDdZd
dl
Zddd�Zdd�Zdd	d
�
Zdd�Zd
d�Zdd�ZdS)z�
Used to manage the thorium register. The thorium register is where compound
values are stored and computed, such as averages etc.
�N�set�list)�set_�list_c	Cs�|id
dd�}|tv
rit|<t
�t|d<tD]=}tjj�|d|�rUz|dd�|
�
}Wnty?


|d�|
�
}Yn
w|durFd}||d	|
<t|d�	|�

q|S)
z�
    Add a value to the named set

    USAGE:

    .. code-block:: yaml

        foo:
          reg.set:
            - add: bar
            - match: my/custom/event
    �T��name�changesZcomment�result�val�tag�dataN�Noner	)
�__reg__r�
__events__�salt�utils�stringutils�
expr_match�get�KeyError�add)rr�match�ret�eventr�r�D/opt/saltstack/salt/lib/python3.10/site-packages/salt/thorium/reg.pyrs"









�


�
rFc
	Cs�|id
dd�}t|
t
�s|
�d�
}
|tv
rit|<gt|d<tD]H}z|dd}Wn
ty8


|d}Yn
wtjj�	|d|�rii}|
D]}	|	|vr_||	||	<|dur_|dd|d	<qGt|d�
|�

q!|d
kr|t|dd|�t|d<|S)a�

    Add the specified values to the named list

    If ``stamp`` is True, then the timestamp from the event will also be added
    if ``prune`` is set to an integer higher than ``0``, then only the last
    ``prune`` values will be kept in the list.

    USAGE:

    .. code-block:: yaml

        foo:
          reg.list:
            - add: bar
            - match: my/custom/event
            - stamp: True
    rTr�
,rr
rZ_stamp�timer
N)�
isinstancer�splitrrrrrrr�append)
rrrZstampZprunerr�
event_data�item�keyrrrr,s2











�





�
�


rc	Cs
|id
dd�}|tv
r!it|<dt|d<dt|d<dt|d<t
D]Z}z|dd}Wn
ty:


|d}Yn
wtjj�|d	|�r}|
|vrYzt|�
}Wn	tyX


Yq#wt|d|7<t|dd
7<t|dt|dt|d<q#|S)aE

    Accept a numeric value from the matched events and store a running average
    of the values in the given register. If the specified value is not numeric
    it will be skipped

    USAGE:

    .. code-block:: yaml

        foo:
          reg.mean:
            - add: data_field
            - match: my/custom/event
    rTrr
r�total�countr
r�
)	rrrrrrr�int�
ValueError)rrrrrr"�comprrr�meanVs0









�




�
 
�
r+c
Cs&|id
dd�}
|tvrt|�
�
|
S)z�
    Clear the namespace from the register

    USAGE:

    .. code-block:: yaml

        clearns:
          reg.clear:
            - name: myregister
    rTr)r�clear�rrrrrr,|s


r,c
Cs |id
dd�}
|tvrt|=|
S)z�
    Delete the namespace from the register

    USAGE:

    .. code-block:: yaml

        deletens:
          reg.delete:
            - name: myregister
    rTr)
rr-rrr�delete�s


r.)Fr
)	�__doc__Zsalt.utils.stringutilsrZ__func_alias__rrr+r,r.rrrr�<module>
s
�
*&