eW~ddlZddlZddlmZddlmZddlZddlmZej dZ GddeZ dS)N)ArgumentParser)Path)SubCommandBaserepo_clicheZdZdZdZ d dededejfdZe d dedefdZ d e fd Z dS) SubCommanddownloadc |jjrQ|jjrE||jj|jj|jjr |jjnddSt dt)Nz/Not enough parameters to identify download file)argschannelnotebookdownload_notebookfilenameloggerinfoNotImplementedError)selfs :lib/python3.11/site-packages/repo_cli/commands/download.pymainzSubCommand.mains 9  (!3 (  " " ! "&*i&8B ""d      KKI J J J%'' 'Nr notebook_namerc|j||}tj|d}t |||d|jdS)NT)allow_redirectswb)apiget_notebook_download_urlrequestsgetopen get_filenamewritecontent)rr rrurlrs rrzSubCommand.download_notebookseh00-HH Ld 3 3 3 T  }h 7 7>>DDQYOOOOOrc@|rgt|}|stj|St d|dt|dz}tjtj |S)NzFile z5 already exists. Please specify a different filename.z.ipynb) ris_fileospathabspathrerrorFileExistsErrorjoingetcwd)rrfiles rr zSubCommand.get_filename$s  &>>D<<>> &wx000 [H[[[&% 8+w||BIKK222r subparserscZ|ddt|_|jdtd|jdtd|jd td dd |j|j dS)Nr z Help string)help descriptionz --channelz Channel name)typer1z --notebookz Notebook namez --filenameFzbfilename to save the report. Can contain full path for report to be saved in a different directory)r3requireddefaultr1)r) add_parser__doc__ subparser add_argumentstr set_defaultsr)rr/s rr6zSubCommand.add_parser2s#.. ]/   ##Kc#OOO ##Ls#QQQ ## u $    ###33333r)N) __name__ __module__ __qualname__namerr:r'r(r staticmethodr rr6rrrr s D ( ( (EIPPP+.P:<'PPPP 3 3C 33 3 3 3\ 34^444444rr) loggingr'argparserpathlibrrrepo_cli.commands.baser getLoggerrrrArrrGs ######111111  : & &64646464646464646464r