o K&b<@sddlZddlZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z mZmZmZmZmZmZddlmZddlmZdZeZGdddejZdS) N)AnyDictListOptionalTupleUnion)aptcontract event_logger exceptionsmessagesstatusutil)base)ApplicationStatusz-32768c sbeZdZdZdZdZdZedee e dffddZ ede fdd Z edee fd d Zde fd d Zeejde fddZd+de de fddZd+ddZd+de ddfddZdeeeejffddZddZ d+dee efdee efde de ffdd Z  d,d!ee d"e d#e ddfd$d%Z!d+de ddfd&d'Z" d-d(e de fd)d*Z#Z$S).RepoEntitlementz*/etc/apt/sources.list.d/ubuntu-{name}.listz$/etc/apt/preferences.d/ubuntu-{name}NFreturncCdSNselfrr(-)?\d+) {}/ubuntuZpin)rr r!r"rDISABLEDr ZNO_APT_URL_FOR_SERVICEr1r,splitrZrun_apt_commandZAPT_POLICY_FAILEDmsgresearchgroupAPT_DISABLED_PINZENABLEDZSERVICE_IS_ACTIVEZSERVICE_NOT_CONFIGURED)rZentitlement_cfgrrepo_urlZprotocolZ repo_pathr:matchrrrapplication_statusds.    z"RepoEntitlement.application_statuscCsP|jj|jd}tddt|dDrdS|sdSt|t|vS)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. r"css|]}|dVqdS)#N) startswith).0linerrr s  z. FT) repo_list_file_tmplr1r"allrZ load_filestripr<bool)rZapt_urlZapt_filerrr_check_apt_url_is_appliedsz)RepoEntitlement._check_apt_url_is_applied orig_accessdeltas allow_enablecs t|||r dS|di}|di}|d}|d}|jd}|r/|r/|} n|\} } | tjj kr=dS| |spt d|j |di} | did} | rh|jj|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. Trrr9rz status-cacheFz5Updating '%s' apt sources list on changed directives.rEz-Installing packages on changed directives: {}, ) package_list)superprocess_contract_deltasr!rZ read_cacheZ"_check_application_status_on_cacherDr rr;rPloggingr0r"rLr1rremove_auth_apt_repor8r.joinr/)rrQrRrSZdelta_entitlementZdelta_directivesZ delta_apt_urlZdelta_packagesZ status_cacherD_Zorig_entitlementZold_url repo_filename __class__rrrWs@            z'RepoEntitlement.process_contract_deltasTrUcleanup_on_failureverbosecCs|s|j}|s dS|jdg}t|sdS|r#tdj|jd|j r/ddi}gd}ni}g}zt j j|jd}t j |||j|dWdStjyW|rV|w) 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_FRONTENDZnoninteractive)z--allow-downgradesz$-o Dpkg::Options::="--force-confdef"z$-o Dpkg::Options::="--force-confold")r$ apt_optionsZ error_msgenv)r$Z messagingr!rZhandle_message_operationsr&r0r1r,apt_noninteractiver ENABLED_FAILEDrrun_apt_install_commandr=r UserFacingErrorr6)rrUr_r`Zmsg_opsrbrar=rrrr/s6   z RepoEntitlement.install_packagescCstd|jjtj}td|jjtj}tj||d|j j |j d}|jj |j }|d di}|d di}| d}|sl|jjd } | d sat|j} | | |j } | ra| d}|sl| }td |j| d } | s|td |j | d} | st|j | d}|std |j |jj|jr|jstd |jtjj |jdj|jj |j d}|jdkrt|| |j|jn t j!"|rt #|g}t j!"tj$s|%dt j!"tj&s|%d|r|st'(d d)|ztj*|dWntjy|+wt,|| |||j-|s+t'(tj.zt/WdStjyB|j+ddw)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 ZhttpZhttps) http_proxy https_proxyrErr obligationsZ resourceTokenZ machineTokenZenableByDefaultzWNo resourceToken present in contract for service %s. Using machine token as credentialsaptKeyztypingrrrrrrZuaclientrr r r r r rZuaclient.entitlementsrZuaclient.statusrrAZget_event_loggerr&Z UAEntitlementrrrrrs $