shell bypass 403
3
���eE� � @ s@ d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZy 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$ ddl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z. dd l/m0Z0m1Z1 dd
l2m3Z3m4Z4 ej5e6�Z7ej8d �Z9ej8d ej:�Z;ej8d
�Z<dZ=d-dd�Z>G dd� de�Z?G dd� de@�ZAG dd� deA�ZBG dd� deA�ZCG dd� de@�ZDG dd � d eA�ZEG d d � d eA�ZFG d d � d eA�ZGG d!d"� d"eA�ZHG d#d$� d$eA�ZIeIeG� eEd%d&d'�d(d)�ZJeJjKZKej8d*�ZLG d+d,� d,e@�ZMdS ).� 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) �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)?mlz https://pypi.python.org/pypic C s | dkr t } t| dd�}|j� S )z�
Return all distribution names known by an index.
:param url: The URL of the index.
:return: A list of all known distribution names.
Ng @)�timeout)�
DEFAULT_INDEXr �
list_packages)�url�client� r* � /usr/lib/python3.6/locators.py�get_all_distribution_names) s r, 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 }x dD ]}||kr
|| }P q
W |d kr0d S t |�}|jdkrpt|j� |�}t|d�rh|j||� n|||< tj| |||||�S )N�location�uri� �replace_header)r. r/ )r �schemer Z get_full_url�hasattrr1 �BaseRedirectHandler�http_error_302) �self�req�fp�code�msg�headersZnewurl�keyZurlpartsr* r* r+ r5 = s
z RedirectHandler.http_error_302N)�__name__�
__module__� __qualname__�__doc__r5 Zhttp_error_301Zhttp_error_303Zhttp_error_307r* r* r* r+ r- 4 s r- c @ s� e Zd ZdZd/Zd0Zd1Zd Zed2 Zd3dd�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zee
e�Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd4d-d.�Zd S )5�LocatorzG
A base class for locators - things that locate distributions.
�.tar.gz�.tar.bz2�.tar�.zip�.tgz�.tbz�.egg�.exe�.whl�.pdfN�defaultc C s, i | _ || _tt� �| _d| _tj� | _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) �_cacher2 r r- �opener�matcherr
�Queue�errors)r6 r2 r* r* r+ �__init__c s
zLocator.__init__c C sX g }xN| j j� sRy| j jd�}|j|� W n | j jk
rD wY nX | j j� qW |S )z8
Return any errors which have occurred.
F)rQ �empty�get�appendZEmpty� task_done)r6 �result�er* r* r+ �
get_errorsv s zLocator.get_errorsc C s | j � dS )z>
Clear any errors which may have been logged.
N)rY )r6 r* r* r+ � clear_errors� s zLocator.clear_errorsc C s | j j� d S )N)rM �clear)r6 r* r* r+ � clear_cache� s zLocator.clear_cachec C s | j S )N)�_scheme)r6 r* r* r+ � _get_scheme� s zLocator._get_schemec C s
|| _ d S )N)r] )r6 �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.
z Please implement in the subclassN)�NotImplementedError)r6 �namer* r* r+ � _get_project� s
zLocator._get_projectc C s t d��dS )zJ
Return all the distribution names known to this locator.
z Please implement in the subclassN)ra )r6 r* r* r+ �get_distribution_names� s z Locator.get_distribution_namesc C sL | j dkr| j|�}n2|| j kr,| j | }n | j� | j|�}|| 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)rM rc rZ )r6 rb rW r* r* r+ � get_project� s
zLocator.get_projectc C sP t |�}tj|j�}d}|jd�}|r6tt|�| j�}|jdkd|j k|||fS )zu
Give an url a score which can be used to choose preferred URLs
for a given project release.
Tz.whl�httpszpypi.python.org)
r � posixpath�basename�path�endswithr$ r# �
wheel_tagsr2 �netloc)r6 r( �trh Z
compatibleZis_wheelr* r* r+ � score_url� s
zLocator.score_urlc C sR |}|rN| j |�}| j |�}||kr(|}||kr@tjd||� ntjd||� |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)rn �logger�debug)r6 �url1�url2rW �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 )r6 �filename� project_namer* r* r+ r � s zLocator.split_filenamec C s dd� }d}t |�\}}}}} }
|
j� jd�r<tjd||
� tj|
�} | rX| j� \} }
nd\} }
|}|r�|d dkr�|dd� }|jd��r2yrt |�}t
|| j �� r�|dkr�d }n ||j |�}|� r�|j |j
|jt||||| d
f�d jd d
� |jD ��d�}W n0 tk
�r. } ztjd|� W Y dd}~X nX n�|j| j��r�tj|� }}x�| jD ]�}|j|��rV|dt|� � }| j||�}|�s�tjd|� nJ|\}}}| �s�|||��r�|||t||||| d
f�d�}|�r�||d< P �qVW |�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 )N)r )Zname1Zname2r* r* r+ � same_project� s z:Locator.convert_url_to_download_info.<locals>.same_projectNzegg=z %s: version hint in fragment: %rr �/z.whlTr0 z, c S s" g | ]}d j t|dd� ���qS )�.� N)�join�list)�.0�vr* r* r+ �
<listcomp> s z8Locator.convert_url_to_download_info.<locals>.<listcomp>)rb �versionrv r( zpython-versionzinvalid path for wheel: %sz No match for project/version: %s)rb r� rv r( zpython-versionz %s_digest)NN���r� )r �lower�
startswithro rp � HASHER_HASH�match�groupsrj r# r$ rk rb r� rv r r| �pyver� Exception�warning�downloadable_extensionsrg rh �lenr )r6 r( rw rx rW r2 rl ri �params�query�frag�m�algo�digestZorigpath�wheel�includerX rv Zextrm rb r� r� r* r* r+ � convert_url_to_download_info� sf
z$Locator.convert_url_to_download_infoc C s4 d}x*dD ]"}d| }||kr
||| f}P q
W |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�sha256�md5z %s_digest)r� r� r* )r6 �inforW r� r< r* r* r+ � _get_digest) s
zLocator._get_digestc C s� |j d�}|j d�}||kr,|| }|j}nt||| jd�}|j}| j|� |_}|d }||d |< |j|d kr�| j|j|�|_|d j|t � �j
|� | |_ |||< 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.
rb r� )r2 r( �digests�urlsN) �pop�metadatar r2 r� r� �
source_urlru �
setdefault�set�add�locator) r6 rW r� rb r� �dist�mdr� r( r* r* r+ �_update_version_data9 s
z Locator._update_version_dataFc C s� d}t |�}|dkr td| ��t| j�}|j|j� | _}tjd|t|�j � | j
|j �}t |�dk�r8g }|j
} x�|D ]|}
|
d
kr�qzyJ|j|
�s�tjd||
� n,|s�| |
�j r�|j|
� ntjd|
|j � W qz tk
r� tjd ||
� Y qzX qzW t |�d
k�rt||jd �}|�r8tjd |� |d } || }|�r�|j�rN|j|_|jdi �j| t� �|_i } |jdi �}
x&|jD ] }||
k�r~|
| | |< �q~W | |_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: %s)r� r� r� )r r r! r2 rO � requirementro rp �typer= re rb r� Z
version_classr� Z
is_prereleaserU r� r� �sortedr< ZextrasrT r� �
download_urlsr� )r6 r� � prereleasesrW �rr2 rO �versionsZslistZvcls�kr� �dZsdr( r* r* r+ �locateP sT
zLocator.locate)rB rC rD rE rF rG )rH rI rJ )rK )rJ )rL )F)r= r> r? r@ �source_extensions�binary_extensions�excluded_extensionsrk r� rR rY rZ r\ r^ r` �propertyr2 rc rd re rn ru r r� r� r� r� r* r* r* r+ rA S s.
FrA 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.
g @)r% N)�superr� rR �base_urlr r) )r6 r( �kwargs)� __class__r* r+ rR � s zPyPIRPCLocator.__init__c C s t | jj� �S )zJ
Return all the distribution names known to this locator.
)r� r) r' )r6 r* r* r+ rd � s z%PyPIRPCLocator.get_distribution_namesc C s i i d�}| j j|d�}x�|D ]�}| j j||�}| j j||�}t| jd�}|d |_|d |_|jd�|_ |jdg �|_
|jd�|_ t |�}|r |d } | d
|_
| j| �|_| |_|||< xB|D ]:} | d
}
| j| �} |d j|t� �j|
� | |d |
< q�W q W |S )
N)r� r� T)r2 rb r� �license�keywords�summaryr r( r� r� )r) Zpackage_releasesZ release_urlsZ release_datar r2 rb r� rT r� r� r� r r� r� r� r� r� r� r� ) r6 rb rW r� r r� �datar� r� r� r( r� r* r* r+ rc � s0
zPyPIRPCLocator._get_project)r= r> r? r@ rR rd rc �
__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 )N)r� r� rR r r� )r6 r( r� )r� r* r+ rR � s zPyPIJSONLocator.__init__c C s t d��dS )zJ
Return all the distribution names known to this locator.
z Not available from this locatorN)ra )r6 r* r* r+ rd � s z&PyPIJSONLocator.get_distribution_namesc C s i i d�}t | jdt|� �}�y�| jj|�}|j� j� }tj|�}t | j
d�}|d }|d |_ |d |_ |j
d�|_|j
dg �|_|j
d �|_t|�}| |_|d
} |||j < x`|d
D ]T}
|
d }|jj|� | j|
�|j|< |d
j|j t� �j|� | j|
�|d |< q�W x�|d
j� D ]�\} } | |j k�r:�q"t | j
d�}
|j |
_ | |
_ t|
�}| |_||| < x\| D ]T}
|
d }|jj|� | j|
�|j|< |d
j| t� �j|� | j|
�|d |< �qpW �q"W W n@ tk
�r } z"| jj t |�� t j d|� W Y d d }~X nX |S )N)r� r� z%s/json)r2 r� rb r� r� r� r� r� r( r� ZreleaseszJSON fetch failed: %s) r r� r rN �open�read�decode�json�loadsr r2 rb r� rT r� r� r� r r� r� r� r� r� r� r� �itemsr� rQ �putr ro � exception)r6 rb rW r( �respr� r� r� r� r� r� r� ZinfosZomd�odistrX r* r* r+ rc � sT
" z PyPIJSONLocator._get_project)r= r> r? r@ rR rd rc r� r* r* )r� r+ r� � s r� c @ s` e Zd ZdZejdejejB ejB �Z ejdejejB �Z
dd� Z ejdej�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 || _ | | _| _| jj| j �}|r0|jd�| _dS )zk
Initialise an instance with the Unicode page contents and the URL they
came from.
r N)r� r� r( �_base�search�group)r6 r� r( r� r* r* r+ rR s
z
Page.__init__z[^a-z0-9$&+,/:;=?@.#%_\\|-]c C s� dd� }t � }x�| jj| j�D ]�}|jd�}|d pZ|d pZ|d pZ|d pZ|d pZ|d }|d
pr|d pr|d }t| j|�}t|�}| jj d
d� |�}|j
||f� q W 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( r2 rl ri r� r� r� r* r* r+ �clean% s zPage.links.<locals>.cleanr0 Zrel1Zrel2Zrel3Zrel4Zrel5Zrel6rq rr Zurl3c S s dt | jd�� S )Nz%%%2xr )�ordr� )r� r* r* r+ �<lambda>3 s z Page.links.<locals>.<lambda>c S s | d S )Nr r* )rm r* r* r+ r� 7 s T)r< �reverse) r� �_href�finditerr� � groupdictr r� r � _clean_re�subr� r� )r6 r� rW r� r� �relr( r* r* r+ �links s
z
Page.linksN)r= r> r? r@ �re�compile�I�S�Xr� r� rR 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 j de j
�Zdd� Zdd� Zdd� Zdd� Zdd� Ze j 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�j� S )N)Zfileobj)�gzipZGzipFiler r� r� )�br* r* r+ r� E s z SimpleScrapingLocator.<lambda>c C s | S )Nr* )r� r* r* r+ r� F s )Zdeflater� ZnoneN�
c sf t t| �jf |� t|�| _|| _i | _t� | _t j
� | _ t� | _ d| _
|| _tj� | _tj� | _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� rR r r� r% � _page_cacher� �_seenr
rP � _to_fetch�
_bad_hosts�skip_externals� num_workers� threading�RLock�_lock�_gplock)r6 r( r% r� r� )r� r* r+ rR I s
z SimpleScrapingLocator.__init__c C sJ g | _ x>t| j�D ]0}tj| jd�}|jd� |j� | j j|� qW dS )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�startrU )r6 �irm r* r* r+ �_prepare_threadsc s
z&SimpleScrapingLocator._prepare_threadsc C s> x| j D ]}| jjd� qW x| j D ] }|j� q$W 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| )r6 rm r* r* r+ �
_wait_threadsp s
z#SimpleScrapingLocator._wait_threadsc C s� i i d�}| j �x || _|| _t| jdt|� �}| jj� | jj� | j � z&t
j d|� | j j
|� | j j� W d | j� X | `W d Q R X |S )N)r� r� z%s/z Queueing %s)r� rW rw r r� r r� r[ r� r� ro rp r� r� r| r� )r6 rb rW r( r* r* r+ rc } s
z"SimpleScrapingLocator._get_projectz<\b(linux-(i\d86|x86_64|arm\w+)|win(32|-amd64)|macosx-?\d+)\bc C s | j j|�S )zD
Does an URL refer to a platform-specific download?
)�platform_dependentr� )r6 r( r* r* r+ �_is_platform_dependent� s z,SimpleScrapingLocator._is_platform_dependentc
C sT | j |�rd}n| j|| j�}tjd||� |rP| j� | j| j|� W dQ 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� rw ro rp r� r� rW )r6 r( r� r* r* r+ �_process_download� s
z'SimpleScrapingLocator._process_downloadc
C s� t |�\}}}}}}|j| j| j | j �r2d}n~| jrL|j| j� rLd}nd|j| j�s^d}nR|d
krld}nD|dkrzd}n6| j|�r�d}n&|j dd�d } | j
� d
kr�d}nd }t j d ||||� |S )z�
Determine whether a link URL from a referring page and with a
particular "rel" attribute should be queued for scraping.
F�homepage�download�httprf �ftp�:r r Z localhostTz#should_queue: %s (%s) from %s -> %s)r� r� )r� rf r� )
r rj r� r� r� r� r� r� r� �splitr� ro rp )
r6 �linkZreferrerr� r2 rl ri �_rW �hostr* r* r+ �
_should_queue� s*
z#SimpleScrapingLocator._should_queuec C s� x�| j j� }z�yz|r�| j|�}|dkr(wx\|jD ]R\}}|| jkr0| jj|� | j|� r0| j|||�r0tj d||� | j j
|� q0W W n2 t k
r� } z| j j
t
|�� W Y dd}~X nX W d| j j� X |sP qW 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� rT �get_pager� r� r� r� r� ro rp r� r� rQ r rV )r6 r( �pager� r� rX r* r* r+ r� � s&
& z SimpleScrapingLocator._fetchc C sX t |�\}}}}}}|dkr:tjjt|��r:tt|�d�}|| jkr`| j| }tj d||� �n�|j
dd�d }d}|| j kr�tj d||� �n�t |d d
id �}�z�y�tj d |� | j
j|| jd
�} tj d|� | j� }
|
jdd�} tj| ��r�| j� } | j� }
|
jd�}|�r"| j| }||
�}
d}tj| �}|�r@|jd�}y|
j|�}
W n tk
�rn |
jd�}
Y nX t |
| �}|| j| < W n� t k
�r� } z |j dk�r�tj d||� W Y dd}~X n� t k
�r } z2tj d||� | j!� | j j"|� W dQ R X W Y dd}~X n2 t#k
�rB } ztj d||� W Y dd}~X nX W d|| 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 %s)r% z
Fetched %sz Content-Typer0 zContent-Encodingzutf-8zlatin-1i� zFetch failed: %s: %s)$r �osri �isdirr r r r� ro rp r� r� r rN r� r% r� rT �HTML_CONTENT_TYPEr� Zgeturlr� �decoders�CHARSETr� r� r� � UnicodeErrorr� r r9 r� r r� r� r� )r6 r( r2 rl ri r� rW r� r7 r� r; Z content_typeZ final_urlr� �encoding�decoderr� rX r* r* r+ r � sZ
&$ z SimpleScrapingLocator.get_pagez<a href=[^>]*>([^<]+)<c C sP t � }| j| j�}|s$td| j ��x&| jj|j�D ]}|j|jd�� q4W |S )zJ
Return all the distribution names known to this locator.
zUnable to get %sr ) r� r r� r � _distname_rer� r� r� r� )r6 rW r r� r* r* r+ rd $ s z,SimpleScrapingLocator.get_distribution_names)Nr� )r= r> r? r@ �zlibZ
decompressr rR r� r� rc r� r� r� r� r� r� r� r� r r rd r� r* r* )r� r+ r� ; s"
;
r� c s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS )
�DirectoryLocatorz?
This class locates distributions in a directory tree.
c sN |j dd�| _tt| �jf |� tjj|�}tjj|�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
rR r ri �abspathr r �base_dir)r6 ri r� )r� r* r+ rR 5 s
zDirectoryLocator.__init__c C s |j | 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 r� )r6 rv �parentr* r* r+ �should_includeF s z DirectoryLocator.should_includec C s� i i d�}x�t j| j�D ]v\}}}xb|D ]Z}| j||�r(t jj||�}tddtt jj|��dddf�}| j ||�}|r(| j
||� q(W | j sP qW |S )N)r� r� r r0 ) r �walkr r ri r| r r r r� r� r ) r6 rb rW �root�dirs�files�fnr( r� r* r* r+ rc N s
z DirectoryLocator._get_projectc C s� t � }x�tj| j�D ]x\}}}xd|D ]\}| j||�r$tjj||�}tddttjj |��dddf�}| j
|d�}|r$|j |d � q$W | j sP qW |S )zJ
Return all the distribution names known to this locator.
r r0 Nrb )
r� r r r r ri r| r r r r� r� r )r6 rW r r r r r( r� r* r* r+ rd ^ s
z'DirectoryLocator.get_distribution_names) r= r> r? r@ rR r rc rd r� r* r* )r� r+ r
0 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 )zJ
Return all the distribution names known to this locator.
z Not available from this locatorN)ra )r6 r* r* r+ rd x s z"JSONLocator.get_distribution_namesc C s� i i d�}t |�}|r�x�|jdg �D ]�}|d dks$|d dkrBq$t|d |d |jd d
�| jd �}|j}|d |_d
|kr�|d
r�d|d
f|_|jdi �|_|jdi �|_|||j < |d j
|j t � �j |d � q$W |S )N)r� r� r ZptypeZsdistZ pyversion�sourcerb r� r� zPlaceholder for summary)r� r2 r( r� r� Z requirements�exportsr� )
r rT r r2 r� r� r� Z dependenciesr r� r� r� r� )r6 rb rW r� r� r� r� r* r* r+ rc ~ s&
"zJSONLocator._get_projectN)r= r> r? r@ rd rc r* r* r* r+ r q 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 rR �
isinstancer �AssertionError�distpath)r6 r r� )r� r* r+ rR � s zDistPathLocator.__init__c C sP | j j|�}|d kr i i d�}n,|j|d|jt|jg�id|jtd g�ii}|S )N)r� r� r� r� )r Zget_distributionr� r� r� )r6 rb r� rW r* r* r+ rc � s z DistPathLocator._get_project)r= r> r? r@ rR rc 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* |j 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 rR )r6 r! r� )r� r* r+ rR � s zAggregatingLocator.__init__c s* t t| �j� x| jD ] }|j� qW d S )N)r� r r\ r! )r6 r� )r� r* r+ r\ � s z AggregatingLocator.clear_cachec C s || _ x| jD ]
}||_qW d S )N)r] r! r2 )r6 r_ r� r* r* r+ r` � s z AggregatingLocator._set_schemec C s� i }x�| j D ]�}|j|�}|r | jr�|jdi �}|jdi �}|j|� |jd�}|r�|r�x6|j� D ]*\}} ||kr�|| | O < qb| ||< qbW |jd�}
|r�|
r�|
j|� q | jd kr�d} n$d} x |D ]}| jj|�r�d} P q�W | r |}P q W |S )Nr� r� TF)r! re r rT �updater� rO r� ) r6 rb rW r� r� r r� Zdfr� r Zdd�foundr* r* r+ rc � s8
z AggregatingLocator._get_projectc C s@ t � }x4| jD ]*}y||j� O }W q tk
r6 Y qX qW |S )zJ
Return all the distribution names known to this locator.
)r� r! rd ra )r6 rW r� r* r* r+ rd � s
z)AggregatingLocator.get_distribution_names)r= r> r? r@ rR r\ r` r� rA r2 �fgetrc rd r� r* r* )r� r+ r � s ,r z https://pypi.python.org/simple/g @)r% �legacy)r2 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! r2 )r6 r� r* r* r+ rR s
zDependencyFinder.__init__c C sv t jd|� |j}|| j|< || j||jf< xD|jD ]:}t|�\}}t jd|||� | jj |t
� �j ||f� q4W dS )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) ro rp r< �
dists_by_name�distsr� �providesr �providedr� r� r� )r6 r� rb �pr� r* r* r+ �add_distribution& s
z!DependencyFinder.add_distributionc C s| t jd|� |j}| j|= | j||jf= xN|jD ]D}t|�\}}t jd|||� | j| }|j ||f� |s0| j|= q0W 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)
ro rp r< r( r) r� r* r r+ �remove)r6 r� rb r, r� �sr* r* r+ �remove_distribution5 s
z$DependencyFinder.remove_distributionc C sB y| j j|�}W n, tk
r< |j� d }| j 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 )r2 rO r" r� )r6 �reqtrO rb r* r* r+ � get_matcherG s z DependencyFinder.get_matcherc C sv | j |�}|j}t� }| j}||krrxL|| D ]@\}}y|j|�}W n tk
r\ d}Y nX |r.|j|� P q.W |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)r2 r< r� r+ r� r" r� ) r6 r1 rO rb rW r+ r� �providerr� r* r* r+ �find_providersW s
z DependencyFinder.find_providersc C s� | j | }t� }x,|D ]$}| j|�}|j|j�s|j|� qW |r^|jd||t|�f� d}nD| j|� | j |= x"|D ]}| j j|t� �j|� qvW | j |� 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� r2 r� r� r� � frozensetr0 r� r- ) r6 r3 �other�problemsZrlistZ unmatchedr/ rO rW r* r* r+ �try_to_replaceo s"
z DependencyFinder.try_to_replaceFc C s i | _ i | _i | _i | _t|p g �}d|krH|jd� |tdddg�O }t|t�rh| }}tj d|� n4| j
j ||d� }}|dkr�t d| ��tj d |� d
|_
t� }t|g�}t|g�}�x�|�r�|j� }|j} | | jkr�| j|� n"| j| }
|
|k�r | j||
|� |j|jB } |j} t� }
||k�rbx2dD ]*}d| }||k�r4|
t|d| �O }
�q4W | | B |
B }�x>|D �]4}| j|�}|�sNtj d|� | j
j ||d�}|dk�r�| �r�| j
j |d
d�}|dk�r�tj d|� |jd|f� n^|j|j }}||f| jk�r|j|� |j|� || k�rN||k�rN|j|� tj d|j� xZ|D ]R}|j} | | jk�r�| jj|t� �j|� n"| j| }
|
|k�rT| j||
|� �qTW �qvW q�W t| jj� �}x.|D ]&}||k|_ |j �r�tj d|j� �q�W tj 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�test�build�devz:%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)r: r; r< ) r+ r) r( r5 r� r. r r ro rp r� r� r Z requestedr� r< r- r9 Z run_requiresZ
meta_requiresZbuild_requires�getattrr4 r� r� Zname_and_versionr� �valuesZbuild_time_dependency)r6 r� Z meta_extrasr� r� r� r8 ZtodoZ
install_distsrb r7 ZireqtsZsreqtsZereqtsr< rX Z all_reqtsr� Z providersr3 �nr r, r) r* r* r+ �find� s�
zDependencyFinder.find)N)NF) r= r> r? r@ rR r- r0 r2 r4 r9 r@ r* r* r* r+ r& s
(r& )N)Nr� �ior r� Zloggingr rg r� r� � ImportErrorZdummy_threadingr r0 r �compatr r r r r r
r r r
r r r4 r r r r Zdatabaser r r r� r �utilr r r r r r r r r r� r! r" r� r# r$ Z getLoggerr= ro r� r� r� r r r& r, r- �objectrA r� r� r� r� r
r r r r' r� ZNAME_VERSION_REr& r* r* r* r+ �<module> sZ D ,
;0E: vA& [