3
P\
@ sP d Z ddlZdZdddgZdd Zedfd dZedfd
dZedfd dZdS ) z Tee function implementations. Ni tee tee_to_filetee_to_bytearrayc c s, x&| j j||dD ]}|| |V qW d S )N)Zamtdecode_content)rawstream)responsecallback chunksizer chunk r /usr/lib/python3.6/tee.py_tee s
r c C s4 dt |ddkpt|tjs$tdt| |j||S )a" Stream the response both to the generator and a file.
This will stream the response body while writing the bytes to
``fileobject``.
Example usage:
.. code-block:: python
resp = requests.get(url, stream=True)
with open('save_file', 'wb') as save_file:
for chunk in tee(resp, save_file):
# do stuff with chunk
.. code-block:: python
import io
resp = requests.get(url, stream=True)
fileobject = io.BytesIO()
for chunk in tee(resp, fileobject):
# do stuff with chunk
:param response: Response from requests.
:type response: requests.Response
:param fileobject: Writable file-like object.
:type fileobject: file, io.BytesIO
:param int chunksize: (optional), Size of chunk to attempt to stream.
:param bool decode_content: (optional), If True, this will decode the
compressed content of the response.
:raises: TypeError if the fileobject wasn't opened with the right mode
or isn't a BytesIO object.
bmode ztee() will write bytes directly to this fileobject, it must be opened with the "b" flag if it is a file or inherit from io.BytesIO.)getattr
isinstanceioBytesIO TypeErrorr write)r Z
fileobjectr
r r r r
r s ( c c s8 t |d$}x t| |||D ]
}|V q W W dQ R X dS )a Stream the response both to the generator and a file.
This will open a file named ``filename`` and stream the response body
while writing the bytes to the opened file object.
Example usage:
.. code-block:: python
resp = requests.get(url, stream=True)
for chunk in tee_to_file(resp, 'save_file'):
# do stuff with chunk
:param response: Response from requests.
:type response: requests.Response
:param str filename: Name of file in which we write the response content.
:param int chunksize: (optional), Size of chunk to attempt to stream.
:param bool decode_content: (optional), If True, this will decode the
compressed content of the response.
wbN)openr )r filenamer
r fdr r r r
r A s c C s" t |tstdt| |j||S )ax Stream the response both to the generator and a bytearray.
This will stream the response provided to the function, add them to the
provided :class:`bytearray` and yield them to the user.
.. note::
This uses the :meth:`bytearray.extend` by default instead of passing
the bytearray into the ``readinto`` method.
Example usage:
.. code-block:: python
b = bytearray()
resp = requests.get(url, stream=True)
for chunk in tee_to_bytearray(resp, b):
# do stuff with chunk
:param response: Response from requests.
:type response: requests.Response
:param bytearray bytearr: Array to add the streamed bytes to.
:param int chunksize: (optional), Size of chunk to attempt to stream.
:param bool decode_content: (optional), If True, this will decode the
compressed content of the response.
z4tee_to_bytearray() expects bytearr to be a bytearray)r bytearrayr r extend)r Zbytearrr
r r r r
r \ s
)__doc__r Z_DEFAULT_CHUNKSIZE__all__r r r r r r r r
s
0