File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/client/ssh/__pycache__/state.cpython-310.pyc
o
�N�g�% � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlZ ddl
Z ddl Z ddl Z ddl
Z ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ e�e�ZG dd� de jj�ZG dd� de jj �Z dd d
�Z dd d �Z dd
d�Z dS )z
Create ssh executor system
� N)�closingc sP e Zd ZdZ d
� fdd� Zddd�Z� fdd�Zd d
� Zd d � Z� Z S )�SSHStatezR
Create a State object which wraps the SSH functions for state operations
Nc s" || _ || _t� j|||d� d S )N)�initial_pillar)�wrapper�context�super�__init__)�self�opts�pillar_overrider r r �� __class__� �I/opt/saltstack/salt/lib/python3.10/site-packages/salt/client/ssh/state.pyr " s zSSHState.__init__c C sn | j | _tj�| j�| _tj�| j�| _tjj| j| jd�}tj�| j|| j| j�| _tj� | j| j�| _
dS )zD
Load up the modules for remote compilation via ssh
)�utilsN) r Z functions�salt�loaderr r
Z serializersZ minion_modsZstatesZrenderZrend)r �data�proxyZlocals_r r r � load_modules. s �zSSHState.load_modulesc sf | j }i }|�|� |�|�di �� |d |d< |d |d< |�d�|d< || _ t� �� }|| _ |S )z�
The opts used during pillar rendering should contain the master
opts in the root namespace. self.opts is the modified minion opts,
containing the original master opts in __master_opts__.
Z__master_opts__�id�saltenvZ pillarenv)r
�update�getr �_gather_pillar)r Z_optsZpopts�pillarr r r r ; s
zSSHState._gather_pillarc C � dS )z(
Stub out check_refresh
Nr )r r �retr r r �
check_refreshQ � zSSHState.check_refreshc C r )z;
Module refresh is not needed, stub it out
Nr �r r r r �module_refreshW r zSSHState.module_refresh)NNNN)NN)
�__name__�
__module__� __qualname__�__doc__r r r r r! �
__classcell__r r r r r s �
r c @ sX e Zd ZdZg Z ddd�Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dS )� SSHHighStatez5
Used to compile the highstate on the master
Nc C sZ || _ tjj�| |� t|||||d�| _tj�| j�| _tj� | j�| _ i | _
g | _ d S )N)r r ) �clientr �state�
BaseHighStater r r Zmatchersr
�topsZ_pydsl_all_declsZ_pydsl_render_stack)r r
r r Zfsclientr r r r r r e s �
zSSHHighState.__init__c C s t jjj�| � d S �N)r r) Z HighState�stack�appendr r r r � push_active} s zSSHHighState.push_activec C r )z'
Stub out load_dynamic
Nr )r �matchesr r r � load_dynamic� r zSSHHighState.load_dynamicc C s� d| j vr t�d� i S tjj�| j | j d �si S i }i }d| j v r(| j d }| jD ]8}|| j �di �vr7q+z|� | j| | j |d�� W q+ t
yc } zt�d||| j d � W Y d}~q+d}~ww |S )z.
Evaluate master_tops locally
r z.Received call for external nodes without an id�grainsZ master_tops)r
r2 z2Top function %s failed with error %s for minion %sN) r
�log�errorr r ZverifyZvalid_idr+ r r � Exception)r r2 r Zfun�excr r r � _master_tops� s0
���zSSHHighState._master_topsc C s | j r
| j �� d S d S r, )r( �destroyr r r r r8 � s �zSSHHighState.destroyc C s | S r, r r r r r � __enter__� s zSSHHighState.__enter__c G s | � � d S r, )r8 )r �_r r r �__exit__� s zSSHHighState.__exit__)NNNNN) r" r# r$ r% r- r r/ r1 r7 r8 r9 r; r r r r r' ^ s
� r' � c
C s� i }| D ]:}t |t�s qd}g }|D ]}|dkr || }n|�d�r#q|�t|| �� q||vr5g ||< |r>|| �|� q|rZ|�d�}|rZ|D ]}|D ]
} || �| g� qNqJ|S )z8
Create a list of file ref objects to reconcile
�baseZ__env__�__�,)�
isinstance�dict�
startswith�extend� salt_refsr. �split)
�chunks�extrasZrefs�chunkr Zcrefsr) Z
extra_refs�env�xr r r �lowstate_file_refs� s2
�
�rK c C sz d}|du rg }t | t�r| �|�r| |vr|�| � t | t�r*| D ]}t||� q"t | t�r;| D ] }t| | |� q1|S )z5
Pull salt file references out of the states
zsalt://N)r@ �strrB r. �listrD rA )r r �proto�compr r r rD � s
rD c C s: t �� }tjj�� }tj�|d�}tj�|d�} tj�|d�}
tjj �
d�gtjj �
d�gtjj �
d�gtjj �
d�gtjj �
d�gtjj �
d �gtjj �
d
�gg} tjj� |d ��} tjj �
|| � W d � n1 smw Y |r�tjj� | d ��} tjj �
|| � W d � n1 s�w Y |r�tjj� |
d ��} tjj �
|| � W d � n1 s�w Y |d u r�d
}z
tj�d|��tj�}
W n ty� tj�dt|���tj�}
Y nw |D ]�}tj�|
d|�}|| �| � tj�||�}tj�|��st�|� || D ]�}|D ]�}tjj �|�d �d�}tj�||�}z
| j|||
d�}W n t�y9 d
}Y nw |�r^tj�||�}tj�|�}tj�|��sVt�|� t�||� �qz
| j |||
d�}W n t�yt d
}Y nw |�r�|D ]1}|t | � |��d � � d�}tj�|||�}tj�|�}tj�|��s�t�|� t�||� �qz�q�q�q
q�zt� � }W n t�y� d }Y nw t�!|� t"t#�$|d���2} tjj�%|�D ]"\} } }|D ]}tj�| |�} | �&| t |�d � �tj�� �q�q�W d � n 1 �s w Y |�rt�!|� t�'|� |S )zf
Generate the execution package from the saltenv file refs and a low state
data structure
z
lowstate.jsonz pillar.jsonzroster_grains.jsonZ_modulesZ_statesZ_grainsZ
_renderersZ
_returners�_outputZ_utilszw+Nr< zsalt-ssh�filesr �/)�cachedirzw:gz)(�tempfileZmkdtempr r rQ Zmkstemp�os�path�join�url�createZfopen�json�dump�rstrip�sep�AttributeErrorrL rC �isdir�makedirs�parse�lstripZ
cache_file�OSError�dirname�shutil�copyZ cache_dir�lenZ get_cachedir�strip�getcwd�chdirr �tarfile�openZos_walk�add�rmtree) Z file_clientrF Z file_refsr Zid_Z
roster_grainsZgendirZ trans_tarZlowfnZpillarfnZroster_grainsfnZ sync_refsZfp_rS r Zcache_dest_rootZenv_root�ref�nameZshortZ
cache_destrV ZtgtZtgt_dirrQ �filename�fn�cwdZtfp�root�dirs�fullr r r �prep_trans_tar� s� � � � � �
�
���
��� �
"�� �
rw )r< r, )NNN)!r% �loggingrU re rk rT �
contextlibr Zsalt.client.sshr Zsalt.client.ssh.shellZ salt.loaderZ salt.minionZ salt.rosterZ
salt.stateZsalt.utils.filesZsalt.utils.jsonZsalt.utils.pathZsalt.utils.stringutilsZsalt.utils.thinZsalt.utils.urlZsalt.utils.verify� getLoggerr" r3 r) �Stater r* r' rK rD rw r r r r �<module> s6
A
R
�