File: //lib/python3/dist-packages/uaclient/__pycache__/apt_news.cpython-38.pyc
U
��Jh�! � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl Z d dl
m Z m Z m
Z
mZ d dlmZmZ d dlmZmZmZ d dlmZ d dlmZ d dlmZm Z m 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+eee) e,d�dd�Z-e j ee j e,d�dd�Z.e/e,d�dd�Z0e*e,d�dd�Z1eee2 ee* d �d d �Z3eed �d d!��Z4eee/ d"�d#d$�Z5eee/ d"�d%d&�Z6e/e/d'�d(d)�Z7ed �d*d+�Z8dS ),� 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 �ee e
ee e
ee ee e
ee e e
d �d
d �Z dS ) �AptNewsMessageSelectors� codenamesF�Zrequired�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 s,
��
�
�r c @ sh e Zd Zede�ededd�ededd�edee��gZddd�e j e
e j e
e e e d �d
d �Z
dS ) �AptNewsMessage�begin�endFr � selectors�linesN)r- r. �r, r- r. r/ c C s || _ || _|| _|| _d S r r0 )r! r, r- r. r/ r" r" r# r$ C 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 k
r0 t�d| � Y dS X t|�}|d krFdS t||�}t|dko`|dk|dk on|dk|dko||dkg�S )Nz Invalid package selector: %rFr )z==�<=�>=)�<r2 )�>r3 )�
ValueError�LOGZwarningr r �any)�package_selectorZ package_nameZversion_operatorZpackage_versionZinstalled_package_versionZversion_comparisonr" r" r# � _does_package_selector_applyQ s& ���r: )�cfgr. �returnc C s� |d kr dS |j d k r*t�� j|j kr*dS |jd k rXt� \}}|d k rJdS ||jkrXdS |jd k rv|jt| �jkrvdS |j d k r�t�
� |j kr�dS |j d k r�t dd� |j D ��s�dS dS )NTFc S s g | ] }t |��qS r" )r: )�.0r9 r" r" r# �
<listcomp>� s �z&do_selectors_apply.<locals>.<listcomp>)
r r Zget_release_infoZseriesr r
r r Z is_attachedr Z
get_dpkg_archr r8 )r; r. Zcloud_idZfailr" r" r# �do_selectors_applyi s2
��r? )r, r- r< c C sV t j �t jj�}|| k r dS | t jdd� }|d ks<||krB|}n|}||krRdS dS )NF� )ZdaysT)r1 �now�timezoneZutcZ timedelta)r, r- rA Zone_month_after_beginZ
end_to_user" r" r# �do_dates_apply� s rC )�cr< c C s t �| �d dkS )Nr �C)� unicodedata�category)rD r" r" r# �is_control_char� s rH )�msgr< c C s^ t | j�dk rdS t | j�dkr$dS | jD ].}tdd� |D ��rF dS t |�dkr* dS q*dS )N� F� c S s g | ] }t |��qS r" )rH )r= rD r" r" r# r> � s z$is_message_valid.<locals>.<listcomp>�M T)�lenr/ r8 )rI �liner" r" r# �is_message_valid� s
rO )r; r r< c C s� |D ]�}zt �|�}W n6 tk
rL } zt�d|� W Y �qW 5 d }~X Y nX t|�sdt�d|� qt|j|j�s�t�d|� qt | |j
�s�t�d|� q| S d S )Nzmsg failed parsing: %rzmsg not valid: %rzmsg dates don't apply: %rz msg selectors don't apply: %r) r+ Z from_dict� Exceptionr7 �debugrO rC r, r- r? r. )r; r Zmsg_dictrI �er" r" r# �select_message� s" rS )r; c C sb t jtjdd� tjtjdd� t�� }tj|| j dtjd�}|�
� t � |j
�}tj|tjd�S )NT)�exist_okZ_apt)�user� )�hashZdestdir)�cls)�os�makedirsr Z
APT_NEWS_PATH�shutil�chown�apt_pkgZAcquireZ AcquireFileZ apt_news_url�runr Z load_fileZdestfile�json�loadsr ZDatetimeAwareJSONDecoder)r; ZacqZ
apt_news_fileZapt_news_contentsr" r" r# �fetch_aptnews_json� s � �ra )r; r< c C s> t | �}t| |�dg ��}t�d|� |d k r:d�|j�S d S )Nr z
using msg: %r�
)ra rS �getr7 rQ �joinr/ )r; Z news_dictrI r" r" r# �fetch_and_process_apt_news� s re c C s� t | �}|j}|j}t�| �}|tjjkr>t� tj
j � t j S t�
tj
j � |tjjkrlt j�|�j|d�S |tjjkr�tj| }|j}|dkr�d}n
|�d�}t j�|�j||d�S dS )z4
:return: str if local news, None otherwise
)�remaining_daysNZUnknownz%d %b %Y)Z expired_daterf )r Zcontract_statusZcontract_remaining_daysr Zget_machine_token_filer ZEXPIRED�valuer �addZNoticeZCONTRACT_EXPIREDr �removeZACTIVE_EXPIRED_SOONZCONTRACT_EXPIRES_SOONZ pluralize�formatZEXPIRED_GRACE_PERIODr Z!CONTRACT_EXPIRY_GRACE_PERIOD_DAYSZcontract_expiry_datetime�strftimeZ CONTRACT_EXPIRED_GRACE_PERIOD)r; Zis_attached_infoZ
expiry_statusrf Zmachine_token_fileZgrace_period_remainingZexp_dtZ
exp_dt_strr" r" r# �local_apt_news� s4
� �
� �rl )�newsr< c C s. d}| � d�D ]}|d�|�7 }q|d7 }|S )Nz#
rb z# {}
)�splitrj )rm �resultrN r" r" r# �format_news_for_apt_update s
rp c
C s� zRt | �}|st| �}|r<tj�|� t|�}tj�|� ntj�� tj�� W nB tk
r� } z$t �
d|� tj�� tj�� W 5 d }~X Y nX d S )Nz2something went wrong while processing apt_news: %r) rl re r Zapt_news_raw_file�writerp Zapt_news_contents_file�deleterP r7 rQ )r; rm Zapt_update_formatted_newsrR r" r" r# �update_apt_news s
rs )9r1 r_ ZloggingrY r[ rF �typingr r r] Zuaclientr r r r Z(uaclient.api.u.pro.status.is_attached.v1r r Z uaclient.aptr
r r Zuaclient.clouds.identityr
Zuaclient.configr Zuaclient.data_typesr r r r r r Zuaclient.filesr r r Z getLoggerZ replace_top_level_logger_namer% r7 r r+ r: r* r? rC r) rH rO �dictrS ra re rl rp rs r" r" r" r# �<module> sL � & � � &