˲j |ddlZddlZddlZddlmZddlmZgdZdadZ d dZ d d Z e d kr e dSdS) N)Path)ZipFile) dbmcursestkinterrtesttests idle_testz__phello__.foo.pyz _bootstrap.pyz_bootstrap_external.pyc^||r|t|dS|S)N) startswithlen)textprefixs AC:\Projects\Historical\venv\Lib\site-packages\torch/utils/_zip.py remove_prefixr s0 v#CKKMM"" Kc|t|tdzz}t|}|jtvrdS|||dS)N/)rstrip_file_dirrname DENY_LISTwrite) file_pathstrip_file_pathzf prepend_strstripped_file_pathpaths r write_to_zipr &sX$}YQT@T'U'UU " # #D yIHHY*+++++rreturnctjd}|ddd|ddd |d d d |dddd|ddd |}|jdz|jz}|ja|j}t|d}t|j D]}tj |rGtj|dzd}t|D]}t!|tdz||ht!|tdz||dS)Nz Zip py source) descriptionpaths*z Paths to zip.)nargshelpz --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 zipr )r'defaultz --zip-namez --zip_namezOutput zip namerw)modez/**/*.pyT) recursive)r)argparseArgumentParser add_argument parse_args install_dirzip_name strip_dirrrrsortedr$osrisdirglobr )parserargs zip_file_namerrpfilesrs rmainr<.s  $ A A AF sAAA /T  @   F    l9JKKK     D$s*T]:M^N"K S ) ) )B DJ   O O 7==   OIa*n===E#E]]   ~3R[  NS0"+ N N N N N O Or__main__)r )r!N) r,r6r4pathlibrzipfilerrrrr r<__name__rrrBs      $ ,,,,$O$O$O$ON zDFFFFFr