PCNUSMSE/transcript_service
0
1o
2 E��h� � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZm Z m3Z4 ddl Z zddl m
Z
ddlmZ dZW n eyA dZY nw d d5lmZ G d d � d ej�ZG d
d� d�ZG dd� d�Ze� Zddedefdd�Zdde6e dedefdd�ZdS ) u] 日志管理模块7 8提供结构化日志记录功能，支持任务跟踪和状态记录。9� N)�Path)�Any�Dict�Optional)�Console)� RichHandlerTF� )�10get_configc s( e Zd ZdZ� fdd�Zdd� Z� ZS )�TaskContextFilteruV 任务上下文过滤器11 12 为日志记录添加任务ID上下文信息。13 c s t � �� d| _d S )N�system)�super�__init__�task_id��self�� __class__� �A/Users/chengpeng/MyProject/transcript_service/src/utils/logger.pyr
s 1415zTaskContextFilter.__init__c C s> t |d�st| dd�|_dS t|dd�du r t| dd�|_dS )u 添加任务ID到日志记录r r NT)�hasattr�getattrr )r �recordr r r �filter" s 16�zTaskContextFilter.filter)�__name__�17__module__� __qualname__�__doc__r
r �
__classcell__r r r r r18 s r19 c @ s� e Zd ZdZd defdd�Zdd� Zdd � Zd20efd d �Zd
d� Z defdd�Z21defdd�Z defdd�Z defdd�Z
defdd�Zdefdd�Zd S ) �Loggeru 日志管理器�transcript_service�namec C s| || _ t� | _| �� t�|�| _t� | _| jj D ]}|�22| j� qt�� }|j D ]}t dd� |j D ��s;|�23| j� q)dS )uZ 初始化日志管理器24 25 Args:26 name: 日志器名称27 c s s � | ]}t |t�V qd S )N)�28isinstancer29 )�.0�fr r r � <genexpr>B s � z"Logger.__init__.<locals>.<genexpr>N)
r r �config�_setup_logging�logging� getLogger�loggerr30 � task_filter�handlers� addFilter�any�filters)r r �handler� root_loggerr r r r
/ s 31 ��zLogger.__init__c C s� | j �� }| j �� d d }|�� rTt|ddd��
}t�|�}W d � n1 s)w Y |�di ��� D ]\}}d|v rKt |t32|d �j �|d< q6t j �
|� dS | �� dS ) u 设置日志配置r% z logging.yaml�r�utf-8)�encodingNr+ �filename)r% � get_logs_dir�get_project_root�exists�open�yaml� safe_load�get�items�strr r r' Z33dictConfig�_setup_default_logging)r �logs_dir� config_file�fileZlogging_configZ handler_nameZhandler_configr r r r& E s 34 �� zLogger._setup_loggingc C s� t rt� }t|dddd�}nt�� }tjddd�}|�|� |�| jj j35r(tj ntj � | j�
� d }tjj|ddd d36�}|�tj � tjddd�}|�|� t�� }|�| jj j37r]tj ntj � |�|� |�|� d S ) u 设置默认日志配置T)�consoleZ show_timeZ show_path�markupz4[%(asctime)s] [%(levelname)s] [%(name)s] %(message)sz%Y-%m-%d %H:%M:%S)�datefmtzapp.logi � � r2 )ZmaxBytesZ backupCountr3 N)�RICH_AVAILABLEr r r' �
StreamHandler� Formatter� setFormatter�setLevelr% �app�debug�DEBUG�INFOr5 r+ ZRotatingFileHandlerr( �38addHandler)r rB Zconsole_handlerZconsole_formatter�log_fileZ file_handler� formatterr0 r r r r> [ s@ ��39 � �40 41z Logger._setup_default_loggingr c C s || j _dS �uR 设置当前任务ID42 43 Args:44 task_id: 任务ID45 N�r* r �r r r r r � set_task_id� s zLogger.set_task_idc C s d| j _dS )� 清除当前任务IDr NrS r r r r �
clear_task_id� s zLogger.clear_task_id�messagec K � | j j||d� dS )� 记录调试信息��extraN�r) rL �r rX �kwargsr r r rL � � z Logger.debugc K rY )� 记录一般信息r[ N�r) �infor^ r r r rc � r` z Logger.infoc K rY )� 记录警告信息r[ N�r) �warningr^ r r r rf � r` zLogger.warningc K rY )� 记录错误信息r[ N�r) �errorr^ r r r ri � r` z Logger.errorc K rY )� 记录严重错误r[ N�r) �criticalr^ r r r rl � r` zLogger.criticalc K rY )u* 记录异常信息（包含堆栈跟踪）r[ N�r) � exceptionr^ r r r rn � r` zLogger.exceptionN�r )r r r r r= r
r& r> rU rW rL rc rf ri rl rn r r r r r , s ,r c @ s� e Zd ZdZd dee defdd�Zdd � Zd46d � Zd efd
d�Z d efdd�Z47d efdd�Z d efdd�Z d efdd�Z
d efdd�Zdefdd�Zdd � ZdS ) �48TaskLoggeruQ 任务日志记录器49 50 为特定任务提供上下文日志记录。51 Nr r � logger_namec C s6 |p t t�� �dd� | _t|�| _| j�| j� dS )u� 初始化任务日志记录器52 53 Args:54 task_id: 任务ID，如果为None则自动生成55 logger_name: 基础日志器名称56 N� )r= �uuid�uuid4r r r) rU )r r rq r r r r
� s 57zTaskLogger.__init__c C s | S )u 进入上下文管理器r r r r r � __enter__� s zTaskLogger.__enter__c C � | j �� dS )u 退出上下文管理器N�r) rW )r �exc_type�exc_val�exc_tbr r r �__exit__� � zTaskLogger.__exit__rX c K � | j j|fi |�� dS )rZ Nr] r^ r r r rL � � zTaskLogger.debugc K r} )ra Nrb r^ r r r rc � r~ zTaskLogger.infoc K r} )rd Nre r^ r r r rf � r~ zTaskLogger.warningc K r} )rg Nrh r^ r r r ri � r~ zTaskLogger.errorc K r} )rj Nrk r^ r r r rl � r~ zTaskLogger.criticalc K r} )u 记录异常信息Nrm r^ r r r rn � r~ zTaskLogger.exceptionc C s | j �|� dS rR )r) rU rT r r r rU � s zTaskLogger.set_task_idc C rv )rV Nrw r r r r rW � r| zTaskLogger.clear_task_id�Nr )r r r r r r= r
ru r{ rL rc rf ri rl rn rU rW r r r r rp � s rp r r �returnc C s t | �S )uo 获取日志实例58 59 Args:60 name: 日志器名称61 62 Returns:63 日志实例64 )r )r r r r �65get_logger� s r� r rq c C s66 t | |�S )u� 获取任务日志实例67 68 Args:69 task_id: 任务ID70 logger_name: 日志器名称71 72 Returns:73 任务日志实例74 )rp )r rq r r r �get_task_logger� s 75 76r� ro r ) r r' Zlogging.configZlogging.handlersrs �pathlibr �typingr r r r9 Z rich.consoler Z rich.loggingr rF � ImportError� core.configr �Filterr77 r rp r) r= r� r� r r r r �<module> s. � ?