3
�<�e� �
@ s d dl Z d dlZd dlZd dlZd dlZd dlZd dljZd dlZd dl Z d dl
Z
yd dl m Z W n d dl m Z Y nX dgZ
dai Ze
j� ZG dd� de�Zdd� Zdd � Zd
d � Zd d� ZG d
d� dej� �Zejejej � ej e� e j ej � dS )� N)�StringIO�getc @ s e Zd Zdd� ZdS )�
LogHandlerc C s || _ || _d S )N)�handler�stream)�selfr r � r �/usr/lib/python3.6/logs.py�__init__ s zLogHandler.__init__N)�__name__�
__module__� __qualname__r
r r r r r s r c C s< t j� }tjtj }d}x t| �D ]}||j|�7 }q"W |S )N� )�randomZ SystemRandom�stringZ
ascii_lettersZdigits�rangeZchoice)Zlength�r�chars�res�ir r r �_random_string s r c
C s� t �� x&td�D ]}td�}|tkrP qW d S t� }tj|�}|j| � tjd�}|j |� t
j |� t ||�}|t|< t
j
d| � |S Q R X 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 Z formatter� log_handlerr r r �log_capture_start$ s"
r% c C sj t �\ y t| }W n tk
r& d S X |jj� }|jj� tj|j� t| = tj d| � |S Q R X 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 st t d krtjd�a tj� j} | jd }|dkr6d}t S |jd�rp|jdd�\}}t j |�}|j
� |j d� |S d S )NZtunedr �__main__ztuned.�.� ZNOTSET) r r Z getLogger�inspectZ currentframe�f_back�f_locals�
startswith�splitZgetChild�remove_all_handlersr )Zcalling_module�name�rootZchildZ child_loggerr r r r E s
c sx e Zd ZdZejd�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||� | jtj� | j� d S )N)�superr6 r
r r �INFO�switch_to_console)r �args�kwargs)� __class__r r r
^ s zTunedLogger.__init__c O s | j tj|f|�|� d S )N)�log�consts�LOG_LEVEL_CONSOLE)r �msgr: r; r r r �consolec s zTunedLogger.consolec C s | j � | j� | j| j� d S )N)�_setup_console_handlerr3 r! �_console_handler)r r r r r9 f s z TunedLogger.switch_to_consolec C s& | j |||� | j� | j| j� d S )N)�_setup_file_handlerr3 r! �
_file_handler)r �filename�maxBytes� backupCountr r r �switch_to_filek s zTunedLogger.switch_to_filec C s" | j }x|D ]}| j|� q W d S )N)�handlersr) )r Z _handlersr r r r r3 r s
z TunedLogger.remove_all_handlersc C s* | j d k rd S tj� | _ | j j| j� d S )N)rC r r r �
_formatter)�clsr r r rB w s
z"TunedLogger._setup_console_handlerc C sj | j d k rd S tjj|�}|dkr&d}tjj|�s<tj|� tjj|t |�t |�d�| _ | j j
| j � d S )Nr r, )rG rH ) rE �os�path�dirname�exists�makedirsr rJ ZRotatingFileHandler�intr rK )rL rF rG rH Z
log_directoryr r r rD s
z TunedLogger._setup_file_handler)r r r
�__doc__r r rK rC rE r
rA r9 r> ZLOG_FILEZLOG_FILE_MAXBYTESZLOG_FILE_COUNTrI r3 � classmethodrB rD �
__classcell__r r )r< r r6 X s
r6 ) �atexitr Zlogging.handlersrM Zos.pathr. Z tuned.constsr> r r Z threadingr �io�__all__r r ZLockr �objectr r r% r* r ZgetLoggerClassr6 Z addLevelNamer? ZLOG_LEVEL_CONSOLE_NAMEZsetLoggerClass�registerZshutdownr r r r �<module> s4
6