AxjUdZddlmZmZddlZddlZddlmZdZgdZ eddZ eee e d <d ee d efd Z d efdZdZGddZdS)a PodWorker | modules | logging.py Log Levels (Level - Value - Description) NOTSET - 0 - No logging is configured, the logging system is effectively disabled. DEBUG - 1 - Detailed information, typically of interest only when diagnosing problems. (Default) INFO - 2 - Confirmation that things are working as expected. WARN - 3 - An indication that something unexpected happened. ERROR - 4 - Serious problem, the software has not been able to perform some function. ) ContextVarTokenN)Optionali)NOTSETTRACEDEBUGINFOWARNERRORrunpod_batch_id)default _batch_idbatch_idreturnc6t|S)z6Set the batch ID associated with the current job task.)rset)rs TC:\Projects\Historical\venv\Lib\site-packages\runpod/serverless/modules/rp_logger.py _set_batch_idrs == " ""tokenc:t|dS)z7Restore the previous batch ID for the current job task.N)rreset)rs r_reset_batch_idrs OOErcVt|tr1|}|tvrt d||St|t r=|dks|t tkrt d|t|St d|)zB Checks the debug level and returns the debug level name. zInvalid debug level: r) isinstancestrupper LOG_LEVELS ValueErrorintlen) log_levels r_validate_log_levelr#!s)S!!OO%% J & &@Y@@AA A)S!!% q==IZ88@Y@@AA A)$$ 8Y88 9 99rc 2eZdZdZdZeejdejddZ dZ dZ dd Z d Z dd eefd Zdd eefd Zdd eefdZdd eefdZdZdd eefdZdS) RunPodLoggerzSingleton class for logging.NRUNPOD_LOG_LEVELRUNPOD_DEBUG_LEVELrcztj$t|t_tjSN)r%_RunPodLogger__instanceobject__new__)clss rr,zRunPodLogger.__new__@s(  " *&,nnS&9&9L #&&rcht||_|d|jdS)zn Set the debug level for logging. Can be set to the name or value of the debug level. zLog level set to N)r#levelinfo)self new_levels r set_levelzRunPodLogger.set_levelEs6 )33  2dj2233333rr c|jdkrdSt|j}|t|kr|dkrdSt|}tx}rd|d|}t |tkr@tdz}t |tz }d|d}|d||z|| dz}tj d r+|||d } ttj | d dS|r|d |}t| dd|d dS)z@ Log message to stdout if RUNPOD_DEBUG is true. rNTIPz [batchId=z] z ...TRUNCATED z CHARACTERS... RUNPOD_ENDPOINT_ID) requestIdmessager/T)flushz | z| )r/rindexrrgetr!MAX_MESSAGE_LENGTHosenvironprintjsondumpsljust) r1r9 message_leveljob_id level_indexrhalf_max_lengthtruncated_amounttruncation_notelog_jsons rlogzRunPodLogger.logMs : ! ! F &&tz22 ))-88 8 8]e=S=S Fg,, }} &8 87(77g77G w<<, , ,0A5O"7||.@@ R0@RRRO(()O;gFVFWFW>XX  :>>. / / %+-XXH $*X&&d 3 3 3 3 F  .--G--G $$Q''44744DAAAArct|}|ddt|dz zz|dz}||d|dS)zv Censors secrets for logging. Replaces everything except the first and last characters with * r*r6z: N)rr!r0)r1 secret_namesecretredacted_secrets rrQzRunPodLogger.secretps[ V )cS[[1_&==r J [55O5566666r request_idc4||d|dS)z debug log rNrLr1r9rSs rdebugzRunPodLogger.debugy  ':.....rc4||d|dS)z info log r NrUrVs rr0zRunPodLogger.info  &*-----rc4||d|dS)z warn log r NrUrVs rwarnzRunPodLogger.warnrZrc4||d|dS)z error log r NrUrVs rerrorzRunPodLogger.errorrXrc2||ddS)z tip log r5NrU)r1r9s rtipzRunPodLogger.tips %     rc4||d|dS)z4 trace log (buffered until flushed) rNrUrVs rtracezRunPodLogger.tracerXr)r Nr))__name__ __module__ __qualname____doc__r*r#r?r@r=r/r,r3rLrQrrrWr0r\r^r`rbrrr%r%6s_&&J     /CW M M    E ''' 444!!!!F777//#//// .. .... .. .... //#//// !!! //#//////rr%)rf contextvarsrrrBr?typingrr>rrr__annotations__rrr#r%rgrrrks   *)))))))  B B B '1z2CT'R'R'R :hsm $RRR#HSM#e#### 5 :::*e/e/e/e/e/e/e/e/e/e/r