o K&b4@shdZddlZddlZddlZddlmZmZddlmZm Z m Z m Z ddl m Z mZmZmZmZmZmZmZmZddlmZejGdddejZejGd d d ejZd ejd eeeffd dZdedefddZ dedeefddZ!d ejde j"j#dedededededefddZ$d ejd ed dfd!d"Z%d ejd ed dfd#d$Z&d ejd e'fd%d&Z(dS)'a1 Update messaging text for use in MOTD and APT custom Ubuntu Advantage messages. Messaging files will be emitted to /var/lib/ubuntu-advantage/message-* which will be sourced by apt-hook/hook.cc and various /etc/update-motd.d/ hooks to present updated text about Ubuntu Advantage service and token state. N)ListTuple)configdefaults entitlementsutil) ANNOUNCE_ESM_TMPL!CONTRACT_EXPIRED_APT_NO_PKGS_TMPLCONTRACT_EXPIRED_APT_PKGS_TMPL"CONTRACT_EXPIRED_GRACE_PERIOD_TMPLCONTRACT_EXPIRED_MOTD_PKGS_TMPLCONTRACT_EXPIRED_SOON_TMPLDISABLED_APT_PKGS_TMPLDISABLED_MOTD_NO_PKGS_TMPLUBUNTU_NO_WARRANTY)ApplicationStatusc@s eZdZdZdZdZdZdZdS)ContractExpiryStatusrN)__name__ __module__ __qualname__NONEACTIVEACTIVE_EXPIRED_SOONEXPIRED_GRACE_PERIODEXPIREDrr@/usr/lib/python3/dist-packages/uaclient/jobs/update_messaging.pyrs rc@s<eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd S)ExternalMessagezmotd-no-packages-apps.tmplzmotd-no-packages-infra.tmplzmotd-packages-apps.tmplzmotd-packages-infra.tmplz$apt-pre-invoke-no-packages-apps.tmplz%apt-pre-invoke-no-packages-infra.tmplz!apt-pre-invoke-packages-apps.tmplz"apt-pre-invoke-packages-infra.tmplz!apt-pre-invoke-esm-service-statuszmotd-esm-service-statuszmotd-esm-announcezubuntu-no-warrantyN)rrrMOTD_APPS_NO_PKGSMOTD_INFRA_NO_PKGSMOTD_APPS_PKGSMOTD_INFRA_PKGSAPT_PRE_INVOKE_APPS_NO_PKGSAPT_PRE_INVOKE_INFRA_NO_PKGSAPT_PRE_INVOKE_APPS_PKGSAPT_PRE_INVOKE_INFRA_PKGSZAPT_PRE_INVOKE_SERVICE_STATUSZMOTD_ESM_SERVICE_STATUS ESM_ANNOUNCErrrrr r!'sr!cfgreturncCs|jstjdfStj}tj}|j}d|kr|kr"nntj|fS| |kr-dkr4nntj|fS|| kr>tj |fStj |fS)z/Return a tuple [ContractExpiryStatus, num_days]r) Z is_attachedrrr!CONTRACT_EXPIRY_GRACE_PERIOD_DAYSCONTRACT_EXPIRY_PENDING_DAYSZcontract_remaining_daysrrrr)r+Z grace_periodZpending_expiryremaining_daysrrr get_contract_expiry_status7s      r0msg tmpl_filecCsB|r t||dSt||drt|dddSdS)zhWrite a template to tmpl_file. When msg is empty, remove both tmpl_file and the generated msg. .tmplN)r write_file remove_fileendswithreplace)r1r2rrr _write_template_or_removeJs   r9msg_dirmsg_template_namescCs"|D] }tdtj||qdS)Nr4)r9ospathjoin)r:r;namerrr _remove_msg_templatesWsr@ent expiry_statusr/ pkgs_file no_pkgs_filemotd_pkgs_filemotd_no_pkgs_filecCsd}} } } |jdd} d| } d| }t}t|d}|rC|jdkrC|d}tjj|d dd }d j|d }nd}tj }| d t j kr|t jkrgtj|j|tjd}|} } } n`|t jkrtj|}tj|j|jd|tjd}|} } } n@|t jkrtj| ||j|jtjd}tj|jtjd} | } tj|j| tjd} n|t jkrtj|j| |||d}tj|j|d} tj|j d}t!| tj||t!|tj||t!| tj||t!| tj||dS)aWrite any related template content for an ESM service. If no content is applicable for the current service state, remove all service-related template files. :param cfg: UAConfig instance for this environment. :param ent: entitlements.base.UAEntitlement, :param expiry_status: Current ContractExpiryStatus enum for attached VM. :param remaining_days: Int remaining days on contrat, negative when expired. :param *_file: template file names to write. r4-_z{{{}_PKG_COUNT}}z{{{}_PACKAGES}}series esm-infrarelease.Zhyphenatedreleasezfor Ubuntu {release} )rKr)titler/urlz%d %b %Y)rNZ expired_dater/rO)pkg_num pkg_namesrNr?rO)rNrO)rNrPrO)rNrPrQ eol_releaserOmessagesN)"r?upperr8formatrget_platform_info is_active_esmrEOL_UA_URL_TMPL BASE_ESM_URLapplication_statusrENABLEDrrr rNZ BASE_UA_URLrr-r Zcontract_expiry_datetimestrftimerr r r rrr<r=r>data_dirr9)r+rArBr/rCrDrErFZpkgs_msgZ no_pkgs_msgZ motd_pkgs_msgZmotd_no_pkgs_msgZ tmpl_prefixZtmpl_pkg_count_varZtmpl_pkg_names_var platform_inforWrK ua_esm_urlrRZgrace_period_remainingr:rrr _write_esm_service_msg_templates]s        r`rIc Cstjj}tjj}tjj}tjj}tjj}tjj}tjj}tj j} tj j} t j |jd} td} | |} tj|jdd}t|pC| j }td}||}t|\}}tj}d}d}t|rd}|tjtjfvrlt }|d|kryd }t }n|tjkrd }t |t j | | |s|d kr|}|rt!||||||| |nt"t j |jd||| |gd |rt!|| ||||||d St"t j |jd||||gd d S) zWrite messaging templates about available esm packages. :param cfg: UAConfig instance for this environment. :param series: string of Ubuntu release series: 'xenial'. rSesm-appsfeatures.allow_betarZ path_to_valuerJFr4rTtrusty)r:r;N)#r!r&valuer(r'r)r"r$r#r%rr<r=r>r]rentitlement_factoryris_config_value_truer+boolis_betar0rr[rWrrrrZrr.r9r`r@)r+rIZapps_no_pkg_fileZ apps_pkg_fileZinfra_no_pkg_fileZinfra_pkg_fileZmotd_apps_no_pkg_fileZmotd_apps_pkg_fileZmotd_infra_no_pkg_fileZmotd_infra_pkg_fileZno_warranty_filer:apps_cls apps_instconfig_allow_betaZ apps_validZ infra_clsZ infra_instrBr/enabled_statusZ msg_esm_appsZ msg_esm_infraZno_warranty_msgrrr write_apt_and_motd_templatess          rnc Cstd}||}tj}|d|k}tj|jdd}t|p"|j }t j |j d}t j |tjj} t} t| d} | rPtjj| ddd d } ntj} t|d k||grjt| d tj| d dSt| dS)a Write human-readable messages if ESM is offered on this LTS release. Do not write ESM announcements on trusty, esm-apps is enable or beta. :param cfg: UAConfig instance for this environment. :param series: string of Ubuntu release series: 'xenial'. rarrbrcrSrIrKrLrGrMrd )rON)rrfrr[rZrrgr+rhrir<r=r>r]r!r*rerVrWrrXrUr8rYallr5rr6) r+rIrjrkrmZapps_not_enabledrlZ apps_not_betar:Z esm_news_filer^rWr_rrr write_esm_announcement_messages, rqcCstdtj|jd}tj|st|t d}t |sEt D]}tj||j }t ||drBt |ddq%dSt||t||tddgdS) auEmit templates and human-readable status messages in msg_dir. These structured messages will be sourced by both /etc/update.motd.d and APT UA-configured hooks. APT hook content will orginate from apt-hook/hook.cc Call esm-apt-hook process-templates to render final human-readable messages. :param cfg: UAConfig instance for this environment. z&Updating UA messages for APT and MOTD.rSrIr3r4Tz&/usr/lib/ubuntu-advantage/apt-esm-hookzprocess-templates)loggingdebugr<r=r>r]existsmakedirsrrVZis_ltsr!rer6r7r8rqrnZsubp)r+r:rIZmsg_enumZmsg_pathrrr update_apt_and_motd_messagesBs"         rv))__doc__enumrrr<typingrrZuaclientrrrrZuaclient.messagesrr r r r r rrrZuaclient.statusruniqueEnumrr!ZUAConfigintr0strr9r@baseZ UAEntitlementr`rnrqrhrvrrrr sN,      c_#