File: //proc/self/root/lib/python3/dist-packages/uaclient/__pycache__/log.cpython-38.pyc
U
��Jh� � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZm Z m
Z
d dl m Z m
Z
mZmZ d dlmZ G dd� dej�ZG dd � d ej�ZG d
d � d ej�Zed �d
d�Zed �dd�Ze e d �dd�Zdd� Ze
ee f ed�dd�Z dd� Z dS )� N)� OrderedDict)�Any�Dict�List�Union)�defaults�secret_manager�system�util)�UAConfigc @ s e Zd ZdZejd�dd�ZdS )�RegexRedactionFilter�,A logging filter to redact confidential info��recordc C s t �t|j��|_dS �NT)r
Zredact_sensitive_logs�str�msg��selfr � r �./usr/lib/python3/dist-packages/uaclient/log.py�filter s zRegexRedactionFilter.filterN��__name__�
__module__� __qualname__�__doc__�logging� LogRecordr r r r r r s r c @ s e Zd ZdZejd�dd�ZdS )�KnownSecretRedactionFilterr
r c C s t j�t|j��|_dS r )r ZsecretsZredact_secretsr r r r r r r s z!KnownSecretRedactionFilter.filterNr r r r r r s r c @ s. e Zd ZdZdZdZdZeje 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 �returnc C s� |� � |_| �|�|_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)r r Zextra_message_dictr( Zlocal_log_recordZfield�valuer r r �format, s,
�
�
zJsonArrayFormatter.formatN) r r r r Zdefault_time_formatZdefault_msec_formatr. r r r r4 r r r r r s
r )r% c C s t �� rt� jS t� S dS )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_filer r r r � get_user_or_root_log_file_pathJ s r8 c C s t j�t�� d�S )z/Gets the correct user log_file storage location�ubuntu-pro.log)�os�path�joinr Zget_user_cache_dirr r r r r7 U s r7 c C sH t �d�} g }| D ]0}t j�d|dtjd�}t j�|�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< r ZUSER_CACHE_SUBDIR�isfile�append)Zuser_directoriesZ log_filesZuser_directoryZ user_pathr r r �get_all_user_log_filesZ s
� r@ c C s\ t �d�} | �t j� t �� }|�t� � |�t j� |�t� � |�t � � | �
|� d S )N� ubuntupro) r � getLogger�setLevel�INFOZ
StreamHandler� setFormatterr � addFilterr r �
addHandler)�loggerZconsole_handlerr r r �setup_journald_loggingn s
rI )� log_levelr6 c C s� t | t�r| �� } t�� s t� }t�d�}|�| � g |_ t
� |�}|� � sh|j
jddd� |jdd� t�|�}|�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.
rA T)�parents�exist_oki� )�modeN)r+ r �upperr
r5 r7 r rB rC Zhandlers�pathlib�Path�exists�parent�mkdirZtouchZ FileHandlerrE r rF r r rG )rJ r6 rH Z
log_file_pathZ file_handlerr r r �setup_cli_loggingy s"
rT c K s d| iS )z�
A helper for passing extra fields to log statements.
Usage:
LOG.info("message string", extra=log.extra(field=something_relevant))
r( r )�kwargsr r r r( � s r( ) r/ r r: rO � collectionsr �typingr r r r Zuaclientr r r r
Zuaclient.configr ZFilterr r Z Formatterr r r8 r7 r@ rI �intrT r( r r r r �<module> s . !