iQdZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl m Z ddl mZGddeeZGd d Zdadad efd Zdd ed efdZy)zt|}|jjddtj|}n#tjtj}|j||jj!|d|_y)Nz %(message)sT)parentsexist_ok)rrlogging getLoggerloggersetLevel_get_logging_level Formatterrparentmkdir FileHandler StreamHandlersysstdout setFormatter addHandlercorrelation_id)selfrrr formatterlog_pathhandlers r__init__zStructuredLogger.__init__s "''-  T44Y?@%%m4  H~H OO ! !$ ! >))(3G++CJJ7GY' w'#rreturnc rtjtjtjtjtjtjtj tj tj tj i}|j|tjS)z.Convert LogLevel enum to logging module level.)rDEBUGr$INFOWARNINGrCRITICALget)r3r level_maps rr(z#StructuredLogger._get_logging_level5si NNGMM MM7<<   goo NNGMM   w//  }}Y 55rlevelc ||jk\S)z@Check if a log level should be logged based on configured level.)r)r3r@s r _should_logzStructuredLogger._should_log@s&&rlog_typemessager2c tjjtj|j|j|j||xs |j d|}t j|tS)z Format log entry as JSON string.) timestamprCr@r&rDr2)default) datetimenowUTCvaluerr2jsondumpsstr)r3rCr@rDr2kwargs log_entrys r _format_logzStructuredLogger._format_logDsf"**..x||< [[ii,C0C0C   zz)S11rc |j|rD|j||||fi|}|j|}|jj ||yy)zInternal logging method.N)rBrQr(r&log)r3rCr@rDr2rO formatted_logrs r_logzStructuredLogger._logRsW   E ",D,,Xug~`Y_`M//6I KKOOI} 5 #rc L|j|tj||fi|y)zLog information.N)rUrr;r3rDr2rCrOs rinfozStructuredLogger.infoZs (HMM7NMfMrc L|j|tj||fi|y)zLog debug information.N)rUrr:rWs rdebugzStructuredLogger.debug^ (HNNG^NvNrc L|j|tj||fi|y)z Log warning.N)rUrr<rWs rwarningzStructuredLogger.warningbs! (H,,g~PPrc L|j|tj||fi|y)zLog error with details.N)rUrrrWs rr zStructuredLogger.errorfr[rc L|j|tj||fi|y)zLog critical error.N)rUrr=rWs rcriticalzStructuredLogger.criticaljs! (H--wQ&Qr exceptionc J| t|}t|jt|tjd}t |t r|j|d<|j|d<|jtjtj||fd|i|y)z$Log an exception with its traceback.N)exception_typeexception_message traceback error_codeexception_detailsra) rNtyperre format_exc isinstancerrfdetailsrUrrr)r3rarDr2rOexception_infos r log_exceptionzStructuredLogger.log_exceptionns ?)nG#9o66!$Y"--/  i .+4+?+?N< (2;2C2CN. / '--. 5) 5-3 5rmethodurlc z|jtjtjd|d||f||d|y)zLog incoming request.z Incoming z request to )rnroN)rUrrrr;)r3rnror2rOs r log_requestzStructuredLogger.log_requestsE '//8==IfX\RUQV2W C(.C C;A Cr status_code response_timec ~|jtjtjd|d|dd|f||d|y)zLog outgoing response.zResponse with status z in .3fs)rrrsN)rUrrrr;)r3rrrsr2rOs r log_responsezStructuredLogger.log_responsesK '""HMM( T-9LAN a-8  aY_ ar event_typec l|jtjtj||fd|i|y)zLog security-related events.rxN)rUrrrr<)r3rxrDr2rOs rlog_security_eventz#StructuredLogger.log_security_events4 '""H$4$4g~ 2& 2*0 2r operationdurationc ~|jtjtjd|d|dd|f||d|y)zLog performance metrics.z Operation z completed in rurv)r{r|N)rUrrrr;)r3r{r|r2rOs rlog_performancez StructuredLogger.log_performancesJ '%%x}}i[xnAF S+4x SKQ Sr workflow_idc n|jtjtj||f||d|y)zLog workflow-specific events.)rrxN)rUrrrr;)r3rrxrDr2rOs rlog_workflow_eventz#StructuredLogger.log_workflow_events9  '""HMM7N K(Z KCI Krc||_y)z0Set the correlation ID for this logger instance.N)r2)r3r2s rset_correlation_idz#StructuredLogger.set_correlation_ids ,r)N)NN)rrrrrr;rNr7intr(boolrBrrQrUrrXrZr]rr r` Exceptionrmrqfloatrwrzr~rrrrrrrs^S8@ W[#S#X#QT#0 6H 6 6''d' +/ 2G 2H 2s 2$' 2=@ 2$(6W6X66 68 ]+ )(IxH rrctTtjdd|d}tjddj}t|}t |||atS)zGet a workflow logger instance.WORKFLOW_LOG_FILEzlogs/z.logrr;)_workflow_loggerrrrrr)rr rrs rget_workflow_loggerrs\990E$t2DE +v6<<> ]+ +D)XF r)r )rrLr$r.rerHenumrpathlibrr entity.enumsrutils.exceptionsrrNrrrrrrrrrrsoB   !) c4 F-F-T+c3Cr