o j>@sdZddlZddlZddlmZzddlmZmZmZm Z Wn e y'Ynwddl m Z GdddejZd d Zd d Zede fd dZeddZdS)zB Helper functionality for interoperability with stdlib `logging`. N)contextmanager)IteratorListOptionalType)tqdmcs(eZdZeffdd ZddZZS)_TqdmLoggingHandlercst||_dSN)super__init__ tqdm_class)selfr  __class__/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/tqdm/contrib/logging.pyr s  z_TqdmLoggingHandler.__init__c CsTz||}|jj||jd|WdSttfy||YdS)N)file)formatr writestreamflushKeyboardInterrupt 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}zF|D]0}t|}t|j}|dur7||j|j|_|jD]}| |q/dd|jD|g|_qdVWt ||D]\}}||_qMdSt ||D]\}}||_q\w)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 Rsz)logging_redirect_tqdm..cSsg|]}t|s|qSrr+)r.r)rrrr0\s ) r$rootr r-r, setFormatter formatterrfilters addFilterzip)loggersr original_handlers_listr/ tqdm_handler orig_handler log_filteroriginal_handlersrrrlogging_redirect_tqdm.s0"    r=c 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`. r7Nr )r7r )copypopr!r=)argskwargs tqdm_kwargsr7r pbarrrrtqdm_logging_redirectes  "rD)__doc__r$r& contextlibrtypingrrrr ImportErrorstdrr!r%r r*r-r=rDrrrrs&    6