U (f} @sddlZddlZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z ddlmZmZmZddlmZGd d d e Zd d Zed e je jeejeede jddede jddede jdgddede jdddgdgdZdS)N) OrderedDict)AnyOptional) exceptionsmessages)AbstractProgress)call_api) ProArgumentProArgumentGroup ProCommand) HelpCategoryc@s(eZdZeeeeeedddZdS)CLIAPIProgress) total_steps done_stepsprevious_step_messagecurrent_step_messagecCs8t}||d<||d<||d<||d<tt|dS)Nrrrr)rprintjsondumps)selfrrrrdr2/usr/lib/python3/dist-packages/uaclient/cli/api.pyprogresss zCLIAPIProgress.progressN)__name__ __module__ __qualname__intrstrrrrrrr s r cKs|jr|jrt|jr:|jdkr:tjs:tj|_|jrHt }nd}t |j |j|j||}t | |jdkr|dSdS)N-Zsuccessr)optionsdatarZCLIAPIOptionsXORDatasysstdinisattyreadZ show_progressr r endpoint_pathrZto_jsonresult)argsZcfgkwargsrr(rrr action_apis"    r+Zapir'Zendpoint)helpmetavarz--show-progress store_true)r,actionz--argsr!*)r,destdefaultnargsz--datar")r,r1r2)Z arguments)r,Z descriptionr/Z help_categoryZargument_groups)rr# collectionsrtypingrrZuaclientrrZ uaclient.apirZuaclient.api.apirZuaclient.cli.commandsr r r Zuaclient.cli.parserr r r+Z CLI_ROOT_APIZ CLI_API_DESCZOTHERZCLI_API_ENDPOINTZCLI_API_SHOW_PROGRESSZ CLI_API_ARGSZ CLI_API_DATAZ api_commandrrrrsX