File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/__pycache__/cp.cpython-310.pyc
Back
o ;j1$ � @ 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(
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings