o h"@szddlmZddlmZmZmZmZddlmZm Z m Z m Z ddl Z ddl Z ddlmZGdddeZGdd d ZdS) )Enum)OptionalDictAnyUnion)ForeBackStyleinitN)datetimec@s eZdZdZdZdZdZdZdS)LogLevelN)__name__ __module__ __qualname__DEBUGINFOSUCCESSWARNINGERRORrrR/var/www/Befach/backend/venv/lib/python3.10/site-packages/crawl4ai/async_logger.pyr s r c@seZdZdZdddddddd d d d Zejejej ej ej ej ej ejejejiZd ejdd d dfdeedededeeeefdeeeefdef ddZdedefddZdedefddZdefdd Z d@d!ededed"eeeefdeeeefd#eef d$d%ZdAdedefd'd(ZdBdedefd*d+ZdCdedefd-d.ZdDdedefd0d1Z dEdedefd3d4Z! 5 6dFd7ed8ed9e"ded:ef d;d<Z# 2 6dGd7ed=eded:efd>d?Z$d S)H AsyncLoggerz Asynchronous logger with support for colored console output and file logging. Supports templated messages with colored components. u→u✓u↓u◆u■u●×u⋯uℹu⚠) INITREADYFETCHSCRAPEEXTRACTCOMPLETErrrrN Tlog_file log_level tag_widthiconscolorsverbosecCs`t||_||_||_|p|j|_|p|j|_||_|r.t j t j t j |dddSdS)ax Initialize the logger. Args: log_file: Optional file path for logging log_level: Minimum log level to display tag_width: Width for tag formatting icons: Custom icons for different tags colors: Custom colors for different log levels verbose: Whether to output to console T)exist_okN)r r%r&r' DEFAULT_ICONSr(DEFAULT_COLORSr)r*osmakedirspathdirnameabspath)selfr%r&r'r(r)r*rrr__init__*s  "zAsyncLogger.__init__tagreturncCsd|d|jdS)z#Format a tag with consistent width.[].)ljustr'r3r5rrr _format_tagJszAsyncLogger._format_tagcCs|j||jdS)z=Get the icon for a tag, defaulting to info icon if not found.r)r(getr;rrr _get_iconNszAsyncLogger._get_iconmessagecCs|jrUtddd}t|jddd5}|tjdtj d}t t D] }t |t r6||d}q)|d|d |d WddS1sNwYdSdS) z.Write a message to the log file if configured.z%Y-%m-%d %H:%M:%S.%fNazutf-8)encodingr7z]  )r%r nowstrftimeopenreplacerRESETr RESET_ALLvarsvalues isinstancestrwrite)r3r? timestampf clean_messagecolorrrr_write_to_fileRs  "zAsyncLogger._write_to_filelevelparams base_colorc Ks |j|jjkr dS|rSz,|jdi|}|r6|D]\} } | |vr5t|| } || | | tj}qWntyR} zd| d}t j }WYd} ~ nd} ~ ww|}|p[|j |} | | |d| |d|tj} |jsz|ddr~t| || dS)a Core logging method that handles message formatting and output. Args: level: Log level for this message message: Message template string tag: Tag for the message params: Parameters to format into the message colors: Color overrides for specific parameters base_color: Base color for the entire message Nz!LOGGING ERROR: Missing parameter z in message template force_verboseFr)valuer&formatitemsrNrHr rJKeyErrorr rr)r<r>r*r=printrT)r3rUr?r5rVr)rWkwargsformatted_messagekeyrS value_strelog_linerrr_log^s2  *zAsyncLogger._logrcK|jtj||fi|dS)zLog a debug message.N)rer rr3r?r5r_rrrdebugzAsyncLogger.debugrcKrf)zLog an info message.N)rer rrgrrrinforizAsyncLogger.inforcKrf)zLog a success message.N)rer rrgrrrsuccessrizAsyncLogger.successrcKrf)zLog a warning message.N)rer rrgrrrwarningrizAsyncLogger.warningrcKrf)zLog an error message.Nrer rrgrrrerrorrizAsyncLogger.errorr 2urlrktiming url_lengthc CsB|j|rtjntjd|||||d|rtjntjtjdddS)aH Convenience method for logging URL fetch status. Args: url: The URL being processed success: Whether the operation was successful timing: Time taken for the operation tag: Tag for the message url_length: Maximum length for URL in log z?{url:.{url_length}}... | Status: {status} | Time: {timing:.2f}s)rprrstatusrq)rsrq)rUr?r5rVr)N)rer rrrGREENREDYELLOW)r3rprkrqr5rrrrr url_statuss zAsyncLogger.url_statusrncCs |jtjd||||dddS)z Convenience method for logging error status. Args: url: The URL being processed error: Error message tag: Tag for the message url_length: Maximum length for URL in log z'{url:.{url_length}}... | Error: {error})rprrrn)rUr?r5rVNrm)r3rprnr5rrrrr error_statuss zAsyncLogger.error_status)NNN)r)r)r)r)r)r ro)rro)%rrr__doc__r,r rr LIGHTBLACK_EXrCYANrrtrrvrrur-rrNintrboolr4r<r>rTrrerhrjrkrlrnfloatrwrxrrrrrs    :  &r)enumrtypingrrrrcoloramarrr r timer.r r rrrrrs