a
    .=ica                     @   s  d dl Z d dlZd dlZd dlmZ d dlmZ g d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 ]PZejerejed ddZeD ]Ze	eed e qqe	eed e qdS )    N)Path)ZipFile)
dbmZcursestkinterr   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   a/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/torch/utils/_zip.pyremove_prefix   s    
r   c                 C   s4   t | td }t|}|jtv 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>   s.   

