o x[h6 @sVUdZddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZed d gZd d gedgeedZeed<eeZdZdZdZde fddZ!de fddZ"d-defddZ#d-ddZ$defdd Z%de d!e&fd"d#Z'de fd$d%Z(de fd&d'Z)d(e*d)eded*ed!df d+d,Z+dS).z1ubuntu_pro: Configure Ubuntu Pro support servicesN)AnyList)urlparse) performancesubputil)Cloud)Config) MetaSchema) PER_INSTANCEzubuntu-advantageubuntu_advantage cc_ubuntu_proubuntu ubuntu_pro)iddistros frequencyactivate_by_schema_keysmetaREDACTEDzUUnable to determine if this is an Ubuntu Pro instance. Fallback to normal Pro attach.) http_proxy https_proxyglobal_apt_http_proxyglobal_apt_https_proxyua_apt_http_proxyua_apt_https_proxy pro_sectioncCsd|vrdS|d}t|ts dt|j}t|t|d|vr&dS|d}t|ts@dt|j}t|t|dS)Nfeaturesz.'ubuntu_pro.features' should be a dict, not a disable_auto_attachzB'ubuntu_pro.features.disable_auto_attach' should be a bool, not a ) isinstancedicttype__name__LOGerror RuntimeErrorbool)rrmsgrr(@/usr/lib/python3/dist-packages/cloudinit/config/cc_ubuntu_pro.pyvalidate_schema_features*s*    r* pro_configc Csg}t|D]:\}}|tvrtd|q|durqzt|}|jdvr.|d|Wqtt fyB|d|Yqw|rOt d d |dS)aValidate user-provided ua:config option values. This function supplements flexible jsonschema validation with specific value checks to aid in triage of invalid user-provided configuration. Note: It does not log/raise config values as they could be urls containing sensitive auth info. @param pro_config: Dictionary of config value under 'ubuntu_pro'. @raises: ValueError describing invalid values provided. z4Not validating unknown ubuntu_pro.config.%s propertyN)httphttpsz-Expected URL scheme http/https for ua:config:zExpected a URL for ua:config:z$Invalid ubuntu_pro configuration: {} ) sorteditemsKNOWN_PRO_CONFIG_PROPSr#warningrschemeappendAttributeError ValueErrorformatjoin)r+errorskeyvalue parsed_urlr(r(r)supplemental_schema_validationEs2  r=c Cs|durdSt|tstdt|jdt|g}t|D]\}}d}i}|dur;t d|ddd|g}n4|d}t d|t d |rW|d t |}n|d |}ddd |g}d |dd |gi}z t j |fi|Wq"t jy}zt|} |dur| |t} ||| fWYd}~q"d}~ww|r|D] \} } td| | qtdddd|DdS)Nz+ubuntu_pro: config should be a dict, not a z%; skipping enabling config parameterszDisabling Pro config for %sproconfigunsetz =REDACTEDzEnabling Pro config %sz\s=set logstringz#Failure enabling/disabling "%s": %sz3Failure enabling/disabling Ubuntu Pro config(s): {}, css|] \}}d|VqdS)z"{}"N)r7).0param_r(r(r) sz!set_pro_config..)rr r%r!r"r=r/r0r#debugresearchescaperProcessExecutionErrorstrreplacerr4r2r7r8) r+ enable_errorsr:r;redacted_key_value subp_kwargs config_cmd key_valueeerr_msgrGr$r(r(r)set_pro_configlsR        rXc CsH|durg}nt|trtd|g}nt|ts%tdt|jg}|r.ddd|g}ndd|g}|ddtg}tdd |z t j |d d h|d Wn%t j yv}zt| |t}d |}t t|t||d}~ww|s{dSgd|}zt j |dd dhd\}} Wnt j y}z tdd ||d}~wwzt|} Wntjy}ztd||d}~wwg} | dgD]} | ddkrtd| dq| | q| r"g} | D].} | d}|dur| |d|d| d}n d| dd| d}t t|qtdd | dS) z=Call ua command line client to attach and/or enable services.NzNubuntu_pro: enable should be a list, not a string; treating as a single enablezIubuntu_pro: enable should be a list, not a %s; skipping enabling servicesr>attachz--no-auto-enablerDzAttaching to Ubuntu Pro. %s r)rcsrCzFailure attaching Ubuntu Pro: )r>enablez --assume-yesz--formatjsonT)capturer\z!Error while enabling service(s): rEzPro response was not json: r9 message_codezservice-already-enabledzService `%s` already enabled.servicezFailure enabling `z`: messagezFailure of type `r!z(Failure enabling Ubuntu Pro service(s): )rrOr#r2listr!r"rrJr8rrNrPrlogexcr%r^loadsJSONDecodeErrorgetr4)tokenr] attach_cmd redacted_cmdrVerr'cmd enable_stdoutrH enable_resprQerrerror_servicesrbr(r(r) configure_pros             rrcloudcCsltdrdSz|jWntyttdwz |jdgWdSty5ttdw)z.Install ubuntu-advantage-tools if not present.r>NzPackage update failedzubuntu-advantage-toolsz(Failed to install ubuntu-advantage-tools) rwhichdistroupdate_package_sources Exceptionrrer#install_packages)rsr(r(r)maybe_install_ua_toolss     ryreturnc Cst|didd}|rdSddlm}ddlm}z!tjddd |}WdW|jS1s5wYW|jS|y[}zt d |t t WYd}~dSd}~ww) NrrFr)UserFacingError)should_auto_attachz6Checking if the instance can be attached to Ubuntu Proalwayslog_modez%Error during `should_auto_attach`: %s) r&rhuaclient.api.exceptionsr{4uaclient.api.u.pro.attach.auto.should_auto_attach.v1r|rTimedr#rJr2ERROR_MSG_SHOULD_AUTO_ATTACH)rrr{r|resultexr(r(r)_should_auto_attach s.       rcCsZ|d}|sd}t|t||d}|r"tdd|t||dddS)Nriz8`ubuntu_pro.token` required in non-Pro Ubuntu instances. enable_betazs>         ' ,d""