OwlCyberSecurity - MANAGER
Edit File: update_messaging.cpython-38.pyc
U Uٵg� � @ 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