o

���g��@s�dZd
dl
Z
d
dlZd
dlmZ
zd
dlmZmZmZm	Z	
Wn	e
y'


Yn
wddlmZ

Gdd�de
j�Zd	d
�Zdd�Zede
fd
d�
�
Zedd��
ZdS)zB
Helper functionality for interoperability with stdlib `logging`.
�N)
�contextmanager)�Iterator�List�Optional�Type�)
�tqdmcs(eZ
dZef
�f
d
d�	Zdd�Z�ZS)�_TqdmLoggingHandlercst��
�
|
|_dS�
N)�super�__init__�
tqdm_class)�selfr
�
�	__class__��G/home/arjun/.local/lib/python3.10/site-packages/tqdm/contrib/logging.pyrs


z_TqdmLoggingHandler.__init__c	CsTz|�|
�
}|j
j||jd
�
|��
WdSttfy


�


|�|
�

YdS)N)
�file)�formatr
�write�stream�flush�KeyboardInterrupt�
SystemExit�handleError)r�record�msgrrr�emits








z_TqdmLoggingHandler.emit)�__name__�
__module__�__qualname__�std_tqdmrr�
__classcell__rrrrr	s�r	c

Cst|t
j�o|jtjtjhvSr
)�
isinstance�logging�
StreamHandlerr�sys�stdout�stderr)
�handlerrrr�_is_console_logging_handler#s

�r*c
Cs|D]
}
t|
�
r|

SqdSr
�
r*)�handlersr)rrr�(_get_first_found_console_logging_handler(s



��r-c	cs��|d
ur	tj
g
}dd�|D�
}z;|D]%}t|
�
}t|j�
}|d
u
r,|�|j�

|j|_dd�|jD�
|g
|_qd
V
Wt||�D]\}}||_qBd
St||�D]\}}||_qQw)a�
    Context manager redirecting console logging to `tqdm.write()`, leaving
    other logging handlers (e.g. log files) unaffected.

    Parameters
    ----------
    loggers  : list, optional
      Which handlers to redirect (default: [logging.root]).
    tqdm_class  : optional

    Example
    -------
    ```python
    import logging
    from tqdm import trange
    from tqdm.contrib.logging import logging_redirect_tqdm

    LOG = logging.getLogger(__name__)

    if __name__ == '__main__':
        logging.basicConfig(level=logging.INFO)
        with logging_redirect_tqdm():
            for i in trange(9):
                if i == 4:
                    LOG.info("console logging redirected to `tqdm.write()`")
        # logging restored
    ```
    Nc
Ssg|]}
|
j�qSr)
r,)�.0�loggerrrr�
<listcomp>Rsz)logging_redirect_tqdm.<locals>.<listcomp>c
Ssg|]}
t|
�
s|
�qSrr+)r.r)rrrr0Zs

�
�)	r$�rootr	r-r,�setFormatter�	formatterr�zip)�loggersr
�original_handlers_listr/�tqdm_handler�orig_handler�original_handlersrrr�logging_redirect_tqdm.s,�"











��
�
�r:c	os��|
��}|�
d
d�}|�
dt�}||i|�
�
�*}t||d��
|V
Wd�
n1s-w



Y
Wd�
dSWd�
dS1sEw



Y
dS)ap

    Convenience shortcut for:
    ```python
    with tqdm_class(*args, **tqdm_kwargs) as pbar:
        with logging_redirect_tqdm(loggers=loggers, tqdm_class=tqdm_class):
            yield pbar
    ```

    Parameters
    ----------
    tqdm_class  : optional, (default: tqdm.std.tqdm).
    loggers  : optional, list.
    **tqdm_kwargs  : passed to `tqdm_class`.
    r5Nr
)r5r
)�copy�popr!r:)�args�kwargs�tqdm_kwargsr5r
�pbarrrr�tqdm_logging_redirectcs�




��
"�rA)�__doc__r$r&�
contextlibr�typingrrrr�ImportError�stdrr!r%r	r*r-r:rArrrr�<module>
s&




�
�4