U
    cc                     @   s   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
mZ d dlmZ d dlmZmZmZ d dlmZ d dlmZmZmZmZmZmZ ejdkrd d	lmZ d
d Zndd Ze dej!Z"dddZ#G dd deZ$dS )    )print_functionN)OrderedDict)log)ZIP_DEFLATEDZipInfoZipFile)
WheelError)urlsafe_b64decode
as_unicodenativeurlsafe_b64encodeas_bytesStringIO)   )TextIOWrapperc                 C   s   t t| dddS )N utf-8)newlineencoding)csvreaderr   )fp r   I/var/www/html/project/venv/lib/python3.8/site-packages/wheel/wheelfile.pyread_csv   s    r   c                 c   s$   t | D ]}dd |D V  q
d S )Nc                 S   s   g | ]}| d qS )r   )decode).0columnr   r   r   
<listcomp>   s     zread_csv.<locals>.<listcomp>)r   r   )r   liner   r   r   r      s    zt^(?P<namever>(?P<name>.+?)-(?P<ver>.+?))(-(?P<build>\d[^-]*))?
     -(?P<pyver>.+?)-(?P<abi>.+?)-(?P<plat>.+?)\.whl$c                 C   s,   t tjd| pt } t| dd S )NZSOURCE_DATE_EPOCHr      )intosenvirongettimegmtime)	timestampr   r   r   get_zipinfo_datetime#   s    r(   c                   @   sR   e Zd ZdZejZdefddZdddZ	dd	 Z
dd
dZdddZdd ZdS )	WheelFilezA ZipFile derivative class that also reads SHA-256 hashes from
    .dist-info/RECORD and checks any read files against those.
    rc              
   C   s  t j|}t|| _|dr,| jd kr:td|t	j
| |||dd d| jd| _| jd | _t | _i | _|dkrd	| j| j< d	| j| jd
 < d	| j| jd < z| | j}W n$ tk
r   td| jY nX | t|D ]}|\}}}	|sq|d\}
}zt|
 W n$ tk
rL   td|
Y nX |
 dkrjtd|
|
t|df| j|< qW 5 Q R X d S )Nz.whlzBad wheel filename {!r}T)compression
allowZip64z{}.dist-infoZnameverz/RECORDr*   )NNz.jwsz.p7szMissing {} file=zUnsupported hash algorithm: {}>   sha1md5z4Weak hash algorithm ({}) is not permitted by PEP 427ascii)r"   pathbasenameWHEEL_INFO_REmatchZparsed_filenameendswithr   formatr   __init__groupZdist_info_pathrecord_pathr   _file_hashes_file_sizesopenKeyErrorr   splithashlibnew
ValueErrorlowerr	   encode)selffilemoder+   r2   recordr   r1   Zhash_sumsize	algorithmr   r   r   r7   1   sH    

 zWheelFile.__init__Nc                    s   d fdd	}t t|tr(|jn||dkr\ds\| jkr\tdtt	
| ||| |dkrds| j \}d k	rt| j|  _ S )Nc                    sR   |d kr j }|  n
| | |  |rN krNtdtd S )NzHash mismatch for file '{}')_eofupdatedigestr   r6   r   )newdataeofZefZef_nameZexpected_hashZrunning_hashZupdate_crc_origr   r   _update_crc^   s    


z#WheelFile.open.<locals>._update_crcr*   /zNo hash found for file '{}')N)r
   
isinstancer   filenamer5   r:   r   r6   r   r   r<   r?   r@   rP   )rD   Zname_or_inforF   pwdrP   rI   r   rO   r   r<   ]   s    
zWheelFile.openc           	      C   s   t d| j| g }t|D ]\}}}|  t|D ]r}tjtj	||}tj
|r8tj||tjjd}|| jkrq8|dr|||f q8| || q8q|  |D ]\}}| || qd S )Nz#creating '%s' and adding '%s' to itrQ   z
.dist-info)loggerinforS   r"   walksortsortedr1   normpathjoinisfilerelpathreplacesepr9   r5   appendwrite)	rD   base_dirZdeferredrootdirnames	filenamesnamer1   arcnamer   r   r   write_filesx   s     

zWheelFile.write_filesc              	   C   s~   t |d}t| }| }W 5 Q R X t|p4|t|jd}t	|j
t|j
B d> |_|ph| j|_| ||| d S )Nrb	date_time   )r<   r"   fstatfilenoreadr   r(   st_mtimestatS_IMODEst_modeS_IFMTexternal_attrr+   compress_typewritestr)rD   rS   rg   rv   fstdatazinfor   r   r   ra      s    zWheelFile.writec                 C   sr   t | ||| t|tr |jn|}td| || jkrn| |}|j	t
t| f| j|< t|| j|< d S )Nzadding '%s')r   rw   rR   r   rS   rU   rV   r9   _default_algorithmrf   r   r   rL   r:   lenr;   )rD   zinfo_or_arcnamebytesrv   fnamehash_r   r   r   rw      s    

zWheelFile.writestrc                    s    j d k	r jdkr jrt }tj|dddd}| fdd j D  |t	 j
ddf tt j
t d	} j|_d
|_ |t|  t  d S )Nw,"
)	delimiter	quotecharlineterminatorc                 3   s.   | ]&\}\}}||d  |  j | fV  qdS )r-   N)r;   )r   r   rI   r   rD   r   r   	<genexpr>   s
   

z"WheelFile.close.<locals>.<genexpr>r   rj   i  )r   rF   r:   r   r   writer	writerowsitemswriterowr6   r9   r   r   r(   r+   rv   ru   rw   r   getvaluer   close)rD   rz   r   r{   r   r   r   r      s    zWheelFile.close)r*   N)NN)N)__name__
__module____qualname____doc__r?   Zsha256r|   r   r7   r<   rh   ra   rw   r   r   r   r   r   r)   *   s   ,




r)   )N)%
__future__r   r   r?   os.pathr"   rerq   sysr%   collectionsr   	distutilsr   rU   zipfiler   r   r   Z	wheel.clir   Z
wheel.utilr	   r
   r   r   r   r   version_infoior   r   compileVERBOSEr3   r(   r)   r   r   r   r   <module>   s,    


