U M[^-@sddlZddlZddlZddlZz$ddlmZmZmZmZm Z m Z Wne k rXYnXddl m Z ddl mZddlmZddl mZddl mZddlmZd ZGd d d ejZdS) N)AnyDictListOptionalTupleUnion)apt) exceptions)base)status)util)ApplicationStatusz-32768cseZdZdZdZdZeddddZeeddd Z iZ ed dd d Z ee j edd dZddeedddZd#ddZddddZddddZd$ddeedfdd Zdddd Zd!d"ZZS)%RepoEntitlementz3/etc/apt/sources.list.d/ubuntu-{name}-{series}.listz-/etc/apt/preferences.d/ubuntu-{name}-{series}NzUnion[int, str, None])returncCsdSNselfrr|ztdddg|jWntjk r8YnXdS)z4Clean up the entitlement without checks or messagingrremoverN)remove_apt_configr ZsubprZProcessExecutionErrorrrrrr(]s zRepoEntitlement._cleanupzTuple[ApplicationStatus, str]cCs|jj|ji}|didi}|d}|sFtjd|jfS|d\}}t ddgt j }t d||}|r|d tkrtjd |jfStjd |jfS) N entitlement directivesaptURLz${} does not have an aptURL directivez://z apt-cachepolicyz(?P(-)?\d+) {}/ubuntuZpinz {} is activez{} is not configured)cfg entitlementsr$namer DISABLEDr"rsplitrr%r ZMESSAGE_APT_POLICY_FAILEDresearchgroupAPT_DISABLED_PINZENABLED)rZentitlement_cfgr/repo_urlZprotocolZ repo_pathr1matchrrrapplication_statuses0    z"RepoEntitlement.application_statuszDict[str, Any]) orig_accessdeltas allow_enablerc st|||rdS|\}}|tjjkr0dStd|j| di}| di dr| di}| di d}|rt d} |j j |j| d} t| |||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. Tz5Updating '%s' apt sources list on changed directives.r.r/r0seriesr4rA)superprocess_contract_deltasr=r r r5logginginfor4r$r get_platform_inforepo_list_file_tmplr"rremove_auth_apt_repor-r ) rr>r?r@r=_Zdelta_entitlementZorig_entitlementZold_urlrA repo_filename __class__rrrD~s.      z'RepoEntitlement.process_contract_deltasc Cs.td}|jj|j|d}|jj|j}|ddi}|d}|sft d|j |jj d}|d}|st d |j|d }|st |j|d }|st d |j|jj|jr@|jst d |jtjj|j d|jj|j|d} |jdkr(t| ||j|jntj| r@t| g} tjtjs^| dtjtjsx| d| rtdd| zt dddg| tj!Wn"t j k r|"YnXt#|||||j$ttj%zt ddgtj&Wn"t j k r(|"YnXdS)zSetup apt config based on the resourceToken and directives. :raise UserFacingError: on failure to setup any aspect of this apt configuration rArBr.r/Z resourceTokenzHNo specific resourceToken present. Using machine token as %s credentialsZ machineTokenaptKeyzs$