o
    [h!                     @   sv  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#de$de!f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*de$de de"de!de!de$dee
e$ e
e# f f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dee fddZdefddZde	fddZ
dS )	
LinkSourcereturnc                 C      t  )z,Returns the underlying link, if there's one.NotImplementedErrorself r   /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/pip/_internal/index/sources.pylink   s   zLinkSource.linkc                 C   r   )z9Candidates found by parsing an archive listing HTML file.r   r   r   r   r   page_candidates"      zLinkSource.page_candidatesc                 C   r   )z,Links found by specifying archives directly.r   r   r   r   r   
file_links&   r   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                   @   s`   e Zd ZdZdeddfddZdddZedee fd	d
Z	ede
eee f fddZdS )_FlatDirectoryToUrlsz"Scans directory and caches resultspathr   Nc                 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__c                 C   s   t | jD ]A}t|j}t|r| j| qz	t|j	d }W n t
y>   z	t|j	d }W n
 ty;   Y Y qw Y nw | 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	   r0   r1   )r   entryurlZproject_filenamer   r   r   _scan_directory8   s"   

z$_FlatDirectoryToUrls._scan_directoryc                 C      | j s|   | jS N)r1   r9   r.   r   r   r   r   r   O      z$_FlatDirectoryToUrls.page_candidatesc                 C   r:   r;   )r1   r9   r0   r   r   r   r   project_name_to_urlsV   r<   z)_FlatDirectoryToUrls.project_name_to_urls)r   N)r    r!   r"   __doc__strr2   r9   r#   r   r   r   r=   r   r   r   r   r+   /   s    
 r+   c                   @   sr   e Zd ZU dZi Zeeef ed< de	dededdfdd	Z
edee fd
dZdefddZdefddZdS )_FlatDirectorySourcezLink 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_urlscandidates_from_pager,   project_namer   Nc                 C   sF   || _ t|| _|| jv r| j| | _d S t|d| _| j| j|< d S )N)r,   )_candidates_from_pager   _project_namerA   _path_to_urlsr+   )r   rB   r,   rC   r   r   r   r2   i   s   

z_FlatDirectorySource.__init__c                 C   s   d S r;   r   r   r   r   r   r   y   s   z_FlatDirectorySource.linkc                 c   s(    | j jD ]}| t|E d H  qd S r;   )rF   r   rD   r   r   r8   r   r   r   r   }   s   z$_FlatDirectorySource.page_candidatesc                 c   s$    | j j| j D ]}t|V  qd S r;   )rF   r=   rE   r   rG   r   r   r   r      s   z_FlatDirectorySource.file_links)r    r!   r"   r>   rA   r   r?   r+   __annotations__CandidatesFromPager2   r#   r   r   r   r$   r   r%   r   r   r   r   r   r@   ^   s    
 
r@   c                   @   X   e Zd ZdZdededdfddZedee fdd	Z	de
fd
dZdef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.
    rB   r   r   Nc                 C      || _ || _d S r;   rD   _linkr   rB   r   r   r   r   r2         
z_LocalFileSource.__init__c                 C      | j S r;   rN   r   r   r   r   r      r   z_LocalFileSource.linkc                 c   s(    t | jjs	d S | | jE d H  d S r;   )r*   rN   r8   rD   r   r   r   r   r         z _LocalFileSource.page_candidatesc                 c   s    t | jjr	d S | jV  d S r;   )r*   rN   r8   r   r   r   r   r      s   z_LocalFileSource.file_linksr    r!   r"   r>   rI   r   r2   r#   r   r   r$   r   r%   r   r   r   r   r   rK      s    	
rK   c                   @   s\   e Zd ZdZdedededdfddZede	e fd	d
Z
defddZdef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.
    rB   page_validatorr   r   Nc                 C   s   || _ || _|| _d S r;   )rD   _page_validatorrN   )r   rB   rV   r   r   r   r   r2      s   
z_RemoteFileSource.__init__c                 C   rQ   r;   rR   r   r   r   r   r      r   z_RemoteFileSource.linkc                 c   s(    |  | js	d S | | jE d H  d S r;   )rW   rN   rD   r   r   r   r   r      rS   z!_RemoteFileSource.page_candidatesc                 c   s    | j V  d S r;   rR   r   r   r   r   r      s   z_RemoteFileSource.file_links)r    r!   r"   r>   rI   PageValidatorr   r2   r#   r   r   r$   r   r%   r   r   r   r   r   rU      s    

rU   c                   @   rJ   )_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.
    rB   r   r   Nc                 C   rL   r;   rM   rO   r   r   r   r2      rP   z_IndexDirectorySource.__init__c                 C   rQ   r;   rR   r   r   r   r   r      r   z_IndexDirectorySource.linkc                 c   s    |  | jE d H  d S r;   rM   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_linksrT   r   r   r   r   rY      s    
rY   locationrB   rV   
expand_dircache_link_parsingrC   c          
      C   s  d }d }t j| rt| }| }n| dr| }t| }nt| r#| }|d u r1d}t||  dS |d u rDt	||t
||dd}	||	fS t j|re|rWt|||d}	||	fS t|t
||dd}	||	fS t j|ryt|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)r\   )rB   rV   r   )rB   r,   rC   )rB   r   z?Location '%s' is ignored: it is neither a file nor a directory.)r3   r,   existsr   
startswithr   r   loggerwarningrU   r   isdirr@   rY   isfilerK   )
rZ   rB   rV   r[   r\   rC   r,   r8   msgsourcer   r   r   build_source   s\   	





re   ),loggingr(   r3   collectionsr   typingr   r   r   r   r   r   Z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    r_   r$   r%   rI   boolrX   r   r?   r*   r+   r@   rK   rU   rY   re   r   r   r   r   <module>   sJ     
/(! 