U 2d@sddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z ee eZe Zedddd d gZGd d d eeZGd ddZdaddZeddddZeddddZeedddZdS)N) namedtuple)Enum)List)defaults event_loggermessagessystemutilNoticeFileDetailsorder_idlabel is_permanentmessagec@seZdZedddddZedddejdZedd d ejdZed d dej dZ ed ddej j dZ eddd ej dZ edddejdZeddd ejdZeddd ejdZeddd ejdZedddejdZeddd ejdZedddddZedd dejdZed!d"d ejdZd#S)$NoticeZreboot_requiredZ10FzSystem reboot required)r r r rZenable_reboot_requiredZ11Zreboot_script_failedZ12TZfips_reboot_requiredZ20Zfips_system_reboot_requiredZ21Zfips_install_out_of_dateZ22Zfips_disable_reboot_requiredZ23Zfips_proc_file_errorZ24Zfips_manual_disable_urlZ25Zwrong_fips_metapackage_on_cloudZlp_lts_reboot_requiredZ30Zcontract_refresh_warningZ40Zoperation_in_progressZ60z"Operation in progress: {operation}Zauto_attach_retry_full_noticeZ70Zauto_attach_total_failureZ71N)__name__ __module__ __qualname__r ZREBOOT_REQUIREDrZENABLE_REBOOT_REQUIRED_TMPLZENABLE_REBOOT_REQUIREDZREBOOT_SCRIPT_FAILEDZFIPS_REBOOT_REQUIRED_MSGZFIPS_REBOOT_REQUIREDZFIPS_SYSTEM_REBOOT_REQUIREDmsgZFIPS_INSTALL_OUT_OF_DATEZFIPS_DISABLE_REBOOT_REQUIREDZFIPS_PROC_FILE_ERRORZNOTICE_FIPS_MANUAL_DISABLE_URLZFIPS_MANUAL_DISABLE_URLZ&NOTICE_WRONG_FIPS_METAPACKAGE_ON_CLOUDZWRONG_FIPS_METAPACKAGE_ON_CLOUDZLIVEPATCH_LTS_REBOOT_REQUIREDZNOTICE_REFRESH_CONTRACT_WARNINGZCONTRACT_REFRESH_WARNINGZOPERATION_IN_PROGRESSZAUTO_ATTACH_RETRY_NOTICEZAUTO_ATTACH_RETRY_FULL_NOTICEZ&AUTO_ATTACH_RETRY_TOTAL_FAILURE_NOTICEZAUTO_ATTACH_RETRY_TOTAL_FAILURErr8/usr/lib/python3/dist-packages/uaclient/files/notices.pyrsrc@s<eZdZeedddZedddZeeddd Zd S) NoticesManager)notice_details descriptionc Cslts*ttdW5QRXdS|jjr8tjntj }d |jj |jj }t tj|||dS)zAdds 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'Trying to add a notice as non-root userN{}-{})r we_are_currently_rootdisable_log_to_consoleLOGwarningvaluer rNOTICES_PERMANENT_DIRECTORYNOTICES_TEMPORARY_DIRECTORYformatr r rZ write_fileospathjoin)selfrr directoryfilenamerrraddns   zNoticesManager.add)rc Csjts*ttdW5QRXdS|jjr8tjntj }d |jj |jj }t tj||dS)ziDeletes a notice file. :param notice_details: Holds details concerning the notice file. z*Trying to remove a notice as non-root userNr)r rrrrrr rrr r!r r rZensure_file_absentr"r#r$)r%rr&r'rrrremoves zNoticesManager.removereturnc stjtjf}g}|D]tjs&qfddtD}|D]}ttj |}|rj| |qBzT| d\}}d}t D]} | j |kr| j|kr| }q|dkrt| |jjWqBtk rttd|W5QRXYqBXqBq||S)zdGets all the notice files currently saved. :returns: List of notice file contents. cs&g|]}tjtj|r|qSr)r"r#isfiler$).0 file_nameZnotice_directoryrr sz'NoticesManager.list..-Nz1Something went wrong while processing notice: {}.)rrr r"r#existslistdirrZ load_filer$appendsplitrr r Exceptionrrr rrrr!sort) r%Znotice_directoriesZnoticesZnotice_file_namesZnotice_file_nameZnotice_file_contentsr r noticenrr/rlistsD     zNoticesManager.listN) rrrrstrr(r)rr:rrrrrms  rcCstdkrtatSN) _notice_clsrrrrr get_noticesr>)rr+cKs$t}|jjf|}|||dSr<)r>rr!r()rkwargsr8rrrrr(sr(cCst}||dSr<)r>r))rr8rrrr)sr)r*cCst}|Sr<)r>r:)r8rrrr:sr:)Zloggingr" collectionsrenumrtypingrZuaclientrrrrr Z getLoggerZreplace_top_level_logger_namerrZget_event_loggerZeventr rrr=r>r(r)r;r:rrrrs$    ]b