x[h6 rUdZddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZed d gZd d gedgeezdZeed<ej4eZdZdZdZde fdZ!de fdZ"d$defdZ#d$dZ$defdZ%de de&fdZ'de fdZ(de fdZ)d e*d!eded"eddf d#Z+y)%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_sectioncLd|vry|d}t|ts8dt|j}tj |t |d|vry|d}t|ts8dt|j}tj |t |y)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)rrmsgr s @/usr/lib/python3/dist-packages/cloudinit/config/cc_ubuntu_pro.pyvalidate_schema_featuresr+*s$:&H h %X''( *  #3H,"#89 )4 0/099: <  #3 1 pro_configcg}t|jD]U\}}|tvrtj d|%|( t |}|j dvr|jd|W|r)tdjdj|y#ttf$r|jd|YwxYw)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_urls r*supplemental_schema_validationr@EsFZ--/0A U , , KKF   ]  A!%J  (99 CC5IA& 3 : :499V;L M   + A MM9#? @ As-B  #CCcZ|yt|ts"tdt|jdt |g}t |jD]\}}d}i}|tjd|ddd|g}nd|d}tjd|tjd |r|d tj|}n|d |}ddd |g}d |dd |gzi} tj|fi||rQ|D]\} } tj'd| | tdj)dj+d|Dy#tj$rA}t|} || j!|t"} |j%|| fYd}~Id}~wwxYw)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): {}, c3DK|]\}}dj|yw)z"{}"N)r:).0param_s r* z!set_pro_config..sM85!&--.Ms )r!r"r'r#r$r@r2r3r%debugresearchescaperProcessExecutionErrorstrreplacerr7r5r:r;) r- enable_errorsr=r>redacted_key_value subp_kwargs config_cmd key_valueeerr_msgrLr&s r*set_pro_configr]ls j$ 'z"++,-3 3  #:.MZ--/01 U! = II3S 97C8J$'5 !2  II.0B Cyy&"e1RYYu%5$67 "e1UG, 5)u M N A H H M}MM    )) 1!fG!-!//%:  #w 0 0  1s*EF*)6F%%F*c0|g}ndt|trtjd|g}n;t|ts+tjdt |j g}|rddd|g}ndd|g}|ddtgz}tjdd j| tj|d d h| |sygd|z} tj|dd dh\}} t!j"|} g} | j'dgD]5} | ddk(rtjd| d%| j)| 7| r~g} | D]Z} | j'd}|| j)|d|d| d}nd| dd| d}tjt|\tddj| zy#tj$rO}t|j|t}d |}tjt|t||d}~wwxYw#tj$r#}tddj|z|d}~wwxYw#t j$$r}td||d}~wwxYw) 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 servicesrBattachz--no-auto-enablerHzAttaching to Ubuntu Pro. %s r)rcsrGzFailure attaching Ubuntu Pro: )rBenablez --assume-yesz--formatjsonT)capturerbz!Error while enabling service(s): rIzPro response was not json: r< message_codezservice-already-enabledzService `%s` already enabled.servicezFailure enabling `z`: messagezFailure of type `r#z(Failure enabling Ubuntu Pro service(s): )r!rTr%r5listr#r$rrOr;rrSrUrlogexcr'rdloadsJSONDecodeErrorgetr7)tokenrc attach_cmd redacted_cmdr[err)cmd enable_stdoutrM enable_resprVerrerror_servicesrhs r* configure_prorxs ~ FC   5   %  0 L ! ! X'95A Xu- cr?hZ/LII+SXXl-CD' *1a&LA  ?& HC99S$QFC q jj/ <!#Mx," ~ "; ; II5s9~ F S! " $&  "Cggi(G"%%g.*7)3s9~6FG)#f+c#i.9IJ KKS ! " 6ii' (  o  % %' V^^E8 ,/t4 C3Q& '  % % /$))F2C C     )- 9  sI+GH5.I.H2#A H--H25I+I&&I+.JJJcloudc@tjdry |jj |jjdgy#t$rt j tdwxYw#t$rt j tdwxYw)z.Install ubuntu-advantage-tools if not present.rBNzPackage update failedzubuntu-advantage-toolsz(Failed to install ubuntu-advantage-tools) rwhichdistroupdate_package_sources Exceptionrrkr%install_packages)rys r*maybe_install_ua_toolsrs zz% ++- %%'?&@A  C01   CCD sAA8%A58%Breturnct|jdijdd}|ryddlm}ddlm} t jdd 5|}ddd|j S#1swYj SxYw#|$r9}tjd |tjtYd}~yd}~wwxYw) Nrr Fr)UserFacingError)should_auto_attachz6Checking if the instance can be attached to Ubuntu Proalwayslog_modez%Error during `should_auto_attach`: %s) r(rnuaclient.api.exceptionsr4uaclient.api.u.pro.attach.auto.should_auto_attach.v1rrTimedr%rOr5ERROR_MSG_SHOULD_AUTO_ATTACH)rr rrresultexs r*_should_auto_attachr s B'++,A5I8     D  *()F  *  $ $$ *  $ $$  92> 01s5BA/B/B4BBC /B??Cc|jd}|s"d}tj|t||jd}|r%tj ddj |t ||jdy)Nroz8`ubuntu_pro.token` required in non-Pro Ubuntu instances. enable_betazrs&8  !--!#.+)+=>? z ,~_0EE jg! & $ 6$ t$ N) s) Xa H %  %T%d%8 A A(d(DBB6B%BtBBr,