U shU4@sdZddlZddlZddlmZddlmZmZmZddlZddlm Z ddl m Z m Z e eZGdddZeZdeeeeefd d d Zdeeeeeefd ddZeedddZeeeefdddZdeeeeeedddZd edddZdS)!z Centralized error logging configuration for the AI Test Case Generator application. This module provides error tracking functionality using MongoDB and email notifications. N)datetime)DictAnyOptional) MongoClient) MONGODB_URI MONGODB_DBc @s,eZdZddZd!eeeeeeefeeeefeeeefdddZd"e eeeefdd d Z d#eeeeeefd d dZ eedddZ eeeefdddZ d$eeeeeedddZe edddZe eeefdddZd%eeeeeeeeeefddd ZdS)& ErrorLoggerc Cszsz@ErrorLogger._send_critical_error_notification..rZapiZhttpZ API_FAILUREmongodbZDATABASE_ERRORauthenticationZ permissionZAUTHENTICATION_ERRORZ SYSTEM_ERROR)r error_messagerr&z&Critical error notification sent for: z,Failed to send critical error notification: N) utils.email_notifierrIr+r,rloweranygetr r rr) rr&rrIZcritical_error_typesr'Zcritical_keywordsZ is_criticalrrrrSrr/sf    z-ErrorLogger._send_critical_error_notification)daysr start_dateend_daterDc Csz|jdkrddiWSddlm}|r|rz>t|d}t|d|dd}d ||d i}||j}Wqtk r} z2td | t||d} d d | ii}W5d} ~ XYqXnt||d} d d | ii}|r||d <|j |} d|iddddididddiig} t |j | } d|iddddididddiig}t |j |}t |j |dddddd d dd}| | |||dWStk r} z(tdt| dt| iWYSd} ~ XYnXdS)a Get a summary of errors from MongoDB Args: days: Number of days to look back (used if start_date/end_date not provided) level: Filter by log level start_date: Start date in YYYY-MM-DD format end_date: End date in YYYY-MM-DD format Returns: Dictionary with error summary statistics NrzError collection not availabler) timedeltaz%Y-%m-%d)r]r)$gtez$ltz0Error parsing date range, falling back to days: rbrz$matchz$groupz$levelz$sum)_idcountz$sortrdz $error_type)rcrrrr ) total_errorserrors_by_levelerrors_by_type recent_errorsZ period_dayszError getting error summary: )r rr`strptimer]rr warningr!Zcount_documentslistZ aggregatefindsortlimitrr)rr]rr^r_r`start_datetime end_datetimeZfilter_criteriarZ cutoff_datergZlevel_pipelinerhZ type_pipelinerirjrrrget_error_summarys^         zErrorLogger.get_error_summary)rNNN)N)r N)NNN)r\NNN)r, __module__ __qualname__rrrrrr$rr0r2r8r=rCr-r/intrsrrrrr s: .    6  r r%cCst||dS)z-Capture an exception with additional context.N) error_loggerr0r%rrrr07sr0r r1cCst|||dS)z3Capture a message with specified level and context.N)rwr2r1rrrr2;sr2r3cCst||dS)zSet a tag for error events.N)rwr8r3rrrr8?sr8r9cCst||dS)z"Set context data for error events.N)rwr=r9rrrr=Csr=r>cCst|||dS)z"Set user context for error events.N)rwrCr>rrrrCGsrCrZ service_namecCs,td|tddtd|dS)z Initialize error logger (compatibility function for Sentry API) Args: service_name: Name of the service for tagging r versionz1.0.0z&Error logger initialized for service: N)rwr8r r rxrrrinit_error_loggerKs  rz)N)r N)NNN)r)__doc__loggingr"rZtypingrrrZpymongorconfig.settingsrr getLoggerr,r r rwrrr0r2r8r=rCrzrrrrs$   &