File: //lib/python3/dist-packages/uaclient/__pycache__/apt_news.cpython-310.pyc
o
S��c� � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlZd dlm Z m
Z
m Z m Z d dl
mZ d dlmZ d dlmZmZmZmZmZmZ d dlmZ d dlmZm Z G d d
� d
e�Z G d d � d e�Z d
e de fdd�Z dedee de!fdd�Z"de j dee j de!fdd�Z#de de!fdd�Z$d
e de!fd d �Z%ded ee& dee fd d �Z'defd!d"�Z(defd#d$�Z)dede!fd%d&�Z*defd'd(�Z+dS ))� N)�List�Optional)�defaults�messages�system�util)�get_cloud_type)�UAConfig)�
BoolDataValue�
DataObject�DatetimeDataValue�Field�StringDataValue� data_list)� state_files)�ContractExpiryStatus�get_contract_expiry_statusc @ sp e Zd Zedee�dd�edee�dd�ededd�gZdddd�dee e
dee e
dee fdd �Z dS )
�AptNewsMessageSelectors� codenamesF��required�clouds�proN�r r r c C s || _ || _|| _d S �Nr )�selfr r r � r �3/usr/lib/python3/dist-packages/uaclient/apt_news.py�__init__" s
z AptNewsMessageSelectors.__init__)
�__name__�
__module__� __qualname__r
r r r
�fieldsr r �str�boolr r r r r r s � �
�
��r c
@ sn e Zd Zede�ededd�ededd�edee��gZddd�de j de
e j de
e de e fd d
�Z
dS ) �AptNewsMessage�begin�endFr � selectors�linesN)r' r( c C s || _ || _|| _|| _d S r )r& r' r( r) )r r& r' r( r) r r r r 6 s
zAptNewsMessage.__init__)r r r! r
r r r r r" �datetimer r r# r r r r r r% . s" � �����r% �msg�returnc C s* d}| j D ] }|d�|�7 }q|d7 }|S )Nz#
z# {}
)r) �format)r+ �result�liner r r �format_messageD s
r0 �cfgr( c C sx |d u rdS |j d urt�� d |j vrdS |jd ur-t� \}}|d ur&dS ||jvr-dS |jd ur:|j| jkr:dS dS )NT�seriesF)r r �get_platform_infor r r � is_attached)r1 r( �cloud_id�failr r r �do_selectors_applyL s
r7 r& r' c C sV t j �t jj�}|| k rdS | t jdd� }|d u s ||kr!|}n|}||kr)dS dS )NF� )�daysT)r* �now�timezone�utc� timedelta)r& r' r: �one_month_after_begin�
end_to_user r r �do_dates_applyd s r@ �cc C s t �| �d dkS )Nr �C)� unicodedata�category)rA r r r �is_control_charv s rE c C s^ t | j�dk r dS t | j�dkrdS | jD ]}tdd� |D ��r# dS t |�dkr, dS qdS )N� F� c S s g | ]}t |��qS r )rE )�.0rA r r r �
<listcomp>� s z$is_message_valid.<locals>.<listcomp>�M T)�lenr) �any)r+ r/ r r r �is_message_validz s
�rM r c C s� |D ]K}zt �|�}W n ty# } z t�d|� W Y d }~qd }~ww t|�s/t�d|� qt|j|j�s=t�d|� qt | |j
�sJt�d|� q| S d S )Nzmsg failed parsing: %rzmsg not valid: %rzmsg dates don't apply: %rz msg selectors don't apply: %r) r% � from_dict� Exception�logging�debugrM r@ r& r' r7 r( )r1 r �msg_dictr+ �er r r �select_message� s& �� rT c C s@ t �� }t j|| jtjd�}|�� t�|j �}t
j |t j
d�S )N)�destdir)�cls)�apt_pkg�Acquire� AcquireFile� apt_news_urlr � UAC_RUN_PATH�runr � load_file�destfile�json�loadsr �DatetimeAwareJSONDecoder)r1 �acq�
apt_news_file�apt_news_contentsr r r �fetch_aptnews_json� s
� �re c
C s� z,t | �}t| |�dg ��}t�d|� |d ur%t|�}tj�|� W d S tj� � W d S t
yJ } zt�d|� tj� � W Y d }~d S d }~ww )Nr z
using msg: %rz2something went wrong while processing apt_news: %r) re rT �getrP rQ r0 r �apt_news_contents_file�write�deleterO )r1 � news_dictr+ �msg_strrS r r r �fetch_and_process_apt_news� s ��rl c C s� t | �\}}|tjkrtj�tjj|d�� dS |tj kr7t
j | }| j j
�d�}tj�tjj||d�� dS |tjkrEtj�tj� dS dS )zB
:return: True if local news was written, False otherwise
)�remaining_daysTz%d %b %Y)� expired_daterm F)r r �ACTIVE_EXPIRED_SOONr rg rh r � CONTRACT_EXPIRES_SOON_APT_NEWSr- �EXPIRED_GRACE_PERIODr �!CONTRACT_EXPIRY_GRACE_PERIOD_DAYS�machine_token_file�contract_expiry_datetime�strftime�&CONTRACT_EXPIRED_GRACE_PERIOD_APT_NEWS�EXPIRED�CONTRACT_EXPIRED_APT_NEWS)r1 �
expiry_statusrm �grace_period_remaining�exp_dtr r r �local_apt_news� s4
��
����
�r| c C s$ t | �}|st�� t| � d S d S r )r| rW �initrl )r1 �local_news_writtenr r r �update_apt_news� s
�r ),r* r_ rP rC �typingr r rW �uaclientr r r r �uaclient.clouds.identityr �uaclient.configr �uaclient.data_typesr
r r r
r r �uaclient.filesr � uaclient.jobs.update_messagingr r r r% r# r0 r$ r7 r@ rE rM �dictrT re rl r| r r r r r �<module> sT ��
���
���
�
%