Edit File: logs.cpython-36.pyc
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 s| t d krtjd�a tj� j} | jd }|dkr6d}t S |jd�rp|jdd�\}}t j |�}|j � |j d� |S dsxt �d S ) NZtunedr �__main__ztuned.�.� ZNOTSETF) r r Z getLogger�inspectZ currentframe�f_back�f_locals� startswith�splitZgetChild�remove_all_handlersr �AssertionError)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)�superr7 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 r: 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)rD r r r � _formatter)�clsr r r rC 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, )rH rI ) rF �os�path�dirname�exists�makedirsr rK ZRotatingFileHandler�intr rL )rM rG rH rI Z log_directoryr r r rE s z TunedLogger._setup_file_handler)r r r �__doc__r r rL rD rF r rB r: r? ZLOG_FILEZLOG_FILE_MAXBYTESZLOG_FILE_COUNTrJ r3 � classmethodrC rE � __classcell__r r )r= r r7 X s r7 ) �atexitr Zlogging.handlersrN Zos.pathr. Z tuned.constsr? r r Z threadingr �io�__all__r r ZLockr �objectr r r% r* r ZgetLoggerClassr7 Z addLevelNamer@ ZLOG_LEVEL_CONSOLE_NAMEZsetLoggerClass�registerZshutdownr r r r �<module> s4 6
Back to File Manager