XylotrechusZ
U
��.e_� � @ sD d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZz d dlZW n e k
rd d dl
ZY nX d dl Z ddl m
Z
ddlmZmZmZmZmZmZmZmZmZmZmZmZm Z m Z m Z ddl m Z m!Z!m"Z" ddl#m$Z$m%Z% ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ dd l0m1Z1m2Z2 dd
l3m4Z4m5Z5 e�6e7�Z8e�9d �Z:e�9d ej;�Z<e�9d
�Z=dZ>d-dd�Z?G dd� de�Z@G dd� deA�ZBG dd� deB�ZCG dd� deB�ZDG dd� deA�ZEG dd � d eB�ZFG d d � d eB�ZGG d d � d eB�ZHG d!d"� d"eB�ZIG d#d$� d$eB�ZJeJeH� eFd%d&d'�d(d)�ZKeKjLZLe�9d*�ZMG d+d,� d,eA�ZNdS ).� N)�BytesIO� )�DistlibException)�urljoin�urlparse�
urlunparse� url2pathname� pathname2url�queue�quote�unescape� string_types� build_opener�HTTPRedirectHandler� text_type�Request� HTTPError�URLError)� Distribution�DistributionPath� make_dist)�Metadata�MetadataInvalidError) �cached_property�parse_credentials� ensure_slash�split_filename�get_project_data�parse_requirement�parse_name_and_version� ServerProxy�normalize_name)�
get_scheme�UnsupportedVersionError)�Wheel�
is_compatiblez^(\w+)=([a-f0-9]+)z;\s*charset\s*=\s*(.*)\s*$z text/html|application/x(ht)?mlzhttps://pypi.org/pypic C s6 | dkr t } t| dd�}z |�� W �S |d�� X dS )z�
Return all distribution names known by an index.
:param url: The URL of the index.
:return: A list of all known distribution names.
N� @��timeout�close)�
DEFAULT_INDEXr �
list_packages)�url�client� r. �@/usr/lib/python3.8/site-packages/pip/_vendor/distlib/locators.py�get_all_distribution_names) s r0 c @ s$ e Zd ZdZdd� Ze Z ZZdS )�RedirectHandlerzE
A class to work around a bug in some Python 3.2.x releases.
c C s� d }dD ]}||kr|| } q"q|d kr.d S t |�}|jdkrnt|�� |�}t|d�rf|�||� n|||< t�| |||||�S )N)�locationZuri� �replace_header)r �schemer Z get_full_url�hasattrr4 �BaseRedirectHandler�http_error_302) �self�req�fp�code�msg�headersZnewurl�keyZurlpartsr. r. r/ r8 @ s
�z RedirectHandler.http_error_302N)�__name__�
__module__� __qualname__�__doc__r8 Zhttp_error_301Zhttp_error_303Zhttp_error_307r. r. r. r/ r1 7 s r1 c @ s� e Zd ZdZdZdZdZdZed Zd)dd �Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zee
e�Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd*d'd(�ZdS )+�LocatorzG
A base class for locators - things that locate distributions.
)z.tar.gzz.tar.bz2z.tarz.zipz.tgzz.tbz)z.eggz.exe�.whl)z.pdfN)rE �defaultc C s, i | _ || _tt� �| _d| _t�� | _dS )a^
Initialise an instance.
:param scheme: Because locators look for most recent versions, they
need to know the version scheme to use. This specifies
the current PEP-recommended scheme - use ``'legacy'``
if you need to support existing distributions on PyPI.
N) �_cacher5 r r1 �opener�matcherr
�Queue�errors)r9 r5 r. r. r/ �__init__f s
zLocator.__init__c C sV g }| j �� sRz| j �d�}|�|� W n | j jk
rD Y qY nX | j �� q|S )z8
Return any errors which have occurred.
F)rK �empty�get�appendZEmpty� task_done)r9 �result�er. r. r/ �
get_errorsy s
zLocator.get_errorsc C s | � � dS )z>
Clear any errors which may have been logged.
N)rS �r9 r. r. r/ � clear_errors� s zLocator.clear_errorsc C s | j �� d S �N)rG �clearrT r. r. r/ � clear_cache� s zLocator.clear_cachec C s | j S rV ��_schemerT r. r. r/ � _get_scheme� s zLocator._get_schemec C s
|| _ d S rV rY )r9 �valuer. r. r/ � _set_scheme� s zLocator._set_schemec C s t d��dS )a=
For a given project, get a dictionary mapping available versions to Distribution
instances.
This should be implemented in subclasses.
If called from a locate() request, self.matcher will be set to a
matcher for the requirement to satisfy, otherwise it will be None.
� Please implement in the subclassN��NotImplementedError)r9 �namer. r. r/ � _get_project� s
zLocator._get_projectc C s t d��dS )�J
Return all the distribution names known to this locator.
r^ Nr_ rT r. r. r/ �get_distribution_names� s z Locator.get_distribution_namesc C sL | j dkr| �|�}n2|| j kr,| j | }n | �� | �|�}|| j |< |S )z�
For a given project, get a dictionary mapping available versions to Distribution
instances.
This calls _get_project to do all the work, and just implements a caching layer on top.
N)rG rb rU )r9 ra rQ r. r. r/ � get_project� s
zLocator.get_projectc C s^ t |�}t�|j�}d}|�d�}|�| j�}|rBtt|�| j�}|j dkd|j
k||||fS )zu
Give an url a score which can be used to choose preferred URLs
for a given project release.
TrE �httpszpypi.org) r � posixpath�basename�path�endswith�downloadable_extensionsr% r$ �
wheel_tagsr5 �netloc)r9 r, �trh Z
compatibleZis_wheelZis_downloadabler. r. r/ � score_url� s
�zLocator.score_urlc C sR |}|rN| � |�}| � |�}||kr(|}||kr@t�d||� nt�d||� |S )a{
Choose one of two URLs where both are candidates for distribution
archives for the same version of a distribution (for example,
.tar.gz vs. zip).
The current implementation favours https:// URLs over http://, archives
from PyPI over those from other locations, wheel compatibility (if a
wheel) and then the archive name.
zNot replacing %r with %rzReplacing %r with %r)ro �logger�debug)r9 �url1�url2rQ �s1�s2r. r. r/ �
prefer_url� s
zLocator.prefer_urlc C s
t ||�S )zZ
Attempt to split a filename in project name, version and Python version.
)r )r9 �filename� project_namer. r. r/ r � s zLocator.split_filenamec C s dd� }d}t |�\}}}}} }
|
�� �d�r<t�d||
� t�|
�} | rX| �� \} }
nd\} }
|}|r�|d dkr�|dd� }|�d ��r>z~t |�}t
|| j �s�t�d
|� nX|dkr�d }n ||j |�}|�r|j |j
|jt||||| d f�d
�dd� |jD ��d�}W n0 tk
�r: } zt�d|� W 5 d}~X Y nX n�|�| j��sZt�d|� n�t�|� }}| jD ]�}|�|��rn|dt|� � }| �||�}|�s�t�d|� nH|\}}}|�r�|||��r�|||t||||| d f�d�}|�r�||d< �q�qn|�r| �r|
|d| < |S )a
See if a URL is a candidate for a download URL for a project (the URL
has typically been scraped from an HTML page).
If it is, a dictionary is returned with keys "name", "version",
"filename" and "url"; otherwise, None is returned.
c S s t | �t |�kS rV )r! )Zname1Zname2r. r. r/ � same_project� s z:Locator.convert_url_to_download_info.<locals>.same_projectNzegg=z %s: version hint in fragment: %r)NN����/rE zWheel not compatible: %sTr3 z, c S s" g | ]}d � t|dd� ���qS )�.� N)�join�list)�.0�vr. r. r/ �
<listcomp> s z8Locator.convert_url_to_download_info.<locals>.<listcomp>)ra �versionrw r, �python-versionzinvalid path for wheel: %szNot downloadable: %sz No match for project/version: %s)ra r� rw r, r� � %s_digest)r �lower�
startswithrp rq � HASHER_HASH�match�groupsrj r$ r% rl ra r� rw r r~ �pyver� Exception�warningrk rg rh �lenr )r9 r, rx ry rQ r5 rm ri �params�query�frag�m�algo�digestZorigpath�wheelZincluderR rw Zextrn ra r� r� r. r. r/ � convert_url_to_download_info� s� �
���
��
z$Locator.convert_url_to_download_infoc C s2 d}dD ]$}d| }||kr||| f} q.q|S )z�
Get a digest from a dictionary by looking at keys of the form
'algo_digest'.
Returns a 2-tuple (algo, digest) if found, else None. Currently
looks only for SHA256, then MD5.
N)Zsha256�md5r� r. )r9 �inforQ r� r? r. r. r/ � _get_digest1 s zLocator._get_digestc C s� |� d�}|� d�}||kr,|| }|j}nt||| jd�}|j}| �|� |_}|d }||d |< |j|d kr�| �|j|�|_|d �|t � ��
|� | |_ |||< dS )z�
Update a result dictionary (the final result from _get_project) with a
dictionary for a specific version, which typically holds information
gleaned from a filename or URL for an archive for the distribution.
ra r� �r5 r, �digests�urlsN) �pop�metadatar r5 r� r� �
source_urlrv �
setdefault�set�add�locator) r9 rQ r� ra r� �dist�mdr� r, r. r. r/ �_update_version_dataA s
z Locator._update_version_dataFc C s� d}t |�}|dkr td| ��t| j�}|�|j� | _}t�d|t|�j � | �
|j �}t |�dk�r2g }|j
} |D ]z}
|
dkr�qxzH|�|
�s�t�d||
� n*|s�| |
�js�|�|
� nt�d|
|j � W qx tk
r� t�d||
� Y qxX qxt |�d k�rt||jd
�}|�r2t�d |� |d } || }|�r�|j�rH|j|_|�d
i ��| t� �|_i } |�di �}
|jD ] }||
k�rv|
| | |< �qv| |_d| _|S )a
Find the most recent distribution which matches the given
requirement.
:param requirement: A requirement of the form 'foo (1.0)' or perhaps
'foo (>= 1.0, < 2.0, != 1.3)'
:param prereleases: If ``True``, allow pre-release versions
to be located. Otherwise, pre-release versions
are not returned.
:return: A :class:`Distribution` instance, or ``None`` if no such
distribution could be located.
NzNot a valid requirement: %rzmatcher: %s (%s)r} �r� r� z%s did not match %rz%skipping pre-release version %s of %szerror matching %s with %rr )r? zsorted list: %srz r� r� )r r r" r5 rI � requirementrp rq �typer@ re ra r� Z
version_classr� Z
is_prereleaserO r� r� �sortedr? ZextrasrN r� �
download_urlsr� )r9 r� � prereleasesrQ �rr5 rI �versionsZslistZvcls�kr� �dZsdr, r. r. r/ �locateX sX
�
zLocator.locate)rF )F)r@ rA rB rC �source_extensions�binary_extensions�excluded_extensionsrl rk rL rS rU rX r[ r] �propertyr5 rb rd re ro rv r r� r� r� r� r. r. r. r/ rD V s.
JrD c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�PyPIRPCLocatorz�
This locator uses XML-RPC to locate distributions. It therefore
cannot be used with simple mirrors (that only mirror file content).
c s* t t| �jf |� || _t|dd�| _dS )z�
Initialise an instance.
:param url: The URL to use for XML-RPC.
:param kwargs: Passed to the superclass constructor.
r& r' N)�superr� rL �base_urlr r- �r9 r, �kwargs�� __class__r. r/ rL � s zPyPIRPCLocator.__init__c C s t | j�� �S �rc )r� r- r+ rT r. r. r/ rd � s z%PyPIRPCLocator.get_distribution_namesc C s i i d�}| j �|d�}|D ]�}| j �||�}| j �||�}t| jd�}|d |_|d |_|�d�|_ |�dg �|_
|�d�|_ t |�}|r |d } | d
|_
| �| �|_| |_|||< |D ]:} | d
}
| �| �} |d �|t� ��|
� | |d |
< q�q |S )
Nr� Tr� ra r� �license�keywords�summaryr r, r� r� )r- Zpackage_releasesZ release_urlsZ release_datar r5 ra r� rN r� r� r� r r� r� r� r� r� r� r� ) r9 ra rQ r� r� r� �datar� r� r� r, r� r. r. r/ rb � s0
zPyPIRPCLocator._get_project�r@ rA rB rC rL rd rb �
__classcell__r. r. r� r/ r� � s r� c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�PyPIJSONLocatorzw
This locator uses PyPI's JSON interface. It's very limited in functionality
and probably not worth using.
c s t t| �jf |� t|�| _d S rV )r� r� rL r r� r� r� r. r/ rL � s zPyPIJSONLocator.__init__c C s t d��dS �rc z Not available from this locatorNr_ rT r. r. r/ rd � s z&PyPIJSONLocator.get_distribution_namesc
C s
i i d�}t | jdt|� �}�z�| j�|�}|�� �� }t�|�}t | j
d�}|d }|d |_ |d |_ |�
d�|_|�
dg �|_|�
d �|_t|�}| |_|d
} |||j < |d
D ]T}
|
d }|j�|� | �|
�|j|< |d
�|j t� ��|� | �|
�|d |< q�|d
�� D ]�\} } | |j k�r4�q t | j
d�}
|j |
_ | |
_ t|
�}| |_||| < | D ]T}
|
d }|j�|� | �|
�|j|< |d
�| t� ��|� | �|
�|d |< �qh�q W n@ tk
�r } z | j� t |�� t � d|� W 5 d }~X Y nX |S )Nr� z%s/jsonr� r� ra r� r� r� r� r� r, r� ZreleaseszJSON fetch failed: %s) r r� r rH �open�read�decode�json�loadsr r5 ra r� rN r� r� r� r r� r� r� r� r� r� r� �itemsr� rK �putr rp � exception)r9 ra rQ r, �respr� r� r� r� r� r� r� ZinfosZomd�odistrR r. r. r/ rb � sT
z PyPIJSONLocator._get_projectr� r. r. r� r/ r� � s r� c @ s` e Zd ZdZe�dejejB ejB �Z e�dejejB �Z
dd� Z e�dej�Z e
dd� �Zd S )
�Pagez4
This class represents a scraped HTML page.
z�
(rel\s*=\s*(?:"(?P<rel1>[^"]*)"|'(?P<rel2>[^']*)'|(?P<rel3>[^>\s
]*))\s+)?
href\s*=\s*(?:"(?P<url1>[^"]*)"|'(?P<url2>[^']*)'|(?P<url3>[^>\s
]*))
(\s+rel\s*=\s*(?:"(?P<rel4>[^"]*)"|'(?P<rel5>[^']*)'|(?P<rel6>[^>\s
]*)))?
z!<base\s+href\s*=\s*['"]?([^'">]+)c C s4 || _ | | _| _| j�| j �}|r0|�d�| _dS )zk
Initialise an instance with the Unicode page contents and the URL they
came from.
r N)r� r� r, �_base�search�group)r9 r� r, r� r. r. r/ rL s
z
Page.__init__z[^a-z0-9$&+,/:;=?@.#%_\\|-]c C s� dd� }t � }| j�| j�D ]�}|�d�}|d pX|d pX|d pX|d pX|d pX|d }|d
pp|d pp|d }t| j|�}t|�}| j� d
d� |�}|�
||f� q t |dd� dd�}|S )z�
Return the URLs of all the links on a page together with information
about their "rel" attribute, for determining which ones to treat as
downloads and which ones to queue for further scraping.
c S s, t | �\}}}}}}t||t|�|||f�S )zTidy up an URL.)r r r )r, r5 rm ri r� r� r� r. r. r/ �clean- s �zPage.links.<locals>.cleanr3 Zrel1Zrel2Zrel3Zrel4Zrel5Zrel6rr rs Zurl3c S s dt | �d�� S )Nz%%%2xr )�ordr� )r� r. r. r/ �<lambda>; � z Page.links.<locals>.<lambda>c S s | d S )Nr r. )rn r. r. r/ r� ? r� T)r? �reverse) r� �_href�finditerr� � groupdictr r� r � _clean_re�subr� r� )r9 r� rQ r� r� �relr, r. r. r/ �links& s$
��� z
Page.linksN)r@ rA rB rC �re�compile�I�S�Xr� r� rL r� r r� r. r. r. r/ r� s � r� c s� e Zd ZdZejdd� dd� d�Zd � fdd � Zd
d � Zd d
� Z dd� Z
e � de j
�Zdd� Zdd� Zdd� Zdd� Zdd� Ze � d�Zd d � Z� ZS ) �SimpleScrapingLocatorz�
A locator which scrapes HTML pages to locate downloads for a distribution.
This runs multiple threads to do the I/O; performance is at least as good
as pip's PackageFinder, which works in an analogous fashion.
c C s t jtt�d��� S )N)Zfileobj)�gzipZGzipFiler r� r� ��br. r. r/ r� M r� z SimpleScrapingLocator.<lambda>c C s | S rV r. r� r. r. r/ r� N r� )Zdeflater� ZnoneN�
c sl t t| �jf |� t|�| _|| _i | _t� | _t �
� | _ t� | _ d| _
|| _t�� | _t�� | _d| _dS )a�
Initialise an instance.
:param url: The root URL to use for scraping.
:param timeout: The timeout, in seconds, to be applied to requests.
This defaults to ``None`` (no timeout specified).
:param num_workers: The number of worker threads you want to do I/O,
This defaults to 10.
:param kwargs: Passed to the superclass.
FN)r� r� rL r r� r( � _page_cacher� �_seenr
rJ � _to_fetch�
_bad_hosts�skip_externals� num_workers� threading�RLock�_lock�_gplock�platform_check)r9 r, r( r� r� r� r. r/ rL Q s
z SimpleScrapingLocator.__init__c C sF g | _ t| j�D ]0}tj| jd�}|�d� |�� | j �|� qdS )z�
Threads are created only when get_project is called, and terminate
before it returns. They are there primarily to parallelise I/O (i.e.
fetching web pages).
)�targetTN) �_threads�ranger� r� ZThread�_fetchZ setDaemon�startrO )r9 �irn r. r. r/ �_prepare_threadsl s
z&SimpleScrapingLocator._prepare_threadsc C s6 | j D ]}| j�d� q| j D ] }|�� q g | _ dS )zu
Tell all the threads to terminate (by sending a sentinel value) and
wait for them to do so.
N)r� r� r� r~ )r9 rn r. r. r/ �
_wait_threadsy s
z#SimpleScrapingLocator._wait_threadsc C s� i i d�}| j �x || _|| _t| jdt|� �}| j�� | j�� | � � z&t � d|� | j
�|� | j
�� W 5 | �
� X | `W 5 Q R X |S )Nr� z%s/z Queueing %s)r� rQ rx r r� r r� rW r� r� r� rp rq r� r� r~ )r9 ra rQ r, r. r. r/ rb � s
z"SimpleScrapingLocator._get_projectz<\b(linux_(i\d86|x86_64|arm\w+)|win(32|_amd64)|macosx_?\d+)\bc C s | j �|�S )zD
Does an URL refer to a platform-specific download?
)�platform_dependentr� )r9 r, r. r. r/ �_is_platform_dependent� s z,SimpleScrapingLocator._is_platform_dependentc C sZ | j r| �|�rd}n| �|| j�}t�d||� |rV| j� | �| j|� W 5 Q R X |S )a%
See if an URL is a suitable download for a project.
If it is, register information in the result dictionary (for
_get_project) about the specific version it's for.
Note that the return value isn't actually used other than as a boolean
value.
Nzprocess_download: %s -> %s) r� r� r� rx rp rq r� r� rQ )r9 r, r� r. r. r/ �_process_download� s
z'SimpleScrapingLocator._process_downloadc
C s� t |�\}}}}}}|�| j| j | j �r2d}n|| jrJ|�| j�sJd}nd|�| j�s\d}nR|dkrjd}nD|dkrxd}n6| �|�r�d}n&|� dd�d } | �
� dkr�d}nd}t � d ||||� |S )
z�
Determine whether a link URL from a referring page and with a
particular "rel" attribute should be queued for scraping.
F)ZhomepageZdownload)Zhttprf Zftp�:r r Z localhostTz#should_queue: %s (%s) from %s -> %s)
r rj r� r� r� r� r� r� r� �splitr� rp rq )
r9 �linkZreferrerr� r5 rm ri �_rQ �hostr. r. r/ �
_should_queue� s0 �
�z#SimpleScrapingLocator._should_queuec
C s� | j �� }z�z�|r�| �|�}|dkr,W W ��q |jD ]j\}}|| jkr2zB| j�|� | �|�s�| �|||�r�t �
d||� | j � |� W q2 t k
r� Y q2X q2W n2 t
k
r� } z| j� t|�� W 5 d}~X Y nX W 5 | j �� X |s q�q dS )z�
Get a URL to fetch from the work queue, get the HTML page, examine its
links for download candidates and candidates for further scraping.
This is a handy method to run in a thread.
NzQueueing %s from %s)r� rN rP �get_pager� r� r� r� r rp rq r� r r� rK r )r9 r, �pager r� rR r. r. r/ r� � s,
� & z SimpleScrapingLocator._fetchc C sX t |�\}}}}}}|dkr:tj�t|��r:tt|�d�}|| jkr`| j| }t� d||� �n�|�
dd�d }d}|| j kr�t� d||� �n�t |d d
id �}�z�z�t� d |� | j
j|| jd
�} t� d|� | �� }
|
�dd�} t�| ��r�| �� } | �� }
|
�d�}|�r"| j| }||
�}
d}t�| �}|�r@|�d�}z|
�|�}
W n tk
�rn |
�d�}
Y nX t |
| �}|| j| < W n� t k
�r� } z |j dk�r�t� d||� W 5 d}~X Y n� t k
�r } z0t� d||� | j!� | j �"|� W 5 Q R X W 5 d}~X Y n2 t#k
�rB } zt� d||� W 5 d}~X Y nX W 5 || j|< X |S )a
Get the HTML for an URL, possibly from an in-memory cache.
XXX TODO Note: this cache is never actually cleared. It's assumed that
the data won't get stale over the lifetime of a locator instance (not
necessarily true for the default_locator).
�filez
index.htmlzReturning %s from cache: %sr r r Nz Skipping %s due to bad host %szAccept-encodingZidentity)r> z Fetching %sr' z
Fetched %sz Content-Typer3 zContent-Encodingzutf-8zlatin-1i� zFetch failed: %s: %s)$r �osri �isdirr r r r� rp rq r r� r rH r� r( r� rN �HTML_CONTENT_TYPEr� Zgeturlr� �decoders�CHARSETr� r� r� � UnicodeErrorr� r r< r� r r� r� r� )r9 r, r5 rm ri r rQ r r: r� r> Z content_typeZ final_urlr� �encoding�decoderr� rR r. r. r/ r � sZ
&$ z SimpleScrapingLocator.get_pagez<a href=[^>]*>([^<]+)<c C sL t � }| �| j�}|s$td| j ��| j�|j�D ]}|�|�d�� q2|S )rc zUnable to get %sr ) r� r r� r � _distname_rer� r� r� r� )r9 rQ r r� r. r. r/ rd 0 s z,SimpleScrapingLocator.get_distribution_names)Nr� )r@ rA rB rC �zlib�
decompressr rL r� r� rb r� r� r� r� r� r� r r� r r rd r� r. r. r� r/ r� C s& �
� !;
r� c s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS )
�DirectoryLocatorz?
This class locates distributions in a directory tree.
c sN |� dd�| _tt| �jf |� tj�|�}tj�|�sDt d| ��|| _
dS )a�
Initialise an instance.
:param path: The root of the directory tree to search.
:param kwargs: Passed to the superclass constructor,
except for:
* recursive - if True (the default), subdirectories are
recursed into. If False, only the top-level directory
is searched,
� recursiveTzNot a directory: %rN) r� r r� r rL r ri �abspathr
r �base_dir)r9 ri r� r� r. r/ rL A s
zDirectoryLocator.__init__c C s |� | j�S )z�
Should a filename be considered as a candidate for a distribution
archive? As well as the filename, the directory which contains it
is provided, though not used by the current implementation.
)rj rk )r9 rw �parentr. r. r/ �should_includeR s z DirectoryLocator.should_includec C s� i i d�}t �| j�D ]t\}}}|D ]Z}| �||�r$t j�||�}tddtt j�|��dddf�}| � ||�}|r$| �
||� q$| j s q�q|S )Nr� r r3 ) r �walkr r ri r~ r r r r� r� r ) r9 ra rQ �root�dirs�files�fnr, r� r. r. r/ rb Z s"
� z DirectoryLocator._get_projectc C s� t � }t�| j�D ]v\}}}|D ]\}| �||�r tj�||�}tddttj� |��dddf�}| �
|d�}|r |� |d � q | j s q�q|S )rc r r3 Nra )
r� r r r r ri r~ r r r r� r� r )r9 rQ r r r r r, r� r. r. r/ rd j s" � z'DirectoryLocator.get_distribution_names) r@ rA rB rC rL r rb rd r� r. r. r� r/ r < s
r c @ s e Zd ZdZdd� Zdd� ZdS )� JSONLocatora
This locator uses special extended metadata (not available on PyPI) and is
the basis of performant dependency resolution in distlib. Other locators
require archive downloads before dependencies can be determined! As you
might imagine, that can be slow.
c C s t d��dS r� r_ rT r. r. r/ rd � s z"JSONLocator.get_distribution_namesc C s� i i d�}t |�}|r�|�dg �D ]�}|d dks"|d dkr@q"t|d |d |�d d
�| jd �}|j}|d |_d
|kr�|d
r�d|d
f|_|�di �|_|�di �|_|||j < |d �
|j t � �� |d � q"|S )Nr� r ZptypeZsdistZ pyversion�sourcera r� r� zPlaceholder for summary)r� r5 r, r� r� Z requirements�exportsr� )
r rN r r5 r� r� r� Z dependenciesr! r� r� r� r� )r9 ra rQ r� r� r� r� r. r. r/ rb � s*
��
zJSONLocator._get_projectN)r@ rA rB rC rd rb r. r. r. r/ r } s r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�DistPathLocatorz�
This locator finds installed distributions in a path. It can be useful for
adding to an :class:`AggregatingLocator`.
c s* t t| �jf |� t|t�s t�|| _dS )zs
Initialise an instance.
:param distpath: A :class:`DistributionPath` instance to search.
N)r� r" rL �
isinstancer �AssertionError�distpath)r9 r% r� r� r. r/ rL � s zDistPathLocator.__init__c C sP | j �|�}|d kr i i d�}n,|j|d|jt|jg�id|jtd g�ii}|S )Nr� r� r� )r% Zget_distributionr� r� r� )r9 ra r� rQ r. r. r/ rb � s �z DistPathLocator._get_project)r@ rA rB rC rL rb r� r. r. r� r/ r" � s
r" c sR e Zd ZdZ� fdd�Z� fdd�Zdd� Zeej j
e�Z dd � Z d
d � Z � Z
S ) �AggregatingLocatorzI
This class allows you to chain and/or merge a list of locators.
c s* |� dd�| _|| _tt| �jf |� dS )a�
Initialise an instance.
:param locators: The list of locators to search.
:param kwargs: Passed to the superclass constructor,
except for:
* merge - if False (the default), the first successful
search from any of the locators is returned. If True,
the results from all locators are merged (this can be
slow).
�mergeFN)r� r' �locatorsr� r& rL )r9 r( r� r� r. r/ rL � s zAggregatingLocator.__init__c s& t t| ��� | jD ] }|�� qd S rV )r� r& rX r( �r9 r� r� r. r/ rX � s
z AggregatingLocator.clear_cachec C s || _ | jD ]
}||_q d S rV )rZ r( r5 )r9 r\ r� r. r. r/ r] � s
z AggregatingLocator._set_schemec C s� i }| j D ]�}|�|�}|r
| jr�|�di �}|�di �}|�|� |�d�}|r�|r�|�� D ]*\}} ||kr�|| | O < q^| ||< q^|�d�}
|r�|
r�|
�|� q
| jd kr�d} n"d} |D ]}| j�|�r�d} q�q�| r
|} q�q
|S )Nr� r� TF)r( re r' rN �updater� rI r� ) r9 ra rQ r� r� r r� Zdfr� r� Zdd�foundr. r. r/ rb � s8
z AggregatingLocator._get_projectc C s< t � }| jD ]*}z||�� O }W q tk
r4 Y q X q |S r� )r� r( rd r` )r9 rQ r� r. r. r/ rd s
z)AggregatingLocator.get_distribution_names)r@ rA rB rC rL rX r] r� rD r5 �fgetrb rd r� r. r. r� r/ r& � s ,r& zhttps://pypi.org/simple/r& r' �legacyr� z1(?P<name>[\w-]+)\s*\(\s*(==\s*)?(?P<ver>[^)]+)\)$c @ sL e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z ddd�Z
dS )�DependencyFinderz0
Locate dependencies for distributions.
Nc C s |pt | _t| jj�| _dS )zf
Initialise an instance, using the specified locator
to locate distributions.
N)�default_locatorr� r" r5 r) r. r. r/ rL * s
zDependencyFinder.__init__c C sr t �d|� |j}|| j|< || j||jf< |jD ]:}t|�\}}t �d|||� | j� |t
� �� ||f� q2dS )z�
Add a distribution to the finder. This will update internal information
about who provides what.
:param dist: The distribution to add.
zadding distribution %szAdd to provided: %s, %s, %sN) rp rq r? �
dists_by_name�distsr� �providesr �providedr� r� r� )r9 r� ra �pr� r. r. r/ �add_distribution2 s
z!DependencyFinder.add_distributionc C sx t �d|� |j}| j|= | j||jf= |jD ]D}t|�\}}t �d|||� | j| }|� ||f� |s.| j|= q.dS )z�
Remove a distribution from the finder. This will update internal
information about who provides what.
:param dist: The distribution to remove.
zremoving distribution %sz Remove from provided: %s, %s, %sN)
rp rq r? r0 r1 r� r2 r r3 �remove)r9 r� ra r4 r� �sr. r. r/ �remove_distributionA s
z$DependencyFinder.remove_distributionc C sB z| j �|�}W n, tk
r< |�� d }| j �|�}Y nX |S )z�
Get a version matcher for a requirement.
:param reqt: The requirement
:type reqt: str
:return: A version matcher (an instance of
:class:`distlib.version.Matcher`).
r )r5 rI r# r )r9 �reqtrI ra r. r. r/ � get_matcherS s z DependencyFinder.get_matcherc C st | � |�}|j}t� }| j}||krp|| D ]B\}}z|�|�}W n tk
rZ d}Y nX |r,|�|� qpq,|S )z�
Find the distributions which can fulfill a requirement.
:param reqt: The requirement.
:type reqt: str
:return: A set of distribution which can fulfill the requirement.
F)r: r? r� r3 r� r# r� ) r9 r9 rI ra rQ r3 r� �providerr� r. r. r/ �find_providersc s
z DependencyFinder.find_providersc C s� | j | }t� }|D ]$}| �|�}|�|j�s|�|� q|rZ|�d||t|�f� d}n@| �|� | j |= |D ]}| j �|t� ��|� qp| � |� d}|S )a�
Attempt to replace one provider with another. This is typically used
when resolving dependencies from multiple sources, e.g. A requires
(B >= 1.0) while C requires (B >= 1.1).
For successful replacement, ``provider`` must meet all the requirements
which ``other`` fulfills.
:param provider: The provider we are trying to replace with.
:param other: The provider we're trying to replace.
:param problems: If False is returned, this will contain what
problems prevented replacement. This is currently
a tuple of the literal string 'cantreplace',
``provider``, ``other`` and the set of requirements
that ``provider`` couldn't fulfill.
:return: True if we can replace ``other`` with ``provider``, else
False.
Z cantreplaceFT)
�reqtsr� r: r� r� r� � frozensetr8 r� r5 ) r9 r; �other�problemsZrlistZ unmatchedr7 rI rQ r. r. r/ �try_to_replace{ s$
�
z DependencyFinder.try_to_replaceFc C s� i | _ i | _i | _i | _t|p g �}d|krH|�d� |tdddg�O }t|t�rh| }}t� d|� n4| j
j ||d� }}|dkr�t d| ��t� d |� d
|_
t� }t|g�}t|g�}|�r�|�� }|j} | | jkr�| �|� n"| j| }
|
|k�r| �||
|� |j|jB } |j} t� }
|�r`||k�r`d D ]*}d | }||k�r4|
t|d
| �O }
�q4| | B |
B }|D �].}| �|�}|�sFt� d|� | j
j ||d�}|dk�r�|�s�| j
j |d
d�}|dk�r�t� d|� |�d|f� n^|j|j }}||f| jk�r|�|� |�|� || k�rF||k�rF|�|� t� d|j� |D ]R}|j} | | jk�rx| j�|t� ��|� n"| j| }
|
|k�rJ| �||
|� �qJ�qpq�t| j�� �}|D ]&}||k|_ |j �r�t� d|j� �q�t� d|� ||fS )a�
Find a distribution and all distributions it depends on.
:param requirement: The requirement specifying the distribution to
find, or a Distribution instance.
:param meta_extras: A list of meta extras such as :test:, :build: and
so on.
:param prereleases: If ``True``, allow pre-release versions to be
returned - otherwise, don't return prereleases
unless they're all that's available.
Return a set of :class:`Distribution` instances and a set of
problems.
The distributions returned should be such that they have the
:attr:`required` attribute set to ``True`` if they were
from the ``requirement`` passed to ``find()``, and they have the
:attr:`build_time_dependency` attribute set to ``True`` unless they
are post-installation dependencies of the ``requirement``.
The problems should be a tuple consisting of the string
``'unsatisfied'`` and the requirement which couldn't be satisfied
by any distribution known to the locator.
z:*:z:test:z:build:z:dev:zpassed %s as requirement)r� NzUnable to locate %rz
located %sT)ZtestZbuildZdevz:%s:z %s_requireszNo providers found for %rzCannot satisfy %rZ unsatisfiedzAdding %s to install_distsz#%s is a build-time dependency only.zfind done for %s) r3 r1 r0 r= r� r6 r# r rp rq r� r� r Z requestedr� r? r5 rA Z run_requiresZ
meta_requiresZbuild_requires�getattrr< r� r� Zname_and_versionr� �valuesZbuild_time_dependency)r9 r� Z meta_extrasr� r� r� r@ ZtodoZ
install_distsra r? ZireqtsZsreqtsZereqtsr? rR Z all_reqtsr� Z providersr; �nr� r4 r1 r. r. r/ �find� s�
�
�
� zDependencyFinder.find)N)NF) r@ rA rB rC rL r5 r8 r: r< rA rE r. r. r. r/ r. % s
(r. )N)Or� �ior r� Zloggingr rg r� r� � ImportErrorZdummy_threadingr r3 r �compatr r r r r r
r r r
r r r7 r r r r Zdatabaser r r r� r r �utilr r r r r r r r r! r� r" r# r� r$ r% Z getLoggerr@ rp r� r� r� r
r r* r0 r1 �objectrD r� r� r� r� r r r" r& r/ r� ZNAME_VERSION_REr. r. r. r. r/ �<module> s^ D,
@0E: zA& [��