edZddlZddlZddlZddlZddlmZddlmZddl m Z ej dZ Gd d e Z dS) a Download artifact download reports from anaconda server conda repo report This command will download a report of all artifact downloads from the anaconda server and save it in the current working directory. If you want to chane the download directory, use the --filename option. N)Path)check_date_format)SubCommandBaserepo_clicFeZdZdZdZ ddZeddZdZdZ dS) SubCommandreportc||jj|jj|jj|jj|jj|jjr |jjnddSN)download_reportargs date_fromdate_to file_type user_nameschannelsfilename)selfs 8lib/python3.11/site-packages/repo_cli/commands/report.pymainzSubCommand.mainsc  I  I  I  I I "&)"4 >DI  $      Nc||krtd|d|d|r|dnd}|r|dnd}||td|d||j|||||}|stddSt| ||||}td||d kr+| tj |d dS|d kr| |d dSdS) Nz&Invalid date range: The starting date z& must be earlier than the ending date .,zDownloading report for z to z'No report available for the given dateszSaving report to jsonzutf-8)encodingcsv) ValueErrorsplitvalidate_channelsloggerinfoapi get_reporterrorr get_filename write_textrdumps) rrrr usernamesrr report_datafilepaths rrzSubCommand.download_report"s w  tttjqttt -6?IOOC(((4 *2<8>>#&&& x((( GGLMMMh)) w 8Y    LLB C C C F)))WiRRSS 222333       ; 7 7'  J J J J J %      g  > > > > > rcL|rgt|}|stj|St d|dtd|d|d|}tjtj |S)NzFile z5 already exists. Please specify a different filename.artifact_download_report__r) ris_fileospathabspathr#r'FileExistsErrorjoingetcwd)rrrrfiles rr(zSubCommand.get_filename?s  &>>D<<>> &wx000 [H[[[&%PyPP7PPYPPw||BIKK222rc|sdSd|jdD}|D]}||vrtd|ddS)NTc.g|]}|d |dS)name).0channels r z0SubCommand.validate_channels..Qs6   v FO   ritemszChannel z does not exist)r% list_channelsr )rr all_channelsr>s rr"zSubCommand.validate_channelsMs 4  81133G<     F FGl** !DG!D!D!DEEE+ F Frc|ddt|_|jdtddd|jdtt jd d |jd td d dgd|jdtd|jdtd|jdtd|j |j dS)Nr zdownload reports)help descriptionz --date_fromz'starting date YYYY-mm-dd for the reportT)typedefaultrDrequiredz --date_toz%Y-%m-%dz"end date YYYY-mm-dd for the report)rFrGrDz --file-typerrzformat of the report)rFrGchoicesrDz --user_nameszcomma separated list of users)rFrDz --channelsz comma separated list of channelsz --filenamezbfilename to save the report. Can contain full path for report to be saved in a different directory)r) add_parser__doc__ subparser add_argumentrdatetimenowstrftimestr set_defaultsr)r subparserss rrJzSubCommand.add_parser[su#.. -7/   ## ": $    ## "%))++44Z@@5 $    ## UO' $    ## 0 $   ## 3 $   ## u $   ###33333rr ) __name__ __module__ __qualname__r;rr staticmethodr(r"rJr<rrr r s D   LP????: 3 3 3\ 3 F F F.4.4.4.4.4rr )rKrNrloggingr2pathlibrutils.validatorsrbaser getLoggerr#r r<rrr]s  000000  : & &t4t4t4t4t4t4t4t4t4t4r