File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/__pycache__/cp.cpython-310.pyc
o
�N�g1$ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Zddl
Zddl Zddl Zddl
ZddlZddlZddlZddlZe�e�ZG dd� dejjj�ZG dd� d�ZdS )z�
The cp module is used to execute the logic used by the salt-cp command
line application, salt-cp is NOT intended to broadcast large files, it is
intended to handle text files.
Salt-cp can be used to distribute configuration files
� Nc @ s e Zd ZdZdd� ZdS )� SaltCPCliz-
Run the salt-cp command line client
c C s | � � t| j�}|�� dS )z!
Execute salt-cp
N)�
parse_args�SaltCPZconfig�run)�selfZcp_� r �?/opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/cp.pyr " s
z
SaltCPCli.runN)�__name__�
__module__� __qualname__�__doc__r r r r r r s r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )r zL
Create a salt cp object, used to distribute simple files with salt
c C s || _ tjj�� | _d S )N)�opts�salt�utils�platform�
is_windows)r r
r r r �__init__0 s zSaltCP.__init__c
C sF | j rd S zttt�|�j�dd � d�W S tttfy" Y d S w )N���� ) r �int�oct�os�stat�st_mode� TypeError�
IndexError�
ValueError)r �pathr r r �_mode4 s �z SaltCP._modec
C s� i }g }zt �|�}W nL tyW } z@|jtjkr(tj�|� d�� t�d� n"|jtj tj
fv rA| � |�||< W Y d}~||fS W Y d}~||fS W Y d}~||fS d}~ww |s_|� |� |D ]}| �
t j�||��\}}|�|� |�|� qa||fS )z3
Get a list of all specified files
z does not exist
�* N)r �listdir�OSError�errno�ENOENT�sys�stderr�write�exit�EINVAL�ENOTDIRr �append�_recurser �join�update�extend) r r �files�
empty_dirsZ sub_paths�exc�fn_�files_� empty_dirs_r r r r+ < s0
�
� ��
zSaltCP._recursec C sH i }t � }| jd D ]}| �|�\}}|�|� |�|� q
|t|�fS )N�src)�setr
r+ r- �sorted)r r/ r0 r2 r3 r4 r r r � _list_filesV s
zSaltCP._list_filesc C sx t j�|�sd|� d�}tj�|d � t�d� tjj �
|d��}|� � }W d � ||iS 1 s3w Y ||iS )zM
Take a path and return the contents of the file as a string
zThe referenced file, z is not available.�
r �rN) r r �isfiler$ r% r& r' r r r/ Zfopen�read)r r2 �errZfp_�datar r r �
_file_dict_ s
��zSaltCP._file_dictc C s` i }| j d D ]&}tj�|�r|�| �|�� qtj�|�r-tjj �
d� |�� t �
d� q|S )zs
Parse the files indicated in opts['src'] and load them into a python
object for transport
r5 zg{} is a directory, only files are supported in non-chunked mode. Use "--chunked" command line argument.� )r
r r r; r- r? �isdirr r � stringutilsZ print_cli�formatr$ r' )r r/ r2 r r r � _load_filesk s �
�zSaltCP._load_filesc C s< | j d r
| �� }n| �� }tj�|| j �dd�| j � dS )z+
Make the salt client call
�chunked�output�nestedN)r
� run_chunked� run_oldstyler rF Zdisplay_output�get)r �retr r r r } s
z
SaltCP.runc C s� | � � | jd g}| jd d|| jd g}| j�dd�}|dur%|�|� tj�| jd ��
}|j|� W d � S 1 s>w Y dS )zO
Make the salt client call in old-style all-in-one call method
�dest�tgtzcp.recv�timeout�selected_target_optionN� conf_file)rD r
rJ r* r �client�get_local_client�cmd)r �arg�argsrO �localr r r rI � s �
$�zSaltCP.run_oldstylec
s� �� � \}}�jd � �jd }�jd }�jd }�j�d�}t|�p1t|�dkp1tt�d� ���|r9tjj j
ntjj j }tjj
��j�j||pJdd �}|d
} � ���fd d �}
i } d
tj �tj��jd ���} |�� D ]�\}
}|
|
�}d}i }||
�jd d�D ]�}t�tjj�|��}|dk}t�d|
|r�|� d�nd|||r�d|� d�nd� |d|||||g|g}|dur�|�|� | j |� }|s�d� |dkr�dnd�jd �}| D ]
}|| � |i �|< q� n$|�� D ]\}}|| � |i �|< |du�r||v�r|||< q�|d7 }q�|�� D ] \}}|| | |< �qqo|D ]>}|
|�}t�d||�r/|� d�nd|� |d|dg|g}|du�rE|�|� | j |� �� D ]
\}}|| � |i �|< �qL�q | W d � S 1 �siw Y dS ) zT
Make the salt client call in the new fasion chunked multi-call way
rL �gziprM rN rO r@ z[\\/]$�glob)Ztgt_type�minionsc s� | �j d v r�rtj�� tj�| ��S � S �j d D ] }tj�| |tj �}|���r-qtj�� tj�|�|� S t� d| � d S )Nr5 z!Failed to find remote path for %s)
r
r r r, �basename�relpath�sep�
startswith�log�error)r2 r r[ �rL Z dest_is_dir�parentr r r �_get_remote_path� s ��
z,SaltCP.run_chunked.<locals>._get_remote_pathz..rP Zsalt_cp_chunk_size)Z
chunk_sizez#Copying %s to %starget '%s' as %s%s� � z (chunk #�)zcp.recv_chunkedNzXPublish failed.{} It may be necessary to decrease salt_cp_chunk_size (current value: {})z File partially transferred.Tz&Creating empty dir %s on %starget '%s') r8 r
rJ �bool�len�re�searchr r Z gzip_utilZ
compress_file� itertoolsZ read_filerY Z CkMinionsZ
check_minionsr r\ rQ rR �items�base64� b64encoderB �to_bytesr^ �debugr* rS rC �
setdefault)r r/ r0 rW rM rN rO �readerZ_resrY rb rK rV r2 �modeZ remote_path�indexZfailed�chunkr* rU �result�msgZminionZ minion_idZ
minion_ret�dirnamer r` r rH � s�
"� ���
� �
���
��
�&�zSaltCP.run_chunkedN)
r r
r r r r r+ r8 r? rD r rI rH r r r r r + s r )r rl r" �loggingr rh r$ Z salt.clientr Z salt.outputZsalt.utils.filesZsalt.utils.gzip_utilZsalt.utils.itertoolsZsalt.utils.minionsZsalt.utils.parsersZsalt.utils.platformZsalt.utils.stringutilsZsalt.utils.verify� getLoggerr r^ r �parsersZSaltCPOptionParserr r r r r r �<module> s(