
    8i                     |    S r SSKrSSKrSSKJr  Sr\R                  " S\< S\< S35      r	S	r
S
 rS rS\
4S jrg)z-Utilities for dealing with streamed requests.    N   )
exceptionsz"[^"\\]*(?:\\.[^"\\]*)*"z;\s*(z|[^\s;=]+)\s*(?:=\s*(z|[^;]+))?\s*i   c                     [         R                  U 5       H?  nUR                  5       u  p#US:X  d  M  [        R                  R                  U5      S   s  $    g )Nfilename   )_OPTION_HEADER_PIECE_REfinditergroupsospathsplit)content_dispositionmatchkvs       v/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/requests_toolbelt/downloadutils/stream.py_get_filenamer      sJ    (112EF||~
?77==#A&&	 G
     c                 R   U=(       a    [         R                  R                  U5      nU(       a  U(       d  UnU$ [        U R                  R                  SS5      5      nU(       d  [        R                  " S5      eU(       a!  [         R                  R                  X5      nU$ UnU$ )a  
Given a response and a path, return a file path for a download.

If a ``path`` parameter is a directory, this function will parse the
``Content-Disposition`` header on the response to determine the name of the
file as reported by the server, and return a file path in the specified
directory.

If ``path`` is empty or None, this function will return a path relative
to the process' current working directory.

If path is a full file path, return it.

:param response: A Response object from requests
:type response: requests.models.Response
:param str path: Directory or file path.
:returns: full file path to download as
:rtype: str
:raises: :class:`requests_toolbelt.exceptions.StreamingError`
zcontent-disposition z'No filename given to stream response to)	r   r   isdirr   headersgetexcStreamingErrorjoin)responser   path_is_dirfilepathresponse_filenames        r   get_download_file_pathr!      s    * .277==.KK O *  !6;
 !$$%NOOww||D<H
 O )HOr   c                    SnSnSnU(       a-  [        [        USS5      5      (       a  SnUn[        USS5      nOT[        X5      n[        R                  R                  U5      (       a  [        R                  " SU-  5      e[        US5      nU R                  US9 H  nUR                  U5        M     U(       d  UR                  5         U$ )	a=  Stream a response body to the specified file.

Either use the ``path`` provided or use the name provided in the
``Content-Disposition`` header.

.. warning::

    If you pass this function an open file-like object as the ``path``
    parameter, the function will not close that file for you.

.. warning::

    This function will not automatically close the response object
    passed in as the ``response`` parameter.

If a ``path`` parameter is a directory, this function will parse the
``Content-Disposition`` header on the response to determine the name of the
file as reported by the server, and return a file path in the specified
directory. If no ``path`` parameter is supplied, this function will default
to the process' current working directory.

.. code-block:: python

    import requests
    from requests_toolbelt import exceptions
    from requests_toolbelt.downloadutils import stream

    r = requests.get(url, stream=True)
    try:
        filename = stream.stream_response_to_file(r)
    except exceptions.StreamingError as e:
        # The toolbelt could not find the filename in the
        # Content-Disposition
        print(e.message)

You can also specify the filename as a string. This will be passed to
the built-in :func:`open` and we will read the content into the file.

.. code-block:: python

    import requests
    from requests_toolbelt.downloadutils import stream

    r = requests.get(url, stream=True)
    filename = stream.stream_response_to_file(r, path='myfile')

If the calculated download file path already exists, this function will
raise a StreamingError.

Instead, if you want to manage the file object yourself, you need to
provide either a :class:`io.BytesIO` object or a file opened with the
`'b'` flag. See the two examples below for more details.

.. code-block:: python

    import requests
    from requests_toolbelt.downloadutils import stream

    with open('myfile', 'wb') as fd:
        r = requests.get(url, stream=True)
        filename = stream.stream_response_to_file(r, path=fd)

    print('{} saved to {}'.format(url, filename))

.. code-block:: python

    import io
    import requests
    from requests_toolbelt.downloadutils import stream

    b = io.BytesIO()
    r = requests.get(url, stream=True)
    filename = stream.stream_response_to_file(r, path=b)
    assert filename is None

:param response: A Response object from requests
:type response: requests.models.Response
:param path: *(optional)*, Either a string with the path to the location
    to save the response content, or a file-like object expecting bytes.
:type path: :class:`str`, or object with a :meth:`write`
:param int chunksize: (optional), Size of chunk to attempt to stream
    (default 512B).
:returns: The name of the file, if one can be determined, else None
:rtype: str
:raises: :class:`requests_toolbelt.exceptions.StreamingError`
FNwriteTnamezFile already exists: %swb)
chunk_size)callablegetattrr!   r   r   existsr   r   openiter_contentr#   close)r   r   	chunksize
pre_openedfdr   chunks          r   stream_response_to_filer1   F   s    n J	BHw566
2vt,)(977>>(##$$%>%IJJ(D!&&)&<
 = 

Or   )__doc__os.pathr   rer   r   r   _QUOTED_STRING_REcompiler   _DEFAULT_CHUNKSIZEr   r!   r1    r   r   <module>r9      sU    3  	   0 **5F5FH   (V ,0;M jr   