U Jh!@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@s*eZdZedddejdZedddddZed d dejdZed d dej dZ ed ddej dZ edddej dZ edddej dZ edddejdZedddejjdZedddejdZedddejdZedddejdZedddddZedd dejdZed!d"dejdZed#d$dejdZd%S)&NoticeZcontract_expired5T)r r r rZreboot_requiredZ10FzSystem reboot requiredZenable_reboot_requiredZ11Zreboot_script_failedZ12Zfips_reboot_requiredZ20Zfips_system_reboot_requiredZ21Zfips_install_out_of_dateZ22Zfips_disable_reboot_requiredZ23Zfips_proc_file_errorZ24Zfips_manual_disable_urlZ25Zwrong_fips_metapackage_on_cloudZlp_lts_reboot_requiredZ30Zoperation_in_progressZ60z"Operation in progress: {operation}Zauto_attach_retry_full_noticeZ70Zauto_attach_total_failureZ71Zlimited_to_releaseZ80N)__name__ __module__ __qualname__r rZCONTRACT_EXPIREDZREBOOT_REQUIREDZENABLE_REBOOT_REQUIRED_TMPLZENABLE_REBOOT_REQUIREDZREBOOT_SCRIPT_FAILEDZFIPS_REBOOT_REQUIRED_MSGZFIPS_REBOOT_REQUIREDZFIPS_SYSTEM_REBOOT_REQUIREDZFIPS_INSTALL_OUT_OF_DATEZFIPS_DISABLE_REBOOT_REQUIREDZFIPS_PROC_FILE_ERRORZtmpl_msgZNOTICE_FIPS_MANUAL_DISABLE_URLZFIPS_MANUAL_DISABLE_URLZ&NOTICE_WRONG_FIPS_METAPACKAGE_ON_CLOUDZWRONG_FIPS_METAPACKAGE_ON_CLOUDZLIVEPATCH_LTS_REBOOT_REQUIREDZOPERATION_IN_PROGRESSZAUTO_ATTACH_RETRY_NOTICEZAUTO_ATTACH_RETRY_FULL_NOTICEZ&AUTO_ATTACH_RETRY_TOTAL_FAILURE_NOTICEZAUTO_ATTACH_RETRY_TOTAL_FAILUREZLIMITED_TO_RELEASErr8/usr/lib/python3/dist-packages/uaclient/files/notices.pyrsrc@sreZdZeedddZedddZeeeddd Zeee d d d Z eed ddZ eedddZ dS)NoticesManager)notice_details descriptioncCs^tstd|jjdS|jjr*tjntj }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.NoticesManager.add(%s) called as non-root userN{}-{})r we_are_currently_rootLOGwarningvaluer r rNOTICES_PERMANENT_DIRECTORYNOTICES_TEMPORARY_DIRECTORYformatr rZ write_fileospathjoin)selfrr directoryfilenamerrraddts$  zNoticesManager.add)rcCs\tstd|jjdS|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 rrrrr r rrrr r rZensure_file_absentr!r"r#)r$rr%r&rrrremoveszNoticesManager.remove)r%returncsfddtDS)zGets 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. cs2g|]*}tjtj|r|r|qSr)r!r"isfiler#_is_valid_notice_file).0 file_namer%r$rr s z9NoticesManager._get_notice_file_names..)r!listdir)r$r%rr.r_get_notice_file_namess z%NoticesManager._get_notice_file_names)r%r-r)cs$|tjkfddtD}||kS)zChecks if the notice file is valid. :param file_name: The name of the notice file. :returns: True if the file is valid, False otherwise. cs&h|]}|jkrd|j|jqS)r)r r r r )r,nZis_permanent_dirrr s z7NoticesManager._is_valid_notice_file..)rrr)r$r%r-Zvalid_file_namesrr3rr+s   z$NoticesManager._is_valid_notice_file)r-r)cCs<|d\}}tD]$}|j|kr|j|kr|jjSqdS)zGets the default message for a notice file. :param file_name: The name of the notice file. :returns: The default message defined in the enum. -)splitrr r rr)r$r-r r noticerrr_get_default_messages z#NoticesManager._get_default_messager)c Cstjtjf}g}|D]}tj|s&q||}|D]h}zttj ||}Wn&t k rvt d|Yq4YnX|r| |q4||}| |q4q||S)zdGets all the notice files currently saved. :returns: List of notice file contents. zPermission error while reading )rrrr!r"existsr1rZ load_filer#PermissionErrorrrappendr9sort)r$Znotice_directoriesZnoticesZnotice_directoryZnotice_file_namesZnotice_file_nameZnotice_file_contentsZdefault_messagerrrlists4     zNoticesManager.listN) rrrrstrr'r(rr1boolr+r9r?rrrrrss   rcCstdkrtatSN) _notice_clsrrrrr get_noticesrD)rr)cKs$t}|jjf|}|||dSrB)rDrr r')rkwargsr8rrrrr'sr'cCst}||dSrB)rDr()rr8rrrr(sr(r:cCst}|SrB)rDr?)r8rrrr? sr?)Zloggingr! collectionsrenumrtypingrZuaclientrrrrr Z getLoggerZreplace_top_level_logger_namerrZget_event_loggerZeventr rrrCrDr'r(r@r?rrrrs&    c