U
    ddg0                     @   sB  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Zddlm	Z	 ddl
mZ ddlmZmZmZmZmZ ddlmZ ddlmZ d	d
dddddddddddddgZe dZG dd dZe Zd-eje eeeejdddZG d d! d!Z G d"d# d#Z!G d$d% d%Z"G d&d' d'Z#e!d(d)Z$e"d*d+d,Z%d+Z&dS ).    N)Path   )set_minimum_tls_version_1_2)Headers)	CertTypesHeaderTypesTimeoutTypesURLTypesVerifyTypes)URL)get_ca_bundle_from_env:zECDHE+AESGCMzECDHE+CHACHA20z
DHE+AESGCMzDHE+CHACHA20zECDH+AESGCMz	DH+AESGCMzECDH+AESzDH+AESz
RSA+AESGCMzRSA+AESz!aNULLz!eNULLz!MD5z!DSSZhttpxc                   @   s   e Zd ZdS )	UnsetTypeN)__name__
__module____qualname__ r   r   1/tmp/pip-unpacked-wheel-fkuvgui5/httpx/_config.pyr   '   s   r   TF)certverify	trust_envhttp2returnc                 C   s   t | |||djS )Nr   r   r   r   )	SSLConfigssl_contextr   r   r   r   create_ssl_context.   s       r   c                   @   s   e Zd ZdZee Zdddddej	e
 eeedddZejdd	d
ZejdddZejdddZejdddZejddddZdS )r   z
    SSL Configuration.
    NTFr   c                C   s&   || _ || _|| _|| _|  | _d S N)r   r   r   r   load_ssl_contextr   )selfr   r   r   r   r   r   r   __init__@   s
    zSSLConfig.__init__r   c                 C   s0   t d| j| j| j| j | jr(|  S |  S )Nz8load_ssl_context verify=%r cert=%r trust_env=%r http2=%r)loggerdebugr   r   r   r   load_ssl_context_verifyload_ssl_context_no_verifyr   r   r   r   r   N   s    zSSLConfig.load_ssl_contextc                 C   s$   |   }d|_tj|_| | |S )zC
        Return an SSL context for unverified connections.
        F)_create_default_ssl_contextcheck_hostnamessl	CERT_NONEverify_mode_load_client_certs)r   contextr   r   r   r%   [   s
    
z$SSLConfig.load_ssl_context_no_verifyc                 C   sH  | j r$| jdkr$t }|dk	r$|| _t| jtjrF| j}| | |S t| jtrZ| j}n*t	| j
 rtt	| j}ntd| j|  }tj|_d|_tjdkrz
d|_W n tk
r   Y nX z
d|_W n tk
r   Y nX | rt|}td| |j|d n*| r:t|}td| |j|d	 | | |S )
zA
        Return an SSL context for verified connections.
        TNzECould not find a suitable TLS CA certificate bundle, invalid path: {}      Fzload_verify_locations cafile=%r)cafilezload_verify_locations capath=%r)capath)r   r   r   
isinstancer)   
SSLContextr,   boolDEFAULT_CA_BUNDLE_PATHr   existsIOErrorformatr'   CERT_REQUIREDr+   r(   sysversion_infopost_handshake_authAttributeErrorhostname_checks_common_nameis_filestrr"   r#   load_verify_locationsis_dir)r   Z	ca_bundler-   Zca_bundle_pathr1   r2   r   r   r   r$   e   sN    






z!SSLConfig.load_ssl_context_verifyc                 C   s|   t t j}t| | jt jO  _|t t jrR| j	rBddgndg}|
| tjdkrxtjd}|rx| jrx||_|S )zy
        Creates the default SSLContext object that's used for both verified
        and unverified connections.
        zhttp/1.1h2r.   SSLKEYLOGFILE)r)   r4   PROTOCOL_TLS_CLIENTr   optionsOP_NO_COMPRESSIONset_ciphersDEFAULT_CIPHERSHAS_ALPNr   set_alpn_protocolsr;   r<   osenvirongetr   keylog_filename)r   r-   Zalpn_idents
keylogfiler   r   r   r'      s    



z%SSLConfig._create_default_ssl_context)r   r   c                 C   s   | j dk	rt| j tr&|j| j d nrt| j tr\t| j dkr\|j| j d | j d d n<t| j trt| j dkr|j| j d | j d | j d d dS )	zF
        Loads client certificates into our SSLContext object
        N)certfile   r   r   )rR   keyfiler/   )rR   rT   password)r   r3   rA   load_cert_chaintuplelen)r   r   r   r   r   r,      s    
zSSLConfig._load_client_certs)r   r   r   __doc__r   certifiwherer6   typingOptionalr   r
   r5   r    r)   r4   r   r%   r$   r'   r,   r   r   r   r   r   9   s"   
8r   c                
   @   s   e Zd ZdZefeeeedejeef ejde	ef ejde	ef ejde	ef ejde	ef dddZ
ejeeje	 f ddd	Zejed
ddZedddZdS )Timeouta  
    Timeout configuration.

    **Usage**:

    Timeout(None)               # No timeouts.
    Timeout(5.0)                # 5s timeout on all operations.
    Timeout(None, connect=5.0)  # 5s timeout on connect, no other timeouts.
    Timeout(5.0, connect=10.0)  # 10s timeout on connect. 5s timeout elsewhere.
    Timeout(5.0, pool=None)     # No timeout on acquiring connection from pool.
                                # 5s timeout elsewhere.
    connectreadwritepoolN)timeoutr`   ra   rb   rc   c                C   sd  t |tr^|tkst|tks"t|tks.t|tks:t|j| _|j| _|j| _|j| _nt |tr|d | _|d | _t	|dk rd n|d | _t	|dk rd n|d | _nt |t
st |t
st |t
st |t
s|| _|| _|| _|| _nlt |t
rtdt |t
r|n|| _t |t
r.|n|| _t |t
rD|n|| _t |t
rZ|n|| _d S )Nr   r   r/   rS      zShttpx.Timeout must either include a default, or set all four parameters explicitly.)r3   r^   UNSETAssertionErrorr`   ra   rb   rc   rW   rX   r   
ValueError)r   rd   r`   ra   rb   rc   r   r   r   r       sD    	



zTimeout.__init__r!   c                 C   s   | j | j| j| jdS )Nr_   r_   r&   r   r   r   as_dict   s
    zTimeout.as_dictotherr   c                 C   s<   t || jo:| j|jko:| j|jko:| j|jko:| j|jkS r   )r3   	__class__r`   ra   rb   rc   r   rk   r   r   r   __eq__  s    



zTimeout.__eq__c              
   C   s`   | j j}t| j| j| j| jhdkr6| d| j dS | d| j d| j d| j d| j d
S )Nr   z	(timeout=)z	(connect=z, read=z, write=z, pool=)rl   r   rX   r`   ra   rb   rc   r   
class_namer   r   r   __repr__  s
    (zTimeout.__repr__)r   r   r   rY   rf   r\   Unionr   r   floatr    DictrA   r]   ri   Anyr5   rn   rr   r   r   r   r   r^      s"   .	r^   c                   @   s^   e Zd ZdZddddeje eje eje dddZej	e
ddd	Zed
ddZdS )Limitsa  
    Configuration for limits to various client behaviors.

    **Parameters:**

    * **max_connections** - The maximum number of concurrent connections that may be
            established.
    * **max_keepalive_connections** - Allow the connection pool to maintain
            keep-alive connections below this point. Should be less than or equal
            to `max_connections`.
    * **keepalive_expiry** - Time limit on idle keep-alive connections in seconds.
    N      @max_connectionsmax_keepalive_connectionskeepalive_expiryc                C   s   || _ || _|| _d S r   ry   )r   rz   r{   r|   r   r   r   r    (  s    zLimits.__init__rj   c                 C   s0   t || jo.| j|jko.| j|jko.| j|jkS r   )r3   rl   rz   r{   r|   rm   r   r   r   rn   3  s    


zLimits.__eq__r!   c                 C   s*   | j j}| d| j d| j d| j dS )Nz(max_connections=z, max_keepalive_connections=z, keepalive_expiry=ro   )rl   r   rz   r{   r|   rp   r   r   r   rr   ;  s     zLimits.__repr__)r   r   r   rY   r\   r]   intrt   r    rv   r5   rn   rA   rr   r   r   r   r   rw     s   rw   c                   @   sl   e Zd Zdddeejejeef  eje dddZ	e
ejejeef  dddZedd	d
ZdS )ProxyN)authheaders)urlr   r   c                C   sd   t |}t|}|jdkr(td||js4|jrN|j|jf}|jd d d}|| _|| _|| _	d S )N)httphttpsZsocks5zUnknown scheme for proxy URL )usernamerU   )
r   r   schemerh   r   rU   	copy_withr   r   r   )r   r   r   r   r   r   r   r    E  s    
zProxy.__init__r!   c                 C   s.   | j d krd S | j d d| j d dfS )Nr   zutf-8r   )r   encoder&   r   r   r   raw_auth[  s    zProxy.raw_authc                 C   sd   | j r| j d dfnd }t| j}|r2d|nd}| jrLdt| jnd}d| | | dS )Nr   z********z, auth= z
, headers=zProxy(ro   )r   rA   r   r   dict)r   r   Zurl_strauth_strZheaders_strr   r   r   rr   d  s
    zProxy.__repr__)r   r   r   r	   r\   r]   TuplerA   r   r    propertybytesr   rr   r   r   r   r   r~   D  s    r~   rx   )rd   d      )rz   r{   )NTTF)'loggingrM   r)   r;   r\   pathlibr   rZ   _compatr   Z_modelsr   _typesr   r   r   r	   r
   Z_urlsr   _utilsr   joinrJ   	getLoggerr"   r   rf   r]   r5   r4   r   r   r^   rw   r~   ZDEFAULT_TIMEOUT_CONFIGZDEFAULT_LIMITSZDEFAULT_MAX_REDIRECTSr   r   r   r   <module>   sd   
     W*+
