U ŒÌJh™ã@søddlZddlZddlZddlZddlmZddlmZmZm Z m Z ddl m Z m Z mZmZddlmZGdd„dejƒZGdd „d ejƒZGd d „d ejƒZed œd d„Zed œdd„Ze ed œdd„Zdd„Ze eefedœdd„Zdd„ZdS)éN)Ú OrderedDict)ÚAnyÚDictÚListÚUnion)ÚdefaultsÚsecret_managerÚsystemÚutil)ÚUAConfigc@s eZdZdZejdœdd„ZdS)ÚRegexRedactionFilterú,A logging filter to redact confidential info©ÚrecordcCst t|jƒ¡|_dS©NT)r Zredact_sensitive_logsÚstrÚmsg©Úselfr©rú./usr/lib/python3/dist-packages/uaclient/log.pyÚfilterszRegexRedactionFilter.filterN©Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚloggingÚ LogRecordrrrrrr sr c@s eZdZdZejdœdd„ZdS)ÚKnownSecretRedactionFilterr rcCstj t|jƒ¡|_dSr)rZsecretsZredact_secretsrrrrrrrsz!KnownSecretRedactionFilter.filterNrrrrrrsrc@s.eZdZdZdZdZdZeje dœdd„Z dS) ÚJsonArrayFormatterzYJson Array Formatter for our logging mechanism Custom made for Pro logging needs z%Y-%m-%dT%H:%M:%Sz%s.%03d)ÚasctimeZ levelnameÚnameZfuncNameÚlinenoÚmessage)rÚreturncCsÄ| ¡|_| |¡|_i}|jr0| |j¡|d<| d¡sJ|jrJ|j|d<|jr`|  |j¡|d<|j  d¡}|r„t |t ƒr„|  |¡tƒ}|jD]}|j  |¡}|||<q||d<t t| ¡ƒ¡S)NÚexc_infoÚ stack_infoÚextra)Z getMessager$Z formatTimer!r&ZformatExceptionÚgetZexc_textr'Z formatStackÚ__dict__Ú isinstanceÚdictÚupdaterÚrequired_fieldsÚjsonÚdumpsÚlistÚvalues)rrZextra_message_dictr(Zlocal_log_recordZfieldÚvaluerrrÚformat,s,  ÿ ÿ     zJsonArrayFormatter.formatN) rrrrZdefault_time_formatZdefault_msec_formatr.rrrr4rrrrr s  r )r%cCst ¡rtƒjStƒSdS)z\ Gets the correct log_file path, adjusting for whether the user is root or not. N)r Úwe_are_currently_rootr Úlog_fileÚget_user_log_filerrrrÚget_user_or_root_log_file_pathJsr8cCstj t ¡d¡S)z/Gets the correct user log_file storage locationúubuntu-pro.log)ÚosÚpathÚjoinr Zget_user_cache_dirrrrrr7Usr7cCsHt d¡}g}|D]0}tj d|dtjd¡}tj |¡r| |¡q|S)z|Gets all the log files for the users in the system Returns a list of all user log files in their home directories. z/homez.cacher9)r:Úlistdirr;r<rZUSER_CACHE_SUBDIRÚisfileÚappend)Zuser_directoriesZ log_filesZuser_directoryZ user_pathrrrÚget_all_user_log_filesZs û  r@cCs\t d¡}| tj¡t ¡}| tƒ¡| tj¡| tƒ¡| t ƒ¡|  |¡dS)NÚ ubuntupro) rÚ getLoggerÚsetLevelÚINFOZ StreamHandlerÚ setFormatterr Ú addFilterr rÚ addHandler)ÚloggerZconsole_handlerrrrÚsetup_journald_loggingns      rI)Ú log_levelr6cCs®t|tƒr| ¡}t ¡s tƒ}t d¡}| |¡g|_ t   |¡}|  ¡sh|j jddd|jddt |¡}| tƒ¡| |¡| tƒ¡| tƒ¡| |¡dS)zoSetup logging to log_file If run as non-root then log_file is replaced with a user-specific log file. rAT)ÚparentsÚexist_oki )ÚmodeN)r+rÚupperr r5r7rrBrCZhandlersÚpathlibÚPathÚexistsÚparentÚmkdirZtouchZ FileHandlerrEr rFr rrG)rJr6rHZ log_file_pathZ file_handlerrrrÚsetup_cli_loggingys"          rTcKsd|iS)z“ A helper for passing extra fields to log statements. Usage: LOG.info("message string", extra=log.extra(field=something_relevant)) r(r)Úkwargsrrrr(šsr()r/rr:rOÚ collectionsrÚtypingrrrrZuaclientrrr r Zuaclient.configr ZFilterr rZ Formatterr rr8r7r@rIÚintrTr(rrrrÚs   .  !