XylotrechusZ
3
�Pf� � @ s� d dl mZ d dlZd dlZd dlZd dlmZmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ d d
lm Z ej e �Z G dd� de�Z dd� Z!ddd�Z"dd� Z#dS )� )�absolute_importN)�Command�SUCCESS)� OrderedDict)�PipXmlrpcTransport)�PyPI)�get_terminal_size)�
indent_log)� CommandError)�NO_MATCHES_FOUND)�parse)�
pkg_resources)�
xmlrpc_clientc s< e Zd ZdZdZdZdZ� fdd�Zdd� Zd d
� Z � Z
S ) �
SearchCommandz@Search for PyPI packages whose name or summary contains <query>.�searchz
%prog [options] <query>zSearch PyPI for packages.c s@ t t| �j||� | jjddddtjdd� | jjd| j� d S )Nz-iz--index�indexZURLz3Base URL of Python Package Index (default %default))�dest�metavar�default�helpr ) �superr �__init__Zcmd_optsZ
add_optionr Zpypi_url�parserZinsert_option_group)�self�args�kw)� __class__� � /usr/lib/python3.6/search.pyr s zSearchCommand.__init__c C sT |s t d��|}| j||�}t|�}d }tjj� r<t� d }t||d� |rPtS t S )Nz)Missing required argument (search query).r )�terminal_width)
r
r �transform_hits�sys�stdout�isattyr �
print_resultsr r )r �optionsr �queryZ pypi_hits�hitsr r r r �run) s
zSearchCommand.runc C sH |j }| j|��.}t||�}tj||�}|j||d�d�}|S Q R X d S )N)�name�summary�or)r Z_build_sessionr r Z ServerProxyr )r r& r% Z index_urlZsessionZ transportZpypir' r r r r 9 s
zSearchCommand.search) �__name__�
__module__� __qualname__�__doc__r) Zusager* r r( r �
__classcell__r r )r r r s r c C s� t � }xv| D ]n}|d }|d }|d }||j� krH|||gd�||< q || d j|� |t|| d �kr ||| d< q W t|j� �S )z�
The list from pypi is really a list of versions. We want a list of
packages with the list of versions stored inline. This converts the
list from pypi into one we can use.
r) r* �version)r) r* �versionsr2 )r �keys�append�highest_version�list�values)r' Zpackages�hitr) r* r1 r r r r B s
r c C sT | sd S |d kr&t dd� | D ��d }dd� tjD �}�x| D �] }|d }|d pVd}|jdd g�d }|d k r�|| d }|d kr�tj||�}d
d|d j|�}d|d||f |f } yvtj| � ||k�r2tj |�}
t
� �H t |d �} |
j | k�rtjd|
j � ntjd|
j � tjd| � W d Q R X W q> t
k
�rJ Y q>X q>W d S )Nc S s. g | ]&}t |d �t |jddg�d � �qS )r) r2 �-� ���)�len�get)�.0r8 r r r �
<listcomp>c s z!print_results.<locals>.<listcomp>� c S s g | ]
}|j �qS r )Z project_name)r> �pr r r r? g s r) r* � r2 r9 r: � �
�
� � z %-*s - %sz%s (%s)zINSTALLED: %s (latest)z
INSTALLED: %sz
LATEST: %sr; )�maxr
Z working_setr= �textwrapZwrap�join�logger�infoZget_distributionr r5 r1 �UnicodeEncodeError) r' Zname_column_widthr Zinstalled_packagesr8 r) r* r1 Z target_width�lineZdistZlatestr r r r$ ^ s>
r$ c C s t | td�S )N)�key)rH �
parse_version)r2 r r r r5 � s r5 )NN)$Z
__future__r Zloggingr! rI Zpip.basecommandr r Z
pip.compatr Z pip.downloadr Z
pip.modelsr Z pip.utilsr Zpip.utils.loggingr Zpip.exceptionsr
Zpip.status_codesr Z pip._vendor.packaging.versionr rP Z pip._vendorr
Zpip._vendor.six.movesr Z getLoggerr, rK r r r$ r5 r r r r �<module> s&
+
&