B
    ØÓY\ž  ã               @   s¾   d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddl	mZ ddlmZ ddlmZ ddl
mZ ddlmZ eZe d	ej¡jZd
Zdd„ ZG dd„ dƒZdS )zWheels support.é    )Úget_platformN)Úparse_version)Úcanonicalize_name)ÚPY3)Ú
pep425tags)Úwrite_requirementsz‚^(?P<project_name>.+?)-(?P<version>\d.*?)
    ((-(?P<build>\d.*?))?-(?P<py_version>.+?)-(?P<abi>.+?)-(?P<platform>.+?)
    )\.whl$zštry:
    __import__('pkg_resources').declare_namespace(__name__)
except ImportError:
    __path__ = __import__('pkgutil').extend_path(__path__, __name__)
c             C   sî   x¸t  | ¡D ]ª\}}}t j || ¡}x6|D ].}t j ||¡}t j |||¡}t  ||¡ q*W xXttt|ƒƒƒD ]D\}	}
t j ||
¡}t j |||
¡}t j 	|¡snt  ||¡ ||	= qnW qW x.t j| ddD ]\}}}|rÜt
‚t  |¡ qÊW dS )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)Zsrc_dirZdst_dirÚdirpathZdirnamesÚ	filenamesÚsubdirÚfÚsrcÚdstÚnÚd© r   ú/lib/python3.7/site-packages/setuptools/wheel.pyÚunpack%   s    
r   c               @   sh   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	e
dd„ ƒZe
dd„ ƒZe
dd„ ƒZdS )ÚWheelc             C   sT   t tj |¡ƒ}|d kr$td| ƒ‚|| _x$| ¡  ¡ D ]\}}t| ||ƒ q8W d S )Nzinvalid wheel name: %r)	Ú
WHEEL_NAMEr	   r   ÚbasenameÚ
ValueErrorÚfilenameÚ	groupdictÚitemsÚsetattr)Úselfr$   ÚmatchÚkÚvr   r   r   Ú__init__=   s    zWheel.__init__c             C   s&   t  | j d¡| j d¡| j d¡¡S )z>List tags (py_version, abi, platform) supported by this wheel.Ú.)Ú	itertoolsÚproductZ
py_versionÚsplitZabiÚplatform)r(   r   r   r   ÚtagsE   s    

z
Wheel.tagsc                s$   t  ¡ ‰ t‡ fdd„|  ¡ D ƒdƒS )z5Is the wheel is compatible with the current platform?c             3   s   | ]}|ˆ krd V  qdS )TNr   )Ú.0Út)Úsupported_tagsr   r   ú	<genexpr>P   s    z&Wheel.is_compatible.<locals>.<genexpr>F)r   Zget_supportedÚnextr2   )r(   r   )r5   r   Úis_compatibleM   s    zWheel.is_compatiblec             C   s,   t j| j| j| jdkrd ntƒ d ¡ d S )NÚany)Úproject_nameÚversionr1   z.egg)Úpkg_resourcesÚDistributionr:   r;   r1   r   Úegg_name)r(   r   r   r   r>   R   s    zWheel.egg_namec             C   sJ   x<|  ¡ D ]0}t |¡}| d¡r
t|ƒ t| jƒ¡r
|S q
W tdƒ‚d S )Nz
.dist-infoz.unsupported wheel format. .dist-info not found)ZnamelistÚ	posixpathÚdirnameÚendswithr   Ú
startswithr:   r#   )r(   ÚzfÚmemberr@   r   r   r   Úget_dist_infoX   s    

zWheel.get_dist_infoc          	   C   s(   t  | j¡}|  ||¡ W dQ R X dS )z"Install wheel as an egg directory.N)ÚzipfileZZipFiler$   Ú_install_as_egg)r(   Údestination_eggdirrC   r   r   r   Úinstall_as_eggb   s    zWheel.install_as_eggc             C   s\   d| j | jf }|  |¡}d| }tj |d¡}|  ||||¡ |  ||¡ |  ||¡ d S )Nz%s-%sz%s.datazEGG-INFO)	r:   r;   rE   r	   r   r   Ú_convert_metadataÚ_move_data_entriesÚ_fix_namespace_packages)r(   rH   rC   Zdist_basenameÚ	dist_infoÚ	dist_dataÚegg_infor   r   r   rG   g   s    
zWheel._install_as_eggc       
         s&  ‡‡fdd„}|dƒ}t | d¡ƒ}t dƒ|  ko>t dƒk n  }|sTtd| ƒ‚t |¡ ˆ |¡ tj |ˆ¡‰tj	j
|ˆt |ˆ¡d‰ d	d
„ ‰tttˆˆ  ¡ ƒƒƒ‰‡ ‡‡fdd„ˆ jD ƒ}t ˆ|¡ t tj |d¡tj |d¡¡ tj	tˆ|dd}	t|	 d¡d tj |d¡ƒ d S )Nc          	      sH   ˆ  t ˆ | ¡¡,}tr&| ¡  d¡n| ¡ }tj ¡  	|¡S Q R X d S )Nzutf-8)
Úopenr?   r   r   ÚreadÚdecodeÚemailÚparserZParserZparsestr)ÚnameÚfpÚvalue)rM   rC   r   r   Úget_metadatas   s    z-Wheel._convert_metadata.<locals>.get_metadataZWHEELzWheel-Versionz1.0z2.0dev0z$unsupported wheel format version: %s)Zmetadatac             S   s   d | _ t| ƒS )N)ZmarkerÚstr)Úreqr   r   r   Úraw_reqŽ   s    z(Wheel._convert_metadata.<locals>.raw_reqc          	      s2   i | ]*}t ‡fd d„tˆˆ  |f¡ƒD ƒƒ|“qS )c             3   s   | ]}|ˆ kr|V  qd S )Nr   )r3   rZ   )Úinstall_requiresr   r   r6   ”   s   z5Wheel._convert_metadata.<locals>.<dictcomp>.<genexpr>)ÚsortedÚmapÚrequires)r3   Zextra)Údistr\   r[   r   r   ú
<dictcomp>’   s   z+Wheel._convert_metadata.<locals>.<dictcomp>ZMETADATAzPKG-INFO)r\   Úextras_require)ZattrsrO   zrequires.txt)r   Úgetr#   r	   ÚmkdirZ
extractallr   r   r<   r=   Zfrom_locationZPathMetadatar   r]   r^   r_   ZextrasÚrenameÚ
setuptoolsÚdictr   Zget_command_obj)
rC   rH   rM   rO   rX   Zwheel_metadataZwheel_versionZwheel_v1rb   Z
setup_distr   )r`   rM   r\   r[   rC   r   rJ   q   s:     



zWheel._convert_metadatac                sê   t j | ˆ ¡‰ t j ˆ d¡}t j |¡r t j | dd¡}t  |¡ xRt  |¡D ]D}| d¡rrt  t j ||¡¡ qNt  t j ||¡t j ||¡¡ qNW t  	|¡ x.t
t jj‡ fdd„dD ƒƒD ]}t|| ƒ q¾W t j ˆ ¡ræt  	ˆ ¡ dS )z,Move data entries to their correct location.ZscriptszEGG-INFOz.pycc             3   s   | ]}t j ˆ |¡V  qd S )N)r	   r   r   )r3   r   )rN   r   r   r6   À   s   z+Wheel._move_data_entries.<locals>.<genexpr>)ÚdataZheadersZpurelibZplatlibN)r	   r   r   r   rd   ÚlistdirrA   Úunlinkre   r   Úfilterr   )rH   rN   Zdist_data_scriptsZegg_info_scriptsÚentryr   r   )rN   r   rK   «   s&    





zWheel._move_data_entriesc          
   C   sª   t j | d¡}t j |¡r¦t|ƒ}| ¡  ¡ }W d Q R X xj|D ]b}t jj|f| d¡žŽ }t j |d¡}t j |¡r@t j |¡s@t|dƒ}| t¡ W d Q R X q@W d S )Nznamespace_packages.txtr-   z__init__.pyÚw)	r	   r   r   r   rP   rQ   r0   ÚwriteÚNAMESPACE_PACKAGE_INIT)rO   rH   Znamespace_packagesrV   ÚmodZmod_dirZmod_initr   r   r   rL   Ç   s    

zWheel._fix_namespace_packagesN)Ú__name__Ú
__module__Ú__qualname__r,   r2   r8   r>   rE   rI   rG   ÚstaticmethodrJ   rK   rL   r   r   r   r   r    ;   s   

:r    )Ú__doc__Zdistutils.utilr   rS   r.   r	   r?   ÚrerF   r<   rf   r   Z!setuptools.extern.packaging.utilsr   Zsetuptools.extern.sixr   r   Zsetuptools.command.egg_infor   ÚtypeZ__metaclass__ÚcompileÚVERBOSEr)   r!   ro   r   r    r   r   r   r   Ú<module>   s*   
