U
    da                  
   @   s0  d dl Z d dlZd dlZd dlmZ d dlmZ ddddddd	d
ddg
Zdd Zdd Z	e
dkr,e jddZejdddd ejddd ejddd ejddd e Zejd ej ZejZeed d!ZejD ]RZejerejed" d#d$ZeD ]Ze	eed e q qe	eed e qdS )%    N)Path)ZipFiledbmZcursestkintertesttestsZ	idle_testz__phello__.foo.pyz_bootstrap.pyz_bootstrap_external.pyc                 C   s   |  |r| t|d  S | S )N)
startswithlen)textprefix r   4/tmp/pip-unpacked-wheel-ua33x9lu/torch/utils/_zip.pyremove_prefix   s    
r   c                 C   s4   t | td }t|}|jtkr$d S || | d S )N/)r   strip_file_dirr   name	DENY_LISTwrite)	file_pathZstrip_file_pathzfZstripped_file_pathpathr   r   r   write_to_zip!   s
    
r   __main__zZip py source)descriptionpaths*zPaths to zip.)nargshelpz--install_dirz#Root directory for all output files)r   z--strip_dirz1The absolute directory we want to remove from zipz
--zip_namezOutput zip namer   w)modez/**/*.pyT)	recursive)argparseglobospathlibr   zipfiler   r   r   r   __name__ArgumentParserparseradd_argument
parse_argsargsZinstall_dirzip_nameZzip_file_nameZ	strip_dirr   r   r   pr   isdirfilesr   r   r   r   r   <module>   sB   

