CasperSecurity
o
��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+dedee) de,fdd�Z-de j dee j de,fdd�Z.de/de,fdd�Z0d e*de,fd d �Z1ded ee2 dee* fd d!�Z3edefd"d#��Z4dedee/ fd$d%�Z5dedee/ fd&d'�Z6d(e/de/fd)d*�Z7defd+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 �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% 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 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_applyQ 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_applyi 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 sb t jtjdd� tjtjdd� t�� }tj|| j dtjd�}|�
� t � |j
�}tj|tjd�S )NT)�exist_ok�_apt)�user� )�hash�destdir)�cls)�os�makedirsr �
APT_NEWS_PATH�shutil�chown�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� )9r1 r{ �loggingro rr rX �typingr r rt �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` ��
�%��
���
� &