o vq iO@sdZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZmZmZmZdd lmZmZd gZGd d d ejZd dZejdeddZej edZ!ddZ"dd Z#e$dkr|e#dSdS)u\ 命令行 Authors: xiangyiqing(xiangyiqing@baidu.com),suoyi@baidu.com Date: 2024/03/05 N)log)pipeline)model_file_download file_download)snapshot_download)convert_patterns)DEFAULT_MAX_WORKERSREPO_TYPE_SUPPORTREPO_TYPE_MODELDEFAULT_DATASET_REVISIONREPO_TYPE_DATASET) upload_file upload_foldermainc@seZdZdZdS)CustomHelpFormatteru% 自定义帮助信息格式 N)__name__ __module__ __qualname____doc__rr[/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/aistudio_sdk/cmdline.pyr src Csttjdtd}|jddd}|jddd}|jd d td d d d|jddtd d gddd|jddd}|}|jddd}|jddtdddd|jddtddd |jd!d"tdd#d |jd$d%td d&gd'd(d|jd)d*td d+d+gd,d|jd-d.t d d/gd0d1d|jd2d3td d4d4d5gd6d|jd7d8d9t d gd:d;|jdtd?d d@dA|jddtd d dBd|jdCdDtd d dEd|jdFdGd}|}|jddGd} | jdHtd@dI| jdJdKdLdMdN|jddOd} | jdPtdQdI| jdddRdS} | jdTdUd} | jdVtd?d dWdA| jdXdYd} | jdZtd[dI| jd\td]dId^d_d`dadbdcdddedf}|jdgdhtdi}djdk dldm| D}|jdnd}|jdodptdi}|jddtddqd |jdJdrtddsdtdu|jddvdLdwdN|jddxt d t | dy|dz|jd)d{td d|d |jd}d~tdi}|jddt ddd |jdJdrtddsdtdu|jddtdi}|jdtddI|jdtddI|jdtddd|jdtddd|jdtdddd|jdddtdd|jddtddd|jd tddd|jdt tdd|jddtdi}|jdtddI|jd\td?dddA|jdtd?dddA|jdttdd|jddtdd|jddtdd|jdtddd|jd tddd|jdt tdtddd|S)u 构建CLI Parser ZPROG)progformatter_classzAI Studio CLI SDKcommand)helpdestconfigu首次使用AI Studio CLI管理任务时, 需要先使用AI Studio账号的访问令牌进行身份认证。 一次认证后，再次使用时无需认证。)rz-tz--tokenFuAI Studio账号的访问令牌)typerequireddefaultrz-lz--log)infodebugru 日志级别)rrr choicesrsubmitu提交SDK产线任务jobz-nz--nameT summit_nameu产线任务名称)rrrrz-pz--pathuE代码包本地路径(文件夹)，要求文件总体积不超过50MB)rrrz-cz--cmdu任务启动命令z-ez--envpaddle2.6_py3.10)zpaddle2.4_py3.7zpaddle2.5_py3.10r'zpaddle3.0_py3.10u*飞桨框架版本, 默认paddle2.6_py3.10z-dz--deviceZv100u硬件资源, 默认v100z-gz--gpus1)ugpu数量, 默认单卡z-payz --paymentZacoinZcouponu?计费方式: * acoin-A币 * coupon-算力点. 默认使用A币z-mz--mount_datasetappendu-数据集挂载, 单个任务最多挂载3个)actionrrr rjobsu查询SDK产线任务query_pipeline_id?u产线id)rnargsr ru 产线名称z-sz--statusu状态stopu停止SDK产线任务stop_pipeline_id)rrz-fz--force store_trueu!强制停止，无需二次确认)r-ru管理SDK产线任务Zjob_idu任务IDu job子命令)rrrZlsu4查询某个 job 的 output 目录下文件夹内容 directoryu输出目录路径cpu7下载某个 job 的 output 目录下的文件到本地 result_fileu结果文件路径 local_pathu本地保存路径u公共领域 (CC0)u署名 (CC BY 4.0)u>署名-非商业性使用-相同方式共享 (CC BY-NC-SA 4.0)u(署名-相同方式共享 (CC BY-SA 4.0)u署名-禁止演绎 (CC-BY-ND)u自由软件基金会 (GPL 2)u署名-允许演绎 (ODC-BY)u其他)r)r*r+datasetu?管理数据集，此命令不在支持，请使用新的命令)rruo数据集许可协议的ID，仅在设置public后生效。默认为1 (公共领域 CC0)。 可选项包括：  css$|] \}}d|d|VqdS)z z: Nr).0kvrrr s"zinit..u数据集操作createu创建数据集u数据集名称z--files+u'本地文件路径，支持多个文件)rrr1rz--publicu是否公开数据集z --licenser))rrr#r rz --descriptionu数据集描述addu上传数据集文件z-idz--idu 数据集iddownloadu 下载文件z--modelu!模型ID，例如 myname/myrepoidz --datasetzkThe id of the dataset to be downloaded. For download, the id of either a model or dataset must be provided.z --revisionNzRevision of the entity.)rr rz --local_dirzNFile will be downloaded to local location specified bylocal_dir, in this case.files*zmSpecify relative path to the repository file(s) to download.(e.g 'tokenizer.json', 'dir/decoder_model.onnx').)rr r1rz --includezEGlob patterns to match files to download.Ignored if file is specified)r1r rrz --excludezLGlob patterns to exclude from files to download.Ignored if file is specified)r1rr rzA User Access Tokenz --max-workersz0The maximum number of workers to download files.uploadu 上传文件repo_idz;The ID of the repo to upload to (e.g. `username/repo-name`)zTOptional, Local path to the file or folder to upload. Defaults to current directory. path_in_repozfOptional, Path of the file or folder in the repo. Defaults to the relative path of the file or folder.z --repo-typezPType of the repo to upload to (e.g. `dataset`, `model`). Defaults to be `model`.)r#r rz'Glob patterns to match files to upload.)r1rrz.Glob patterns to exclude from files to upload.z--commit-messagez,The message of commit. Default to be `None`.r+r*z1The number of workers to use for uploading files.)rZcli_logargparseArgumentParserradd_subparsers add_parser add_argumentstrintjoinitemslistkeysrr r minos cpu_count)parserZsubparser_aistudiorr$Zsubparser_submitZ submit_jobr.r2Zsubparser_stopZstop_jobr%Z subparser_jobZjob_lsZjob_cpZlicense_mappingr>Z license_helpZdatasets_createrDrFrGrJrrrinit&s            r\ZAISTUDIO_CACHE_HOMEHOMEr z/.cache/aistudio/.auth/tokencCsPt|ttd}|t|Wdn1swYttddS)z# save to separate location wNi)printopen TOKEN_FILEwriterRrYchmod)tokenfrrr save_tokens  rgc Cst}tjdd}t|z||}WnYdSt|dddkr|js,Jd|jddks8Jd|jdd}|j|_|j durVt j |rV||_ ||_ nM|j durht j |rh||_ d |_ n;|j durutd |d |j durt j |j r|j |_ t j |j |_ n|j dur|j |_ d |_ n|j |_ |j |_ t j |j rt|j |j |j|j|j|jd n*t j |j rt|j|j |j |j|jt|jt|j|j|jd nt|j dtd|jdSt|drE|jrEt|jdkrt|j|jd|j|j |jddSt|jdkr-t!|j|j |j|j|j|jddSt!|j|j |jt|jt|j|j|jddSt|dr|j"r|j rV|j nt#}t|jdkrqt$|j"|jd|j|t%|jdn-t|jdkrt!|j"||j|j|j|jdnt!|j"||jt|jt|j|j|jdtd|j"ddSd|vrt&'|dSd|vrt&(|dSd|vrt&)|dSd|vr|j*st+j,ddd st-.d!dSt-.d"t&/|dSd#|vrt&0|dSd$|vrd%|vrt&1|dSd&|vr$d'|vr$t-2d(t&3|dSd)|vr:d'|vr:t-2d(t&4|dSt-.d*dS)+u CLI入口r)NrrJz`repo_id` is required/z1repo_id should be in format of username/repo-name.'zD' is not a local file or folder. Please set `local_path` explicitly.r)Zpath_or_fileobjrLrK repo_typecommit_messagere) rKZ folder_pathrLrmrlallow_patternsignore_patterns max_workersrez is not a valid local pathzFinished uploading to modelr) local_dirrevisionre)rKrsrrrnrpre)rKrsrrrnrorprer>)rrrsrlrez& Successfully Downloaded from dataset z. rer&r/r3zDo you want to continue?Fr^zAborted.z Confirmed.r5r7r8namerHz&This command is not supported any moreidu无效的命令)5r\sysargvr` parse_argsgetattrrKcountsplitr8rYpathisfilerLisdir ValueErrorbasenamer rlrmrerrincludeexcluderphasattrrqlenrHrrrrsrr>r rr rZ set_configrDqueryforceclickZconfirmrr!r2Zlist_output_filesZdownload_output_fileerrorZcreate_datasetZadd_file)r[args repo_nameZdataset_revisionrrrrs0                     __main__)%rrvrMrrYZ aistudio_sdkrZaistudio_sdk.sdkrZaistudio_sdk.file_downloadrrZaistudio_sdk.snapshot_downloadrZaistudio_sdk.utils.utilrZaistudio_sdk.configrr r r r Zaistudio_sdk.hubr r__all__RawTextHelpFormatterrr\getenvZ cache_homer| expanduserrbrgrrrrrrs6    1 *