U Mh @s~ddlZddlZddlZddlmZddlmZddddddd d d d g Zd addZ dddZ ddddZ e dkrze dS)N)Path)ZipFiledbmZcursestkintertestZtestsZ idle_testz__phello__.foo.pyz _bootstrap.pyz_bootstrap_external.pycCs||r|t|dS|S)N) startswithlen)textprefixr B/var/www/html/venv/lib/python3.8/site-packages/torch/utils/_zip.py remove_prefix s rcCs8|t|td}t|}|jtkr(dS|||dS)N/)rstrip_file_dirrname DENY_LISTwrite) file_pathZstrip_file_pathzf prepend_strZstripped_file_pathpathr r r write_to_zip&s  r)returncCstjdd}|jdddd|jddd d |jd d d d |jddddd|jdddd |}|jd|j}|ja|j}t |dd}t |j D]Z}t j |rtj|ddd}t |D]}t|td||dqqt|td||dqdS)Nz Zip py source) descriptionpaths*z Paths to zip.)nargshelpz --install-dirz --install_dirz#Root directory for all output files)rz --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)rdefaultz --zip-namez --zip_namezOutput zip namerw)modez/**/*.pyT) recursive)r)argparseArgumentParser add_argument parse_argsZ install_dirzip_nameZ strip_dirrrrsortedrosrisdirglobr)parserargsZ zip_file_namerrpfilesrr r r main.sF     r0__main__)r) r#r+r)pathlibrzipfilerrrrrr0__name__r r r r s*   '