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