File: //lib/python3/dist-packages/uaclient/files/__pycache__/notices.cpython-38.pyc
U
��Jh�! � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z m
Z
m Z m Z m
Z
e �e
�e��Ze
�� Zedddd d
g�ZG d d � d ee�ZG d
d� d�Zdadd� Zedd�dd�Zedd�dd�Zee d�dd�Z dS )� N)�
namedtuple)�Enum)�List)�defaults� event_logger�messages�system�util�NoticeFileDetails�order_id�label� is_permanent�messagec @ s* e Zd Zedddejd�Zeddddd�Zed d
dejd�Zed d dej d�Z ed
ddej
d�Z edddej d�Z edddej
d�Z
edddejd�Zedddejjd�Zedddejd�Zedddejd�Zedddejd�Zed d dd d�Zed d dejd�Zed!d"dejd�Zed#d$dejd�Zd%S )&�NoticeZcontract_expired�5T)r r r
r Zreboot_requiredZ10FzSystem reboot requiredZenable_reboot_requiredZ11Zreboot_script_failedZ12Zfips_reboot_requiredZ20Zfips_system_reboot_requiredZ21Zfips_install_out_of_dateZ22Z fips_disable_reboot_requiredZ23Zfips_proc_file_errorZ24Zfips_manual_disable_urlZ25Z wrong_fips_metapackage_on_cloudZlp_lts_reboot_requiredZ30Zoperation_in_progressZ60z"Operation in progress: {operation}Z auto_attach_retry_full_noticeZ70Zauto_attach_total_failureZ71Zlimited_to_releaseZ80N) �__name__�
__module__� __qualname__r
r ZCONTRACT_EXPIREDZREBOOT_REQUIREDZENABLE_REBOOT_REQUIRED_TMPLZENABLE_REBOOT_REQUIREDZREBOOT_SCRIPT_FAILEDZFIPS_REBOOT_REQUIRED_MSGZFIPS_REBOOT_REQUIREDZFIPS_SYSTEM_REBOOT_REQUIREDZFIPS_INSTALL_OUT_OF_DATEZ FIPS_DISABLE_REBOOT_REQUIREDZFIPS_PROC_FILE_ERRORZtmpl_msgZ NOTICE_FIPS_MANUAL_DISABLE_URLZFIPS_MANUAL_DISABLE_URLZ&NOTICE_WRONG_FIPS_METAPACKAGE_ON_CLOUDZ WRONG_FIPS_METAPACKAGE_ON_CLOUDZ LIVEPATCH_LTS_REBOOT_REQUIREDZOPERATION_IN_PROGRESSZAUTO_ATTACH_RETRY_NOTICEZ AUTO_ATTACH_RETRY_FULL_NOTICEZ&AUTO_ATTACH_RETRY_TOTAL_FAILURE_NOTICEZ AUTO_ATTACH_RETRY_TOTAL_FAILUREZLIMITED_TO_RELEASE� r r �8/usr/lib/python3/dist-packages/uaclient/files/notices.pyr s� ����������������r c @ sr e Zd Zeed�dd�Zed�dd�Zeee d�dd �Zeee d
�d d �Z
eed
�dd�Z ee d�dd�Z dS )�NoticesManager)�notice_details� descriptionc C s^ t �� s t�d|jj� dS |jjr*tjntj }d�
|jj |jj�}t �
tj�||�|� dS )z�Adds a notice file. If the notice is found,
it overwrites it.
:param notice_details: Holds details concerning the notice file.
:param description: The content to be written to the notice file.
z.NoticesManager.add(%s) called as non-root userN�{}-{})r �we_are_currently_root�LOG�warning�valuer r
r �NOTICES_PERMANENT_DIRECTORY�NOTICES_TEMPORARY_DIRECTORY�formatr r Z
write_file�os�path�join)�selfr r � directory�filenamer r r �addt s$ ��� � �zNoticesManager.add)r c C s\ t �� s t�d|jj� dS |jjr*tjntj }d�
|jj |jj�}t �
tj�||�� dS )ziDeletes a notice file.
:param notice_details: Holds details concerning the notice file.
z1NoticesManager.remove(%s) called as non-root userNr )r r r r r r r
r r r r r r Zensure_file_absentr! r" r# )r$ r r% r& r r r �remove� s ��� �zNoticesManager.remove)r% �returnc s � �fdd�t �� �D �S )z�Gets the list of notice file names in the given directory.
:param directory: The directory to search for notice files.
:returns: List of notice file names.
c s2 g | ]*}t j�t j�� |��r��� |�r|�qS r )r! r" �isfiler# �_is_valid_notice_file)�.0� file_name�r% r$ r r �
<listcomp>� s �z9NoticesManager._get_notice_file_names.<locals>.<listcomp>)r! �listdir)r$ r% r r. r �_get_notice_file_names� s �z%NoticesManager._get_notice_file_names)r% r- r) c s$ |t jk� � fdd�tD �}||kS )z�Checks if the notice file is valid.
:param file_name: The name of the notice file.
:returns: True if the file is valid, False otherwise.
c s& h | ] }|j � krd �|j|j��qS )r )r
r r r )r, �n�Zis_permanent_dirr r � <setcomp>� s
�z7NoticesManager._is_valid_notice_file.<locals>.<setcomp>)r r r )r$ r% r- Zvalid_file_namesr r3 r r+ � s
�z$NoticesManager._is_valid_notice_file)r- r) c C s< |� d�\}}tD ]$}|j|kr|j|kr|jj S qdS )z�Gets the default message for a notice file.
:param file_name: The name of the notice file.
:returns: The default message defined in the enum.
�-� )�splitr r r r r )r$ r- r r �noticer r r �_get_default_message� s
z#NoticesManager._get_default_message�r) c
C s� t jt jf}g }|D ]�}tj�|�s&q| �|�}|D ]h}zt�tj� ||��}W n& t
k
rv t � d| � Y q4Y nX |r�|�
|� q4| �|�}|�
|� q4q|�� |S )zdGets all the notice files currently saved.
:returns: List of notice file contents.
z Permission error while reading )r r r r! r" �existsr1 r Z load_filer# �PermissionErrorr r �appendr9 �sort)r$ Znotice_directoriesZnoticesZnotice_directoryZnotice_file_namesZnotice_file_nameZnotice_file_contentsZdefault_messager r r �list� s4 �
��
�zNoticesManager.listN)
r r r r �strr' r( r r1 �boolr+ r9 r? r r r r r s s �
r c C s t d krt� a t S �N)� _notice_clsr r r r r �
get_notice� s rD )r r) c K s$ t � }| jjf |�}|�| |� d S rB )rD r r r' )r �kwargsr8 r r r r r' � s r' c C s t � }|�| � d S rB )rD r( )r r8 r r r r( s r( r: c C s t � } | �� S rB )rD r? )r8 r r r r? s r? ) Zloggingr! � collectionsr �enumr �typingr Zuaclientr r r r r Z getLoggerZ replace_top_level_logger_namer r Zget_event_loggerZeventr
r r rC rD r' r( r@ r? r r r r �<module> s&
�c