U
    
Hav                     @   sB  d Z ddlZddlZddlZddlmZmZmZm	Z	m
Z
mZmZ dddddgZed	Zd
ZedZG dd deZdeeef e
e ee	e e	e dddZdeeef e
e ee	e edddZee	d  ddddZeeeef dddZG dd dZG dd dZG dd dZeeeef ZG dd dZdS )z	 PEP 610     N)AnyDictIterableOptionalTypeTypeVarUnion	DirectUrlDirectUrlValidationErrorDirInfoArchiveInfoVcsInfoTzdirect_url.jsonz.^\$\{[A-Za-z0-9-_]+\}(:\$\{[A-Za-z0-9-_]+\})?$c                   @   s   e Zd ZdS )r
   N)__name__
__module____qualname__ r   r   C/tmp/pip-unpacked-wheel-tx790h60/pip/_internal/models/direct_url.pyr
      s   )dexpected_typekeydefaultreturnc                 C   s4   || kr|S | | }t ||s0td||||S )z3Get value from dictionary and verify expected type.z-{!r} has unexpected type for {} (expected {}))
isinstancer
   formatr   r   r   r   valuer   r   r   _get   s    
  r   c                 C   s(   t | |||}|d kr$t| d|S )Nz must have a value)r   r
   r   r   r   r   _get_required)   s    r   InfoType)infosr   c                 C   sF   dd | D } | st dt| dkr.t d| d d k	s>t| d S )Nc                 S   s   g | ]}|d k	r|qS Nr   ).0infor   r   r   
<listcomp>3   s      z#_exactly_one_of.<locals>.<listcomp>z/missing one of archive_info, dir_info, vcs_info   z1more than one of archive_info, dir_info, vcs_infor   )r
   lenAssertionError)r    r   r   r   _exactly_one_of2   s    r(   )kwargsr   c                  K   s   dd |   D S )z Make dict excluding None values.c                 S   s   i | ]\}}|d k	r||qS r!   r   )r"   kvr   r   r   
<dictcomp>B   s       z _filter_none.<locals>.<dictcomp>)items)r)   r   r   r   _filter_none@   s    r.   c                   @   sp   e Zd ZdZdeeee ee ee ddddZeeeee	f  ed  dddZ
eee	f d	d
dZdS )r   vcs_infoN)vcs	commit_idrequested_revisionresolved_revisionresolved_revision_typer   c                 C   s"   || _ || _|| _|| _|| _d S r!   r0   r2   r1   r3   r4   )selfr0   r1   r2   r3   r4   r   r   r   __init__H   s
    zVcsInfo.__init__r   r   c              	   C   sF   |d krd S | t |tdt |tdt|tdt|tdt|tddS )Nr0   r1   r2   r3   r4   )r0   r1   r2   r3   r4   )r   strr   clsr   r   r   r   
_from_dictV   s    




zVcsInfo._from_dictr   c                 C   s   t | j| j| j| j| jdS )Nr5   )r.   r0   r2   r1   r3   r4   r6   r   r   r   _to_dictb   s    zVcsInfo._to_dict)NNN)r   r   r   namer9   r   r7   classmethodr   r   r<   r?   r   r   r   r   r   E   s      "c                   @   s`   e Zd ZdZdee ddddZeeeee	f  ed  dddZ
eee	f d	d
dZdS )r   archive_infoN)hashr   c                 C   s
   || _ d S r!   rC   )r6   rC   r   r   r   r7   o   s    zArchiveInfo.__init__r8   c                 C   s   |d krd S | t |tddS )NrC   rD   )r   r9   r:   r   r   r   r<   u   s    zArchiveInfo._from_dictr=   c                 C   s   t | jdS )NrD   )r.   rC   r>   r   r   r   r?   {   s    zArchiveInfo._to_dict)N)r   r   r   r@   r   r9   r7   rA   r   r   r<   r?   r   r   r   r   r   l   s    "c                   @   s\   e Zd ZdZdeddddZeeee	e
f  ed  ddd	Zee	e
f d
ddZdS )r   dir_infoFN)editabler   c                 C   s
   || _ d S r!   rF   )r6   rF   r   r   r   r7      s    zDirInfo.__init__r8   c                 C   s"   |d krd S | t |tddddS )NrF   F)r   rG   )r   boolr:   r   r   r   r<      s
    zDirInfo._from_dictr=   c                 C   s   t | jp
d dS )NrG   )r.   rF   r>   r   r   r   r?      s    zDirInfo._to_dict)F)r   r   r   r@   rH   r7   rA   r   r   r9   r   r<   r?   r   r   r   r   r      s    "c                   @   s   e Zd Zdeeee ddddZeedddZeedd	d
Z	ddddZ
eeeef d dddZeeef dddZeed dddZedddZdS )r	   N)urlr#   subdirectoryr   c                 C   s   || _ || _|| _d S r!   )rI   r#   rJ   )r6   rI   r#   rJ   r   r   r   r7      s    zDirectUrl.__init__)netlocr   c                 C   sR   d|kr|S | dd\}}t| jtr@| jjdkr@|dkr@|S t|rN|S |S )N@r%   git)splitr   r#   r   r0   
ENV_VAR_REmatch)r6   rK   	user_passZnetloc_no_user_passr   r   r   _remove_auth_from_netloc   s    


z"DirectUrl._remove_auth_from_netlocr=   c                 C   s<   t j| j}| |j}t j|j||j|j	|j
f}|S )zurl with user:password part removed unless it is formed with
        environment variables as specified in PEP 610, or it is ``git``
        in the case of a git URL.
        )urllibparseurlsplitrI   rR   rK   
urlunsplitschemepathqueryfragment)r6   purlrK   surlr   r   r   redacted_url   s    zDirectUrl.redacted_urlc                 C   s   |  |   d S r!   )	from_dictto_dictr>   r   r   r   validate   s    zDirectUrl.validater8   c                 C   sR   t t|tdt|tdttt|tdtt|tdt	t|tdgdS )NrI   rJ   rB   rE   r/   )rI   rJ   r#   )
r	   r   r9   r   r(   r   r<   dictr   r   r:   r   r   r   r^      s    

zDirectUrl.from_dictc                 C   s&   t | j| jd}| j || jj< |S )N)rI   rJ   )r.   r]   rJ   r#   r?   r@   )r6   resr   r   r   r_      s    zDirectUrl.to_dict)sr   c                 C   s   |  t|S r!   )r^   jsonloads)r;   rc   r   r   r   	from_json   s    zDirectUrl.from_jsonc                 C   s   t j|  ddS )NT)	sort_keys)rd   dumpsr_   r>   r   r   r   to_json   s    zDirectUrl.to_json)N)r   r   r   r9   r   r   r7   rR   propertyr]   r`   rA   r   r   r^   r_   rf   ri   r   r   r   r   r	      s"    
)N)N)__doc__rd   reurllib.parserS   typingr   r   r   r   r   r   r   __all__r   ZDIRECT_URL_METADATA_NAMEcompilerO   	Exceptionr
   r9   r   r   r(   r.   r   r   r   r   r	   r   r   r   r   <module>   sJ   $
 
    
   	'