File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/pydsl.cpython-310.pyc
o
�N�g�6 � @ s� d Z ddlmZ ddlmZ ddlmZ ed� � �Z
G dd� de �Z G dd � d e
�Zd
aG d d � d �ZG d
d� d�ZG dd� d�Zdd� ZG dd� d�Zd
S )a�
:maintainer: Jack Kuan <kjkuan@gmail.com>
:maturity: new
:platform: all
A Python DSL for generating Salt's highstate data structure.
This module is intended to be used with the `pydsl` renderer,
but can also be used on its own. Here's what you can do with
Salt PyDSL::
# Example translated from the online salt tutorial
apache = state('apache')
apache.pkg.installed()
apache.service.running() \
.watch(pkg='apache',
file='/etc/httpd/conf/httpd.conf',
user='apache')
if __grains__['os'] == 'RedHat':
apache.pkg.installed(name='httpd')
apache.service.running(name='httpd')
apache.group.present(gid=87).require(apache.pkg)
apache.user.present(uid=87, gid=87,
home='/var/www/html',
shell='/bin/nologin') \
.require(apache.group)
state('/etc/httpd/conf/httpd.conf').file.managed(
source='salt://apache/httpd.conf',
user='root',
group='root',
mode=644)
Example with ``include`` and ``extend``, translated from
the online salt tutorial::
include('http', 'ssh')
extend(
state('apache').file(
name='/etc/httpd/conf/httpd.conf',
source='salt://http/httpd2.conf'
),
state('ssh-server').service.watch(file='/etc/ssh/banner')
)
state('/etc/ssh/banner').file.managed(source='salt://ssh/banner')
Example of a ``cmd`` state calling a python function::
def hello(s):
s = "hello world, %s" % s
return dict(result=True, changes=dict(changed=True, output=s))
state('hello').cmd.call(hello, 'pydsl!')
� )�uuid4)� HighState�� OrderedDictz_listen require watch prereq use listen_in require_in watch_in prereq_in use_in onchanges onfailc @ s e Zd ZdS )�
PyDslErrorN)�__name__�
__module__� __qualname__� r
r
�D/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/pydsl.pyr a s r c @ s e Zd Zdd� ZdS )�Optionsc C s
| � |�S �N)�get)�self�namer
r
r � __getattr__f �
zOptions.__getattr__N)r r r r r
r
r
r r e s r Nc @ sn e Zd Zdd� Zedd� �Zedd� �Zdd� Zd d
� Zd d � Z ddd�Z
dd� Z dd� Z dd� Z
dd� Zd
S )�Slsc C sR || _ || _g | _t�� j| _g | _g | _t � | _
g | _ || _ t�� s't
d��d S )Nz+PyDSL only works with a running high state!)r �saltenv�includesr �
get_activeZbuilding_highstate�included_highstate�extends�declsr �options�funcs� rendered_slsr )r �slsr r r
r
r �__init__n s �z Sls.__init__c C �
t �� jS r
)r r Z_pydsl_all_decls��clsr
r
r �
get_all_decls| �
zSls.get_all_declsc C r r
)r r Z_pydsl_render_stackr r
r
r �get_render_stack� r# zSls.get_render_stackc K � | j �|� d S r
)r �update)r r r
r
r �set� � zSls.setc O s\ d|v r |� d� |�d| j�}|�dd�r%|D ]
}| j�||f� qd S t�� }td u r4|�|� � �a| j
}g }|D ]b}|� d|� �} | | j vro| j � |� |�
||| j t�\}
} |�||
| � | rjtd�| ���|�|� d|� �} | |vr~|�d � q;|| d D ]}
t|
t�r�tt|
��d kr�|�|
d � nq�q;|s�d S t|�d
kr�|d S |S ) N�envr ZdelayedF�:�
�_slsmod_� stateconf�slsmod� r )�popr r r �appendr r � SLS_MATCHESZ top_matchesZget_topr r �addZ render_state�merge_included_statesr �joinZclean_duplicate_extends�
isinstance�dict�next�iter�len)r Z sls_names�kwsr ZinclZ HIGHSTATE� highstateZslsmodsr Zr_envZhistates�errors�state_id�argr
r
r �include� sH
�
�
�z Sls.includec G s| | j js| �� r td��|D ]-}|jj}| j�| �� � |�� t
| j �}t | j �D ]}|d8 }|j
|kr:| j |= nq)qd S )Nz7Cannot extend() after the ordered option was turned on!r/ )r �ordered� last_funcr �mod� _state_idr r1 r" r0 r: r �reversed�_id)r Z state_funcs�fr> �i�declr
r
r �extend� s
���z
Sls.extendNc C sV |sdt � � �}z| �� | W S ty* t|� | �� |< }| j�|� | Y S w )N�.)�_uuidr" �KeyError�StateDeclarationr r1 )r �id�sr
r
r �state� s �z Sls.statec C s | j r| j d S d S )N���)r �r r
r
r rB � s z
Sls.last_funcc C r% r
)r r1 )r Z statefuncr
r
r �
track_func� r( zSls.track_funcc C s� d| j }| �|�jj|d� | �� |= t� }| jr$dd� | jD �|d< | jr=t� |d< }| jD ] }|jdd�||j < q1| j
D ] }|�� ||j < q@| j rbg }t �
� �|| j |� |rbtd�|���|S ) Nr, )r. c S s g | ]
}|d |d i�qS )r r/ r
)�.0�tr
r
r �
<listcomp>� s z$Sls.to_highstate.<locals>.<listcomp>r@ rJ ��contextr+ )r rQ r- r' r" r r r �_reprrF r r r r r4 r r5 )r r. Z slsmod_idr< rJ �extrI r= r
r
r � to_highstate� s(
�zSls.to_highstatec
C s� |� � D ]T\}}| �|�}|� � D ]F\}}d|v r"|�dd�\}}ntdd� |D ��}|�|� t||�}i } |D ]}
t|
t�rNtt|
� � ��\} } | | | < q9||fi | �� qqd S )NrK r/ c s s � | ]
}t |t�r|V qd S r
)r6 �str)rU �xr
r
r � <genexpr>� s � z%Sls.load_highstate.<locals>.<genexpr>) �itemsrQ �rsplitr8 �remove�getattrr6 r7 r9 )
r r< ZsidrI rP �modname�args�funcnamerC Z
named_argsr^ �k�vr
r
r �load_highstate� s"
���zSls.load_highstater
)r r r r � classmethodr" r$ r' r@ rJ rQ rB rT r\ ri r
r
r
r r m s
/
r c @ sD e Zd Zdd� Zdd� ZeZdd� Zdd� Zdd
d �Zdd
d�Z d S )rN c C s || _ g | _d S r
)rF �_mods)r rO r
r
r r s
zStateDeclaration.__init__c C s: | j D ] }|j|kr| S qt|| j�}| j �|� |S r
)rk �_name� StateModulerF r1 )r r �mr
r
r r s
� z StateDeclaration.__getattr__c C � | j S r
)rF rS r
r
r �__str__ � zStateDeclaration.__str__c C s
t | j�S r
)r9 rk rS r
r
r �__iter__ r zStateDeclaration.__iter__Nc s t � fdd�| D ��S )Nc 3 s � | ]}|� � �V qd S r
)rZ )rU rn rX r
r r_ s � z)StateDeclaration._repr.<locals>.<genexpr>r �r rY r
rX r rZ s zStateDeclaration._reprTc C sD t �� d }| j|�� v r[|�� }|r-| jd j|ur-td�| jd j | j|j
|j
j ���|�� � | j� |j
�| � | jd j�� | jD ]}z |j�|j� W qF tyZ Y qFw t�� jjd | j| �� i�}t|t�sytd�d�|����t|�� dd� d �}|r�|D ]\}}|d
s�dd l }td �|� ||i����q�|S ) NrR zVCannot run state({}: {}) that is required by a runtime state({}: {}), at compile time.r z
state.highz-An error occurred while running highstate: {}z; c S s | d d S )Nr/ Z __run_num__r
�rV r
r
r �<lambda>> s z+StateDeclaration.__call__.<locals>.<lambda>)�key�resultz.Failed executing low state at compile time:
{}) r r$ rF r" rB rk �_funcr �formatrl rC rD r0 r rb �_remove_auto_requirer �
ValueErrorr r rQ Z functionsrZ r6 r7 r5 �sortedr` �pprintZpformat) r �checkr rB rn rw rg rh r} r
r
r �__call__ sV
��
� �
�� ���zStateDeclaration.__call__r
)T)
r r r r r � __getitem__rp rr rZ r r
r
r
r rN s
rN c @ s6 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
d �Zd S )
rm c C s || _ || _d | _d S r
)rD rl rx )r r Z parent_declr
r
r r M s
zStateModule.__init__c C s� | j r*|| j jkr | j S |tvr$| j jr td�|| j���|| j _| j S t| j |�S |tv r;td | � | _ }t||�S t|| � | _ }|S )Nz?Multiple state functions({}) not allowed in a state module({})!)rx r �
REQUISITESr ry rl rc �
StateFunction)r r rG r
r
r r R s"
�
zStateModule.__getattr__c O s t | |��||�S r
)rc � configure)r Z_fnamere r; r
r
r r h � zStateModule.__call__c C ro r
)rl rS r
r
r rp k rq zStateModule.__str__Nc C s | j | j�|�fS r
)rl rx rZ rs r
r
r rZ n r� zStateModule._reprr
)r r r r r r rp rZ r
r
r
r rm L s rm c s � fdd�}|S )Nc s< |D ] }| � � |d � q|�� D ] }| j � g|�R � q| S r
)� referencer` )r re r; rC Zmod_refrt r
r �reqs s
z&_generate_requsite_method.<locals>.reqr
)rV r� r
rt r �_generate_requsite_methodr s r� c @ s^ e Zd Zdd� Zdd� Zdd� Zddd �Zd
d � Zd d
� Ze � Z
e D ]Z e
e �e
e < q"[
[ dS )r� c C sf || _ || _g | _d | _t�� d }|jjr1|�� }|r*| � |j � t
| j�d | _|� | � d S d S )NrR r/ ) rC r re �
require_indexr r$ r rA rB Zrequirer: rT )r r Z
parent_modr Zlast_fr
r
r r ~ s �zStateFunction.__init__c C s" | j d ur| j| j = d | _ d S d S r
)r� re rS r
r
r rz � s
�z"StateFunction._remove_auto_requirec O s | � ||� | S r
)r� )r re r; r
r
r r � s zStateFunction.__call__Nc C sB | j s|dkrtd| jj� ���| j s|dkr| jS | j g| j S )NrJ z(No state function specified for module: )r r rC rl re rs r
r
r rZ � s �zStateFunction._reprc C s� t |�}|r7|d }| jjdkr.| jdv r.t|�r.|j|d< t||dd � |d�}|dd �= t|d d�|d< |�� D ] \}}|�||i� q;| j �
|� | S )Nr �cmd)�callZ wait_callr/ )�funcre r; )r ) �listrC rl r �callabler r7 r` r1 re rJ )r re r; �firstrg rh r
r
r r� � s
�
zStateFunction.configurec C sP t |t�r |j}n|r
|std|� d|� ���| j�|t|�t|�igi� d S )Nz+Invalid a requisite reference declaration! z: )r6 rm rD r re r1 r] )r �req_typerC �refr
r
r r� � s
"zStateFunction.referencer
)r r r r rz r rZ r� r� �locals�nsr� r� r� r
r
r
r r� } s
r� )�__doc__Zuuidr rL Z
salt.stater Zsalt.utils.odictr r' �splitr� � Exceptionr r7 r r2 r rN rm r� r� r
r
r
r �<module> s" U � I&