gK`ddlZddlZddlZddlmZmZddlmZmZm Z ddl m Z m Z m Z mZmZmZmZmZddlmZmZmZmZmZmZmZmZmZddlmZmZddl m!Z!ejDejFe$Z%dd d Z&d Z'd Z(d Z)e jTZ+GddZ,de-fdZ.ddZ/de-fdZ0y)N) lru_cachewraps)AnyDictOptional)apt event_logger exceptionshttpmessagessnapsystemutil) APT_NEWS_URLBASE_CONTRACT_URLBASE_LIVEPATCH_URLBASE_SECURITY_URLBASE_VULNERABILITY_DATA_URLCONFIG_DEFAULTSCONFIG_FIELD_ENVVAR_ALLOWLISTDEFAULT_CONFIG_FILEDEFAULT_DATA_DIR) state_filesuser_config_file) safe_loadnametype)availableResourcesresourceEntitlements_unset) http_proxy https_proxyapt_http_proxyapt_https_proxyua_apt_http_proxyua_apt_https_proxyglobal_apt_http_proxyglobal_apt_https_proxyupdate_messaging_timermetering_timerapt_news apt_news_urlvulnerability_data_url_prefixlxd_guest_attach) contract_urldata_dirfeatureslog_file log_level security_urlr-settings_overrides ua_config livepatch_urlc eZdZdZdZdZ d1deeee fdee jdeeddfd Z e defd Ze defd Ze defd Ze deefd Zej$defdZe deefdZej$defdZe deefdZej$defdZe deefdZej$defdZe defdZej$defdZe eddeefdZej$defdZe eddeefdZej$defdZe defdZej$defdZe defdZej$defd Ze defd!Zej$defd"Ze defd#Zej$defd$Ze defd%Z e j$defd&Z e defd'Z!e!j$defd(Z!e de jDfd)Z#e#j$de jDfd*Z#e d+Z$e d,Z%e defd-Z&e d.Z'd/Z(d0Z)y)2UAConfig)r%r&)r'r()r#r$Ncfg user_configseriesreturncp|rd|_||_d|_n1t|_t |j\|_|_|r||_n$ t j j|_d|jvrOt jji|j j|jdd|_||_y#t$rO}tjd|tjdt j|_Yd}~d}~wwxYw)NzError loading user config)exc_infozUsing default config valuesr6T) optional_type_errors_become_null)cfg_pathr: invalid_keysget_config_path parse_configr;rread ExceptionLOGwarningUserConfigData from_dictto_dictr<)selfr:r;r<es 1/usr/lib/python3/dist-packages/uaclient/config.py__init__zUAConfig.__init__Vs   DMDH $D +-DM*6t}}*E 'DHd' *D  E#3#?#?#D#D#F  $(( "/>>HHG4##++-G+1FG15 I D    E 7! D 9:#3#B#B#D   Es#C D5&AD00D5cB|jjdtS)Nr/)r:getrrMs rOr/zUAConfig.contract_urlyxx||N,=>>cB|jjdtS)Nr4)r:rRrrSs rOr4zUAConfig.security_url}rTrUcB|jjdtS)Nr7)r:rRrrSs rOr7zUAConfig.livepatch_urlsxx||O-?@@rUc.|jjSN)r;r!rSs rOr!zUAConfig.http_proxys***rUvaluecx||j_tjj|jyrY)r;r!rwriterMrZs rOr!zUAConfig.http_proxys*&+#$$**4+;+;6$$**4+;+;Return a dictionary of any features provided in uaclient.conf.r1zDUnexpected uaclient.conf features value. Expected dict, but found %s)r:rR isinstancedictrHrI)rMr1s rOr1zUAConfig.featuresPsC88<< + (D) 3  rUcf dD]S}t||}|tjd|(t|tr|dks>t j |||js |jr,|js |jrt jtjd|jtjtjd|jtjtjd|jtjtjd|jtjtjd|j tj"tjd|j$tj&|js |jrDt)j*|j|jt(j,j.n[|js |jrCt)j*|j|jt(j,j0g}t3j4rt3j6|j |j$|j s#t3j8t2j:s/|j$s4t3j8t2j<r|j?dddl m!}dd l"m#}||jH}tKd |D}|r|jM|j |j$|j s|j9|j:s'|j$s,|j9|j<r|j?d tO|dkDr:d jQ|}tStTjVjY| ddl-m.} |j^t`jbjdk7r,| |jfstStTjhytjjljotkjp|j^y)N)r)r*z1No config set for %s, default value will be used.rkeyrZr httpsr ) livepatch)_enabled_servicesc3@K|]}|jdk(s|yw)rN)r).0ents rO z*UAConfig.process_config..s  sxx;/FC srz, )services) _is_attached) guest_attach)9rrHdebugrintr InvalidPosIntConfigValuer'r(r%r&InvalidProxyCombinationConfigr validate_proxyPROXY_VALIDATION_APT_HTTP_URLPROXY_VALIDATION_APT_HTTPS_URLr!PROXY_VALIDATION_SNAP_HTTP_URLr"PROXY_VALIDATION_SNAP_HTTPS_URLrsetup_apt_proxy AptProxyScopeGLOBALUACLIENTr is_snapd_installedconfigure_snap_proxyget_config_option_valueHTTP_PROXY_OPTIONHTTPS_PROXY_OPTIONappenduaclientr-uaclient.api.u.pro.status.enabled_services.v1renabled_servicesanyconfigure_livepatch_proxylenjoinprintr !PROXY_DETECTED_BUT_NOT_CONFIGUREDrp(uaclient.api.u.pro.status.is_attached.v1rr.rrr is_attached-WARNING_LXD_GUEST_ATTACH_SET_BUT_NOT_ATTACHEDrlxd_pro_config_filer\ LXDProConfig) rMproprZservices_with_proxiesrrrlivepatch_enabledrrs rOprocess_configzUAConfig.process_config_s DD$'E} G s+uqy 99E   & &$*E*E  " "d&=&=::< <    & &  . .    ' '  / /  D**D,N,N     # #  / /  DOOT%H%H   T%%t'K'K   % %)D)D   **++!!((   # #t'>'>   &&''!!**  !#  " " $  % %doot7G7G HOO001G1GH$$001H1HI%,,V4& -T2CC +     / /!1!1 OO55//$$5500&,,[9 $ % )yy!67H ::AA%B  J  ! !%5%H%H%L%L L &22 (HH I  + + 1 1((d6K6KL rUcB|j0t|jD]}tjd|d|jvrUtjdtjdtjdtjdyy)Nz&Ignoring invalid uaclient.conf key: %sr6z)legacy "ua_config" found in uaclient.confzPlease do the following:z 1. run `sudo pro config set field=value` for each field/value pair present under "ua_config" in /etc/ubuntu-advantage/uaclient.confzQ 2. Delete "ua_config" and all sub-fields in /etc/ubuntu-advantage/uaclient.conf)rCsortedrHrIr:)rM invalid_keys rOwarn_about_invalid_keysz UAConfig.warn_about_invalid_keyss    (%d&7&78   ! !  !F?c???c??AsAA+HSM++===,Xc],,===3HSM33===28C=22=s==s #))=3=*=t x}  !!=3="= t   ""=C=#=  ""=C=#= =C==d  =$=!=3%%=s=&=$ __=d==c =#=="2"E"E =&6&I&I== ::!!E#EE  tlrUr9r=cTtjjd}|r|StS)z4Get config path to be used when loading config dict.UA_CONFIG_FILE)osenvironrRr) config_files rOrDrDs$**..!12K rUc`tjt}|s t}tj d|t j j|r-|jttj|i}t jjD]\}}|j}|jds(|dd}|jdr|dd}|j!drTt j j|rttj|}nt#j$|d |vr||i|d <||d |<|t&vs|||<|j|d |vr%t j j)|d |d <d D]4}t+j,||rt#j.||| t1|j3j5t6}|D]}|j9|||fS) aParse known Pro config file Attempt to find configuration in cwd and fallback to DEFAULT_CONFIG_FILE. Any missing configuration keys will be set to CONFIG_DEFAULTS. Values are overridden by any environment variable with prefix 'UA_'. @param config_path: Fullpath to pro configfile. If unspecified, use DEFAULT_CONFIG_FILE. @return: Dict of configuration values. z%Using client configuration file at %sua_N features_ yaml)filepathr1r0)r/r4r)copyrrDrHrrpathexistsupdaterr load_fileritemslower startswithendswithr InvalidFeatureYamlConfigValuer expanduserr is_service_urlInvalidURLConfigValuesetkeys differenceVALID_UA_CONFIG_KEYSpop) config_pathr:env_keysrrZ field_namefeature_field_namerCrs rOrErEs ))O $C %' II5{C ww~~k" 9V--k:;<Hjj&&(- Uiik >>% QRJ$$[1%/^" >>&)ww~~e, )&*:*:5*A B(FF%*S('95&AC O:?C O$6755',$5-6JJxS'',,S_=J/L""3s8,22s#c(K KLsxxz?--.BCL#     rU override_keycfd}|S)aDecorator used to override function return by config settings. To identify if we should override the function return, we check if the config object has the expected override key, we use it has, we will use the key value as the function return. Otherwise we will call the function normally. @param override_key: key to be looked for in the settings_override entry in the config dict. If that key is present, we will return its value as the function return. c2tfd}|S)Nct\}}|jdijt}|tk7r dk(r|dfS|SS)Nr5 cloud_type)rErRUNSET_SETTINGS_OVERRIDE_KEY)r:_value_overridefrs rOnew_fz>apply_config_settings_override..wrapper..new_fAs[!^FC WW%92>BB9N!<<</*D11%%3JrU)r)rrrs` rOwrapperz/apply_config_settings_override..wrapper@s q    rUr)rrs` rOapply_config_settings_overrider3s" NrUrY)1rrr functoolsrrtypingrrrrrr r r r r rruaclient.defaultsrrrrrrrrruaclient.filesrr uaclient.yamlr getLoggerreplace_top_level_logger_namerrHMERGE_ID_KEY_MAPrUA_CONFIGURABLE_KEYSrget_event_loggerrmr9rrDrErrrUrOrs  &&&      9#g:::8DE!"'$  & %%']]@ <~rU