o f6 @sVUdZddlZddlZddlZddlmZmZddlmZddl m Z m Z ddl m Z ddlmZddlmZdd lmZd Zed d gZd dgedgeedZeed<eeZdZdZdZde fddZ!de fddZ"d.defddZ#d.ddZ$de fd d!Z%de d"e&fd#d$Z'de fd%d&Z(de fd'd(Z)d)e*d*ede d+ed"df d,d-Z+dS)/z1ubuntu_pro: Configure Ubuntu Pro support servicesN)AnyList)urlparse)subputil)Cloud)Config) MetaSchema) PER_INSTANCEzhttps://ubuntu.com/prozubuntu-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_validationFs2  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)r6).0param_r'r'r( sz!set_pro_config..)rrr$r r!r<r.r/r"debugresearchescaperProcessExecutionErrorstrreplacerr3r1r6r7) r* enable_errorsr9r:redacted_key_value subp_kwargs config_cmd key_valueeerr_msgrFr#r'r'r(set_pro_configmsR        rWc 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-enablerCzAttaching to Ubuntu Pro. %s r)rcsrBzFailure attaching Ubuntu Pro: )r=enablez --assume-yesz--formatjsonT)capturer[z!Error while enabling service(s): rDzPro response was not json: r8 message_codezservice-already-enabledzService `%s` already enabled.servicezFailure enabling `z`: messagezFailure of type `r z(Failure enabling Ubuntu Pro service(s): )rrNr"r1listr r!rrIr7rrMrOrlogexcr$r]loadsJSONDecodeErrorgetr3)tokenr\ attach_cmd redacted_cmdrUerrr&cmd enable_stdoutrG enable_resprPerror_servicesrar'r'r( configure_pros             rpcloudcCsltdrdSz|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 Exceptionrrdr"install_packages)rqr'r'r(maybe_install_ua_toolss     rwreturnc Cst|didd}|rdSddlm}ddlm}z tjtj d|d}W|jS|yG}zt d |t t WYd}~dSd}~ww) NrrFr)UserFacingError)should_auto_attachz6Checking if the instance can be attached to Ubuntu Pro)logfuncr&funcz%Error during `should_auto_attach`: %s) r%rguaclient.api.exceptionsry4uaclient.api.u.pro.attach.auto.should_auto_attach.v1rzrlog_timer"rIr1ERROR_MSG_SHOULD_AUTO_ATTACH)rrryrzresultexr'r'r(_should_auto_attach s(     rcCsZ|d}|sd}t|t||d}|r"tdd|t||dddS)Nrhz8`ubuntu_pro.token` required in non-Pro Ubuntu instances. enable_betaz|durFt d YdSYdS|yd}zd |j } t | t| |d}~ww) Nr)AlreadyAttachedErrorry)FullAutoAttachOptionsfull_auto_attachr\r)r\rzAttaching to Ubuntu Prooptions)r{r&r|kwargszThe instance is already attached to Pro. Leaving enabled services untouched. Ignoring config directives ubuntu_pro: enable and enable_betaz!Error during `full_auto_attach`: )r}rry2uaclient.api.u.pro.attach.auto.full_auto_attach.v1rrrgrrr"rIr1r&r#r$) rrryrrr\rrrr&r'r'r( _auto_attach8s6       rnamecfgargscCs2d}tt|}|r)t|dkrtdd|tdd|||d}d|vr;|r7td|d|d}|durGtd|dSt |t s]d t |j }t |t|d |vrld }t |t|t|t|d t|td t|rt|dS|dhkst|dSdS)Nr^zLUnable to configure Ubuntu Pro. Multiple deprecated config keys provided: %srDz^Deprecated configuration key(s) provided: %s. Expected "ubuntu_pro"; will attempt to continue.rrz;Ignoring deprecated key %s and preferring ubuntu_pro configz=Skipping module named %s, no 'ubuntu_pro' configuration foundz%'ubuntu_pro' should be a dict, not a commandszPDeprecated configuration "ubuntu-advantage: commands" provided. Expected "token"r>zETo discover more log info, please check /var/log/ubuntu-advantage.logr)rcDEPRECATED_KEYS intersectionlenr$r7r"r1rIrrr r!r#rwrWrgr)rrkeysr)rrrqrr deprecatedr&r'r'r(handle^sd       r)N),__doc__r]loggingrJtypingrr urllib.parser cloudinitrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemar cloudinit.settingsr PRO_URLrArrcr__annotations__ getLoggerr!r"rrr0rr)r<rWrprwr%rrrrNrr'r'r'r(s@         ' ,d"&