Google
Edit File: logs.cpython-39.pyc
a �]�i� � @ 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�stringZ 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 r/ 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 �msgr@ rA r r r �consolec s zTunedLogger.consolec C s | � � | �� | �| j� d S r )�_setup_console_handlerr8 r% �_console_handler)r r r r r? f s z TunedLogger.switch_to_consolec C s& | � |||� | �� | �| j� d S r )�_setup_file_handlerr8 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 r8 r s z TunedLogger.remove_all_handlersc C s* | j d urd S t�� | _ | j �| j� d S r )rJ r r r# � _formatter)�clsr r r rI 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 r1 )rN rO ) rL �os�path�dirname�exists�makedirsr rQ ZRotatingFileHandler�intr# rR )rS rM rN rO Z log_directoryr r r rK s �z TunedLogger._setup_file_handler)r r r �__doc__r r" rR rJ rL r rH r? rE ZLOG_FILEZLOG_FILE_MAXBYTESZLOG_FILE_COUNTrP r8 � classmethodrI rK � __classcell__r r rB r r< X s � r< ) �atexitr Zlogging.handlersrT Zos.pathr3 Z tuned.constsrE r r � threadingr �io�__all__r$ r �Lockr �objectr r r* r/ r ZgetLoggerClassr< Z addLevelNamerF ZLOG_LEVEL_CONSOLE_NAMEZsetLoggerClass�registerZshutdownr r r r �<module> s4 6