o 38h@sddlZddlZddlZddlZddlZddlZddlZddlmZm Z m Z GdddZ e Z Gdddej Zdeded ed efd d ZdS) N)DictAnyOptionalc@seZdZdZddedefddZddZd d Zd d Zd e ee ffddZ ddededededede e ee ff ddZ dS) LogForwarderz Forwards logs to the Node.js API for real-time frontend display Uses a background thread to avoid blocking the main thread Napi_urlapi_keycCsT|ptjdd|_|ptjd|_t|_d|_d|_ t t |_ |dS)z Initialize the log forwarder Args: api_url: URL of the Node.js API logs endpoint api_key: Internal API key for authentication NODEJS_API_URLz"http://localhost:5000/internal-apiINTERNAL_API_KEYFN)osenvirongetrrqueueQueue log_queue is_running worker_threadlogging getLogger__name__loggerstart)selfrrr8/var/www/html/minaions-tender/ai-engine/log_forwarder.py__init__s   zLogForwarder.__init__cCs<|jsd|_tj|jdd|_|j|jddSdS)z'Start the worker thread to process logsT)targetdaemonzLogForwarder startedN)r threadingThread _process_logsrrrinforrrrr"s  zLogForwarder.startcCs.d|_|jr|jjdd|jddSdS)zStop the worker threadFtimeoutzLogForwarder stoppedN)rrjoinrr r!rrrstop*s zLogForwarder.stopc Cs|jrEz!z |jjdd}Wn tjyYWqw|||jWnty?}z|j d|WYd}~nd}~ww|jsdSdS)z4Process logs from the queue and send them to the APIr#zError processing log: N) rrr r Empty _send_log task_done Exceptionrerror)r log_entryerrrr1s zLogForwarder._process_logsr-c Csz*tj|jd||jdddd}|jdkr(|jd|jd|jWd SWd StyF}z|jd |WYd }~d Sd }~ww) za Send log to API Args: log_entry: Log entry to send z /logs/entryzapplication/json)zx-internal-api-keyz Content-Typer")jsonheadersr$zFailed to send log to API:  zError sending log to API: N) requestspostrr status_coderwarningtextr+)rr-responser.rrrr)Cs   " zLogForwarder._send_logr tenant_id process_id process_typemessageleveldetailsc Cs4||||||pitdtd}|j|dS)am Add log to queue Args: tenant_id: Tenant ID process_id: Process ID (tender_id, analysis_id, etc.) process_type: Process type (discovery, analysis, bid_generation) message: Log message level: Log level (debug, info, warning, error) details: Additional details z%Y-%m-%dT%H:%M:%S.%fZ)r9r:r;r<r=r> timestampN)timestrftimegmtimerput)rr9r:r;r<r=r>r-rrrlogZs zLogForwarder.log)NN)r N)r __module__ __qualname____doc__strrrr&rrrr)rrDrrrrr srcs4eZdZd dededeffdd ZddZZS) ApiLogHandlerNr9r:r;cs t||_||_||_dS)N)superrr9r:r;)rr9r:r; __class__rrrxs  zApiLogHandler.__init__cCs|jr |jr |js dS|j}||}|j|j|jd}|j r2t |j d|d<| |j |d<t j |j|j|j|||ddS)N) logger_namepathliner' exception traceback)r9r:r;r<r=r>)r9r:r; levelnamelowerformatnamepathnamelinenoexc_inforHformatException log_forwarderrD)rrecordr=r<r>rrremit~s&   zApiLogHandler.emit)NNN)rrErFrHrr\ __classcell__rrrKrrIwsrIrMr9r:r;cCsDt|}t|||}|tjtd}|||||S)z Set up API logging for a specific logger Args: logger_name: Logger name tenant_id: Tenant ID process_id: Process ID process_type: Process type Returns: logger: Configured logger z %(message)s)rrrIsetLevelINFO Formatter setFormatter addHandler)rMr9r:r;rhandler formatterrrrsetup_api_loggings     re)r rr3r/rr r@typingrrrrrZHandlerrIrHrerrrrsj(