o j@s&ddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZ d$dede ed ed ed e f d d Z     d%dedede ede eded eded dfddZd&de deded efddZd'dede eded e efddZded ed!efd"d#ZdS)(N) expanduser)AnyIterableListOptional)tqdmTurl start_byte block_size progress_barreturnc cs,tjj|dd}tj|}t|dd}Wdn1s$wY||kr/dStj|}|r?d||jd<tj|G}t dd d || d $}d } | |} | s\n| V| t | 7} | t | qTWdn1swwYWddSWddS1swYdS) aDStream url by chunk Args: url (str): Url. start_byte (int or None, optional): Start streaming at that point (Default: ``None``). block_size (int, optional): Size of chunks to stream (Default: ``32 * 1024``). progress_bar (bool, optional): Display a progress bar (Default: ``True``). HEADmethodContent-LengthNz bytes={}-RangeBTi)unit unit_scale unit_divisortotaldisabler) urllibrequestRequesturlopenintinfogetformatheadersrreadlenupdate) r r r r reqresponseurl_sizeupointerpbar num_byteschunkr-E/home/kuhnn/.local/lib/python3.10/site-packages/TTS/utils/download.py stream_urls:    Pr/sha256Fdownload_folderfilename hash_value hash_typeresumecCstjj|dd}tj|}|p|ptj|}tj ||} |r4tj | r4d} tj | } n|sCtj | rCt d | d} d} |r~| t|ddkr~t| d } t| ||rh WddSWdn1srwYt d | t| | } t|| |d D]}| |qWdn1swYt| d  } |rt| ||st d | WddSWddS1swYdS) aPDownload file to disk. Args: url (str): Url. download_folder (str): Folder to download file. filename (str or None, optional): Name of downloaded file. If None, it is inferred from the url (Default: ``None``). hash_value (str or None, optional): Hash for url (Default: ``None``). hash_type (str, optional): Hash type, among "sha256" and "md5" (Default: ``"sha256"``). progress_bar (bool, optional): Display a progress bar (Default: ``True``). resume (bool, optional): Enable resuming download (Default: ``False``). rrabz6{} already exists. Delete the file manually and retry.wbNrrrbzBThe hash of {} does not match. Delete the file manually and retry.)r r )rrrrr get_filenameospathbasenamejoinexistsgetsize RuntimeErrorr!rr open validate_filer/write)r r1r2r3r4r r5r&req_infofilepathmode local_sizefile_objfpointerr,r-r-r. download_url8s>     "rJrHcCsR|dkr t}n |dkrt}nt |d}|sn||q||kS)a8Validate a given file object with its hash. Args: file_obj: File object to read from. hash_value (str): Hash for url. hash_type (str, optional): Hash type, among "sha256" and "md5" (Default: ``"sha256"``). Returns: bool: return True if its a valid file, else False. r0md5Ti)hashlibr0rK ValueErrorr#r% hexdigest)rHr3r4 hash_funcr,r-r-r.rBms     rB from_pathto_path overwritecCsn|dur tj|}zNt|d>}td|g}|D](}tj||j}| r>| |tj |r>td||s>q| ||q|WdWS1sRwYWn tj ybYnwzFt|d5}td||}|D]}tj||}tj |rtd||sqw| ||qwWd|WS1swY|WStjyYtdw)aExtract archive. Args: from_path (str): the path of the archive. to_path (str or None, optional): the root path of the extraced files (directory of from_path) (Default: ``None``) overwrite (bool, optional): overwrite existing files (Default: ``False``) Returns: list: List of paths to extracted files even if not overwritten. NrzOpened tar file %s.z%s already extracted.zOpened zip file %s.z2 > [!] only supports tar.gz, tgz, and zip achives.)r:r;dirnametarfilerAloggingrr=nameisfileappendr>extract ReadErrorzipfileZipFilenamelist BadZipFileNotImplementedError)rPrQrRtarfilesfile_ file_pathzfiler-r-r.extract_archivesP     &       rf dataset_path dataset_name output_pathc Csztj||}zddl}|jtd|d|jj||ddWdSty<tdtjt dd YdSw) ajDownload dataset from kaggle. Args: dataset_path (str): This the kaggle link to the dataset. for example vctk is 'mfekadu/english-multispeaker-corpus-for-voice-cloning' dataset_name (str): Name of the folder the dataset will be saved in. output_path (str): Path of the location you want the dataset folder to be saved to. rNz Downloading z...T)r;unzipz][!] in order to download kaggle datasets, you need to have a kaggle api token stored in your ~z.kaggle/kaggle.json) r:r;r=kaggleapi authenticateprintdataset_download_filesOSErrorr)rgrhri data_pathrlr-r-r.download_kaggle_datasets   rs)NrT)NNr0TF)r0)NF)rLrVr:rUrurllib.requestr\os.pathrtypingrrrrtorch.utils.model_zoorstrrboolr/rJrBrfrsr-r-r-r.s^   + 5$2