File: //lib/python3/dist-packages/uaclient/jobs/__pycache__/update_messaging.cpython-310.pyc
o
S��c!8 � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
ddl m Z m
Z
mZmZmZmZ ddlmZ ddlmZ dd lmZmZmZmZmZ d
Z d Z d Z d
Z dZ ej!G dd� dej"��Z#ej!G dd� dej"��Z$dZ%de j&de
e#e'f fdd�Z(edd�de
e)e)f fdd��Z*de)d e)fd d �Z+d e)d e e) fd!d"�Z,de j&d#ej-j.d$e#d%e'd&e)d'e)d(e)d)e)fd*d+�Z/de j&d,e)ddfd-d.�Z0de j&d,e)ddfd/d0�Z1de j&de2fd1d2�Z3d3d4� Z4de j&fd5d6�Z5dS )7a%
Update messaging text for use in MOTD and APT custom Ubuntu Pro 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 Pro service and token state.
� N)� lru_cache)�exists)�List�Tuple)�config�contract�defaults� entitlements�system�util)�identity)�ApplicationStatus)�ANNOUNCE_ESM_APPS_TMPL�'CONTRACT_EXPIRED_MOTD_GRACE_PERIOD_TMPL�"CONTRACT_EXPIRED_MOTD_NO_PKGS_TMPL� CONTRACT_EXPIRED_MOTD_PKGS_TMPL� CONTRACT_EXPIRED_MOTD_SOON_TMPLzhttps://ubuntu.com/16-04z https://ubuntu.com/azure/proz https://ubuntu.com/16-04/azurezhttps://ubuntu.com/aws/prozhttps://ubuntu.com/gcp/proc @ s e Zd ZdZdZdZdZdZdS )�ContractExpiryStatusr � � � � N)�__name__�
__module__� __qualname__�NONE�ACTIVE�ACTIVE_EXPIRED_SOON�EXPIRED_GRACE_PERIOD�EXPIRED� r r �@/usr/lib/python3/dist-packages/uaclient/jobs/update_messaging.pyr " s r c @ s8 e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d 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-announceN)r r r �MOTD_APPS_NO_PKGS�MOTD_INFRA_NO_PKGS�MOTD_APPS_PKGS�MOTD_INFRA_PKGS�APT_PRE_INVOKE_APPS_NO_PKGS� APT_PRE_INVOKE_INFRA_NO_PKGS�APT_PRE_INVOKE_APPS_PKGS�APT_PRE_INVOKE_INFRA_PKGS� APT_PRE_INVOKE_SERVICE_STATUS�MOTD_ESM_SERVICE_STATUS� ESM_ANNOUNCEr r r r! r" , s r" z6/usr/lib/update-notifier/update-motd-updates-available�cfg�returnc C s� | j stjdfS tj}tj}| jj}d| kr |kr#n ntj|fS | | kr.dk r5n ntj |fS || k r?tj
|fS tj |fS )z/Return a tuple [ContractExpiryStatus, num_days]r ) � is_attachedr r r �!CONTRACT_EXPIRY_GRACE_PERIOD_DAYS� CONTRACT_EXPIRY_PENDING_DAYS�machine_token_file�contract_remaining_daysr r r r )r. � grace_period�pending_expiry�remaining_daysr r r! �get_contract_expiry_status@ s
r8 )�maxsizec C s� t �� \} }t�� d }d}d}d}d}| d ur*| �d�}| �d�}| �d�}| }|dk}| d u r9|s9tjdfS | d u s?|rE|rEtdfS |rM|sMtd fS |rU|rUt d
fS |r]|s]t
d fS |re|set d fS tjdfS )
N�seriesF�aws�gce�azure�xenial� z
for 16.04z on Azurez for 16.04 on Azurez on AWSz on GCP) r �get_cloud_typer
�get_platform_info�
startswithr � BASE_UA_URL�XENIAL_ESM_URL�
AZURE_PRO_URL�AZURE_XENIAL_URL� AWS_PRO_URL� GCP_PRO_URL)�cloud�_r: �is_aws�is_gcp�is_azure�non_azure_cloud� is_xenialr r r! �get_contextual_esm_info_urlS s2
rP �msg� tmpl_filec C sB | r
t �|| � dS t �|� |�d�r t �|�dd�� dS dS )zhWrite a template to tmpl_file.
When msg is empty, remove both tmpl_file and the generated msg.
�.tmplr? N)r
�
write_file�ensure_file_absent�endswith�replace)rQ rR r r r! �_write_template_or_removeu s
�rX �msg_dir�msg_template_namesc C s" |D ] }t dtj�| |�� qd S )Nr? )rX �os�path�join)rY rZ �namer r r! �_remove_msg_templates� s �r_ �ent�
expiry_statusr7 � pkgs_file� no_pkgs_file�motd_pkgs_file�motd_no_pkgs_filec C s d } } }
} |j �� �dd�} d�| �}
|�� d tjkrW|tjkr-t j|d�}
|
} n*|tj
krHt j | }| j
j�d�}tj||d�}
|
} n|tjkrWtj|
|j d �}
t} tj�| jd
�}t| tj�||�� t|tj�||�� t| tj�||�� t|
tj�||�� d S ) a� Write 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.
r? �-rJ z{{{}_PKG_COUNT}}r )r7 z%d %b %Y)� expired_dater7 )�pkg_num�service�messagesN)r^ �upperrW �format�application_statusr
�ENABLEDr r r r r r1 r3 �contract_expiry_datetime�strftimer r r r r[ r\ r] �data_dirrX )r. r` ra r7 rb rc rd re �pkgs_msg� no_pkgs_msg�
motd_pkgs_msg�motd_no_pkgs_msg� tmpl_prefix�tmpl_pkg_count_var�grace_period_remaining�exp_dtrY r r r! � _write_esm_service_msg_templates� sF
�
���
���rz r: c C sR t jj}t jj}t jj}t jj}t jj}t jj}t jj}t j j} t
j � | j
d�}
tj| dd�} | | �} tj| jdd�}
t|
pA| j �}tj| dd�}|| �}t| �\}}tj}d}d}t�|�rr|�� d |krkd }n|tj krrd }|sv|}|r�t | |||||| |� n
t |
||| |gd
� |r�t | | ||||||� d S t |
||||gd
� d S ) z�Write messaging templates about available esm packages.
:param cfg: UAConfig instance for this environment.
:param series: string of Ubuntu release series: 'xenial'.
rj �esm-apps�r. r^ �features.allow_beta�r �
path_to_valuez esm-infraFr T)rY rZ N) r" r' �valuer) r( r* r# r% r$ r& r[ r\ r] rq r �entitlement_factoryr �is_config_value_truer. �bool�is_betar8 r
rn r
�
is_active_esmrm r r2 rz r_ )r. r: �apps_no_pkg_file�
apps_pkg_file�infra_no_pkg_file�infra_pkg_file�motd_apps_no_pkg_file�motd_apps_pkg_file�motd_infra_no_pkg_file�motd_infra_pkg_filerY �apps_cls� apps_inst�config_allow_beta�
apps_valid� infra_cls�
infra_instra r7 �enabled_status� msg_esm_apps�
msg_esm_infrar r r! � write_apt_and_motd_templates� s� �
� ��
� �
�r� c C s� t j| dd�}|| �}tj}|�� d |k}tj| jdd�}t|p$|j �}t
j � | j
d�}t
j � |tjj�} |rN|rNt� \}
} t�| dtj|
d� � d S t�| � d S )
a Write human-readable messages if ESM is offered on this LTS release.
Do not write ESM announcements if esm-apps is enabled or beta.
:param cfg: UAConfig instance for this environment.
:param series: string of Ubuntu release series: 'xenial'.
r{ r| r r} r~ rj �
)�urlN)r r� r
rn rm r r� r. r� r� r[ r\ r] rq r" r- r� rP r
rT r rl rU ) r. r: r� r� r� �apps_not_enabledr� �
apps_not_betarY �
esm_news_filer� rJ r r r! � write_esm_announcement_message s" �
�r� c
C s t �d� tj�| jd�}tj�|�st�|� t� � d }t�
|�sEt D ] }tj�||j �}t�
|� |�d�rBt�
|�dd�� q%dS t| �\}}|tjtjfvrWt| � t| |� t| |� z t�dg� W dS ty� } zt �dt|�� W Y d }~dS d }~ww )
ac Emit 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 apt-esm-hook to render final human-readable
messages.
:param cfg: UAConfig instance for this environment.
z.Updating Ubuntu Pro messages for APT and MOTD.rj r: rS r? Tz&/usr/lib/ubuntu-advantage/apt-esm-hookz failed to run apt-esm-hook: %sN)�logging�debugr[ r\ r] rq r �makedirsr
rA �is_ltsr" r� rU rV rW r8 r r r �update_contract_expiryr� r� �subp� Exception�str)r. rY r: �msg_enum�msg_pathra rJ �excr r r! � update_apt_and_motd_messages7 s:
� �
���r� c
C s\ t t�r%z t�tdg� W n ty$ } z t�| � W Y d } ~ nd } ~ ww t�g d�� d S )Nz--force)�sudo� systemctl�restartzmotd-news.service)r �UPDATE_NOTIFIER_MOTD_SCRIPTr
r� r� r� � exception)r� r r r! � refresh_motde s ��r� c C s� | j }|�dd�}|�di ��di ��dd �}t�| �}|�||�}|�di ��di ��dd �}|r7t�|�n| jj}| jj|krQ||d d d< | j� |� d S d S )N� machineTokenr? �machineTokenInfo� contractInfo�id� effectiveTo)
�
machine_token�getr �UAContractClient�get_updated_contract_infor �parse_rfc3339_dater3 ro �write)r. �
orig_tokenr� � contract_id�contract_client�resp� resp_expiry�
new_expiryr r r! r� t s2
�
�
� ��
���r� )6�__doc__�enumr� r[ � functoolsr �os.pathr �typingr r �uaclientr r r r r
r �uaclient.cloudsr �(uaclient.entitlements.entitlement_statusr
�uaclient.messagesr r r r r rD rE rF rG rH �unique�Enumr r" r� �UAConfig�intr8 r� rP rX r_ �base�
UAEntitlementrz r� r� r� r� r� r� r r r r! �<module> sj ��
�!
�������
�>T .