File: //lib/python3/dist-packages/uaclient/__pycache__/apt_news.cpython-310.pyc
o
�ϴfD! � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dlZd dl m
Z
m Z m Z m
Z
d dlmZmZ d dlmZmZmZ d dlmZ d dlmZ d dlmZmZm Z m Z m Z m Z d d l m!Z!m"Z"m#Z# e�$e
�%e&��Z'G d
d � d e�Z(G d d
� d
e�Z)dd� Z*dedee( de+fdd�Z,de j dee j de+fdd�Z-de.de+fdd�Z/d e)de+fd d �Z0ded ee1 dee) fd d!�Z2edefd"d#��Z3dedee. fd$d%�Z4dedee. fd&d'�Z5d(e.de.fd)d*�Z6defd+d,�Z7dS )-� N)�List�Optional)�defaults�messages�system�util)�ContractExpiryStatus� _is_attached)�ensure_apt_pkg_init�get_pkg_version�version_compare)�get_cloud_type)�UAConfig)�
BoolDataValue�
DataObject�DatetimeDataValue�Field�StringDataValue� data_list)�
machine_token�notices� state_filesc @ s� e Zd Zedee�dd�edee�dd�ededd�edee�dd�edeee��dd�gZdddddd �dee e
dee e
dee dee e
dee e e
f
d
d �Z dS ) �AptNewsMessageSelectors� codenamesF��required�clouds�pro�
architectures�packagesN�r r r r r c C s" || _ || _|| _|| _|| _d S �Nr )�selfr r r 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 s2 ��
�
�
��
��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/ r0 )r" r- r. r/ r0 r# r# r$ r% B s
zAptNewsMessage.__init__)r& r' r( r r r r r r) �datetimer r r* r% r# r# r# r$ r, : s" � �����r, c C s� z| \}}}W n t y t�d| � Y dS w t|�}|d u r"dS t||�}t|dko/|dv |dk o6|dv |dko=|dv g�S )Nz Invalid package selector: %rFr )z==�<=�>=)�<r2 )�>r3 )�
ValueError�LOG�warningr r �any)�package_selector� package_name�version_operator�package_version�installed_package_version�version_comparisonr# r# r$ � _does_package_selector_applyP s& ����r@ �cfgr/ �returnc C s� |d u rdS |j d urt�� j|j vrdS |jd ur,t� \}}|d ur%dS ||jvr,dS |jd ur;|jt| �jkr;dS |j d urIt�
� |j vrIdS |j d urZt dd� |j D ��sZdS dS )NTFc S � g | ]}t |��qS r# )r@ )�.0r: r# r# r$ �
<listcomp>� s ��z&do_selectors_apply.<locals>.<listcomp>)
r r �get_release_info�seriesr r
r r � is_attachedr �
get_dpkg_archr r9 )rA r/ �cloud_id�failr# r# r$ �do_selectors_applyh s2
��rL r- r. c C sV t j �t jj�}|| k rdS | t jdd� }|d u s ||kr!|}n|}||kr)dS dS )NF� )�daysT)r1 �now�timezone�utc� timedelta)r- r. rO �one_month_after_begin�
end_to_user# r# r$ �do_dates_apply� s rU �cc C s t �| �d dkS )Nr �C)� unicodedata�category)rV r# r# r$ �is_control_char� s rZ �msgc C s^ t | j�dk r dS t | j�dkrdS | jD ]}tdd� |D ��r# dS t |�dkr, dS qdS )N� F� c S rC r# )rZ )rD rV r# r# r$ rE � s z$is_message_valid.<locals>.<listcomp>�M T)�lenr0 r9 )r[ �liner# r# r$ �is_message_valid� s
�ra 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� Exceptionr7 �debugra rU r- r. rL r/ )rA r �msg_dictr[ �er# r# r$ �select_message� s& �� rg 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)rA �acq�
apt_news_file�apt_news_contentsr# r# r$ �fetch_aptnews_json� s � �r} c C s> t | �}t| |�dg ��}t�d|� |d ur d�|j�S d S )Nr z
using msg: %r�
)r} rg �getr7 rd �joinr0 )rA � news_dictr[ r# r# r$ �fetch_and_process_apt_news� s r� c C s� t | �}|j}|j}t�| �}|tjjkr t� tj
j � t j S t�
tj
j � |tjjkr6t j�|�j|d�S |tjjkr[tj| }|j}|du rKd}n|�d�}t j�|�j||d�S dS )z4
:return: str if local news, None otherwise
)�remaining_daysN�Unknownz%d %b %Y)� expired_dater� )r �contract_status�contract_remaining_daysr �get_machine_token_filer �EXPIRED�valuer �add�Notice�CONTRACT_EXPIREDr �remove�ACTIVE_EXPIRED_SOON�CONTRACT_EXPIRES_SOON� pluralize�format�EXPIRED_GRACE_PERIODr �!CONTRACT_EXPIRY_GRACE_PERIOD_DAYS�contract_expiry_datetime�strftime� CONTRACT_EXPIRED_GRACE_PERIOD)rA �is_attached_info�
expiry_statusr� �machine_token_file�grace_period_remaining�exp_dt�
exp_dt_strr# r# r$ �local_apt_news� s4
� �
��r� �newsc C s. d}| � d�D ] }|d�|�7 }q|d7 }|S )Nz#
r~ z# {}
)�splitr� )r� �resultr` 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� r� r �apt_news_raw_file�writer� �apt_news_contents_file�deleterc r7 rd )rA r� �apt_update_formatted_newsrf r# r# r$ �update_apt_news s
��r� )8r1 rw �loggingrm rX �typingr r rp �uaclientr r r r �(uaclient.api.u.pro.status.is_attached.v1r r � uaclient.aptr
r r �uaclient.clouds.identityr
�uaclient.configr �uaclient.data_typesr r r r r r �uaclient.filesr r r � getLogger� replace_top_level_logger_namer& r7 r r, r@ r+ rL rU r* rZ ra �dictrg r} r� r� r� r� r# r# r# r$ �<module> s^ ��
�%��
���
� &