U )d@sdZddlmZddlZddlZddlmZzddlmZm Z m Z m Z Wne k r\YnXddl 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__8/tmp/pip-unpacked-wheel-u8lm3wqg/tqdm/contrib/logging.pyr sz_TqdmLoggingHandler.__init__c Cs\z(||}|jj||jd|Wn.ttfk rBYn||YnXdS)N)file)formatrwritestreamflushKeyboardInterrupt SystemExit handleError)rrecordmsgrrremits  z_TqdmLoggingHandler.emit)__name__ __module__ __qualname__std_tqdmr r __classcell__rrrrr sr cCst|tjo|jtjtjhkSr ) isinstancelogging StreamHandlerrsysstdoutstderr)handlerrrr_is_console_logging_handler%s r+cCs|D]}t|r|SqdSr r+)handlersr*rrr(_get_first_found_console_logging_handler*sr.c cs|dkrtjg}dd|D}zZ|D]J}t|}t|j}|dk rV||j|j|_dd|jD|g|_q$dVW5t||D]\}}||_qXdS)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%rootzipr-r r. setFormatter formatterr)loggersrZoriginal_handlers_listr0Zoriginal_handlersZ tqdm_handlerZ orig_handlerrrrlogging_redirect_tqdm0s$"   r7c osX|}|dd}|dt}|||$}t||d |VW5QRXW5QRXdS)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