U
    M8c                     @   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z.z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@@@accept-encodinghost
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'}
    ,r   r
   z
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   8/tmp/pip-unpacked-wheel-1wxsa54m/urllib3/util/request.pymake_headers   s.    *


r   c              	   C   sR   |dk	rt | | n:t| dddk	rNz|  }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rNz|| 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ImportErrorbrotliobjectr#   r   r&   r   r   r   r   r   <module>   s0         
H