o
    d!                     @   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Zddl	m
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mZ ddlmZ ed	ejjZd
Zejdddd Zdd Zej dd Z!G dd dZ"dS )zWheels support.    N)get_platform)Versionsys_tags)canonicalize_name)write_requirements_egg_basename)_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__)
)maxsizec                   C   s   dd t  D S )Nc                 S   s   h | ]
}|j |j|jfqS  )interpreterabiplatform.0tr   r   D/app/.heroku/python/lib/python3.10/site-packages/setuptools/wheel.py	<setcomp>%   s    z&_get_supported_tags.<locals>.<setcomp>r   r   r   r   r   _get_supported_tags    s   r   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_dirZdst_dirdirpathdirnames	filenamessubdirfsrcdstndr   r   r   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_traces>   s   r1   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)selfr6   matchkvr   r   r   __init__M   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splitr   r   r:   r   r   r   tagsU   s
   


z
Wheel.tagsc                 C   s   t dd |  D dS )z2Is the wheel compatible with the current platform?c                 s   s    | ]
}|t  v rd V  qdS )TN)r   r   r   r   r   	<genexpr>_   s    z&Wheel.is_compatible.<locals>.<genexpr>F)nextrE   rD   r   r   r   is_compatible]   s   zWheel.is_compatiblec                 C   s&   t | j| j| jdkrd nt dd S )Nany)r   z.egg)r   project_nameversionr   r   rD   r   r   r   egg_namea   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   r5   )r:   zfmemberrO   r   r   r   get_dist_infoh   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ZipFiler6   _install_as_egg)r:   destination_eggdirrR   r   r   r   install_as_eggr   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   rT   r   r   r   _convert_metadata_move_data_entries_fix_namespace_packages)r:   rX   rR   Zdist_basename	dist_info	dist_dataegg_infor   r   r   rW   w   s   
zWheel._install_as_eggc                    sH  dd l }fdd}|d}t|d}td|  ko#tdk n  }|s.td| t| tj||jj	||
|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 )Nr   c                    sT    t | }| d}tj |W  d    S 1 s#w   Y  d S )Nzutf-8)	openrN   r   readdecodeemailparserParserparsestr)namefpvalue)r^   rR   r   r   get_metadata   s   $z-Wheel._convert_metadata.<locals>.get_metadataZWHEELzWheel-Versionz1.0z2.0dev0z$unsupported wheel format version: %s)metadatac                 S   s   d | _ t| S N)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   )r   rp   )install_requiresr   r   
<listcomp>   s    z6Wheel._convert_metadata.<locals>.<dictcomp>.<listcomp>)maprequires)r   extra)distrr   rq   r   r   
<dictcomp>   s    z+Wheel._convert_metadata.<locals>.<dictcomp>METADATAzPKG-INFO)rr   extras_require)attrsr`   zrequires.txt)pkg_resourcesparse_versiongetr5   r	   r   r   r   Distributionfrom_locationPathMetadatar   rt   ru   extrasrename
setuptoolsdictr1   r   get_command_obj)rR   rX   r^   r`   r|   rk   wheel_metadatawheel_versionZwheel_v1rz   Z
setup_distr   )rw   r^   rr   rq   rR   r   r[      sN   

"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.scriptsrZ   z.pycc                 3   s    | ]
}t j |V  qd S rm   )r   r   r   )r   r*   r_   r   r   rF      s
    
z+Wheel._move_data_entries.<locals>.<genexpr>)dataheaderspurelibplatlibN)r   r   r   r   mkdirlistdirrP   unlinkr   r    filterr+   )rX   r_   Zdist_data_scriptsZ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.txtr?   z__init__.pyw)
r   r   r   r   ra   rb   rC   r   writeNAMESPACE_PACKAGE_INIT)r`   rX   namespace_packagesri   modZmod_dirZmod_initr   r   r   r]      s&   

zWheel._fix_namespace_packagesN)__name__
__module____qualname__r>   rE   rH   rL   rT   rY   rW   staticmethodr[   r\   r]   r   r   r   r   r2   K   s    


;
r2   )#__doc__rd   r@   	functoolsr   rN   rerU   
contextlibdistutils.utilr   r   Z#setuptools.extern.packaging.versionr   r}   Z setuptools.extern.packaging.tagsr   Z!setuptools.extern.packaging.utilsr   Zsetuptools.command.egg_infor   r   Zsetuptools.archive_utilr	   compileVERBOSEr;   r3   r   	lru_cacher   r+   contextmanagerr1   r2   r   r   r   r   <module>   s<    


