File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/pyobjects.cpython-310.pyc
o
�N�g�* � @ s� d Z ddlZddlZddlmZ dZe�e�ZG dd� de �Z
G dd� de
�Z G d d
� d
e
�Z G d d � d �Z
G d
d� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� de�Zdd� ZG dd � d ed �ZdS ) z�
:maintainer: Evan Borgstrom <evan@borgstrom.ca>
Pythonic object interface to creating state data, see the pyobjects renderer
for more documentation.
� N�� OrderedDict) �listenZ onchangesZonfail�requireZwatchZuseZ listen_inZ onchanges_inZ onfail_inZ
require_inZwatch_inZuse_inc @ � e Zd ZdS )�StateExceptionN��__name__�
__module__� __qualname__� r r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/pyobjects.pyr � r c @ r )�DuplicateStateNr r r r r
r # r r c @ r )�InvalidFunctionNr r r r r
r ' r r c @ s� e Zd ZdZe� Zg Zg Ze� ZdZ e
dd� �Z e
dd� �Z e
dd� �Z
e
dd
d ��Ze
d d
� �Ze
dd� �Ze
dd� �Ze
dd� �ZdS )�RegistryzK
The StateRegistry holds all of the states that have been created.
Tc C s t � | _g | _g | _t � | _d S �N)r �states�
requisites�includes�extends��clsr r r
�empty6 s zRegistry.emptyc G s | j sd S | j|7 _d S r )�enabledr )r �argsr r r
�include= s zRegistry.includec C sV t dd� | j�� D ��}| jr| j|d< | jr%t dd� | j�� D ��|d< | �� |S )Nc S � g | ]\}}||f�qS r r ��.0�id_Zstates_r r r
�
<listcomp>F � z&Registry.salt_data.<locals>.<listcomp>r c S r r r r r r r
r! M r" �extend)r r �itemsr r r )r r r r r
� salt_dataD s
�zRegistry.salt_dataFc C s� | j sd S |r | j}n| j}||v r#|j|| v r"td�||j���nt� ||< | jrG| jD ]}|j|j vr<g |j |j< |j |j �
|� � q.|| � |� � d S )Nz*A state with id ''{}'', type ''{}'' exists) r r r � full_funcr �formatr r � requisite�kwargs�append�update)r r �stater# �attr�reqr r r
�addT s( ���
z Registry.addc C s | j ||dd� d S )NT)r# )r/ )r r r, r r r
r# q s zRegistry.extendc C s t |�S r )� StateExtend)r �namer r r
� make_extendu s zRegistry.make_extendc C s | j sd S | j�|� d S r )r r r* )r r( r r r
�push_requisitey s zRegistry.push_requisitec C s | j sd S | jd= d S )N���)r r r r r r
�
pop_requisite� s zRegistry.pop_requisiteN)F)r r
r �__doc__r r r r r r � classmethodr r r% r/ r# r2 r3 r5 r r r r
r + s.
r c @ s e Zd Zdd� ZdS )r0 c C �
|| _ d S r )r1 )�selfr1 r r r
�__init__� �
zStateExtend.__init__N)r r
r r: r r r r
r0 � s r0 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�StateRequisitec C s || _ || _|| _d S r )r( �moduler )r9 r( r= r r r r
r: � s
zStateRequisite.__init__c C � | j | jiS r )r= r �r9 r r r
�__call__� � zStateRequisite.__call__c C s t �| � d S r )r r3 r? r r r
� __enter__� s zStateRequisite.__enter__c C � t �� d S r �r r5 �r9 �type�value� tracebackr r r
�__exit__� rA zStateRequisite.__exit__N)r r
r r: r@ rB rI r r r r
r< � s
r< c @ s, e Zd ZdZd
dd�Zdd� Zd dd �ZdS ) � StateFactorya�
The StateFactory is used to generate new States through a natural syntax
It is used by initializing it with the name of the salt module::
File = StateFactory("file")
Any attribute accessed on the instance returned by StateFactory is a lambda
that is a short cut for generating State objects::
File.managed('/path/', owner='root', group='root')
The kwargs are passed through to the State object
Nc C s || _ |d u r g }|| _d S r )r= � valid_funcs)r9 r= rK r r r
r: � s
zStateFactory.__init__c s4 �j r� �j vrtd�� �j���� �fdd�}|S )Nz=The function '{}' does not exist in the StateFactory for '{}'c s t | �j� fi |��S r )�Stater= )r r) ��funcr9 r r
�
make_state� s z,StateFactory.__getattr__.<locals>.make_state)rK r r' r= )r9 rN rO r rM r
� __getattr__� s ��zStateFactory.__getattr__r c C s t || j|�S )zJ
When an object is called it is being used as a requisite
)r< r= )r9 r r( r r r
r@ � s zStateFactory.__call__r )r )r r
r r6 r: rP r@ r r r r
rJ � s
rJ c @ sP e Zd ZdZdd� Zedd� �Zedd� �Zdd � Zd
d � Z d d
� Z
dd� Z dS )rL z�
This represents a single item in the state tree
The id_ is the id of the state, the func is the full name of the salt
state (i.e. file.managed). All the keyword args you pass in become the
properties of your state.
c K s� || _ || _|| _tD ] }||v r*zt|| � W q ty) || g||< Y q w q || _t| j t�rBt �
| j j | � | j j | _ nt � | j | � t
d| j| j �| _d S )Nr )r r= rN �
REQUISITES�iter� TypeErrorr) �
isinstancer0 r r# r1 r/ r< r( )r9 r r= rN r) r- r r r
r: � s" �� zState.__init__c sH | j � tD ]}|� v rdd� � | D �� |< q� fdd�t� �� �D �S )Nc S s g | ] }t |t�r |� n|�qS r )rT r< )r r. r r r
r! � s ��z State.attrs.<locals>.<listcomp>c s g | ]}|� | i�qS r r )r �k�r) r r
r! � r" )r) rQ �sorted�keys)r9 r- r rV r
�attrs� s
��z State.attrsc C s | j �d| j��S �N�.)r= rN r? r r r
r& � s zState.full_funcc C s | j �d| j�d| j��S )Nz = �:)r r& rY r? r r r
�__str__� s z
State.__str__c C r> r )r& rY r? r r r
r@ rA zState.__call__c C s t �| j� d S r )r r3 r( r? r r r
rB s zState.__enter__c C rC r rD rE r r r
rI rA zState.__exit__N) r r
r r6 r: �propertyrY r& r] r@ rB rI r r r r
rL � s
rL c @ s e Zd ZdZdd� Zdd� ZdS )�
SaltObjectz�
Object based interface to the functions in __salt__
.. code-block:: python
:linenos:
Salt = SaltObject(__salt__)
Salt.cmd.run(bar)
c C r8 r )�_salt)r9 �saltr r r
r: r; zSaltObject.__init__c s G � �fdd�d�}|� S )Nc s e Zd Z� �fdd�ZdS )z+SaltObject.__getattr__.<locals>.__wrapper__c s* z �j � � d|� � W S ty t�w rZ )r` �KeyError�AttributeError)ZwselfrN ��modr9 r r
rP s
�z7SaltObject.__getattr__.<locals>.__wrapper__.__getattr__N)r r
r rP r rd r r
� __wrapper__ s rf r )r9 re rf r rd r
rP s zSaltObject.__getattr__N)r r
r r6 r: rP r r r r
r_ s
r_ c s< e Zd ZdZedd� �Zdd� Z� fdd�Zdd � Z� Z S )
�MapMetazk
This is the metaclass for our Map class, used for building data maps based
off of grain data.
c C s t � S r r )�metaclsr1 �basesr r r
� __prepare__+ s zMapMeta.__prepare__c C s t �| |||�}|�� |_|S r )rF �__new__rX �__ordered_attrs__)r r1 ri rY �cr r r
rk / s
zMapMeta.__new__c s | � � t� �|||� d S r )�__set_attributes__�superr: )r r1 ri Znmspc�� __class__r r
r: 4 s zMapMeta.__init__c
C s� g }t � }| jD ]B}|d dkrq| j| }t�|�s qt|dd�}|�|� t|d|�}i }|jD ]}|d dkrA|j| ||< q2|�|||f� qzGt| j d�sXt
� d� n9g } | j D ]'}t t
|��D ] \}
}z|d |kry| �|� d ||
< W qe ty� Y qew q]| �dd � |D �� | }W n ty� Y nw i } |r�tjd
|� } |D ]\}}}|| vr�q�| | |kr�| �|� q�t| d �r�tjd | j�}
|
r�| �|
� | D ]
}t| || | � q�d S )
Nr �_Z __grain__Z os_familyZ __match__�__iter__z'pyobjects: priority must be an iterablec S s g | ]}|d ur|�qS r r )r �xr r r
r! l r" z.MapMeta.__set_attributes__.<locals>.<listcomp>z grains.item�merge�
pillar.get)�setrl �__dict__�inspectZisclass�getattrr/ r* �hasattr�priority�log�error�list� enumeraterS r# rc �Map�__salt__r+ ru �setattr)r Z
match_infoZ
grain_targets�itemZfiltZgrain�matchZ match_attrsr1 Znew_match_info�indexrY Z
grain_valsZpillarr r r
rn 8 sl
�
� �� � �
�
�zMapMeta.__set_attributes__)
r r
r r6 r7 rj rk r: rn �
__classcell__r r rp r
rg % s
rg c O s t �d� i S )Nz-Map needs __salt__ set before it can be used!)r} r~ )�arU r r r
� need_salt� s
r� c @ s e Zd Zeed�ZdS )r� )zgrains.filter_byrv N)r r
r r� r� r r r r
r� � s r� )� metaclass)r6 ry �loggingZsalt.utils.odictr rQ � getLoggerr r} � Exceptionr r r r r0 r< rJ rL r_ rF rg r� r� r r r r
�<module> s$
]+D`