
    .Ph                     t    d Z ddlZddlZddlmZ dZ ej        deded          Z	d	Z
d
 Zd Zde
fdZdS )z-Utilities for dealing with streamed requests.    N   )
exceptionsz"[^"\\]*(?:\\.[^"\\]*)*"z;\s*(z|[^\s;=]+)\s*(?:=\s*(z|[^;]+))?\s*i   c                     t                               |           D ]F}|                                \  }}|dk    r't          j                            |          d         c S Gd S )Nfilename   )_OPTION_HEADER_PIECE_REfinditergroupsospathsplit)content_dispositionmatchkvs       f/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/requests_toolbelt/downloadutils/stream.py_get_filenamer      se    (112EFF ' '||~~1
??7==##A&&&&  4    c                    |ot           j                            |          }|r|s|}nct          | j                            dd                    }|st          j        d          |r!t           j                            ||          }n|}|S )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    * .27==..K )K ))  !6;;
 
 ! 	P$%NOOO 	)w||D*;<<HH )HOr   c                    d}d}d}|r4t          t          |dd                    rd}|}t          |dd          }nVt          | |          }t          j                            |          rt          j        d|z            t          |d          }| 	                    |          D ]}|
                    |           |s|                                 |S )	a5  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566 "
2vt,,)(D997>>(## 	K$%>%IJJJ(D!!&&)&<<  
 



Or   )__doc__os.pathr   rer   r   r   _QUOTED_STRING_REcompiler   _DEFAULT_CHUNKSIZEr   r!   r1    r   r   <module>r9      s    3 3  				             0 $"**5F5F5F5F5F5FH      ( ( (V ,0;M j j j j j jr   