Your IP : 216.73.216.52
U
8�-d� � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
m Z m Z ddl
mZ ddlmZ ddlmZmZ dd lmZ d
Zd ZejG d d
� d
ej��Zed�dd�Z eeee f d�dd�Z ee d�dd�Z dd� Z!dS )a%
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)�exists)�Tuple)�contract�defaults�messages�system)�_updates)�UAConfig)�ESMAppsEntitlement�ESMInfraEntitlement)�ApplicationStatuszmotd-contract-statusz6/usr/lib/update-notifier/update-motd-updates-availablec @ 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
)�cfgc C s� | j }|�dd�}|�di ��di ��dd �}t�| �}|�||�}|�di ��di ��dd �}|d k r�|| jjkr�||d d d< | j�|� d S )NZ machineToken� ZmachineTokenInfoZ contractInfo�idZ effectiveTo)�
machine_token�getr ZUAContractClientZget_updated_contract_info�machine_token_file�contract_expiry_datetime�write)r Z
orig_tokenr Z contract_idZcontract_clientZrespZ resp_expiryr r r �update_contract_expiry& s@ � ��
� � ���
��
�r$ )r �returnc C s� | j stjdfS tj}tj}| jj}|dkrBt� d� tj
| fS d| krV|krdn n
tj |fS | | krzdk r�n n
tj |fS || k r�tj
|fS tj
|fS )z/Return a tuple [ContractExpiryStatus, num_days]r Nz:contract effectiveTo date is null - assuming it is expired)� is_attachedr
r r �!CONTRACT_EXPIRY_GRACE_PERIOD_DAYSZ CONTRACT_EXPIRY_PENDING_DAYSr! Zcontract_remaining_days�loggingZwarningr r r r )r Z grace_periodZpending_expiry�remaining_daysr r r �get_contract_expiry_statusA s"
�
r* c C s� | j s
dS t�d� tj�| jdt�}t| �\}}|t j
t j t j fkrZt
| � t| �\}}|t jt jfkrxt�|� �n&|t j
kr�t�|tjj|d�� �n|t j kr�tj| }| jj}|dkr�d}n
|�d�}t�|tjj||d�� n�|t j k�r�d }d
}t� � �r6t | �� � \} }
| t j k�rld }t!| �j"j#}n6t�$� �rlt%| �� � \} }
| t j k�rld }t!| �j"j&}|d
k�r�t�|tj'� nt�|tj(j||d
�� dS )z�Emit human-readable status message used by motd.
Used by /etc/update.motd.d/91-contract-ua-esm-status
:param cfg: UAConfig instance for this environment.
Fz&Updating Ubuntu Pro messages for MOTD.r )r) NZUnknownz%d %b %Y)Z expired_dater) zn/ar z esm-infrazesm-apps)�pkg_num�serviceT))r& r( �debug�os�path�joinZdata_dir� MOTD_CONTRACT_STATUS_FILE_NAMEr* r
r r r r$ r r r Zensure_file_absentZ
write_filer ZCONTRACT_EXPIRES_SOON_MOTD�formatr r' r! r" �strftimeZ"CONTRACT_EXPIRED_GRACE_PERIOD_MOTDZ is_current_series_active_esmr Zapplication_statusr ZENABLED� api_u_pro_packages_updates_v1ZsummaryZnum_esm_infra_updatesZis_current_series_ltsr
Znum_esm_apps_updatesZ CONTRACT_EXPIRED_MOTD_NO_PKGSZCONTRACT_EXPIRED_MOTD_PKGS) r Z motd_contract_status_msg_pathZ
expiry_statusr) Zgrace_period_remainingZexp_dtZ
exp_dt_strr, r+ Zesm_infra_status�_Zesm_apps_statusr r r �update_motd_messages\ s�
� � �
��
�
��
�
�
���r6 c
C sL t t�rHzt�tdg� W n, tk
rF } zt�| � W 5 d } ~ X Y nX d S )Nz--force)r �UPDATE_NOTIFIER_MOTD_SCRIPTr Zsubp� Exceptionr( Z exception)�excr r r � refresh_motd� s
r: )"�__doc__�enumr( r. �os.pathr �typingr Zuaclientr r r r Z&uaclient.api.u.pro.packages.updates.v1r r4 Zuaclient.configr Zuaclient.entitlementsr
r Z(uaclient.entitlements.entitlement_statusr r1 r7 �unique�Enumr
r$ �intr* �boolr6 r: r r r r �<module> s* �
� X