o 3go&@s<ddlmZddlZddlZddlZddlZddlmZddlm Z m Z d0ddZ d Z d Z e jd d d ZeZe eedfZd1ddZd2ddZd3ddZddd4ddZedd5ddZGd d!d!ejZGd"d#d#ejZddd6d%d&Zd'd(Zed)kre e sd*d+l m!Z!e!d,d-d.Z"e!d,d/dZ#dSdS)7) annotationsN) lru_cache) TYPE_CHECKINGUnionreturnNonec Csddlm}dd}zdjWnty#}z|}WYd}~nd}~ww|dtd}|d|dt|d |d |||d t d }| d | d| d|d|||dtd  tj| ddS)Nr)CallablecSst|tjdS)N)file)printsysstderr)sr)/home/garg/my-data/HPI/my/core/logging.pysztest..z+ Logging module's defaults are not great:default_loggerztFor example, this should be logged as error. But it's not even formatted properly, doesn't have logger name or levelzY The reason is that you need to remember to call basicConfig() first. Let's do it now:zkOK, this is better. But the default format kinda sucks, I prefer having timestamps and the file/line numberzr Also exception logging is kinda lame, doesn't print traceback by default unless you remember to pass exc_info:zq With make_logger you get a reasonable logging format, colours (via colorlog library) and other neat things:testzdefault level is INFOz"... so this shouldn't be displayedzwarnings are easy to spot!z/ Exceptions print traceback by default now:zt You can (and should) use it via regular logging.getLogger after that, e.g. let's set logging level to DEBUG nowz)... now debug messages are also displayed)typingrwhatever Exceptionlogging getLoggererror basicConfig exception make_loggerinfodebugwarningsetLevelDEBUG)rMeexlllrrrr s4          rINFOzX{start}[%(levelname)-7s %(asctime)s %(name)s %(filename)s:%(lineno)-4d]{end} %(message)sstartendlevelLevelIshLevelcCs*|durtjSt|tr|Stt|SN)rNOTSET isinstanceintgetattrupperr+rrrmklevel:s  r5 Level | NonecCs>tjdd}|durt|Stjdd}|durtjSdS)NLOGGING_COLLAPSECOLLAPSE_DEBUG_LOGS)osenvirongetr5rr )clrrrget_collapse_levelBsr=namestrcCsd}tj||dptj||ddd}|dur t|Sdtjvr,ttjdSdtjvrBddlm}|dttjdSdS) NLOGGING_LEVEL_._LOGGING_LEVEL_HPIHPI_LOGSr)mediumzNThe HPI_LOGS environment variable is deprecated, use LOGGING_LEVEL_HPI instead)r9r:r;replacer5my.core.warningsrE)r>PREFIXlvlrErrr get_env_levelNs,   rJr4loggerstr | logging.LoggercCsft|tr t|}|durt}t|j}|dur|}nt|}|jtj kr+| |t |jddS)z, Wrapper to simplify logging setup. N)r>) r0r?rr DEFAULT_LEVELrJr>r5r+r/r_setup_handlers_and_formatters)rKr+ env_levelrIrrr setup_loggerds     rPcCst|}|tt}|dustjst}nt |d}| |d|_ zddl }Wnt yDtjdddtt}Ynwtjddd }|jrW||}ntt}||dS) N)maxlevelFrz>You might want to 'pip install colorlog' for nice colored logs stacklevelz %(log_color)sz %(reset)sr()rr addFilterAddExceptionTracebackr=r r isatty StreamHandlerCollapseLogsHandler addHandler propagatecolorlogModuleNotFoundErrorwarningswarn FormatterFORMAT_NOCOLORFORMATformatstreamColoredFormatter setFormatter)r>rKcollapse_levelhandlerr\ formatter FORMAT_COLORrrrrN}s&          rNc@seZdZdddZdS) rVrecordlogging.LogRecordrboolcCsH|jdkr"|j}t|tr"|jdus|jdkr"t|||jf}||_dS)NERROR)NNNT) levelnamemsgr0 BaseExceptionexc_infotype __traceback__)selfrkexcrrrrrfilters  zAddExceptionTraceback.filterN)rkrlrrm)__name__ __module__ __qualname__rwrrrrrVsrVcsHeZdZUdZdZded<ejZded<dfd d Z dd dZ Z S)rYz Collapses subsequent debug log lines and redraws on the same line. Hopefully this gives both a sense of progress and doesn't clutter the terminal as much? Frmlastr-rQrrcstj|i|||_dSr.)super__init__rQ)rurQargskwargs __class__rrr}s zCollapseLogsHandler.__init__rkrlc CszL||}|j|jkod|v}|r|jr|jdn |jr%|jd||_td\}}|j|dtd|t ||rBdnd| WdS| |YdS)N z r r') rclevelnorQr{rdwriter9get_terminal_sizemaxlenflush handleError)rurkrpcurcolumnsrBrrremits   .zCollapseLogsHandler.emit)rQr-rr)rkrlrr) rxryrz__doc__r{__annotations__rr rQr}r __classcell__rrrrrYs  rYlogging.LoggercCst|}t||d|S)Nr4)rrrP)r>r+rKrrrrs  rcCsddlm}tjdddur|Szddl}Wnty+tjddd|YSwt |dd}|dur8|S| }t |d||S)Nr)MockENLIGHTEN_ENABLEzAYou might want to 'pip install enlighten' for a nice progress barrRrSINSTANCE) unittest.mockrr9r:r; enlightenr]r^r_r2 get_managersetattr)rrinstancerrr get_enlightens      r__main__rR) deprecatedzuse make_logger insteadcOt|i|Sr.rr~rrrr LazyLoggerrcOrr.rrrrrrKr)rr)r+r,rr-)rr6)r>r?rr6)rKrLr+r,rr)r>r?rr)r>r?r+r,rr)$ __future__rrr9r r^ functoolsrrrrrrMrbrcrar1r-r?r,r5r=rJrPrNFilterrVrXrYrrrxcompatrrrKrrrrs@   &    * "