3
Pf @ s d dl mZ d dlZd dlZd dlZd dlZy d dlZW n ek
rT d dlZY nX d dl m
Z
d dl m Z yd dl
mZ W n ek
r dZY nX ej Zd e_ejdddZd d
ZG d d d ejZd
d ZG dd dejZG dd dejj Z G dd dej Z dS ) )absolute_importN)WINDOWS)
ensure_dir)colorama c c s. t j| 7 _z
dV W dt j| 8 _X dS )zv
A context manager which will cause the log output to be indented for any
log messages emitted inside it.
N)
_log_state indentation)Znum r /usr/lib/python3.6/logging.py
indent_log s
r c C s t tddS )Nr r )getattrr r r r r
get_indentation) s r
c @ s e Zd Zdd ZdS )IndentingFormatterc C s, t jj| |}djdd |jdD }|S )z
Calls the standard formatter, but will indent all of the log messages
by our current indentation level.
c S s g | ]}d t | qS ) )r
).0liner r r
6 s z-IndentingFormatter.format..T)logging Formatterformatjoin
splitlines)selfrecordZ formattedr r r
r / s
zIndentingFormatter.formatN)__name__
__module__ __qualname__r r r r r
r - s r c s fdd}|S )Nc s dj t | tjjg S )Nr )r listr ZStyleZ RESET_ALL)Zinp)colorsr r
wrapped= s z _color_wrap..wrappedr )r r r )r r
_color_wrap< s r! c @ sT e Zd Zer2ejeejjfej eejj
fgZ ng Z dddZ dd Z
dd ZdS ) ColorizedStreamHandlerNc C s( t jj| | tr$tr$tj| j| _d S )N)r
StreamHandler__init__r r AnsiToWin32stream)r r& r r r
r$ N s z ColorizedStreamHandler.__init__c C sR t sdS t| jt js | jn| jj}t|dr:|j r:dS tjj ddkrNdS dS )NFisattyTZTERMZANSI)
r
isinstancer& r% r hasattrr' osenvironget)r Z real_streamr r r
should_colorT s z#ColorizedStreamHandler.should_colorc C sB t jj| |}| j r>x&| jD ] \}}|j|kr ||}P q W |S )N)r r# r r- COLORSlevelno)r r msglevelZcolorr r r
r i s
z ColorizedStreamHandler.format)N)r r r r r ZERRORr! ZForeZREDZWARNINGZYELLOWr. r$ r- r r r r r
r" B s
r" c @ s e Zd Zdd ZdS )BetterRotatingFileHandlerc C s t tjj| j tjjj| S )N) r r* pathdirnameZ baseFilenamer handlersRotatingFileHandler_open)r r r r
r7 w s z BetterRotatingFileHandler._openN)r r r r7 r r r r
r2 u s r2 c @ s e Zd Zdd Zdd ZdS )MaxLevelFilterc C s
|| _ d S )N)r1 )r r1 r r r
r$ ~ s zMaxLevelFilter.__init__c C s |j | jk S )N)r/ r1 )r r r r r
filter s zMaxLevelFilter.filterN)r r r r$ r9 r r r r
r8 | s r8 )r ) Z
__future__r
contextlibr Zlogging.handlersr* Z threading ImportErrorZdummy_threadingZ
pip.compatr Z pip.utilsr Z pip._vendorr ExceptionZlocalr r contextmanagerr r
r r r! r# r" r5 r6 r2 Filterr8 r r r r
s0
3