File: //proc/1233/cwd/usr/lib/python3/dist-packages/uaclient/__pycache__/apt_news.cpython-310.pyc
o
nHJe$ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dlZd dl m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZ d dlmZ d dlmZmZ d d lmZmZm Z m Z m Z m Z d d
l m!Z! e�"e
�#e$��Z%G d d � d e�Z&G d
d� de�Z'dedee& de(fdd�Z)de j dee j de(fdd�Z*de+de(fdd�Z,de'de(fd d �Z-ded ee. dee' fd d �Z/edefd!d"��Z0dedee+ fd#d$�Z1dedee+ fd%d&�Z2d'e+de+fd(d)�Z3defd*d+�Z4dS ),� N)�List�Optional)�defaults�messages�system�util)� _is_attached)�ensure_apt_pkg_init)�get_cloud_type)�UAConfig)�ContractExpiryStatus�get_contract_expiry_status)�
BoolDataValue�
DataObject�DatetimeDataValue�Field�StringDataValue� data_list)� state_filesc @ 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 8 s
zAptNewsMessage.__init__)r! r" r# r r r r r r$ �datetimer r r% r r r r r r' 0 s" � �����r' �cfgr* �returnc C sz |d u rdS |j d urt�� j|j vrdS |jd ur,t� \}}|d ur%dS ||jvr,dS |jd ur;|jt| �jkr;dS dS )NTF) r r �get_release_info�seriesr r
r r � is_attached)r- r* �cloud_id�failr r r �do_selectors_applyF s
r4 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) r7 �one_month_after_begin�
end_to_user r r �do_dates_apply^ s r= �cc C s t �| �d dkS )Nr �C)� unicodedata�category)r> r r r �is_control_charp s rB �msgc 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 )rB )�.0r> r r r �
<listcomp>{ s z$is_message_valid.<locals>.<listcomp>�M T)�lenr+ �any)rC �liner r r �is_message_validt s
�rL 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�LOG�debugrL r= r( r) r4 r* )r- r �msg_dictrC �er r r �select_message� s& �� rS c C sR t jtjdd� t�� }tj|| jdtjd�}|�� t �
|j �}t j
|tjd�S )NT)�exist_ok� )�hash�destdir)�cls)�os�makedirsr � UAC_RUN_PATH�apt_pkg�Acquire� AcquireFile� apt_news_url�runr � load_file�destfile�json�loadsr �DatetimeAwareJSONDecoder)r- �acq�
apt_news_file�apt_news_contentsr r r �fetch_aptnews_json� s � �ri c C s> t | �}t| |�dg ��}t�d|� |d ur d�|j�S d S )Nr z
using msg: %r�
)ri rS �getrO rP �joinr+ )r- � news_dictrC r r r �fetch_and_process_apt_news� s rn c C s� t | �\}}|tjkrtj�|�j|d�S |tjkr:tj | }| j
j }|du r*d}n|� d�}tj
�|�j||d�S |tjkrBtjS dS )z4
:return: str if local news, None otherwise
)�remaining_daysN�Unknownz%d %b %Y)� expired_datero )r
r �ACTIVE_EXPIRED_SOONr �CONTRACT_EXPIRES_SOON� pluralize�format�EXPIRED_GRACE_PERIODr �!CONTRACT_EXPIRY_GRACE_PERIOD_DAYS�machine_token_file�contract_expiry_datetime�strftime� CONTRACT_EXPIRED_GRACE_PERIOD�EXPIRED�CONTRACT_EXPIRED)r- �
expiry_statusro �grace_period_remaining�exp_dt�
exp_dt_strr r r �local_apt_news� s*
�
�
��
r� �newsc C s. d}| � d�D ] }|d�|�7 }q|d7 }|S )Nz#
rj z# {}
)�splitru )r� �resultrK r r r �format_news_for_apt_update� s
r� c
C s� z,t | �}|s t| �}|r tj�|� t|�}tj�|� W d S tj�� tj�� W d S tyO } zt �
d|� tj�� tj�� W Y d }~d S d }~ww )Nz2something went wrong while processing apt_news: %r) r� rn r �apt_news_raw_file�writer� �apt_news_contents_file�deleterN rO rP )r- r� �apt_update_formatted_newsrR r r r �update_apt_news� s
��r� )5r, rc �loggingrY r@ �typingr r r\ �uaclientr r r r �(uaclient.api.u.pro.status.is_attached.v1r � uaclient.aptr �uaclient.clouds.identityr
�uaclient.configr �uaclient.contractr r
�uaclient.data_typesr r r r r r �uaclient.filesr � getLogger� replace_top_level_logger_namer! rO r r' r& r4 r= r% rB rL �dictrS ri rn r� r� r� r r r r �<module> s^ ��
���
���
�