OwlCyberSecurity - MANAGER
Edit File: notices.cpython-38.pyc
U Uٵg�! � @ 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