File: //usr/lib/python3.9/site-packages/tuned/__pycache__/logs.cpython-39.pyc
a
,�g� � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlZd dl Z d dl
Z
zd dl m Z W n d dl m Z Y n0 dgZ
dai Ze
�� ZG dd� de�Zdd� Zdd � Zd
d � Zd d� ZG d
d� de�� �Ze�ejej � e� e� e � ej � dS )� N)�StringIO�getc @ s e Zd Zdd� ZdS )�
LogHandlerc C s || _ || _d S �N)�handler�stream)�selfr r � r �./usr/lib/python3.9/site-packages/tuned/logs.py�__init__ s zLogHandler.__init__N)�__name__�
__module__� __qualname__r r r r r
r s r c C s8 t �� }tjtj }d}t| �D ]}||�|�7 }q |S )N� )�randomZ SystemRandom�string�
ascii_letters�digits�range�choice)�length�r�chars�res�ir r r
�_random_string s r c C s� t �� td�D ]}td�}|tvr q8qW d � d S t� }t�|�}|�| � t�d�}|� |� t
� |� t ||�}|t|< t
�
d| � |W d � S 1 s�0 Y d S )N�
� z%%(levelname)-8s %(name)s: %(message)szAdded log handler %s.)�log_handlers_lockr r � log_handlersr �logging�
StreamHandler�setLevel� Formatter� setFormatter� root_logger�
addHandlerr �debug)Z log_levelr �tokenr r � formatter� log_handlerr r r
�log_capture_start$ s$
�
r+ c C s� t �v z t| }W n ty2 Y W d � d S 0 |j�� }|j�� t�|j� t| = t� d| � |W d � S 1 s�0 Y d S )NzRemoved log handler %s.)
r r �KeyErrorr �getvalue�closer% �
removeHandlerr r' )r( r* Zcontentr r r
�log_capture_finish8 s
r0 c C s| t d u rt�d�a t�� j} | jd }|dkr6d}t S |�d�rp|�dd�\}}t � |�}|�
� |� d� |S dsxJ �d S ) NZtunedr �__main__ztuned.�.� ZNOTSETF) r% r Z getLogger�inspectZ currentframe�f_back�f_locals�
startswith�splitZgetChild�remove_all_handlersr" )Zcalling_module�name�root�childZ child_loggerr r r
r E s
c sx e Zd ZdZe�d�ZdZdZ� fdd�Z dd� Z
dd � Z e j
e je jfd
d �Zd d
� Zedd� �Zedd� �Z� ZS )� TunedLoggerz!Custom TuneD daemon logger class.z1%(asctime)s %(levelname)-8s %(name)s: %(message)sNc s. t t| �j|i |�� | �tj� | �� d S r )�superr= r r" r �INFO�switch_to_console)r �args�kwargs�� __class__r r
r ^ s zTunedLogger.__init__c O s | j tj|g|�R i |�� d S r )�log�consts�LOG_LEVEL_CONSOLE)r �msgrA rB r r r
�consolec s zTunedLogger.consolec C s | � � | �� | �| j� d S r )�_setup_console_handlerr9 r& �_console_handler)r r r r
r@ f s z TunedLogger.switch_to_consolec C s& | � |||� | �� | �| j� d S r )�_setup_file_handlerr9 r& �
_file_handler)r �filename�maxBytes� backupCountr r r
�switch_to_filek s zTunedLogger.switch_to_filec C s | j }|D ]}| �|� q
d S r )�handlersr/ )r Z _handlersr r r r
r9 r s z TunedLogger.remove_all_handlersc C s* | j d urd S t�� | _ | j �| j� d S r )rK r r! r$ �
_formatter)�clsr r r
rJ w s
z"TunedLogger._setup_console_handlerc C sj | j d urd S tj�|�}|dkr&d}tj�|�s<t�|� tjj|t |�t |�d�| _ | j �
| j � d S )Nr r2 )rO rP ) rM �os�path�dirname�exists�makedirsr rR ZRotatingFileHandler�intr$ rS )rT rN rO rP Z
log_directoryr r r
rL s
�z TunedLogger._setup_file_handler)r r
r �__doc__r r# rS rK rM r rI r@ rF ZLOG_FILEZLOG_FILE_MAXBYTESZLOG_FILE_COUNTrQ r9 � classmethodrJ rL �
__classcell__r r rC r
r= X s
�
r= ) �atexitr Zlogging.handlersrU Zos.pathr4 Z tuned.constsrF r r � threadingr �io�__all__r% r �Lockr �objectr r r+ r0 r ZgetLoggerClassr= Z addLevelNamerG ZLOG_LEVEL_CONSOLE_NAMEZsetLoggerClass�registerZshutdownr r r r
�<module> s4
6