Your IP : 216.73.216.14
U
8�-dN � @ 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 dlmZmZmZmZmZmZ d dlmZ d dlm Z m Z G d d
� d
e�Z G d d � d e�Z e e d
�dd�Z!eee e"d�dd�Z#e j ee j e"d�dd�Z$e e"d�dd�Z%e e"d
�dd�Z&eee' ee d�d d �Z(ed �d d �Z)ed �d!d"�Z*ee"d#�d$d%�Z+ed �d&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 @ sl e Zd Zedee�dd�edee�dd�ededd�gZdddd�ee e
ee e
ee d�dd �Z dS )
�AptNewsMessageSelectors� codenamesF�Zrequired�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 @ 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 r) )r r% r& r' r( r r r r 7 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_messageE s
r0 )�cfgr' r, c C sx |d kr dS |j d k r,t�� d |j kr,dS |jd k rZt� \}}|d k rLdS ||jkrZdS |jd k rt|j| jkrtdS dS )NTZseriesF)r r Zget_platform_infor r r Z is_attached)r1 r' Zcloud_idZfailr r r �do_selectors_applyM s
r2 )r% r& r, c C sV t j �t jj�}|| k r dS | t jdd� }|d ks<||krB|}n|}||krRdS dS )NF� )ZdaysT)r* �now�timezoneZutcZ timedelta)r% r& r4 Zone_month_after_beginZ
end_to_user r r �do_dates_applye s r6 )�cr, c C s t �| �d dkS )Nr �C)� unicodedata�category)r7 r r r �is_control_charw s r; 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 )r; )�.0r7 r r r �
<listcomp>� s z$is_message_valid.<locals>.<listcomp>�M T)�lenr( �any)r+ r/ r r r �is_message_valid{ s
rC )r1 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� Exception�logging�debugrC r6 r% r& r2 r' )r1 r Zmsg_dictr+ �er r r �select_message� s" rH )r1 c C sP t jtjdd� t�� }tj|| jtjd�}|�� t �
|j �}t j
|tjd�S )NT)�exist_ok)Zdestdir)�cls)�os�makedirsr Z UAC_RUN_PATH�apt_pkgZAcquireZ AcquireFileZ apt_news_url�runr Z load_fileZdestfile�json�loadsr ZDatetimeAwareJSONDecoder)r1 ZacqZ
apt_news_fileZapt_news_contentsr r r �fetch_aptnews_json� s � �rQ c
C s� zRt | �}t| |�dg ��}t�d|� |d k rFt|�}tj�|� n
tj� � W n8 t
k
r� } zt�d|� tj� � W 5 d }~X Y nX d S )Nr z
using msg: %rz2something went wrong while processing apt_news: %r) rQ rH �getrE rF r0 r �apt_news_contents_file�write�deleterD )r1 Z news_dictr+ Zmsg_strrG r r r �fetch_and_process_apt_news� s rV )r1 r, c C s� t | �\}}|tjkr0tj�tjj|d�� dS |tj kr�t
j | }| j j
}|dkrZd}n
|�d�}tj�tjj||d�� dS |tjkr�tj�tj� dS dS )zB
:return: True if local news was written, False otherwise
)�remaining_daysTNZUnknownz%d %b %Y)Z expired_daterW F)r r ZACTIVE_EXPIRED_SOONr rS rT r Z CONTRACT_EXPIRES_SOON_APT_NEWSr- ZEXPIRED_GRACE_PERIODr Z!CONTRACT_EXPIRY_GRACE_PERIOD_DAYSZmachine_token_fileZcontract_expiry_datetime�strftimeZ&CONTRACT_EXPIRED_GRACE_PERIOD_APT_NEWSZEXPIREDZCONTRACT_EXPIRED_APT_NEWS)r1 Z
expiry_statusrW Zgrace_period_remainingZexp_dtZ
exp_dt_strr r r �local_apt_news� s8
��
�
��
�rY c C s t | �}|s t�� t| � d S r )rY rM ZinitrV )r1 Zlocal_news_writtenr r r �update_apt_news� s rZ )-r* rO rE rK r9 �typingr r rM Zuaclientr r r r Zuaclient.clouds.identityr Zuaclient.configr Zuaclient.data_typesr
r r r
r r Zuaclient.filesr Z uaclient.jobs.update_messagingr r r r$ r" r0 r# r2 r6 r; rC �dictrH rQ rV rY rZ r r r r �<module> sB � � � '