U
    •Ê¹j§  ã                
   @   s`  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d„Z	e
dk�r\e jdd�Zejdddd� ejdddd� ejdddd� ejd d!d"dd#� ejd$d%d&d� e ¡ Zejd' ej ZejZejZeed(d)�ZeejƒD ]^Zej e¡�rFejed* d+d,�ZeeƒD ]Ze	eed' eed-� �q(qüe	eed' eed-� qüdS )/é    N)ÚPath)ÚZipFileÚdbmZcursesÚtkinterÚtestZtestsZ	idle_testz__phello__.foo.pyz_bootstrap.pyz_bootstrap_external.pyc                 C   s   |   |¡r| t|ƒd … S | S )N)Ú
startswithÚlen)ÚtextÚprefix© r   úO/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/utils/_zip.pyÚremove_prefix   s    
r   Ú c                 C   s8   |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ÚzfÚprepend_strZ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--install_dirz#Root directory for all output files)r   z--strip-dirz--strip_dirz1The absolute directory we want to remove from zipz--prepend-strz--prepend_strz7A string to prepend onto all paths of a file in the zip)r   Údefaultz
--zip-namez
--zip_namezOutput zip namer   Úw)Úmodez/**/*.pyT)Ú	recursive)r   )r   )Ú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   Úsortedr   Úpr   ÚisdirÚfilesr   r   r   r   r   Ú<module>   sP   ñ

   ÿ