o ui@s2ddlZddlZddlZGdddZddZdS)Nc@s0eZdZdZdZedddZedddZdS)Loggerz7 Logger class for managing logging operations. NcCs|jdur |||_|jS)aM Get the logger instance with the specified name. If it doesn't exist, create and cache it. Args: cls (type): The class type. name (str, optional): The name of the logger. Defaults to None, which uses the class name. Returns: logging.Logger: The logger instance. N)_logger init_logger)clsnamer7/home/ubuntu/sommelier/podcast-pipeline/utils/logger.py get_loggers zLogger.get_loggerc Cs|dur"d}dtjvr|dtjd}dtjvr"|dtjd}td|t|}|tjtdt }td t }tj d |d d t d }t d |d|d|d}| |||Gdddtj }|d }t} | ||| |S)a Initialize the logger, including file and console logging. Args: cls (type): The class type. name (str, optional): The name of the logger. Defaults to None. Returns: logging.Logger: The initialized logger instance. NmainSELF_ID_IDCUDA_VISIBLE_DEVICES_GPUzInitialize logger for z%Y-%m-%dz%H-%M-%Szlogs/T)exist_okz4%(asctime)s - %(name)s - %(levelname)s - %(message)s/-z.logcs eZdZdZfddZZS)z*Logger.init_logger..ColorFormatterzS Custom log formatter to add color to specific log levels. cs|jtjkrdt|jd|_n2|jtjkr"dt|jd|_n!|jtjkr3dt|jd|_n|jtjkrCdt|jd|_t |S)a Format the log record with color based on log level. Args: record (logging.LogRecord): The log record to format. Returns: str: The formatted log message. zzzzz) levelnologgingERRORstrmsgWARNINGINFODEBUGsuperformat)selfrecord __class__rrrHs     z1Logger.init_logger..ColorFormatter.format)__name__ __module__ __qualname____doc__r __classcell__rrrrColorFormatterCsr%)osenvironprintr getLoggersetLevelrtimestrftime localtimemakedirs Formatter FileHandler setFormatter addHandler StreamHandler) rrloggerlog_datelog_time formatterfhr%color_formatterchrrrr s4         zLogger.init_logger)N)r r!r"r#r classmethodr rrrrrr s rcsfdd}|S)z Decorator to log the execution time of a function. Args: func (callable): The function whose execution time is to be logged. Returns: callable: The wrapper function that logs the execution time of the original function. csDt}|i|}t}tdjd||d|S)Nz Function z took z seconds to execute)r+rr debugr )argskwargs start_timeresultend_timefuncrrwrapperqsztime_logger..wrapperr)rCrDrrBr time_loggerfs  rE)rr+r&rrErrrrs  ]