Edit File: logcapture.cpython-37.pyc
B r9�`�$ � @ s� d Z ddlZddlmZ ddlZddlmZ ddlmZmZm Z yddl m Z W n e k rp ddl m Z Y nX e� e�ZG dd� de�ZG d d � d e�ZG d d � d e�ZdS ) a This plugin captures logging statements issued during test execution. When an error or failure occurs, the captured log messages are attached to the running test in the test.capturedLogging attribute, and displayed with the error failure output. It is enabled by default but can be turned off with the option ``--nologcapture``. You can filter captured logging statements with the ``--logging-filter`` option. If set, it specifies which logger(s) will be captured; loggers that do not match will be passed. Example: specifying ``--logging-filter=sqlalchemy,myapp`` will ensure that only statements logged via sqlalchemy.engine, myapp or myapp.foo.bar logger will be logged. You can remove other installed logging handlers with the ``--logging-clear-handlers`` option. � N)�Handler)�Plugin)�anyp�ln�safe_str)�StringIOc @ sL e Zd Zdd� Zdd� Zee�Zdd� Zdd� Zee�Zd d � Zd d � Z d S )� FilterSetc C s | � |�\| _| _d S )N)� _partition� inclusive� exclusive)�selfZfilter_components� r �H/opt/alt/python37/lib/python3.7/site-packages/nose/plugins/logcapture.py�__init__! s zFilterSet.__init__c C sH g g }}x4| D ],}|� d�r2|�|dd � � q|�|� qW ||fS )N�-� )� startswith�append)� componentsr r Z componentr r r r % s zFilterSet._partitionc C s | sdS | � |�o | �|� S )z-returns whether this record should be printedT)�_allow�_deny)r �recordr r r �allow/ s zFilterSet.allowc s � fdd�}t ttt|| ���S )zNreturn the bool of whether `record` starts with any item in `matchers`c s � | kp� � | d �S )N�.)r )�key)r r r �record_matches_key: s z0FilterSet._any_match.<locals>.record_matches_key)r �bool�list�map)Zmatchersr r r )r r � _any_match7 s zFilterSet._any_matchc C s | j s dS | �| j |�S )NT)r r )r r r r r r ? s zFilterSet._allowc C s | j s dS | �| j |�S )NF)r r )r r r r r r D s zFilterSet._denyN) �__name__� __module__� __qualname__r r � staticmethodr r r r r r r r r s r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�MyMemoryHandlerc C s4 t �| � t�||�}| �|� t|�| _g | _d S )N)r r �logging� Formatter� setFormatterr � filterset�buffer)r � logformat� logdatefmt�filters�fmtr r r r K s zMyMemoryHandler.__init__c C s | j �| �|�� d S )N)r) r �format)r r r r r �emitQ s zMyMemoryHandler.emitc C s d S )Nr )r r r r �flushS s zMyMemoryHandler.flushc C s g | _ d S )N)r) )r r r r �truncateU s zMyMemoryHandler.truncatec C s | j �|j�rt�| |�S d S )N)r( r �namer �filter)r r r r r r3 W s zMyMemoryHandler.filterc C s | j �� }|d= |S )N�lock)�__dict__�copy)r �stater r r � __getstate__Z s z MyMemoryHandler.__getstate__c C s | j �|� t�� | _d S )N)r5 �update� threading�RLockr4 )r r7 r r r � __setstate__^ s z MyMemoryHandler.__setstate__N) r r! r" r r/ r0 r1 r3 r8 r< r r r r r$ J s r$ c @ s� e Zd ZdZdZdZdZdZdZdZ dZ d gZ d d � Z d d � Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� ZdS )"� LogCapturea Log capture plugin. Enabled by default. Disable with --nologcapture. This plugin captures logging statements issued during test execution, appending any output captured to the error or failure output, should the test fail or raise an error. TZNOSE_NOLOGCAPTURE� logcapturei� z$%(name)s: %(levelname)s: %(message)sNFz-nosec C s� |j dd|�| j� ddd� |j ddd|�d �p6| jd d d � |j d dd|�d�pX| jd dd � |j ddd|�d�ddd � |j dddddd� |j ddd d d d� d S ) z&Register commandline options. z--nologcapture� store_falser> z^Disable logging capture plugin. Logging configuration will be left intact. [NOSE_NOLOGCAPTURE])�action�default�dest�helpz--logging-format�store�logcapture_formatZNOSE_LOGFORMATZFORMATzvSpecify custom format to print statements. Uses the same format as used by standard logging handlers. [NOSE_LOGFORMAT])r@ rB rA �metavarrC z--logging-datefmt�logcapture_datefmtZNOSE_LOGDATEFMTz�Specify custom date/time format to print statements. Uses the same format as used by standard logging handlers. [NOSE_LOGDATEFMT]z--logging-filter�logcapture_filtersZNOSE_LOGFILTERZFILTERa� Specify which statements to filter in/out. By default, everything is captured. If the output is too verbose, use this option to filter out needless output. Example: filter=foo will capture statements issued ONLY to foo or foo.what.ever.sub but not foobar or other logger. Specify multiple loggers with comma: filter=foo,bar,baz. If any logger name is prefixed with a minus, eg filter=-foo, it will be excluded rather than included. Default: exclude logging messages from nose itself (-nose). [NOSE_LOGFILTER] z--logging-clear-handlers� store_trueF�logcapture_clearz Clear all other logging handlersz--logging-level�NOTSET�logcapture_levelz Set the log level to captureN)� add_option�get�env_optr* r+ )r �parser�envr r r �optionss s6 zLogCapture.optionsc C sP || _ |jr|jrd| _|j| _|j| _|j| _ |j | _ |j rL|j � d�| _dS )zConfigure plugin. F�,N)�confr> Z loggingConfig�enabledrE r* rG r+ rJ �clearrL �loglevelrH �splitr, )r rR rT r r r � configure� s zLogCapture.configurec C s� t �� }| jrrt|d�r2x|jD ]}|�|� q W x>tt jjj � � �D ](}t|d�rFx|jD ]}|�|� q\W qFW x,|jd d � D ]}t |t �r�|j� |� q�W |�| j� t| dd�}|�tt |�� d S )N�handlersrW rK )r% � getLoggerrV �hasattrrZ � removeHandlerr �Logger�manager� loggerDict�values� isinstancer$ �remove� addHandler�handler�getattr�setLevel)r Z root_loggerre �loggerrW r r r �setupLoghandler� s zLogCapture.setupLoghandlerc C s | � � dS )z7Set up logging handler before test run begins. N)�start)r r r r �begin� s zLogCapture.beginc C s t | j| j| j�| _| �� d S )N)r$ r* r+ r, re ri )r r r r rj � s zLogCapture.startc C s d S )Nr )r r r r �end� s zLogCapture.endc C s | � � dS )z0Clear buffers and handlers before test. N)ri )r �testr r r � beforeTest� s zLogCapture.beforeTestc C s | j �� dS )z"Clear buffers after test. N)re r1 )r rm r r r � afterTest� s zLogCapture.afterTestc C s | � ||�S )z5Add captured log messages to failure output. )� formatError)r rm �errr r r � formatFailure� s zLogCapture.formatFailurec C s2 | � � |_}|s|S |\}}}|| �||�|fS )z3Add captured log messages to error output. )�formatLogRecordsZcapturedLogging�addCaptureToErr)r rm rq �recordsZec�ev�tbr r r rp � s zLogCapture.formatErrorc C s t tt| jj��S )N)r r r re r) )r r r r rs � s zLogCapture.formatLogRecordsc C s$ d� t|�td�g| td�g �S )N� z >> begin captured logging <<z>> end captured logging <<)�joinr r )r rv ru r r r rt � s zLogCapture.addCaptureToErr)r r! r" �__doc__rU rO r2 �scorer* r+ rV r, rR rY ri rk rj rl rn ro rr rp rs rt r r r r r= c s* . r= )rz r% r r: Znose.plugins.baser Z nose.utilr r r �ior � ImportErrorr[ r �log�objectr r$ r= r r r r �<module> s *