U Jh<@sRddlZddlmZddlmZmZmZmZmZm Z ddl m Z m Z ddl mZmZmZddlmZddlmZddlmZmZdd lmZd d d d ddddddddddddddddddd d!d"d#d$gZeeeee eeefeefd%d&d'Zeeee eeefeefd(d)d*Zd0eeeeeee ed+d,d-ZGd.d/d/ZdS)1N) import_module)AnyCallableDictListOptionalTuple)AbstractProgresserrors)APIData APIResponseErrorWarningObject)UAConfig)IncorrectFieldTypeError)API_UNKNOWN_ARGWARN_NEW_VERSION_AVAILABLE)check_for_new_versionzu.pro.version.v1z,u.pro.attach.auto.configure_retry_service.v1z%u.pro.attach.auto.full_auto_attach.v1z'u.pro.attach.auto.should_auto_attach.v1z%u.pro.attach.guest.get_guest_token.v1zu.pro.attach.magic.initiate.v1zu.pro.attach.magic.revoke.v1zu.pro.attach.magic.wait.v1z'u.pro.attach.token.full_token_attach.v1zu.pro.detach.v1zu.pro.packages.summary.v1zu.pro.packages.updates.v1zu.pro.security.cves.v1z!u.pro.security.fix.cve.execute.v1zu.pro.security.fix.cve.plan.v1z!u.pro.security.fix.usn.execute.v1zu.pro.security.fix.usn.plan.v1z'u.pro.security.status.livepatch_cves.v1z(u.pro.security.status.reboot_required.v1zu.pro.services.dependencies.v1zu.pro.services.disable.v1zu.pro.services.enable.v1z u.pro.status.enabled_services.v1zu.pro.status.is_attached.v1zu.apt_news.current_news.v1zu.security.package_manifest.v1zu.unattended_upgrades.status.v1)optionsfieldsreturnc Csi}g}|D]|}z|d\}}Wn tk rBtj|dYnX|rL|sXtj|d||kr|ttj|djtj id|||<q ||fS)N=argtitlecodemeta) split ValueErrorr ZAPIBadArgsFormatappendr rformatmsgname)rrkwargswarningsZoptionkvr'2/usr/lib/python3/dist-packages/uaclient/api/api.py_process_options+s&   r))datarrcCsi}g}zt|}Wn$tjjk r:tj|dYnX|D]8\}}||krt|tt j |dj t j id|||<qD||fS)N)r*rr) jsonloadsdecoderZJSONDecodeErrorr ZAPIJSONDataFormatErroritemsrr rr r!r")r*rr#r$Z json_datar%r&r'r'r( _process_dataHs"  r/) endpoint_pathrr*cfgprogress_objectrc Cs6|tkrttj|dStd|}|j}g}|jrndd|jjD}z@|r`t||\} } n|rtt ||\} } n ig} } | | Wn2tj k r} zt| WYSd} ~ XYnXz|j | }Wn>t k r } zttj| j|dWYSd} ~ XYnXz*|jr*|j|||d} n |||} Wn2tk rj} zt| WYSd} ~ XYnXnx|sz|rttj|dSz&|jr|j||d} n ||} Wn2tk r} zt| WYSd} ~ XYnXt} | r|ttj| djtjidt|j| j|t|j| | jd d S) N)endpointz uaclient.api.cSsg|] }|jqSr')key).0fr'r'r( uszcall_api..)rr3)r2)versionr)typeZ attributesr)Z_schema_versionr$r*) VALID_ENDPOINTSr Z error_outZAPIInvalidEndpointrr3 options_clsrr)r/extendZAPIErrorZ from_dictrZ APIMissingArgr4supports_progressfn ExceptionZAPINoArgsForEndpointrrr rr r!r"r r8r$r r)r0rr*r1r2moduler3Zoption_warningsrr#r$eresultZ new_versionr'r'r(call_apibs    "    rCc@s"eZdZdeeeedddZdS) APIEndpointF)r8r"r>r=cCs"||_||_||_||_||_dS)N)r8r"r>r;r=)selfr8r"r>r;r=r'r'r(__init__s zAPIEndpoint.__init__N)F)__name__ __module__ __qualname__strrboolrFr'r'r'r(rDsrD)N) r+ importlibrtypingrrrrrrZ uaclient.apir r Zuaclient.api.data_typesr r r Zuaclient.configrZuaclient.data_typesrZuaclient.messagesrrZuaclient.versionrr:rJr)r/rCrDr'r'r'r(sl         T