
    JTh	                     v    S SK r S SKrS SKrS SKJr  S SKJr  / SQrSqS r	S
S jr
SS jr\S	:X  a  \" 5         gg)    N)Path)ZipFile)
dbmcursestkinterr   testtests	idle_testz__phello__.foo.pyz_bootstrap.pyz_bootstrap_external.py c                 N    U R                  U5      (       a  U [        U5      S  $ U $ )N)
startswithlen)textprefixs     H/var/www/auris/envauris/lib/python3.13/site-packages/torch/utils/_zip.pyremove_prefixr       s'    vCKM""K    c                     U[        U [        S-   5      -   n[        U5      nUR                  [        ;   a  g UR                  X5        g )N/)r   strip_file_dirr   name	DENY_LISTwrite)	file_pathstrip_file_pathzfprepend_strstripped_file_pathpaths         r   write_to_zipr    &   s@    $}YQT@T'UU"#DyyIHHY+r   c            	         [         R                  " SS9n U R                  SSSS9  U R                  SSS	S
9  U R                  SSSS
9  U R                  SSSSS9  U R                  SSSS
9  U R                  5       nUR                  S-   UR
                  -   nUR                  qUR                  n[        USS9n[        UR                  5       Hw  n[        R                  R                  U5      (       a>  [        R                  " US-   SS9n[        U5       H  n[!        U[        S-   XCS9  M     Me  [!        U[        S-   XCS9  My     g )Nz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 zipr   )r&   defaultz
--zip-namez
--zip_namezOutput zip namer   w)modez/**/*.pyT)	recursive)r   )argparseArgumentParseradd_argument
parse_argsinstall_dirzip_name	strip_dirr   r   r   sortedr#   osr   isdirglobr    )parserargszip_file_namer   r   pfilesr   s           r   mainr;   .   sO   $$AF
sA
/T   @  
 F	   l9JKD$$s*T]]:M^^N""K	S	)BDJJ77==IIa*n=E#E]	~3R + NS0"N  r   __main__)r   )returnN)r+   r5   r3   pathlibr   zipfiler   r   r   r   r    r;   __name__ r   r   <module>rB      sG      	  
	$ ,$ON zF r   