
    "f                     `    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y)    N)Pathc                     t        j                  d|  d      j                  d      }t        |      rt	        |j                  d      d         S dS )Nz
gsutil du Tshellzutf-8 r   )
subprocesscheck_outputdecodelenevalsplit)urlss     6C:\work2\python\work_new2\yolov7\utils\google_utils.pygsutil_getsizer      sH    *SE 2$?FFwOA$'F4Q 11    c                    t        t        |       j                         j                  dd      j	                               } | j                         s$	 t        j                  d| d      j                         }|d   D cg c]  }|d   	 }}|d   }| j                  }||v r|  d| d}d}	 d| d| d| }	t        d|	 d|  d       t        j                   j#                  |	|        | j                         r| j%                         j&                  dkD  sJ 	 | j                         r| j%                         j&                  dk  r | j/                  d
       t        d|        t        d       y y y c c}w #  g d}t        j                  d	d
      j                         j                         d   }Y 'xY w# t(        $ rV}
t        d|
        |sJ d       d| d| }	t        d|	 d|  d       t+        j,                  d|	 d|         Y d }
~
d }
~
ww xY w# | j                         r| j%                         j&                  dk  r | j/                  d
       t        d|        t        d        Y y xY w)N' zhttps://api.github.com/repos/z/releases/latestassetsnametag_name)z	yolov7.ptzyolov7-tiny.ptz
yolov7x.ptzyolov7-d6.ptzyolov7-e6.ptzyolov7-e6e.ptzyolov7-w6.ptzgit tagTr   z2 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 
missing_okzERROR: Download failure: )r   strstripreplacelowerexistsrequestsgetjsonr   r	   r
   r   r   printtorchhubdownload_url_to_filestatst_size	Exceptionossystemunlink)filereporesponsexr   tagr   msg	redundantr   es              r   attempt_downloadr7      si   D	!))#r288:;D;;=	V||&CD6IY$Z[``bH)1();<);Aai);F<:&C yy6>FLTFR\]CI+D61DSE4&QSEdV378		..sD9{{})<)<s)BBB)B {{}		(;(;c(AKK4K05cU;<b	'   =	V7F)))4@GGIOOQRTUC  6(,- 7"77y7vVD6JSEdV378		HSEdV4556 {{}		(;(;c(AKK4K05cU;<b	sQ   .E< E7E< /A,F> H  7E< <<F;>	HAHH  HH   AI<c                    t        j                          }t        |      }t        d      }t        d|  d| dd       |j                  d       |j                  d       t	        j
                         d	k(  rd
nd}t        j
                  d|  d|        t        j                  j                  d      rdt                d|  d| }n	d| d|  d}t        j
                  |      }|j                  d       |dk7  r|j                  d       t        d       |S |j                  dk(  r5t        dd       t        j
                  d|        |j                          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%   r.   platformr-   r,   pathr!   	get_tokensuffix)idr/   tr9   outr   rs          r   gdrive_downloadrG   8   si   		A:D(^F	Gt4PTvUY
Z`bcKK4K 
MMTM" ??$	1%{CIIOPRtSWX[W\]^	ww~~hMik]Z^_a^bbghlgmnTF"KB4qQ
		!A
MMTM" 	Avt$  {{fo2&
		IdV$%	F499;?3'r
*+Hr   c                     t        |       5 }|D ]$  }d|v s|j                         d   c cd d d        S  	 d d d        y# 1 sw Y   yxY w)Ndownloadr   r   )openr   )r9   flines      r   rA   rA   [   sK    	fDT!zz|B''( ( 
 	 
 s   
A A A  A	)r   )zWongKinYiu/yolov7)r   ztmp.zip)z./cookie)r,   r?   r   r>   pathlibr   r"   r&   r   r7   rG   rA    r   r   <module>rO      s2    
      2"J Fr   