.Ph HdZddlZdZgdZdZedfdZedfdZedfdZdS) zTee function implementations.Ni)tee tee_to_filetee_to_bytearrayc#fK|j||D]}|||VdS)N)amtdecode_content)rawstream)responsecallback chunksizerchunks c/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/requests_toolbelt/downloadutils/tee.py_teer sU$$4B%DD cdt|ddvs)t|tjst dt ||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. bmodeztee() 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 TypeErrorrwrite)r fileobjectr rs rrrs`P 7:vr22 2 2 z2: . . 3788 8 **I~ F FFrc#Kt|d5}t||||D]}|V ddddS#1swxYwYdS)aStream 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 rfdrs rrrAs, h  2y.AA  EKKKK s ;??cxt|tstdt||j||S)axStream 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 bytearrayrrextend)r bytearrr rs rrr\sC8 gy ) )%$%% % '.)^ D DDr)__doc__r_DEFAULT_CHUNKSIZE__all__rrrrrrr*s##  4 4 4);.G.G.G.Gb/A#63E$(EEEEEEr