o
    0ήc                      @   s   d 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
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d	ejjZd
Zdd Zejdd ZG dd dZdS )zWheels support.    N)get_platform)parse_version)sys_tags)canonicalize_name)write_requirements)_unpack_zipfile_objz^(?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 ]Q\}}}t j|| }|D ]}t j||}t j|||}t || qttt|D ]"\}	}
t j||
}t j|||
}t j	|sUt || ||	= q3qt j| ddD ]\}}}|rgJ t 
| q^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rmdir)src_dirdst_dirdirpathdirnames	filenamessubdirfsrcdstnd r   7/tmp/pip-target-vg8gfxp4/lib/python/setuptools/wheel.pyunpack    s$   	r!   c               	   c   s>    ddl m}  | | j}zdV  W | | dS | | w )z*
    Temporarily disable info traces.
    r   )logN)	distutilsr"   set_thresholdWARN)r"   savedr   r   r    disable_info_traces6   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   sP   t tj|}|d u rtd| || _|  D ]
\}}t| || qd S )Nzinvalid wheel name: %r)	
WHEEL_NAMEr	   r   basename
ValueErrorfilename	groupdictitemssetattr)selfr,   matchkvr   r   r    __init__E   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product
py_versionsplitabiplatformr0   r   r   r    tagsM   s
   


z
Wheel.tagsc                    s0   t dd t D  t fdd|  D dS )z5Is the wheel is compatible with the current platform?c                 s   s     | ]}|j |j|jfV  qd S N)interpreterr:   r;   .0tr   r   r    	<genexpr>W   s    
z&Wheel.is_compatible.<locals>.<genexpr>c                 3   s    | ]	}| v rd V  qdS )TNr   r@   supported_tagsr   r    rC   Y   s    F)setr   nextr=   r<   r   rD   r    is_compatibleU   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versionr;   z.egg)pkg_resourcesDistributionrJ   rK   r;   r   egg_namer<   r   r   r    rN   [   s   zWheel.egg_namec                 C   sF   |  D ]}t|}|drt|t| jr|  S qtd)Nz
.dist-infoz.unsupported wheel format. .dist-info not found)namelist	posixpathdirnameendswithr   
startswithrJ   r+   )r0   zfmemberrQ   r   r   r    get_dist_infoa   s   

zWheel.get_dist_infoc                 C   s>   t | j}| || W d   dS 1 sw   Y  dS )z"Install wheel as an egg directory.N)zipfileZipFiler,   _install_as_egg)r0   destination_eggdirrT   r   r   r    install_as_eggk   s   "zWheel.install_as_eggc                 C   s\   d| j | jf }| |}d| }tj|d}| |||| | || | || d S )Nz%s-%sz%s.dataEGG-INFO)	rJ   rK   rV   r	   r   r   _convert_metadata_move_data_entries_fix_namespace_packages)r0   rZ   rT   dist_basename	dist_info	dist_dataegg_infor   r   r    rY   p   s   
zWheel._install_as_eggc           
         s@  fdd}|d}t |d}t d|  kot dk n  }|s*td| t| tj|tjj	|t
|d d	d
 tt   fdd jD }t| ttj|dtj|d tjt|dd}	t  t|	dd tj|d W d    d S 1 sw   Y  d S )Nc                    sT    t | }| d}tj |W  d    S 1 s#w   Y  d S )Nzutf-8)	openrP   r   readdecodeemailparserParserparsestr)namefpvalue)ra   rT   r   r    get_metadata|   s   $z-Wheel._convert_metadata.<locals>.get_metadataWHEELzWheel-Versionz1.0z2.0dev0z$unsupported wheel format version: %s)metadatac                 S   s   d | _ t| S r>   )markerstr)reqr   r   r    raw_req   s   z(Wheel._convert_metadata.<locals>.raw_reqc              	      s.   i | ]}|fd dt  |fD qS )c                    s   g | ]}| vr|qS r   r   )rA   rs   )install_requiresr   r    
<listcomp>   s    z6Wheel._convert_metadata.<locals>.<dictcomp>.<listcomp>)maprequires)rA   extra)distru   rt   r   r    
<dictcomp>   s    z+Wheel._convert_metadata.<locals>.<dictcomp>METADATAzPKG-INFO)ru   extras_require)attrsrc   zrequires.txt)r   getr+   r   r	   r   r   rL   rM   from_locationPathMetadatar   rw   rx   extrasrename
setuptoolsdictr'   r   get_command_obj)
rT   rZ   ra   rc   rn   wheel_metadatawheel_versionwheel_v1r}   
setup_distr   )rz   ra   ru   rt   rT   r    r]   z   sL   

"zWheel._convert_metadatac                    s   t j|   t j d}t j|rNt j| dd}t | t |D ]"}|dr8t t j|| q&t t j||t j|| q&t 	| t
t jj fdddD D ]}t||  q\t j rqt 	  dS dS )z,Move data entries to their correct location.scriptsr\   z.pycc                 3   s    | ]
}t j |V  qd S r>   )r	   r   r   )rA   r   rb   r   r    rC      s
    
z+Wheel._move_data_entries.<locals>.<genexpr>)dataheaderspurelibplatlibN)r	   r   r   r   mkdirlistdirrR   unlinkr   r   filterr!   )rZ   rb   dist_data_scriptsegg_info_scriptsentryr   r   r   r    r^      s,   


zWheel._move_data_entriesc              	   C   s   t j| d}t j|rkt|}|  }W d    n1 s"w   Y  |D ]C}t jj|g|dR  }t j|d}t j|sJt | t j|sjt|d}|t	 W d    n1 sew   Y  q)d S d S )Nznamespace_packages.txtr5   z__init__.pyw)
r	   r   r   r   rd   re   r9   r   writeNAMESPACE_PACKAGE_INIT)rc   rZ   namespace_packagesrl   modmod_dirmod_initr   r   r    r_      s&   

zWheel._fix_namespace_packagesN)__name__
__module____qualname__r4   r=   rH   rN   rV   r[   rY   staticmethodr]   r^   r_   r   r   r   r    r(   C   s    


9
r(   )__doc__rg   r6   r	   rP   rerW   
contextlibdistutils.utilr   rL   r   r    setuptools.extern.packaging.tagsr   !setuptools.extern.packaging.utilsr   setuptools.command.egg_infor   setuptools.archive_utilr   compileVERBOSEr1   r)   r   r!   contextmanagerr'   r(   r   r   r   r    <module>   s8    
