U 2d>@sddlZddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZmZmZmZmZmZmZmZddlmZddlmZeZeeeZGdddej Z!dS) N)exists)AnyDictListOptionalTupleUnion)aptcontract event_logger exceptionshttpmessagessystemutil)base)ApplicationStatuscs,eZdZdZdZdZdZdZdZe e e e dfdddZ e ee dd d Zedd d Ze eje dd dZd$eedddZd%ddZeeeejfdddZddZd&ee efee efeedfdd Z d'eee eeddddZ!d(edddd Z"d)eed!d"d#Z#Z$S)*RepoEntitlementz*/etc/apt/sources.list.d/ubuntu-{name}.listz$/etc/apt/preferences.d/ubuntu-{name}z {}/ubuntuNF)returncCsdSNselfrrs  z!RepoEntitlement._check_for_rebootcCsdSrrrrrr repo_key_fileFszRepoEntitlement.repo_key_file)silentrcCs|j|d|jrb|jrbt|jdkr6dd|jnd}td|tt j j|j dn*| tt j j|j d|jdd d S) zEnable specific entitlement. @return: True on success, False otherwise. @raises: UserFacingError on failure to install suggested packages r&rz:  zSkipping installing packages{}titleinstall)Z operationT)setup_apt_configZsupports_access_onlyZ access_onlylenr!joinr#infoformatrZACCESS_ENABLED_TMPLr+install_packagesZ ENABLED_TMPLZ_check_for_reboot_msg)rr&Z packages_strrrr_perform_enableKs    zRepoEntitlement._perform_enablecCs"t|dr||j|ddS)Nremove_packagesr'T)hasattrr4remove_apt_config)rr&rrr_perform_disableas  z RepoEntitlement._perform_disablecCstjtjj|jdf}|j}|didi}|d}|sTtjtjj|jdfSt j tj j d}t |j||}|rtjtjj|jdf}|jr|jD]*}t |stjtjj|j|dfSq|S)Nr*rraptURL) error_msg)servicepackage)rDISABLEDrZSERVICE_NOT_CONFIGUREDr1r+rrZNO_APT_URL_FOR_SERVICEr Zget_apt_cache_policyZAPT_POLICY_FAILEDmsgresearch repo_url_tmplZENABLEDZSERVICE_IS_ACTIVEcheck_packages_are_installedr!Z is_installedZ SERVICE_DISABLED_MISSING_PACKAGEname)rZcurrent_statusrrrepo_urlZpolicymatchr;rrrapplication_statusgs>     z"RepoEntitlement.application_statuscCsP|jj|jd}tddt|dDr6dS|s>dSt|t|kS)zCheck if apt url delta should be applied. :param apt_url: string containing the apt url to be used. :return: False if apt url is already found on the source file. True otherwise. rBcss|]}|dVqdS)#N) startswith).0linerrr sz. FT) repo_list_file_tmplr1rBallrZ load_filestripsplitbool)rZapt_urlZapt_filerrr_check_apt_url_is_appliedsz)RepoEntitlement._check_apt_url_is_applied) orig_accessdeltas allow_enablercs t|||rdS|di}|di}|d}|d}|jd}|r^|r^|} n |\} } | tjkrxdS| |st d|j |di} | did} | r|j j|j d } t| | |||rt d d ||j|d dS) a1Process any contract access deltas for this entitlement. :param orig_access: Dictionary containing the original resourceEntitlement access details. :param deltas: Dictionary which contains only the changed access keys and values. :param allow_enable: Boolean set True if allowed to perform the enable operation. When False, a message will be logged to inform the user about the recommended enabled service. :return: True when delta operations are processed; False when noop. Trrr8rz status-cacheFz5Updating '%s' apt sources list on changed directives.rFz-Installing packages on changed directives: {}, ) package_list)superprocess_contract_deltasrcfgZ read_cacheZ"_check_application_status_on_cacherErr<rRLOGr0rBrMr1r remove_auth_apt_repor6r-r/r2)rrSrTrUZdelta_entitlementZdelta_directivesZ delta_apt_urlZdelta_packagesZ status_cacherE_Zorig_entitlementZold_url repo_filename __class__rrrYs@            z'RepoEntitlement.process_contract_deltasT)rWcleanup_on_failureverbosercCs|s |j}|sdS|jdg}t|s.dS|rFtdj|jd|j r`ddi}ddd g}nd}g}z(t j j|jd}t j |||j|d Wn$tjk r|r|YnXdS) a)Install contract recommended packages for the entitlement. :param package_list: Optional package list to use instead of self.packages. :param cleanup_on_failure: Cleanup apt files if apt install fails. :param verbose: If true, print messages to stdout NZ pre_installzInstalling {title} packagesr*ZDEBIAN_FRONTENDZnoninteractivez--allow-downgradesz$-o Dpkg::Options::="--force-confdef"z$-o Dpkg::Options::="--force-confold")r! apt_optionsr9override_env_vars)r!Z messagingrrZhandle_message_operationsr#r0r1r+apt_noninteractiverENABLED_FAILEDr run_apt_install_commandr=r UserFacingErrorr6)rrWrarbZmsg_opsrdrcr=rrrr2s:  z RepoEntitlement.install_packagescCsd}d}d}|jjs|jjrNtd|jjtj}td|jjtj}tjj }n@|jj s^|jj rtd|jj tj}td|jj tj}tjj }tj |||d|jj|jd}|j}|ddi}|ddi}|d } | s>|jjd } |d s&t|j} | | |j} | r&| d } | s>| } td |j|d } | s`td|j|d}|s|t|j|d}|std|j|jj|jr|jstd|jt j!j|jdj"|j#j|jd}t$|||j|jg}t%tj&s|'dt%tj(s*|'d|r|sLt)*dd+|ztj,|dWn"tjk r~|-YnXt.||j/|| ||j0|st)*t j1z t2Wn&tjk r|j-ddYnXdS)zSetup apt config based on the resourceToken and directives. Also sets up apt proxy if necessary. :raise UserFacingError: on failure to setup any aspect of this apt configuration Nr Zhttps) http_proxy https_proxyZ proxy_scoperFrr obligationsZ resourceTokenZ machineTokenZenableByDefaultzWNo resourceToken present in contract for service %s. Using machine token as credentialsaptKeyz6Ubuntu Pro server provided no aptKey directive for {}.r8Zsuitesz%Empty {} apt suites directive from {}z:Cannot setup apt pin. Empty apt repo origin value '{}'. {}r*zapt-transport-httpszca-certificateszInstalling prerequisites: {}rV)r!F)run_apt_update)3rZZglobal_apt_http_proxyZglobal_apt_https_proxyr Zvalidate_proxyZPROXY_VALIDATION_APT_HTTP_URLZPROXY_VALIDATION_APT_HTTPS_URLr Z AptProxyScopeZGLOBALZua_apt_http_proxyZua_apt_https_proxyZUACLIENTZsetup_apt_proxyrMr1rBrr machine_tokenr ZUAContractClientZget_resource_machine_accessr[Zwarningr+r rhMissingAptURLDirectiveZ contract_urlroriginrrfr=repo_pref_file_tmplZadd_ppa_pinningrZAPT_METHOD_HTTPS_FILEappendZCA_CERTIFICATES_FILEr#r0r/rgr6Zadd_auth_apt_repor@r%APT_UPDATING_LISTSrun_apt_update_command)rr&rirjZscoper^Z resource_cfgrrktokenrnZclientZmachine_accessrlrCZ repo_suitesrepo_pref_fileZprerequisite_pkgsrrrr-s                   z RepoEntitlement.setup_apt_config)rmr&c Cstj}|jj|jd}|jjj|j di}| di}| d}|sXt |jt |||jt |||jr|jj|jd}t||r|sttjt dS)zRemove any repository apt configuration files. :param run_apt_update: If after removing the apt update command after removing the apt files. rFrrr8N)rZget_release_infoseriesrMr1rBrZZmachine_token_fileZ entitlementsrr ror r\r%Zremove_apt_list_filesrrqZensure_file_absentr#r0rrsrt) rrmr&rwr^rZaccess_directivesrCrvrrrr6s&       z!RepoEntitlement.remove_apt_config)F)F)F)NTT)F)TF)%__name__ __module__ __qualname__rMrqr@rprerApropertyrintstrrrr!rQr$abcabstractmethodr%r3r7rrrrZ NamedMessagerErRrrrYr2r-r6 __classcell__rrr_rrsV  (  ?  3r)"r~r Zloggingr>os.pathrtypingrrrrrrZuaclientr r r r r rrrZuaclient.entitlementsrZ(uaclient.entitlements.entitlement_statusrZget_event_loggerr#Z getLoggerZreplace_top_level_logger_namerxr[Z UAEntitlementrrrrrs  (