U
+ �a` � @ s� d Z ddlmZ ddlmZ ddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl m
Z
ddlmZ ddlmZ ddlmZ e �d e j�jZd
Zd d � ZG d
d� d�ZdS )zWheels support.� )� get_platform)�logN)�
parse_version)�sys_tags)�canonicalize_name)�write_requirementsz�^(?P<project_name>.+?)-(?P<version>\d.*?)
((-(?P<build>\d.*?))?-(?P<py_version>.+?)-(?P<abi>.+?)-(?P<platform>.+?)
)\.whl$z8__import__('pkg_resources').declare_namespace(__name__)
c C s� t �| �D ]�\}}}t j�|| �}|D ].}t j�||�}t j�|||�}t �||� q&ttt|���D ]D\} }
t j�||
�}t j�|||
�}t j� |�sft �||� || = qfq
t j| dd�D ] \}}}|r�t
�t � |� q�dS )zDMove everything under `src_dir` to `dst_dir`, and delete the former.T)�topdownN) �os�walk�path�relpath�join�renames�reversed�list� enumerate�exists�AssertionError�rmdir) �src_dirZdst_dir�dirpath�dirnames� filenames�subdir�f�src�dst�n�d� r �8/tmp/pip-install-1bd_1mtk/setuptools/setuptools/wheel.py�unpack s
r! c @ sh e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
dd� �Z e
dd� �Z e
dd� �Z
dS )�Wheelc C sP t tj�|��}|d kr$td| ��|| _|�� �� D ]\}}t| ||� q6d S )Nzinvalid wheel name: %r) �
WHEEL_NAMEr r �basename�
ValueError�filename� groupdict�items�setattr)�selfr& �match�k�vr r r �__init__6 s zWheel.__init__c C s& t �| j�d�| j�d�| j�d��S )z>List tags (py_version, abi, platform) supported by this wheel.�.)� itertools�product�
py_version�split�abi�platform�r* r r r �tags> s
�z
Wheel.tagsc s0 t dd� t� D ��� t� fdd�| �� D �d�S )z5Is the wheel is compatible with the current platform?c s s | ]}|j |j|jfV qd S �N)Z interpreterr4 r5 ��.0�tr r r � <genexpr>H s z&Wheel.is_compatible.<locals>.<genexpr>c 3 s | ]}|� krd V qdS )TNr r9 ��supported_tagsr r r< J s F)�setr �nextr7 r6 r r= r �
is_compatibleF s �zWheel.is_compatiblec C s, t j| j| j| jdkrd nt� d��� d S )N�any)� project_name�versionr5 z.egg)�
pkg_resources� DistributionrC rD r5 r �egg_namer6 r r r rG L s ��zWheel.egg_namec C sJ |� � D ]4}t�|�}|�d�rt|��t| j��r| S qtd��d S )Nz
.dist-infoz.unsupported wheel format. .dist-info not found)�namelist� posixpath�dirname�endswithr �
startswithrC r% )r* �zf�memberrJ r r r �
get_dist_infoR s
��
zWheel.get_dist_infoc C s( t �| j��}| �||� W 5 Q R X dS )z"Install wheel as an egg directory.N)�zipfile�ZipFiler& �_install_as_egg)r* �destination_eggdirrM r r r �install_as_egg\ s zWheel.install_as_eggc C s\ d| j | jf }| �|�}d| }tj�|d�}| �||||� | �||� | �||� d S )Nz%s-%sz%s.data�EGG-INFO) rC rD rO r r r
�_convert_metadata�_move_data_entries�_fix_namespace_packages)r* rS rM Z
dist_basename� dist_info� dist_data�egg_infor r r rR a s
zWheel._install_as_eggc sL ��fdd�}|d�}t |�d��}t d�| ko>t d�k n }|sTtd| ��t�|� ��|� tj�|���tj j
|�t� |��d�� d d
� �t t
t�� �� ����� ��fd d �� jD �}t��|� t�tj�|d
�tj�|d�� tj t�|d�d�} tjj}
t�tj� z t| �d�d tj�|d�� W 5 t�|
� X d S )Nc
sH �� t�� | ���,}|�� �d�}tj�� �|�W 5 Q R � S Q R X d S )Nzutf-8) �openrI r
�read�decode�email�parser�Parser�parsestr)�name�fp�value)rY rM r r � get_metadatam s z-Wheel._convert_metadata.<locals>.get_metadata�WHEELz
Wheel-Versionz1.0z2.0dev0z$unsupported wheel format version: %s)�metadatac S s d | _ t| �S r8 )�marker�str)�reqr r r �raw_req� s z(Wheel._convert_metadata.<locals>.raw_reqc
s2 i | ]*}|t �fd d�t�� �|f��D ���qS )c 3 s | ]}|� kr|V qd S r8 r )r: rk )�install_requiresr r r< � s �z5Wheel._convert_metadata.<locals>.<dictcomp>.<genexpr>)�sorted�map�requires)r: �extra)�distrm rl r r �
<dictcomp>� s
� �z+Wheel._convert_metadata.<locals>.<dictcomp>�METADATAzPKG-INFO)rm �extras_require)�attrsr[ z requires.txt)r �getr% r �mkdir�
extractallr r
rE rF �
from_location� PathMetadatar rn ro rp �extras�rename�
setuptools�dictr � _global_log� threshold�
set_threshold�WARNr �get_command_obj) rM rS rY r[ rf Zwheel_metadata�
wheel_versionZwheel_v1ru Z
setup_distZ
log_thresholdr )rr rY rm rl rM r rV k sT ��
�� ��� �zWheel._convert_metadatac s� t j�| � �� t j�� d�}t j�|�r�t j�| dd�}t �|� t �|�D ]D}|�d�rpt �t j�||�� qLt �t j�||�t j�||�� qLt � |� t
t jj� fdd�dD ��D ]}t || � q�t j�� �r�t � � � dS )z,Move data entries to their correct location.�scriptsrU z.pycc 3 s | ]}t j�� |�V qd S r8 )r r r
)r: r �rZ r r r< � s �z+Wheel._move_data_entries.<locals>.<genexpr>)�data�headers�purelib�platlibN) r r r
r rx �listdirrK �unlinkr} r �filterr! )rS rZ Zdist_data_scriptsZegg_info_scripts�entryr r r� r rW � s. �
�
� zWheel._move_data_entriesc
C s� t j�| d�}t j�|�r�t|��}|�� �� }W 5 Q R X |D ]l}t jj|f|�d��� }t j�|d�}t j�|�s~t �|� t j�|�s>t|d��}|�t � W 5 Q R X q>d S )Nznamespace_packages.txtr/ z __init__.py�w)
r r r
r r\ r] r3 rx �write�NAMESPACE_PACKAGE_INIT)r[ rS Znamespace_packagesrd �modZmod_dirZmod_initr r r rX � s �
z Wheel._fix_namespace_packagesN)�__name__�
__module__� __qualname__r. r7 rA rG rO rT rR � staticmethodrV rW rX r r r r r" 4 s
?
r" )�__doc__�distutils.utilr � distutilsr r_ r0 r rI �rerP rE r~ r Z setuptools.extern.packaging.tagsr Z!setuptools.extern.packaging.utilsr Zsetuptools.command.egg_infor �compile�VERBOSEr+ r# r� r! r" r r r r �<module> s, ��