a
    þdú  ã                   @   sB   d dl Z d dlZd dlZd dlZdZd
dd„Zdd„ Zdd	„ ZdS )é    Nzahttps://nvlabs-fi-cdn.nvidia.com/stylegan2-ada-pytorch/pretrained/metrics/inception-2015-12-05.ptú./c              	   C   s€   t j | d¡}t j |¡s|tj t¡F}t|dƒ}t	 
||¡ W d   ƒ n1 sT0    Y  W d   ƒ n1 sr0    Y  |S )Nzinception-2015-12-05.ptÚwb)ÚosÚpathÚjoinÚexistsÚurllibÚrequestÚurlopenÚinception_urlÚopenÚshutilÚcopyfileobj)ZfpathZinception_pathÚresponseÚf© r   úb/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/cleanfid/downloads_helper.pyÚcheck_download_inception   s
    Hr   c              	   C   s¨   t j |¡}t j | |¡}t j |¡s¤t j| dd td|› ƒ tj 	|¡F}t
|dƒ}t ||¡ W d   ƒ n1 s|0    Y  W d   ƒ n1 sš0    Y  |S )NT)Úexist_okzdownloading statistics to r   )r   r   Úbasenamer   r   ÚmakedirsÚprintr   r	   r
   r   r   r   )Zlocal_folderÚurlÚnameÚ
local_pathr   r   r   r   r   Úcheck_download_url   s    Hr   c                 C   sž   dd„ }d}t  ¡ }|j|d| idd}||ƒ}|rN| |dœ}|j||dd}d}t|d	ƒ.}	| |¡D ]}
|
rh|	 |
¡ qhW d   ƒ n1 s0    Y  d S )
Nc                 S   s*   | j  ¡ D ]\}}| d¡r
|  S q
d S )NZdownload_warning)ÚcookiesÚitemsÚ
startswith)r   ÚkeyÚvaluer   r   r   Úget_confirm_token0   s    

z0download_google_drive.<locals>.get_confirm_tokenz+https://drive.google.com/uc?export=downloadÚidT)ÚparamsÚstream)r"   Zconfirmi €  r   )ÚrequestsÚSessionÚgetr   Úiter_contentÚwrite)Zfile_idZout_pathr!   ÚURLÚsessionr   Útokenr#   Z
CHUNK_SIZEr   Úchunkr   r   r   Údownload_google_drive/   s    
r.   )r   )	r   Úurllib.requestr   r%   r   r   r   r   r.   r   r   r   r   Ú<module>   s   
