o eZ @sddlZddlmZddlmZddlmZddlmZddl m Z m Z zddl m Z ddlmZWn ey=ed wd ed ed efd dZdee efd efddZdS)N)md5)Path) copyfileobj)Union) VIDEO_ASSETS VideoAssets)get)tqdmz| Please install requests and tqdm to download assets or install supervision with assets pip install supervision[assets] filenameoriginal_md5_hashreturncCs\tj|sdSt|d}|}t|}Wd||kS1s%wY||kS)a/ Check if the MD5 hash of a file matches the original hash. Parameters: filename (str): The path to the file to be checked as a string. original_md5_hash (str): The original MD5 hash to compare against. Returns: bool: True if the hashes match, False otherwise. FrbN)ospathexistsopenreadr hexdigest)r r file file_contentscomputed_md5_hashrL/var/www/myenv/lib/python3.10/site-packages/supervision/assets/downloader.pyis_md5_hash_matchings   r asset_namec Csvt|tr|jn|}t|s|tvrtd|dtt|dddd}|t |j dd}t| }|j jdddtj|jd|d d d +}|d  }t||Wd n1sewYWd |SWd |S1s}wY|St|rt|t|dstdt|t|St|d|SdddtD}td|d)a Download a specified asset if it doesn't already exist or is corrupted. Parameters: asset_name (Union[VideoAssets, str]): The name or type of the asset to be downloaded. Returns: str: The filename of the downloaded asset. Example: ```python >>> from supervision.assets import download_assets, VideoAssets >>> download_assets(VideoAssets.VEHICLES) "vehicles.mp4" ``` z Downloading z assets rT)streamallow_redirectszContent-Length)parentsexist_okrz#a351fb)totaldesccolourwbNz#File corrupted. Re-downloading... z asset download complete. z, css|]}|jVqdS)N)value).0assetrrr Ysz"download_assets..z2Invalid asset. It should be one of the following: .) isinstancerr%rrrprintrraise_for_statusintheaders expanduserresolveparentmkdirr wrapattrrawrrrrremovedownload_assetsjoin ValueError)rr response file_size folder_pathraw_respr valid_assetsrrrr6+sB        r6)rhashlibrpathlibrshutilrtypingrsupervision.assets.listrrrequestsr tqdm.autor ImportErrorr8strboolrr6rrrrs