o i @spddlZddlZddlZddlmZddlmZgdZdaddZ ddd d Z dd d Z e d kr6e dSdS)N)Path)ZipFile) dbmcursestkinterrtesttests idle_testz__phello__.foo.pyz _bootstrap.pyz_bootstrap_external.pycCs||r |t|dS|S)N) startswithlen)textprefixrR/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/utils/_zip.py remove_prefix!s rreturncCs8|t|td}t|}|jtvrdS|||dS)N/)rstrip_file_dirrname DENY_LISTwrite) file_pathstrip_file_pathzf prepend_strstripped_file_pathpathrrr write_to_zip's  rc Cstjdd}|jdddd|jddd d |jd d d d |jddddd|jdddd |}|jd|j}|ja|j}t |dd<}t |j D]-}t j |rptj|ddd}t |D] }t|td||dqbqMt|td||dqMWddS1swYdS)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_namerrpfilesrrrrmain/sB    "r8__main__)r )rN) r(r2r0pathlibrzipfilerrrrrr8__name__rrrrs    &