PCNUSMSE/transcript_service
0
1o
2 E��h � @ s� d Z ddlZddlmZ ddlmZmZmZ ddlZddl m3Z4 ddl m Z ddl m
Z
G dd � d e �ZG d5d � d e �ZG d d
� d
e �ZG dd� de �ZG dd� de �ZG dd� d�Ze� adefdd�Zddee defdd�ZdS )uv 配置管理模块6 7提供应用程序配置的加载和管理功能。8支持多环境配置和环境变量覆盖。9� N)�Path)�Any�Dict�Optional)�Field)� BaseSettings)�SettingsConfigDictc @ s� e Zd ZU dZedddddd�ZdZeed < d10Z eed < dZ11e ed < d
Z eed< dZ
eed< dZeed< dZeed< dZeed< dS )� AppConfigu 应用程序配置ZAPP_�.env�utf-8F�ignore�Z12env_prefixZenv_fileZenv_file_encodingZcase_sensitive�extrau 音频转文字服务�namez1.0.0�version�debugz 127.0.0.1�hosti� �portl �
max_file_size�d �max_files_count� �concurrent_tasksN)�__name__�13__module__� __qualname__�__doc__r � model_configr �str�__annotations__r r �boolr r �intr r r � r" r" �@/Users/chengpeng/MyProject/transcript_service/src/core/config.pyr s" 14 � r c @ s� e Zd ZU dZedddddd�Zedd d15�Zee d < edd d16�Z17ee d
< eddd18�Z ee d< eddd19�Z ee d< dZ
ee d< dZee d< dZee d< dZee d< dS )� OSSConfigu OSS配置ZOSS_r20 r Fr r
.u OSS服务端点�� description�endpointu 访问密钥ID�
access_key_idu 访问密钥密码�access_key_secretu 存储桶名称� bucket_name�, �upload_timeout� �url_expire_hoursz21temp/audio� temp_prefix� �auto_cleanup_daysN)r r r r r r r r' r r r( r) r* r, r! r. r/ r1 r" r" r" r# r$ $ s" 22 � r$ c @ s� e Zd ZU dZedddddd�Zedd d23�Zee d < d Z24ee d
< dZ ee d< dZ e
e d< dZe
e d< dZe
e d< ddgZee e d< dS )�DashScopeConfigu 阿里云百炼API配置Z25DASHSCOPE_r26 r Fr r
.u API密钥r% �api_keyz%https://dashscope.aliyuncs.com/api/v1�base_urlz
paraformer-v2�modelr+ �timeout� � max_retriesr � retry_delay�zh�en�language_hintsN)r r r r r r r r3 r r r4 r5 r6 r! r8 r9 r< �listr" r" r" r# r2 8 s 27 � r2 c @ sH e Zd ZU dZedddddd�ZdZeed < d28Z eed < d Z29eed
< dS )�30TaskConfigu 任务配置ZTASK_r31 r Fr r
� �status_check_intervali �max_processing_timei� �32queue_sizeN) r r r r r r r@ r! r rA rB r" r" r" r# r> K s 33 � r> c @ sT e Zd ZU dZedddddd�ZdZeed < d34Z eed < d Z35eed
< dZ e ed< dS )�
LoggingConfigu 日志配置ZLOGGING_r36 r Fr r
�INFO�levelZ37structured�formatZ10MB�
file_max_sizer � backup_countN)
r r r r r r rE r r rF rG rH r! r" r" r" r# rC Z s 38 � rC c @ sd e Zd ZdZddee fdd�Zdeeef fdd�Z de39fd d40�Z de41fd d �Z de42fd
d�Z
dS )�Configu 配置管理器N� environmentc C s~ |pt �dd�| _| �� | _tdi | j�di ���| _t� | _ t43� | _ t di | j�di ���| _
tdi | j�di ���| _dS )uz 初始化配置管理器44 45 Args:46 environment: 环境名称（development/production）47 Z ENVIRONMENTZ development�app�task�loggingNr" )�os�getenvrJ � _load_configZ _config_datar �getrK r$ �ossr2 Z dashscoper> rL rC rM )�selfrJ r" r" r# �__init__m s 48 zConfig.__init__�returnc C sx t t�jjjd d }|| j� d� }|�� s td|� ���t|ddd��
}t�|�W d � S 1 s5w Y dS ) u 加载配置文件�configZ environmentsz.yamlu 配置文件不存在: �rr )�encodingN) r �__file__�parentrJ �exists�FileNotFoundError�open�yamlZ safe_load)rS Z49config_dirZ config_file�filer" r" r# rP � s $�zConfig._load_configc C s t t�jjjS )u 获取项目根目录)r rY rZ )rS r" r" r# �get_project_root� s zConfig.get_project_rootc C � | � � d }|jdd� |S )u 获取日志目录�logsT��exist_ok�r` �mkdir)rS �logs_dirr" r" r# � get_logs_dir� � zConfig.get_logs_dirc C ra )u 获取临时文件目录�tempTrc re )rS Ztemp_dirr" r" r# � get_temp_dir� ri zConfig.get_temp_dir�N)r r r r r r rT r r rP r r` rh rk r" r" r" r# rI j s rI rU c C s t S )u 获取配置实例)rV r" r" r" r# �50get_config� s rm rJ c C s t | �atS )u 重新加载配置)rI rV )rJ r" r" r# �
reload_config� s rn rl )r rN �pathlibr �typingr r r r^ Zpydanticr Zpydantic_settingsr r r r$ r2 r> rC rI rV rm r rn r" r" r" r# �<module> s" 5