Edit File: locators.cpython-36.opt-1.pyc
3 )�]bE� � @ 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 |� || _dS )zs Initialise an instance. :param distpath: A :class:`DistributionPath` instance to search. N)r� r rR �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)r0 r< r� r) r� r" r� ) r6 r/ 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� r0 r� r� r� � frozensetr. r� r+ ) r6 r1 �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)r8 r9 r: ) r) r' r& r3 r� r, � isinstancer ro rp r� r� r Z requestedr� r< r+ r7 Z run_requiresZ meta_requiresZbuild_requires�getattrr2 r� r� Zname_and_versionr� �valuesZbuild_time_dependency)r6 r� Z meta_extrasr� r� r� r6 ZtodoZ install_distsrb r5 ZireqtsZsreqtsZereqtsr< rX Z all_reqtsr� Z providersr1 �nr r* r' r* r* r+ �find� s� zDependencyFinder.find)N)NF) r= r> r? r@ rR r+ r. r0 r2 r7 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& [