o i @sddlZddlZddlZddlZddlZddlmZddlZddl Zzddl Z dZ Wn e y5dZ Ynwzddl Z e durHGddde j ZdZWn e yVdZYnwddlmZdd lmZeejZdZd ejd efd d Zd ejfddZejejjejjfddded ejd dfddZ d ejfddZ!ej"d e#eeffddZ$ded efddZ%    d'dede&de&dedBde&d ejf dd Z'Gd!d"d"e(Z)ded efd#d$Z*ded e&fd%d&Z+dS)(N)castTFc @s.eZdZ d dedededBddfddZdS) DownloadProgressBarNbbsizetsizereturncCs&|dur||_||||jdS)N)totalupdaten)selfrrrr K/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/pydicom/data/download.py update_toszDownloadProgressBar.update_to)rrN)__name__ __module__ __qualname__intrr r r rrsrr)retry) warn_and_logfpathrcCsd}t}t|d0}||}t|dkr1||||}t|dksWd|SWd|S1s>wY|S)zReturn the SHA256 checksum for the file at `fpath`. Parameters ---------- fpath : pathlib.Path The absolute path to the file that is to be checksummed. Returns ------- str The SHA256 checksum of the file. irbrN)hashlibsha256openreadlenr hexdigest)r BLOCKSIZEhasherfbufr r rcalculate_file_hash/s        r"cCstjd}|jdd|S)zReturn the path to the pydicom config directory, creating it if required The config directory will be named ``.pydicom`` and will be created in the local user's home directory. z.pydicomTexist_ok)pathlibPathhomemkdir) config_dirr r rget_config_dirGs r*z*Installing the `requests` package may help)exc_msgurlc CsLt|}trqtrNtj|dd}t|jdd}t|d$}t j |j dd|ddd | d d d D]}| |q3Wd n 1sEwYd Sd St|}t|d}| |j Wd d S1sjwYd Strtddd | d d d}tjj|||jdWd d S1swYd Stj||d S)zDownload the file at `url` to `fpath` with a progress bar. Parameters ---------- url : str The URL to download the file from. fpath : pathlib.Path The absolute path where the file will be written to. T)streamzcontent-lengthrwbi) chunk_sizeBr/)r unit unit_scaleminitersdescN)r3r4r5r6) reporthook)osfspath HAVE_REQUESTSUSE_PROGRESS_BARrequestsgetrheadersrtqdm iter_contentsplitwritecontentrurllibrequest urlretriever) r,rfilenamertotal_size_in_bytesfiledatar tr r rdownload_with_progressSs:     $ ""rMcCstd}|jdd|S)z@Return the path to the cache directory, creating it if required.rKTr#)r*r()data_dirr r r get_data_dirs  rOcCsHttd}ttttft|WdS1swYdS)z>Return a dict containing the URL mappings from ``urls.json```.z urls.jsonN)rHERErdictstrjsonload)url_filer r r get_url_maps$rVrGcCs:ddtD}z||WStytdw)aReturn the download URL corresponding to `filename`. The filename:URL mappings are located in the ``urls.json`` file. Parameters ---------- filename : str The filename of the file to get the corresponding URL for. Returns ------- str The download URL corresponding to `filename`. Raises ------ ValueError If `filename` is not in the ``urls.json`` record. cSi|] \}}||qSr lower.0kvr r r zget_url..z:The file provided isn't within pydicom's urls.json record.)rVitemsrYKeyError ValueError)rGurlsr r rget_urls  rd check_hashredownload_on_hash_mismatchquietcCstrtdt|}|r&|r&zt|Wn ty%|Ynw|s7|dur2t|}t |||r_zt |}Wn tyL| YSw|s_|r[|t |ddSt d| S)aReturn the absolute path to the cached file with `filename`. If the file isn't available in the cache then it will be downloaded. Parameters ---------- filename : str The filename of the file to return the path to. check_hash : bool, optional ``True`` to perform a SHA256 checksum on the file, ``False`` otherwise. redownload_on_hash_mismatch : bool, optional ``True`` to redownload the file on checksum failure, ``False`` otherwise. url : str, optional The file's corresponding download URL Returns ------- pathlib.Path The absolute path to the file. z No network!NF)rfz2The file on disk does not match the recorded hash.)_SIMULATE_NETWORK_OUTAGE RuntimeErrorrOjoinpathexistsget_cached_filehash NoHashFoundunlinkrdrMdata_file_hash_checkresolvedata_path_with_downloadrb)rGrerfr,rgfilepath hash_agreesr r rrqs6         rqc@s eZdZdS)rmN)rrrr r r rrmsrmcCsxttd}ttttft|}dd|D}Wdn1s&wYz||WSt y;t w)zReturn the SHA256 checksum of a cached file. Parameters ---------- filename : str The filename of the cached file to calculate the checksum for. Returns ------- str The SHA256 checksum of the cached file. hashes.jsoncSrWr rXrZr r rr^r_z'get_cached_filehash..N) rrPrrQrRrSrTr`rYrarm)rG hash_filehashesr r rrls  rlc Cst|}t|}t|}z t|}W||kStyctdtt d }t |}Wdn1s9wY|||<tt dd}t j ||dddWd1s]wYw)a+Return ``True`` if the SHA256 checksum of the cached file is correct. Parameters ---------- filename : str The filename of the cached file to check. Returns ------- bool ``True`` if the cached file has the correct checksum, ``False`` otherwise. z4Hash not found in hashes.json. File will be updated.rtNwT)indent sort_keys) r8r9rOrjr"rlrmrrrPrSrTdump)rGrrcalculated_filehashcached_filehashrurvr r rros&      ro)TTNT), functoolsrrSr8r%typingrurllib.requestrD urllib.errorr<r: ImportErrorr?rr;r pydicom.miscrr&__file__rpparentrPrhrRr"r*error HTTPErrorURLErrorrMrO lru_cacherQrVrdboolrqrarmrlror r r rsr      ( ?