
    |f                     b    d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlZddZddZ	ddZ
dd
ZdS )    N)Path c                     t          j        d|  d                              d          }t          |          r(t	          |                    d          d                   ndS )Nz
gsutil du T)shellzutf-8 r   )
subprocesscheck_outputdecodelenevalsplit)urlss     */var/www/html/yolov7/utils/google_utils.pygsutil_getsizer      sZ     2S 2 2$???FFwOOA$'FF14Q   1    WongKinYiu/yolov7c                 (   t          t          |                                                               dd                                                    } |                                 s&	 t          j        d| d                                          }d |d         D             }|d         }n#  g d}d	}Y nxY w| j	        }||v r|  d
| d}d}	 d| d| d| }t          d| d|  d           t          j                            ||            |                                 r|                                 j        dk    sJ nm# t           $ r`}	t          d|	            |s
J d            d| d| }t          d| d|  d           t#          j        d| d|             Y d }	~	nd }	~	ww xY w|                                 r|                                 j        dk     r(|                     d           t          d|            t          d           d S # |                                 r|                                 j        dk     r(|                     d           t          d|            t          d            Y d S xY wd S d S )N'r   zhttps://api.github.com/repos/z/releases/latestc                     g | ]
}|d          S )name ).0xs     r   
<listcomp>z$attempt_download.<locals>.<listcomp>   s    <<<Aai<<<r   assetstag_name)z	yolov7.ptzyolov7-tiny.ptz
yolov7x.ptzyolov7-d6.ptzyolov7-e6.ptzyolov7-e6e.ptzyolov7-w6.ptzv0.1z2 missing, try downloading from https://github.com/z
/releases/Fzhttps://github.com/z/releases/download//zDownloading z to z...g    .AzDownload error: zNo secondary mirrorzhttps://storage.googleapis.com/z/ckpt/zcurl -L z -o T
missing_okzERROR: Download failure: )r   strstripreplacelowerexistsrequestsgetjsonr   printtorchhubdownload_url_to_filestatst_size	Exceptionossystemunlink)
filereporesponser   tagr   msg	redundantr   es
             r   attempt_downloadr:      s   D		!!))#r2288::;;D;;== 	|$ZD$Z$Z$Z[[``bbH<<();<<<F:&CC	7 7 7F CCCy6>>]]T]]]CIQDQQSQQ4QQ7S77d777888	..sD999{{}}B)<s)B)B)B)B)B 6 6 6,,,--- 77"7777JJJDJJ7S77d777888	4S44d44555555556 {{}} =		(;c(A(AKK4K000;c;;<<<b				 {{}} =		(;c(A(AKK4K000;c;;<<<b			?  >s@   +AB0 0B:A4E	 H  	
F3AF.)H  .F33H   A*Jtmp.zipc                 f   t          j                     }t          |          }t          d          }t          d|  d| dd           |                    d           |                    d           t	          j                    d	k    rd
nd}t          j        d|  d|            t          j                            d          rdt                       d|  d| }n	d| d|  d}t          j        |          }|                    d           |dk    r'|                    d           t          d           |S |j
        dk    r<t          dd           t          j        d|            |                                 t          dt          j                     |z
  dd           |S )Ncookiez;Downloading https://drive.google.com/uc?export=download&id=z as z... r   )endTr   WindowsNULz	/dev/nullz?curl -c ./cookie -s -L "drive.google.com/uc?export=download&id=z" > z?curl -Lb ./cookie "drive.google.com/uc?export=download&confirm=z&id=z" -o zcurl -s -L -o z) "drive.google.com/uc?export=download&id="r   zDownload error z.zipzunzipping... z	unzip -q zDone (z.1fzs))timer   r)   r2   platformr1   r0   pathr%   	get_tokensuffix)idr3   tr=   outr   rs          r   gdrive_downloadrK   9   s   	A::D(^^F	
Z
Z
ZPT
Z
Z
Z`bccccKK4K   
MMTM""" ?$$	11%%{CI]PR]]X[]]^^^	w~~h Rnikknn_annhlnnQTQQBQQQ
	!A
MMTM""" 	Avvt$$$    {fo2&&&&
	$d$$%%%	
*49;;?
*
*
*
*+++Hr   ./cookiec                     t          |           5 }|D ].}d|v r(|                                d         c cd d d            S /	 d d d            n# 1 swxY w Y   dS )Ndownloadr   )openr   )r=   flines      r   rE   rE   \   s    	f ( 	( 	(DT!!zz||B'''( ( ( ( ( ( ( (!	(( ( ( ( ( ( ( ( ( ( ( ( ( ( ( 2s   $AAAA)r   )r   )r   r;   )rL   )r0   rC   r   rB   pathlibr   r&   r*   r   r:   rK   rE   r   r   r   <module>rT      s    
			              2 2 2 2# # # #L       F     r   