a
    d                     @   sH  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d	Z	e
d
krDe 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jD ]ZZejer.ejed" d#d$ZeD ]Ze	eed eed% qqe	eed eed% 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   Y/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/utils/_zip.pyremove_prefix   s    
r    c                 C   s8   |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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   r   pr   isdirfilesr   r   r   r   r   <module>   s6   


