U
    dd8                     @   s  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	 d dl
mZ d dlZddlmZ ejrvddlmZ ddd	Zed
d edD  eddd e D ZdSejeef eeje edddZdTejeef eje edddZdedddZ eedddZ!eeedddZ"d Z#e#d! Z$e#d" Z%eeje d#d$d%Z&eje d&d'd(Z'eej(ej)eef  dd)d*Z*eeje d+d,d-Z+d.d/hZ,ej-ej.ej/ej/f  ej0ej.ej/ej/f  d0d1d2Z1d3eje2 d4d5d6Z3d3d3ed7d8d9Z4d3d3ed:d;d<Z5ej)eeje f d&d=d>Z6dUejeef eedd@dAZ7dVejeef eeddBdCZ8eej/ej/dDdEdFZ9eeddGdHZ:eje eje dIdJdKZ;ej<eje2 dLdMdNZ=G dOdP dPZ>G dQdR dRZ?dS )W    N)Path)
getproxies   )PrimitiveDataURLz%22z\\)"\c                 C   s$   i | ]}|d krt |d|qS )   z%{:02X})chrformat.0c r   0/tmp/pip-unpacked-wheel-fkuvgui5/httpx/_utils.py
<dictcomp>   s       r       |c                 C   s   g | ]}t |qS r   )reescaper   r   r   r   
<listcomp>   s     r   )valuelowerencodingreturnc                 C   s.   t | tr| }n| |pd}|r*| S |S )zE
    Coerce str/bytes into a strictly byte-wise HTTP header key.
    ascii)
isinstancebytesencoder   )r   r   r   Zbytes_valuer   r   r   normalize_header_key   s    
r    )r   r   r   c                 C   s   t | tr| S | |pdS )zG
    Coerce str/bytes into a strictly byte-wise HTTP header value.
    r   )r   r   r   r   r   r   r   r   normalize_header_value,   s    
r"   r   )r   r   c                 C   s,   | dkrdS | dkrdS | dkr$dS t | S )z
    Coerce a primitive data type into a string value.

    Note that we prefer JSON-style 'true'/'false' for boolean values here.
    TtrueFfalseN )strr   r   r   r   primitive_value_to_str7   s    r(   )r   r   c                 C   s*   zt |  W n tk
r$   Y dS X dS )z7
    Return `True` if `encoding` is a known codec.
    FT)codecslookupLookupError)r   r   r   r   is_known_encodingF   s
    r,   )namer   r   c                 C   s6   t jt tddd}t||}|  d| d S )z;
    Encode a name/value pair within a multipart form.
    )matchr   c                 S   s   t | d S )Nr   )!_HTML5_FORM_ENCODING_REPLACEMENTSgroup)r.   r   r   r   replacerV   s    z#format_form_param.<locals>.replacerz="r   )typingMatchr&   _HTML5_FORM_ENCODING_REsubr   )r-   r   r1   r   r   r   format_form_paramQ   s    r6             )datar   c                 C   s   | d d }|t jt jfkr dS |d d t jkr6dS |d d t jt jfkrRdS |t}|dkrhdS |dkr|d d d tkrd	S |d
d d tkrdS |dkr|d d t	krdS |d
d  t	krdS d S )N   zutf-32r9   z	utf-8-sigr8   zutf-16r   utf-8z	utf-16-ber   z	utf-16-lez	utf-32-bez	utf-32-le)
r)   BOM_UTF32_LEBOM_UTF32_BEBOM_UTF8BOM_UTF16_LEBOM_UTF16_BEcount_null_null2_null3)r:   sample	nullcountr   r   r   guess_json_utfc   s*    
rH   r   c                  C   sT   dt jkr(tt jd } |  r(t| S dt jkrPtt jd }| rPt|S d S )NZSSL_CERT_FILEZSSL_CERT_DIR)osenvironr   is_filer&   is_dir)Zssl_fileZssl_pathr   r   r   get_ca_bundle_from_env   s    

rN   c           	   
   C   s   g }d}|  |} | s|S td| D ]}z|dd\}}W n tk
r\   |d }}Y nX d| di}|dD ]F}z|d\}} W n tk
r   Y  qY nX |  ||| |< qv|| q&|S )	a7  
    Returns a list of parsed link headers, for more info see:
    https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Link
    The generic syntax of those is:
    Link: < uri-reference >; param1=value1; param2="value2"
    So for instance:
    Link; '<http:/.../front.jpeg>; type="image/jpeg",<http://.../back.jpeg>;'
    would return
        [
            {"url": "http:/.../front.jpeg", "type": "image/jpeg"},
            {"url": "http://.../back.jpeg"},
        ]
    :param value: HTTP Link entity-header field
    :return: list of parsed link headers
    z '"z, *<;r   r%   urlz<> '"=)stripr   split
ValueErrorappend)	r   linksreplace_charsvalrP   paramslinkparamkeyr   r   r   parse_header_links   s&    
r]   )content_typer   c                 C   s   t j }| |d< |jd dS )Nzcontent-type)failobj)emailmessageMessageget_content_charset)r^   msgr   r   r   parse_content_type_charset   s    
re   authorizationzproxy-authorization)itemsr   c                 c   s8   | D ].\}}t | tkr(td|d}||fV  qd S )Nz[secure])match_type_of)to_strr   SENSITIVE_HEADERSto_bytes_or_str)rg   kvr   r   r   obfuscate_sensitive_headers   s    rn   r   )rP   r   c                 C   s"   | j d k	r| j S ddd| jS )NP     )httphttps)portgetscheme)rP   r   r   r   port_or_default   s    
rv   )rP   otherr   c                 C   s(   | j |j ko&| j|jko&t| t|kS )z@
    Return 'True' if the given URLs share the same origin.
    )ru   hostrv   )rP   rw   r   r   r   same_origin   s
    
ry   )rP   locationr   c                 C   s<   | j |j krdS | jdko:t| dko:|jdko:t|dkS )zA
    Return 'True' if 'location' is a HTTPS upgrade of 'url'
    Frq   ro   rr   rp   )rx   ru   rv   )rP   rz   r   r   r   is_https_redirect   s    


r{   c                  C   s   t  } i }dD ]6}| |r| | }d|kr0|nd| || d< qdd | dddD }|D ]&}|d	krzi   S |rfd
|d| < qf|S )z+Gets proxy information from the environment)rq   rr   allz://zhttp://c                 S   s   g | ]}|  qS r   )rR   )r   rx   r   r   r   r      s     z+get_environment_proxies.<locals>.<listcomp>nor%   ,*Nzall://*)r   rt   rS   )Z
proxy_infoZmountsru   hostnameZno_proxy_hostsr   r   r   get_environment_proxies   s    
r   r<   c                 C   s   t | tr| |S | S Nr   r&   r   r!   r   r   r   to_bytes  s    r   c                 C   s   t | tr| S | |S r   )r   r&   decoder!   r   r   r   ri     s    ri   )r   rh   r   c                 C   s   t |tr| S |  S r   r   )r   rh   r   r   r   rk     s    rk   c                 C   s0   | d | d   krdkr,n n| dd S | S )Nr   r   r   r   r'   r   r   r   unquote  s    r   )filenamer   c                 C   s   | rt | d pdS d S )Nr   zapplication/octet-stream)	mimetypes
guess_type)r   r   r   r   guess_content_type  s    r   )streamr   c                 C   sx   z|   }t|j}W nZ ttfk
rr   z$|  }| dtj}| | W n ttfk
rl   Y Y dS X Y nX |S )zs
    Given a file-like stream object, return its length in number of bytes
    without reading it into memory.
    r   N)	filenorJ   fstatst_sizeAttributeErrorOSErrortellseekSEEK_END)r   fdlengthoffsetr   r   r   peek_filelike_length!  s    r   c                   @   sR   e Zd ZedddZddddZddddZedd	d
ZedddZdS )TimerrI   c                    sZ   t  }|dkr dd l}| S |dkrFdd l}tt| I d H S dd l	}|
  S )Ntrior   curio)sniffioZcurrent_async_libraryr   current_timer   r2   castfloatZclockasyncioZget_event_looptime)selfZlibraryr   r   r   r   r   r   	_get_time;  s    zTimer._get_timeNc                 C   s   t  | _d S r   r   perf_counterstartedr   r   r   r   
sync_startJ  s    zTimer.sync_startc                    s   |   I d H | _d S r   r   r   r   r   r   r   async_startM  s    zTimer.async_startc                 C   s   t  }|| j S r   r   r   nowr   r   r   sync_elapsedP  s    zTimer.sync_elapsedc                    s   |   I d H }|| j S r   r   r   r   r   r   async_elapsedT  s    zTimer.async_elapsed)	__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r   r   :  s
   r   c                   @   s~   e Zd ZdZeddddZdeddd	Zee	j
eeef d
ddZed
ddZd edddZe	jedddZdS )
URLPatterna  
    A utility class currently used for making lookups against proxy keys...

    # Wildcard matching...
    >>> pattern = URLPattern("all")
    >>> pattern.matches(httpx.URL("http://example.com"))
    True

    # Witch scheme matching...
    >>> pattern = URLPattern("https")
    >>> pattern.matches(httpx.URL("https://example.com"))
    True
    >>> pattern.matches(httpx.URL("http://example.com"))
    False

    # With domain matching...
    >>> pattern = URLPattern("https://example.com")
    >>> pattern.matches(httpx.URL("https://example.com"))
    True
    >>> pattern.matches(httpx.URL("http://example.com"))
    False
    >>> pattern.matches(httpx.URL("https://other.com"))
    False

    # Wildcard scheme, with domain matching...
    >>> pattern = URLPattern("all://example.com")
    >>> pattern.matches(httpx.URL("https://example.com"))
    True
    >>> pattern.matches(httpx.URL("http://example.com"))
    True
    >>> pattern.matches(httpx.URL("https://other.com"))
    False

    # With port matching...
    >>> pattern = URLPattern("https://example.com:1234")
    >>> pattern.matches(httpx.URL("https://example.com:1234"))
    True
    >>> pattern.matches(httpx.URL("https://example.com"))
    False
    N)patternr   c                 C   s  ddl m} |r.d|kr.td| d| d||}|| _|jdkrJdn|j| _|jd	kr`dn|j| _|j| _|jr|jd	krd | _n|jd
rt	
|jdd  }t	d| d| _nV|jd	rt	
|jdd  }t	d| d| _n t	
|j}t	d| d| _d S )Nr   r   :zUProxy keys should use proper URL forms rather than plain scheme strings. Instead of "z", use "z://"r|   r%   r   z*.r8   z^.+\.$z^(.+\.)?^)_urlsr   rT   r   ru   rx   rs   
host_regex
startswithr   r   compile)r   r   r   rP   domainr   r   r   __init__  s(    zURLPattern.__init__r   )rw   r   c                 C   sV   | j r| j |j krdS | jr8| jd k	r8| j|js8dS | jd k	rR| j|jkrRdS dS )NFT)ru   rx   r   r.   rs   r   rw   r   r   r   matches  s    zURLPattern.matchesrI   c                 C   s4   | j dk	rdnd}t| j }t| j }|||fS )z
        The priority allows URLPattern instances to be sortable, so that
        we can match from most specific to least specific.
        Nr   r   )rs   lenrx   ru   )r   Zport_priorityZhost_priorityZscheme_priorityr   r   r   priority  s    zURLPattern.priorityc                 C   s
   t | jS r   )hashr   r   r   r   r   __hash__  s    zURLPattern.__hash__c                 C   s   | j |j k S r   )r   r   r   r   r   __lt__  s    zURLPattern.__lt__c                 C   s   t |to| j|jkS r   )r   r   r   r   r   r   r   __eq__  s    zURLPattern.__eq__)r   r   r   __doc__r&   r   boolr   propertyr2   Tupleintr   r   r   Anyr   r   r   r   r   r   Y  s   )r   )N)N)r<   )r<   )@r)   email.messager`   r   rJ   r   r   r2   pathlibr   urllib.requestr   r   _typesr   TYPE_CHECKINGr   r   r/   updateranger   joinkeysr4   Unionr&   r   r   Optionalr    r"   r(   r,   r6   rC   rD   rE   rH   rN   ListDictr]   re   rj   Iterabler   AnyStrIteratorrn   r   rv   ry   r{   r   r   ri   rk   r   r   r   r   r   r   r   r   r   r   <module>   st   
    %	&