File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/gzip_util.cpython-310.pyc
o
�N�gn � @ sb d Z ddlZddlZddlZG dd� dej�Zddd�Zdd d
�Zdd d �Zd
d� Z ddd�Z
dS )zp
salt.utils.gzip
~~~~~~~~~~~~~~~
Helper module for handling gzip consistently between 2.7+ and 2.6-
� Nc @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�GzipFileN� c C s t j�| ||||� d S )N)�gzipr �__init__)�self�filename�mode�
compresslevel�fileobj� r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/gzip_util.pyr s zGzipFile.__init__c C s | S )z+Context management protocol. Returns self.r )r r r r � __enter__ s zGzipFile.__enter__c G s | � � dS )z+Context management protocol. Calls close()N)�close)r �argsr r r �__exit__ s zGzipFile.__exit__)NNr N)�__name__�
__module__� __qualname__r r
r r r r r r
s
r �rbr c C s& t tjd�r
t�| ||�S t| ||�S )Nr
)�hasattrr r �open)r r r r r r r s r c C s. t tjd�rtjd|| |d�S td|| |d�S )Nr
� )r r r
r )r r r )r
r r r r r � open_fileobj" s ��r c C s^ t �� }t|d|��}t| t�s| �t�} |�| � W d � n1 s$w Y |�� }|S )z@
Returns the data compressed at gzip level compression.
�wbN) �io�BytesIOr �
isinstance�bytes�encodeZ__salt_system_encoding__�write�getvalue)�datar �buf�ogz�
compressedr r r �compress, s
�r% c C sD t �| �}t|d��}|�� }|W d � S 1 sw Y d S )Nr )r r r �read)r! r" ZigzZuncr r r �
uncompress9 s
$�r' � c c s � z t |�}||kr t�W n ty td��w z^||kret�� }t|d|��,}z
|�| �|��}W n tyL tj j
� | d�} |�| �|��}Y nw W d � n1 sWw Y |� � V ||ks W z| �
� W dS tyw Y dS w z| �
� W w ty� Y w w )a�
Generator that reads chunk_size bytes at a time from a file/filehandle and
yields the compressed result of each read.
.. note::
Each chunk is compressed separately. They cannot be stitched together
to form a compressed file. This function is designed to break up a file
into compressed chunks for transport and decompression/reassembly on a
remote host.
z chunk_size must be an integerr r N)�int�
ValueErrorr r r r r& �AttributeError�salt�utils�filesZfopenr r )Zfh_r Z
chunk_sizeZ
bytes_readr" r# r r r �
compress_file@ sD � � � �� �
�� �� �r/ )r r )r )r r( ) �__doc__r r Zsalt.utils.filesr, r r r r% r' r/ r r r r �<module> s