B
    @
d                 @   s   d dl mZ d dlmZ ddlmZ ddlmZmZ dZ	e
ddd	gZd
Zy.yd dlZW n ek
rt   d dlZY nX W n ek
r   Y n
X ed7 Ze ZdddZdd Zdd ZdS )    )absolute_import)	b64encode   )UnrewindableBodyError)binteger_typesz@@@SKIP_HEADER@@@zaccept-encodinghostz
user-agentzgzip,deflateNz,brc             C   s   i }|r6t |trnt |tr*d|}nt}||d< |rB||d< | rNd|d< |rldtt|d |d< |rdtt|d |d	< |rd
|d< |S )a  
    Shortcuts for generating request headers.

    :param keep_alive:
        If ``True``, adds 'connection: keep-alive' header.

    :param accept_encoding:
        Can be a boolean, list, or string.
        ``True`` translates to 'gzip,deflate'.
        List will get joined by comma.
        String will be used as provided.

    :param user_agent:
        String representing the user-agent you want, such as
        "python-urllib3/0.6"

    :param basic_auth:
        Colon-separated username:password string for 'authorization: basic ...'
        auth header.

    :param proxy_basic_auth:
        Colon-separated username:password string for 'proxy-authorization: basic ...'
        auth header.

    :param disable_cache:
        If ``True``, adds 'cache-control: no-cache' header.

    Example::

        >>> make_headers(keep_alive=True, user_agent="Batman/1.0")
        {'connection': 'keep-alive', 'user-agent': 'Batman/1.0'}
        >>> make_headers(accept_encoding=True)
        {'accept-encoding': 'gzip,deflate'}
    ,zaccept-encodingz
user-agentz
keep-alive
connectionzBasic zutf-8authorizationzproxy-authorizationzno-cachezcache-control)
isinstancestrlistjoinACCEPT_ENCODINGr   r   decode)
keep_aliveaccept_encoding
user_agent
basic_authproxy_basic_authdisable_cacheheaders r   9/tmp/pip-install-587nrtaf/urllib3/urllib3/util/request.pymake_headers   s*    *


r   c          	   C   sR   |dk	rt | | n:t| dddk	rNy|  }W n ttfk
rL   t}Y nX |S )z
    If a position is provided, move file to that point.
    Otherwise, we'll attempt to record a position for future use.
    Ntell)rewind_bodygetattrr   IOErrorOSError_FAILEDTELL)bodyposr   r   r   set_file_positione   s    
r$   c          	   C   st   t | dd}|dk	rNt|trNy|| W qp ttfk
rJ   tdY qpX n"|tkr`tdntdt| dS )z
    Attempt to rewind body to a certain position.
    Primarily used for request redirects and retries.

    :param body:
        File-like object that supports seek.

    :param int pos:
        Position to seek to in file.
    seekNzAAn error occurred when rewinding request body for redirect/retry.zRUnable to record file position for rewinding request body during a redirect/retry.z4body_pos must be of type integer, instead it was %s.)	r   r   r   r   r    r   r!   
ValueErrortype)r"   body_pos	body_seekr   r   r   r   w   s    r   )NNNNNN)
__future__r   base64r   
exceptionsr   Zpackages.sixr   r   ZSKIP_HEADER	frozensetZSKIPPABLE_HEADERSr   Z
brotlicffiZ_unused_module_brotliImportErrorZbrotliobjectr!   r   r$   r   r   r   r   r   <module>   s.        
B