U
    ¸mœdÜ  ã                   @   s¢   d Z ddlZddlZddlmZ zddlmZmZmZm	Z	 W n e
k
rP   Y nX ddlmZ G dd„ dejƒZd	d
„ Zdd„ Zedefdd„ƒZedd„ ƒZdS )zB
Helper functionality for interoperability with stdlib `logging`.
é    N)Úcontextmanager)ÚIteratorÚListÚOptionalÚTypeé   )Útqdmc                       s(   e Zd Zef‡ fdd„	Zdd„ Z‡  ZS )Ú_TqdmLoggingHandlerc                    s   t t| ƒ ¡  || _d S ©N)Úsuperr	   Ú__init__Ú
tqdm_class)Úselfr   ©Ú	__class__© úM/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/tqdm/contrib/logging.pyr      s    z_TqdmLoggingHandler.__init__c              	   C   s\   z(|   |¡}| jj|| jd� |  ¡  W n. ttfk
rB   ‚ Y n   |  |¡ Y nX d S )N)Úfile)Úformatr   ÚwriteÚstreamÚflushÚKeyboardInterruptÚ
SystemExitÚhandleError)r   ÚrecordÚmsgr   r   r   Úemit   s    
z_TqdmLoggingHandler.emit)Ú__name__Ú
__module__Ú__qualname__Ústd_tqdmr   r   Ú__classcell__r   r   r   r   r	      s   þr	   c                 C   s   t | tjƒo| jtjtjhkS r
   )Ú
isinstanceÚloggingÚStreamHandlerr   ÚsysÚstdoutÚstderr)Úhandlerr   r   r   Ú_is_console_logging_handler#   s    ÿr*   c                 C   s   | D ]}t |ƒr|  S qd S r
   ©r*   )Úhandlersr)   r   r   r   Ú(_get_first_found_console_logging_handler(   s    r-   c              	   c   sš   | dkrt jg} dd„ | D ƒ}zZ| D ]J}t|ƒ}t|jƒ}|dk	rV| |j¡ |j|_dd„ |jD ƒ|g |_q$dV  W 5 t| |ƒD ]\}}||_q„X dS )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                 S   s   g | ]
}|j ‘qS r   )r,   )Ú.0Úloggerr   r   r   Ú
<listcomp>R   s     z)logging_redirect_tqdm.<locals>.<listcomp>c                 S   s   g | ]}t |ƒs|‘qS r   r+   )r.   r)   r   r   r   r0   Z   s   ÿ)	r$   ÚrootÚzipr,   r	   r-   ÚsetFormatterÚ	formatterr   )Úloggersr   Zoriginal_handlers_listr/   Zoriginal_handlersZtqdm_handlerZorig_handlerr   r   r   Úlogging_redirect_tqdm.   s$    "
ÿþ
r6   c               
   o   sX   |  ¡ }| dd¡}| dt¡}|| |Ž�$}t||d�� |V  W 5 Q R X W 5 Q R X dS )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`.
    r5   Nr   )r5   r   )ÚcopyÚpopr!   r6   )ÚargsÚkwargsZtqdm_kwargsr5   r   Zpbarr   r   r   Útqdm_logging_redirectc   s    r;   )Ú__doc__r$   r&   Ú
contextlibr   Útypingr   r   r   r   ÚImportErrorZstdr   r!   r%   r	   r*   r-   r6   r;   r   r   r   r   Ú<module>   s"   þ4