o _c@sdZddlmZddlZddlZddlmZzddlmZm Z m Z m Z Wn e y-Ynwddl mZGdd d ejZd d Zd d ZedefddZeddZdS)zB Helper functionality for interoperability with stdlib `logging`. )absolute_importN)contextmanager)IteratorListOptionalType)tqdmcs(eZdZeffdd ZddZZS)_TqdmLoggingHandlercstt|||_dSN)superr __init__ tqdm_class)selfr __class__?/usr/local/lib/python3.10/dist-packages/tqdm/contrib/logging.pyr s z_TqdmLoggingHandler.__init__c CsTz||}|jj||jd|WdSttfy||YdS)N)file)formatrwritestreamflushKeyboardInterrupt SystemExit handleError)rrecordmsgrrremits z_TqdmLoggingHandler.emit)__name__ __module__ __qualname__std_tqdmr r __classcell__rrrrr sr cCst|tjo|jtjtjhvSr ) isinstancelogging StreamHandlerrsysstdoutstderr)handlerrrr_is_console_logging_handler%s r+cCs|D] }t|r |SqdSr r+)handlersr*rrr(_get_first_found_console_logging_handler*s r.c cs|dur tjg}dd|D}z;|D]%}t|}t|j}|dur,||j|j|_dd|jD|g|_qdVWt||D]\}}||_qBdSt||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 ``` NcSsg|]}|jqSr)r-).0loggerrrr Tsz)logging_redirect_tqdm..cSsg|]}t|s|qSrr,)r/r*rrrr1\s ) r%rootr r.r- setFormatter formatterrzip)loggersrZoriginal_handlers_listr0Z tqdm_handlerZ orig_handlerZoriginal_handlersrrrlogging_redirect_tqdm0s,"  r7c os|}|dd}|dt}||i|*}t||d |VWdn1s-wYWddSWddS1sEwYdS)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`. r6Nr)r6r)copypopr"r7)argskwargsZ tqdm_kwargsr6rZpbarrrrtqdm_logging_redirectes  "r<)__doc__ __future__rr%r' contextlibrtypingrrrr ImportErrorZstdr r"r&r r+r.r7r<rrrrs(     4