File: //opt/saltstack/salt/lib/python3.10/site-packages/libcloud/utils/__pycache__/files.cpython-310.pyc
o
;j� � @ sv d dl Z d dlZd dlmZ d dlmZmZ dZg d�Zddd�Z d e
d
edee
de
f fdd
�Zdd� Z
dd� ZdS )� N)� Generator)�b�nexti� )�read_in_chunks�exhaust_iterator�guess_file_mime_typeFc c s � |pt }z| j}|f}W n ty t}| f}Y nw td�}d}|r*t|�dkr�|sMzt||� �}t|�dkr>||7 }nd}W n tyL d}Y nw t|�dkr^|r\|r\td�V dS |ryt||d�E dH }|rx|d|� V ||d� }n|V td�}|r*t|�dks*dS dS )a:
Return a generator which yields data in chunks.
:param iterator: An object which implements an iterator interface
or a File like object with read method.
:type iterator: :class:`object` which implements iterator interface.
:param chunk_size: Optional chunk size (defaults to CHUNK_SIZE)
:type chunk_size: ``int``
:param fill_size: If True, make sure chunks are exactly chunk_size in
length (except for last chunk).
:type fill_size: ``bool``
:param yield_empty: If true and iterator returned no data, only yield empty
bytes object
:type yield_empty: ``bool``
TODO: At some point in the future we could use byte arrays here if version
>= Python 3. This should speed things up a bit and reduce memory usage.
� Fr TN)�data�
chunk_size)�
CHUNK_SIZE�read�AttributeErrorr r �len�
StopIteration�_optimized_chunked_generator) �iteratorr
Z fill_sizeZyield_empty�get_data�argsr �empty�chunk� r �H/opt/saltstack/salt/lib/python3.10/site-packages/libcloud/utils/files.pyr sF �
�
��
��r r r
�returnc c sP � d}|| t | �k r | ||| � V ||7 }|| t | �k s| |d � } | S )Nr )r )r r
Zchunk_startr r r r Y s ��r c C s� t d�}zt t| ��}W n
ty t d�}Y nw t|�dkr@||7 }zt t| ��}W n
ty9 t d�}Y nw t|�dks |S )aM
Exhaust an iterator and return all data returned by it.
:type iterator: :class:`object` which implements iterator interface.
:param iterator: An object which implements an iterator interface
or a File like object with read method.
:rtype ``str``
:return Data returned by the iterator.
r r )r r r r )r r r r r r r e s ���r c C s"