a  d @stddlZddlZddlZddlmZmZddlmZddlmZddl m Z ddZ d d Z dd d Z dddZdS)N)download_url_to_fileget_dir)tqdm)urlparse) sizeof_fmtc Cst}d}d|i}|j||dd}t|}|rH||d<|j||dd}|j||dddid}d |jvrt|jd d d }nd }t|||d S) zDownload files from google drive. Ref: https://stackoverflow.com/questions/25010369/wget-curl-large-file-from-google-drive # noqa E501 Args: file_id (str): File id. save_path (str): Save path. z*https://docs.google.com/uc?export=downloadidT)paramsstreamZconfirmRangez bytes=0-2)r r headersz Content-Range/rN)requestsSessiongetget_confirm_tokenr intsplitsave_response_content) Zfile_idZ save_pathsessionURLr responsetokenZresponse_file_size file_sizerd/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/utils/download_util.pydownload_file_from_google_drive s  rcCs*|jD]\}}|dr |Sq dS)NZdownload_warning)cookiesitems startswith)rkeyvaluerrrr*s  rc Cs|dur(tt||dd}t|}nd}t|dv}d}||D]F}||7}|dur~|d|dt|d||rF||qF|dur| Wdn1s0YdS)Nchunk)totalunitwbrrz Download z / ) rmathceilropen iter_contentupdateset_descriptionwriteclose) rZ destinationr chunk_sizeZpbarZreadable_file_sizefZdownloaded_sizer#rrrr1s    rTcCs|durt}tj|d}tj|ddt|}tj|j}|durL|}tjtj||}tj|st d|d|dt ||d|d|S) aPLoad file form http url, will download models if necessary. Ref:https://github.com/1adrianb/face-alignment/blob/master/face_alignment/utils.py Args: url (str): URL to be downloaded. model_dir (str): The path to save the downloaded model. Should be a full path. If None, use pytorch hub_dir. Default: None. progress (bool): Whether to show the download progress. Default: True. file_name (str): The downloaded file name. If None, use the file name in the url. Default: None. Returns: str: The path to the downloaded file. NZ checkpointsT)exist_okzDownloading: "z" to  )Z hash_prefixprogress) rospathjoinmakedirsrbasenameabspathexistsprintr)urlZ model_dirr3 file_nameZhub_dirpartsfilenameZ cached_filerrrload_file_from_urlFs r@)Nr")NTN)r'r4rZ torch.hubrrr urllib.parsermiscrrrrr@rrrrs