U
    „~$h¸!  ã                
   @   sh  d dl Z d dlZd dlZd dlmZ d dlmZmZmZm	Z	m
Z
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lmZmZ d dlmZ e  e¡Zee Zee Zeegee f Z eege!f Z"G d	d
„ d
ƒZ#e$e!dœdd„Z%G dd„ dƒZ&G dd„ de#ƒZ'G dd„ de#ƒZ(G dd„ de#ƒZ)G dd„ de#ƒZ*e$e e"e!e!e$ee
e$ e
e# f dœdd„Z+dS )é    N)Údefaultdict)ÚCallableÚDictÚIterableÚListÚOptionalÚTuple)ÚInvalidSdistFilenameÚInvalidWheelFilenameÚcanonicalize_nameÚparse_sdist_filenameÚparse_wheel_filename)ÚInstallationCandidate)ÚLink)Úpath_to_urlÚurl_to_path)Úis_urlc                   @   s>   e Zd Zeee dœdd„ƒZedœdd„Ze	dœdd„Z
dS )	Ú
LinkSource©Úreturnc                 C   s
   t ƒ ‚dS )z,Returns the underlying link, if there's one.N©ÚNotImplementedError©Úself© r   ú?/tmp/pip-unpacked-wheel-g27rp152/pip/_internal/index/sources.pyÚlink   s    zLinkSource.linkc                 C   s
   t ƒ ‚dS )z9Candidates found by parsing an archive listing HTML file.Nr   r   r   r   r   Úpage_candidates"   s    zLinkSource.page_candidatesc                 C   s
   t ƒ ‚dS )z,Links found by specifying archives directly.Nr   r   r   r   r   Ú
file_links&   s    zLinkSource.file_linksN)Ú__name__Ú
__module__Ú__qualname__Úpropertyr   r   r   ÚFoundCandidatesr   Ú
FoundLinksr   r   r   r   r   r      s   r   )Úfile_urlr   c                 C   s   t j| dd�d dkS )NF)Ústrictr   z	text/html)Ú	mimetypesÚ
guess_type)r%   r   r   r   Ú_is_html_file+   s    r)   c                   @   sb   e Zd ZdZeddœdd„Zddœdd„Zeee dœd	d
„ƒZ	ee
eee f dœdd„ƒZdS )Ú_FlatDirectoryToUrlsz"Scans directory and caches resultsN)Úpathr   c                 C   s    || _ g | _ttƒ| _d| _d S )NF)Ú_pathÚ_page_candidatesr   ÚlistÚ_project_name_to_urlsÚ_scanned_directory)r   r+   r   r   r   Ú__init__2   s    
z_FlatDirectoryToUrls.__init__r   c                 C   s¢   t  | j¡D ]Š}t|jƒ}t|ƒr0| j |¡ qzt|j	ƒd }W nB t
k
r„   zt|j	ƒd }W n tk
r~   Y Y qY nX Y nX | j|  |¡ qd| _dS )zrScans directory once and populates both page_candidates
        and project_name_to_urls at the same time
        r   TN)ÚosÚscandirr,   r   r+   r)   r-   Úappendr   Únamer
   r   r	   r/   r0   )r   ÚentryÚurlZproject_filenamer   r   r   Ú_scan_directory8   s    
z$_FlatDirectoryToUrls._scan_directoryc                 C   s   | j s|  ¡  | jS ©N)r0   r8   r-   r   r   r   r   r   O   s    z$_FlatDirectoryToUrls.page_candidatesc                 C   s   | j s|  ¡  | jS r9   )r0   r8   r/   r   r   r   r   Úproject_name_to_urlsV   s    z)_FlatDirectoryToUrls.project_name_to_urls)r   r    r!   Ú__doc__Ústrr1   r8   r"   r   r   r   r:   r   r   r   r   r*   /   s   r*   c                   @   sl   e Zd ZU dZi Zeeef ed< e	eeddœdd„Z
eee dœdd	„ƒZedœd
d„Zedœdd„ZdS )Ú_FlatDirectorySourcezûLink source specified by ``--find-links=<path-to-dir>``.

    This looks the content of the directory, and returns:

    * ``page_candidates``: Links listed on each HTML file in the directory.
    * ``file_candidates``: Archives in the directory.
    Ú_paths_to_urlsN)Úcandidates_from_pager+   Úproject_namer   c                 C   sD   || _ t|ƒ| _|| jkr(| j| | _nt|d�| _| j| j|< d S )N)r+   )Ú_candidates_from_pager   Ú_project_namer>   Ú_path_to_urlsr*   )r   r?   r+   r@   r   r   r   r1   i   s    

z_FlatDirectorySource.__init__r   c                 C   s   d S r9   r   r   r   r   r   r   y   s    z_FlatDirectorySource.linkc                 c   s&   | j jD ]}|  t|ƒ¡E d H  qd S r9   )rC   r   rA   r   ©r   r7   r   r   r   r   }   s    z$_FlatDirectorySource.page_candidatesc                 c   s"   | j j| j D ]}t|ƒV  qd S r9   )rC   r:   rB   r   rD   r   r   r   r   �   s    z_FlatDirectorySource.file_links)r   r    r!   r;   r>   r   r<   r*   Ú__annotations__ÚCandidatesFromPager1   r"   r   r   r   r#   r   r$   r   r   r   r   r   r=   ^   s   
ûr=   c                   @   sT   e Zd ZdZeeddœdd„Zeee dœdd„ƒZ	e
dœd	d
„Zedœdd„ZdS )Ú_LocalFileSourceaC  ``--find-links=<path-or-url>`` or ``--[extra-]index-url=<path-or-url>``.

    If a URL is supplied, it must be a ``file:`` URL. If a path is supplied to
    the option, it is converted to a URL first. This returns:

    * ``page_candidates``: Links listed on an HTML file.
    * ``file_candidates``: The non-HTML file.
    N©r?   r   r   c                 C   s   || _ || _d S r9   ©rA   Ú_link©r   r?   r   r   r   r   r1   �   s    z_LocalFileSource.__init__r   c                 C   s   | j S r9   ©rJ   r   r   r   r   r   ˜   s    z_LocalFileSource.linkc                 c   s&   t | jjƒsd S |  | j¡E d H  d S r9   )r)   rJ   r7   rA   r   r   r   r   r   œ   s    z _LocalFileSource.page_candidatesc                 c   s   t | jjƒrd S | jV  d S r9   )r)   rJ   r7   r   r   r   r   r   ¡   s    z_LocalFileSource.file_links©r   r    r!   r;   rF   r   r1   r"   r   r   r#   r   r$   r   r   r   r   r   rG   †   s   ürG   c                   @   sV   e Zd ZdZeeeddœdd„Zee	e dœdd„ƒZ
edœd	d
„Zedœdd„ZdS )Ú_RemoteFileSourcez¸``--find-links=<url>`` or ``--[extra-]index-url=<url>``.

    This returns:

    * ``page_candidates``: Links listed on an HTML file.
    * ``file_candidates``: The non-HTML file.
    N)r?   Úpage_validatorr   r   c                 C   s   || _ || _|| _d S r9   )rA   Ú_page_validatorrJ   )r   r?   rO   r   r   r   r   r1   °   s    z_RemoteFileSource.__init__r   c                 C   s   | j S r9   rL   r   r   r   r   r   º   s    z_RemoteFileSource.linkc                 c   s&   |   | j¡sd S |  | j¡E d H  d S r9   )rP   rJ   rA   r   r   r   r   r   ¾   s    z!_RemoteFileSource.page_candidatesc                 c   s   | j V  d S r9   rL   r   r   r   r   r   Ã   s    z_RemoteFileSource.file_links)r   r    r!   r;   rF   ÚPageValidatorr   r1   r"   r   r   r#   r   r$   r   r   r   r   r   rN   §   s   
û
rN   c                   @   sT   e Zd ZdZeeddœdd„Zeee dœdd„ƒZ	e
dœd	d
„Zedœdd„ZdS )Ú_IndexDirectorySourcez·``--[extra-]index-url=<path-to-directory>``.

    This is treated like a remote URL; ``candidates_from_page`` contains logic
    for this by appending ``index.html`` to the link.
    NrH   c                 C   s   || _ || _d S r9   rI   rK   r   r   r   r1   Î   s    z_IndexDirectorySource.__init__r   c                 C   s   | j S r9   rL   r   r   r   r   r   Ö   s    z_IndexDirectorySource.linkc                 c   s   |   | j¡E d H  d S r9   rI   r   r   r   r   r   Ú   s    z%_IndexDirectorySource.page_candidatesc                 C   s   dS )Nr   r   r   r   r   r   r   Ý   s    z _IndexDirectorySource.file_linksrM   r   r   r   r   rR   Ç   s   ürR   )Úlocationr?   rO   Ú
expand_dirÚcache_link_parsingr@   r   c          
      C   s   d }d }t j | ¡r"t| ƒ}| }n$|  d¡r:| }t| ƒ}nt| ƒrF| }|d krbd}t || ¡ dS |d krˆt	||t
||d�d�}	||	fS t j |¡rÄ|r¨t|||d�}	nt|t
||d�d�}	||	fS t j |¡rìt|t
||d�d�}	||	fS t d| ¡ |d fS )	Nzfile:zVLocation '%s' is ignored: it is either a non-existing path or lacks a specific scheme.)NN)rU   )r?   rO   r   )r?   r+   r@   )r?   r   z?Location '%s' is ignored: it is neither a file nor a directory.)r2   r+   Úexistsr   Ú
startswithr   r   ÚloggerÚwarningrN   r   Úisdirr=   rR   ÚisfilerG   )
rS   r?   rO   rT   rU   r@   r+   r7   ÚmsgÚsourcer   r   r   Úbuild_sourceá   sZ    	

ÿ
ýý
þ
þþr^   ),Úloggingr'   r2   Úcollectionsr   Útypingr   r   r   r   r   r   Úpip._vendor.packaging.utilsr	   r
   r   r   r   Úpip._internal.models.candidater   Úpip._internal.models.linkr   Úpip._internal.utils.urlsr   r   Úpip._internal.vcsr   Ú	getLoggerr   rX   r#   r$   rF   ÚboolrQ   r   r<   r)   r*   r=   rG   rN   rR   r^   r   r   r   r   Ú<module>   s:    
/(! ø