a XyaI@sLddlmZddlmZddlmZddlZddlmZGddde Z dS))mktime)parse) OrderedDictN)DestionationPathExistsc@sXeZdZdZddZdddZddd Zdd d Zd d ZdddZ ddZ ddZ dS) Downloaderz7 Download files from your Anaconda repository. cCs||_||_||_dSN) aserver_apiusernamenotebook)selfrr r r Glib/python3.9/site-packages/binstar_client/utils/notebook/downloader.py__init__szDownloader.__init__.FcCs||_||||Sr)output ensure_outputdownload_files)r package_typesrforcer r r __call__szDownloader.__call__cCsb||_|t}|D]@}|dp,d}||vr|||rP|||d<qt|dq|S)z This additional method was created to better handle the log output as files are downloaded one by one on the commands/download.py. typebasename)rrr list_filesget can_downloadr)r rrrfilesfpkg_typer r r list_download_filess  zDownloader.list_download_filescCsbg}|D]L}|dpd}||vr |||rL||||dq t|dq t|S)Nrrr)rrrdownloadappendrsorted)r rrrrrr r r r*s   zDownloader.download_filescCs|d}|j|j|j|d|}tjtj|s`zttj|Wnt y^Yn0t tj |j |d*}| dD]}||qWdn1s0YdS)z. Download file into location. rversionwbiN)rr r r ospathexistsdirnamemakedirsOSErroropenjoinrZ iter_contentwrite)r distfilenameZrequests_handleZfdoutchunkr r r r 7s zDownloader.downloadcCs"tjtj|j|d p |S)z Can download if location/file does not exist or if force=True :param dist: :param force: :return: True/False r)r%r&r'r,r)r r.rr r r rJszDownloader.can_downloadcCstj|jst|jdS)z2 Ensure output's directory exists N)r%r&r'rr))r r r r rSszDownloader.ensure_outputc Csg}i}|j|j|jd}|D]2}|d|vrF||d|q"|g||d<q"|D]d\}}z|t|dddWq^ty|t|dddYq^||dYq^0q^|S)zY List available files in a project (aka notebook). :return: list rrcSs t|dS)Nr#)intxr r r lz'Downloader.list_files..)keycSstt|dS)NZ upload_time)rrZ timetupler2r r r r4or5)rpackager r r!itemsmax ValueError)r rtmprrrZversionsr r r rZs"  zDownloader.list_filesN)rF)rF)F)F) __name__ __module__ __qualname____doc__rrrrr rrrr r r r r s    r) timerZdateutil.parserr collectionsrr%Zbinstar_client.errorsrobjectrr r r r s