a d @sddlmZddlZddlZddlmZddlZddlZddl Z ddl Z ddl m Z e e ddZeeszeeWneyYn0dddZdd d Zdd d ZdS))print_functionN)download~z .cache/gdowncsjdur dt}t|d4tfdddD]}||q4Wdn1sX0Y|S)Nirbcs S)N)read blocksizefr^/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/gdown/cached_download.pyzmd5sum..r)hashlibmd5openiterupdate hexdigest)filenamer hashblockrr r md5sums *rFcCspt|trt|dks$td||s6td|t|}||kr\|sXtd|dStd||dS)N zMD5 must be 32 chars: {}zComputing MD5: {}zMD5 matches: {}Tz*MD5 doesn't match: actual: {} expected: {}) isinstancestrlen ValueErrorformatprintrAssertionError)rrquietr Z md5_actualrrr assert_md5sum!s r"c Ks|dur8|dddddddd }tt|}t|r\|s\|sXtd ||St|r|rzt|||d |WSty}zt|t j d WYd}~n d}~00ttd }zt t |WntyYn0tjtd}zt|d} |s6d} |rd| |} n d| } t| t j d t|| fd|i|t|t| |Wdn1sz0YWn tyt|Yn0|rt|||d |dur|||S)a#Cached download from URL. Parameters ---------- url: str URL. Google Drive URL is also supported. path: str, optional Output filename. Default is basename of URL. md5: str, optional Expected MD5 for specified file. quiet: bool Suppress terminal output. Default is False. postprocess: callable Function called with filename as postprocess. kwargs: dict Keyword arguments to be passed to `download`. Returns ------- path: str Output filename. N/z-SLASH-:z-COLON-=z-EQUAL-?z -QUESTION-zFile exists: {})r!)fileZ_dl_lock)dirdlzCached Downloadingz{}: {}z{}...r!)replaceospjoin cache_rootexistsrrr"r sysstderrosmakedirsdirnameOSErrortempfilemkdtemprfilelockZFileLockshutilmove Exceptionrmtree) urlpathrr! postprocesskwargseZ lock_pathZ temp_rootZ temp_pathmsgrrr cached_download3sZ  $      0  rB)N)FN)NNNFN) __future__rrr1os.pathr=r+r8r/r5r7rr, expanduserr-r.r2r4rr"rBrrrr s$