3
sjM @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z dd Z dd Z
dd Zdd Zd
d ZdS ) Nc C s t jdd} | jddd | jdddd tjjd
| j }d }yttf t | W n4 t
k
r } ztd j | d
}W Y d d }~X nX t j
d
d S )Nz,Recreate wheel of package with given RECORD.) description record_pathzPath to RECORD file)helpz-oz --output-dirz>Dir where to place the wheel, defaults to current working dir.outdir)r destdefaultr z
Failed: {} )argparseArgumentParser add_argumentospathcurdir
parse_argsprintrewheel_from_recordvars
BaseExceptionformatsysexit)parsernsZretcodee r /usr/lib64/python3.6/__init__.pyrun
s r c s g }t jtjtjg}dd |D }x| D ]}d}xd|D ]\tjj|djtj fddtj
D } fdd|D }t |dkr8|d }q8W |dk r|j tjj|d q*|j d q*W |S )
zReturn list of paths to RECORD files for system-installed projects.
If a project is not installed, the resulting list contains None instead
of a path to its RECORD
c S s g | ]}t jj|r|qS r )r r
exists).0spr r r
<listcomp>( s z'find_system_records.<locals>.<listcomp>Nz-[^\{0}]+\.dist-infoc s g | ]}t jj |qS r )r r
join)r p)r r r r . s c s g | ]}t j |r|qS r )rematch)r c) dist_info_rer r r 0 s r r ZRECORD)
sitegetsitepackagesr base_prefixbase_exec_prefixr r
r! r seplistdirlenappend)ZprojectsZrecordsZ sys_sitepackZprojectr
Z
candidatesZfilteredr )r& r r find_system_records s
r/ c C s t jjt jj| }| t|d jt jj}t||\}}t| }t jj||d }t j
|dt j d}t j
} tj| }
x:|D ]2\} } }
|jt jj|| | d |
j| | |
g qW |
j|ddg |j|| j |j |jS )zcRecreates a whee of package with given record_path and returns path
to the newly created wheel.Nz.whlw)modeZ compression)Zarcname )r r
dirnamer- stripr+ get_records_to_packget_wheel_namer! zipfileZZipFileZ ZIP_DEFLATEDioStringIOcsvwriterwriteZwriterowZwritestrgetvalueclosefilename)r r site_dirrecord_relpathto_writeto_omitZnew_wheel_nameZnew_wheel_pathZ new_wheelZ
new_recordr; fZsha_hashsizer r r r : s
r c C s t jjt jj| d}tj|dd}tjj j |j
}W dQ R X t jjt jj| d}tj|dd}tjj j |j
}W dQ R X |j d}|j d}d} g }
x&|j d D ]} |
j
| jd
d qW d j|
}
|j d jd
d
d \} }
td||| |
| |
g}d
jt|S )z.Return proper name of the wheel, without .whl.ZWHEELzutf-8)encodingNZMETADATANameZVersionr2 ZTag-r .r )r r
r! r3 codecsopenemailr ZParserZparsestrreadgetZget_allr. splitfilterlist)r Zwheel_info_pathZwheel_info_fileZ
wheel_infoZ
metadata_pathZ
metadata_fileZmetadataZ distributionversionZ build_tagZlang_tagtZabi_tagZplat_tagZto_joinr r r r6 V s
r6 c
C s t jj| |}tj|dd}|j }W dQ R X t jj|jt jj}|dt d d }g }g }x|j
D ]}|j d} t | dkrnt jj | d s| d j
d s| d jd
s| d jd s| d |ks| d j
|r|j| q|j| qnqnW ||fS ) a Accepts path of sitedir and path of RECORD file relative to it.
Returns two lists:
- list of files that can be written to new RECORD straight away
- list of files that shouldn't be written or need some processing
(pyc and pyo files, scripts)
zutf-8)rF Nz dist-infodata,rJ r z..z.pycz.pyo)r r
r! rK rL rN r3 r4 r+ r-
splitlinesrP isabs
startswithendswithr. )
r@ rA Zrecord_file_pathZ record_fileZrecord_contentsZdata_dirrB rC lZsplr r r r5 n s"
r5 )r rK r: Z email.parserrM r r8 r# r'
subprocessr r7 r r/ r r6 r5 r r r r <module> s