File: //usr/lib/python3/dist-packages/uaclient/timer/__pycache__/update_messaging.cpython-38.pyc
U
��Jh� � @ s� d Z ddlZddlZddlmZ ddlmZmZmZm Z m
Z
ddl m Z
ddlmZ ddlmZmZ ddlmZ dd lmZmZ d
Zd Ze�e
�e ��Z ed �d
d�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)�contract�defaults�messages�system�util)�_updates)�_enabled_services)�ContractExpiryStatus� _is_attached)�UAConfig)�
machine_token�noticeszmotd-contract-statusz6/usr/lib/update-notifier/update-motd-updates-available)�cfgc C s� t �| �}|j }|�di ��di ��dd �}t�| �}|�|�dd�|�}|�di ��di ��dd �}|d k r�||jkr�||d d d< |�|� d S )NZmachineTokenInfoZ contractInfo�idZ machineToken� Z effectiveTo)r
�get_machine_token_file�getr ZUAContractClientZget_contract_machine�contract_expiry_datetime�write)r �machine_token_fileZ
orig_tokenZ contract_idZcontract_clientZrespZ resp_expiry� r �A/usr/lib/python3/dist-packages/uaclient/timer/update_messaging.py�update_contract_expiry s@
� ��
� � �����
�r )r �returnc C s t | �}|jsdS t�d� tj�| jdt�}|j }|j
}t � | �}|t
jjt
jjt
jjfkrzt| � t | �}|j }|j
}|t
jjt
jjfkr�t�tjj� t�|� �nj|t
jjkr�t�tjj� t�|t j � |�j |d�d � �n,|t
jjk�rRt�tjj� t j!| }|j"}|dk�r$d}n
|�#d�}t�|t j$� |�j ||d �d � n�|t
jjk�rt�%tjj� d
} d }
d d
� t&| �j'D �} t�(� �r�d| k�r�d} t)| �j*j+}
n$t�,� �r�d| k�r�d} t)| �j*j-}
|
d k�r�t�|t jd � n"t�|t j.� |
�j |
| d�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 )�remaining_daysz
NZUnknownz%d %b %Y)Z expired_dater zn/ar c S s h | ]
}|j �qS r )�name)�.0�sr r r � <setcomp>� s z'update_motd_messages.<locals>.<setcomp>z esm-infrazesm-apps)�pkg_num�serviceT)/r Z is_attached�LOG�info�os�path�joinZdata_dir� MOTD_CONTRACT_STATUS_FILE_NAMEZcontract_statusZcontract_remaining_daysr
r r
ZACTIVE_EXPIRED_SOON�valueZEXPIRED_GRACE_PERIODZEXPIREDr ZACTIVEZNONEr �removeZNoticeZCONTRACT_EXPIREDr Zensure_file_absentZ
write_filer ZCONTRACT_EXPIRES_SOONZ pluralize�formatr Z!CONTRACT_EXPIRY_GRACE_PERIOD_DAYSr �strftimeZ CONTRACT_EXPIRED_GRACE_PERIOD�addr Zenabled_servicesZ is_current_series_active_esm� api_u_pro_packages_updates_v1ZsummaryZnum_esm_infra_updatesZis_current_series_ltsZnum_esm_apps_updatesZCONTRACT_EXPIRED_WITH_PKGS) r Zis_attached_infoZ motd_contract_status_msg_pathZ
expiry_statusr r Zgrace_period_remainingZexp_dtZ
exp_dt_strr! r Zenabled_services_namesr r r �update_motd_messages; s�
�
�� ����
����
�
�
�
� ��� r. 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
r2 )"�__doc__Zloggingr$ �os.pathr Zuaclientr r r r r Z&uaclient.api.u.pro.packages.updates.v1r r- Z-uaclient.api.u.pro.status.enabled_services.v1r Z(uaclient.api.u.pro.status.is_attached.v1r
r Zuaclient.configr Zuaclient.filesr
r r' r/ Z getLoggerZ replace_top_level_logger_name�__name__r" r �boolr. r2 r r r r �<module> s �i