103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
"""
|
||
数据库配置模块
|
||
使用pydantic进行配置验证和管理
|
||
"""
|
||
|
||
from typing import Optional, List
|
||
from pydantic_settings import BaseSettings
|
||
from functools import lru_cache
|
||
from pydantic import PostgresDsn, field_validator
|
||
from pydantic_core.core_schema import FieldValidationInfo
|
||
|
||
|
||
class Settings(BaseSettings):
|
||
"""应用配置类"""
|
||
|
||
RTSP_ENABLED: bool = True
|
||
|
||
# API配置
|
||
API_V1_PREFIX: str = "/api/v1"
|
||
PROJECT_NAME: str = "algorithm-service"
|
||
PROJECT_VERSION: str = "1.0.0"
|
||
PROJECT_DESCRIPTION: str = "algorithm-service"
|
||
BACKEND_CORS_ORIGINS: List[str] = ["http://localhost:3000", "http://localhost:8000"]
|
||
|
||
|
||
# 数据库配置
|
||
DATABASE_HOST: str = "localhost"
|
||
DATABASE_PORT: int = 5432
|
||
DATABASE_USER: str = "postgres"
|
||
DATABASE_PASSWORD: str = "yipai123"
|
||
DATABASE_NAME: str = "pmms"
|
||
DATABASE_SCHEMA: str = "public"
|
||
|
||
# 连接池配置
|
||
DATABASE_POOL_SIZE: int = 10
|
||
DATABASE_MAX_OVERFLOW: int = 20
|
||
DATABASE_POOL_RECYCLE: int = 3600 # 连接回收时间（秒）
|
||
DATABASE_ECHO: bool = False # SQL日志，生产环境设为False
|
||
|
||
# 应用配置
|
||
APP_NAME: str = "SurFaceFeature API"
|
||
APP_VERSION: str = "1.0.0"
|
||
DEBUG: bool = False
|
||
|
||
# 日志配置
|
||
LOG_LEVEL: str = "INFO"
|
||
LOG_FILE: Optional[str] = None
|
||
|
||
# 异步配置
|
||
ASYNC_MODE: bool = False
|
||
|
||
# 资源文件夹配置
|
||
FACE_REGISTER_IMAGE_RESOURCE_DIR: str = "D:/ruoyi/uploadPath/face"
|
||
VIDEO_RESOURCE_DIR: str = "D:/ruoyi/uploadPath/video"
|
||
FACE_CAL_FEATURE_TIMEOUT_HOURS: int = 10
|
||
FACE_MODEL_VERSION: int = 0 #insight_face_buffalo_l
|
||
FACE_USE_GPU: bool = True
|
||
FACE_USE_NPU: bool = False
|
||
SUR_CONFIG_TYPE_FACE: int = 0
|
||
SUR_CONFIG_SCOPE_GLOBAL: int = 0
|
||
|
||
# JWT配置（预留）
|
||
SECRET_KEY: str = "your-secret-key-here-change-in-production"
|
||
ALGORITHM: str = "HS256"
|
||
ACCESS_TOKEN_EXPIRE_MINUTES: int = 30
|
||
|
||
@property
|
||
def DATABASE_URL(self) -> str:
|
||
"""构建数据库连接URL"""
|
||
return f"postgresql://{self.DATABASE_USER}:{self.DATABASE_PASSWORD}@{self.DATABASE_HOST}:{self.DATABASE_PORT}/{self.DATABASE_NAME}"
|
||
|
||
@property
|
||
def ASYNC_DATABASE_URL(self) -> str:
|
||
"""构建异步数据库连接URL"""
|
||
return f"postgresql+asyncpg://{self.DATABASE_USER}:{self.DATABASE_PASSWORD}@{self.DATABASE_HOST}:{self.DATABASE_PORT}/{self.DATABASE_NAME}"
|
||
|
||
@field_validator("DATABASE_POOL_SIZE")
|
||
def validate_pool_size(cls, v):
|
||
"""验证连接池大小"""
|
||
if v < 1:
|
||
raise ValueError("DATABASE_POOL_SIZE must be at least 1")
|
||
if v > 100:
|
||
raise ValueError("DATABASE_POOL_SIZE cannot exceed 100")
|
||
return v
|
||
|
||
class Config:
|
||
env_file = ".env"
|
||
env_file_encoding = "utf-8"
|
||
case_sensitive = False
|
||
extra = "ignore"
|
||
|
||
|
||
@lru_cache()
|
||
def get_settings() -> Settings:
|
||
"""
|
||
获取配置单例
|
||
使用lru_cache避免重复加载.env文件
|
||
"""
|
||
return Settings()
|
||
|
||
|
||
# 导出配置实例
|
||
settings = get_settings() |