core/conf.py

import os
from dotenv import load_dotenv
from functools import lru_cache
from typing import Any, Literal
from pydantic import model_validator
from pydantic_settings import BaseSettings, SettingsConfigDict

from core.path_conf import BASE_PATH

load_dotenv()


class Settings(BaseSettings):

    model_config = SettingsConfigDict(
        env_file=f"{BASE_PATH}/.env",
        env_file_encoding="utf-8",
        extra="ignore",
        case_sensitive=True,
    )

    ENVIRONMENT: Literal["dev", "pro"]

    API_BASE_URL: str
    API_BASE_URL_VATO: str
    API_BASE_URL_ACCESS_TOKEN: str

    # DATABASE_TYPE: Literal['mysql', 'postgresql']
    # DATABASE_HOST: str
    # DATABASE_PORT: int
    # DATABASE_USER: str
    # DATABASE_PASSWORD: str

    # MODEl NER
    NER_MODEL_DIR: str = "app/ner/models/ner"
    TASK_NAME: str = "token-classification"
    ACCELERATOR: str = "ort"
    DEVICE: str = "cpu"
    TRUNCATE: bool = True
    MAX_LENGTH: int = 512

    # FastAPI
    FASTAPI_API_V1_PATH: str = "/api/v1"
    FASTAPI_TITLE: str = "FastAPI"
    FASTAPI_VERSION: str = "0.0.1"
    FASTAPI_DESCRIPTION: str = "FastAPI"
    FASTAPI_DOCS_URL: str = "/docs"
    FASTAPI_REDOC_URL: str = "/redoc"
    FASTAPI_OPENAPI_URL: str | None = "/openapi"
    FASTAPI_STATIC_FILES: bool = True

    DATABASE_ECHO: bool = False
    DATABASE_POOL_ECHO: bool = False
    DATABASE_SCHEMA: str = "fba"
    DATABASE_CHARSET: str = "utf8mb4"

    CORS_ALLOWED_ORIGINS: list[str] = [
        "http://127.0.0.1:8000",
        "http://localhost:5173",
    ]
    CORS_EXPOSE_HEADERS: list[str] = [
        "X-Request-ID",
    ]

    MIDDLEWARE_CORS: bool = True
    MIDDLEWARE_ACCESS: bool = True

    DATETIME_TIMEZONE: str = "Asia/Ho_Chi_Minh"
    DATETIME_FORMAT: str = "%Y-%m-%d %H:%M:%S"

    UPLOAD_READ_SIZE: int = 1024
    UPLOAD_IMAGE_EXT_INCLUDE: list[str] = ["jpg", "jpeg", "png", "gif", "webp"]
    UPLOAD_IMAGE_SIZE_MAX: int = 5 * 1024 * 1024  # 5 MB
    UPLOAD_VIDEO_EXT_INCLUDE: list[str] = ["mp4", "mov", "avi", "flv"]
    UPLOAD_VIDEO_SIZE_MAX: int = 20 * 1024 * 1024  # 20 MB

    TRACE_ID_REQUEST_HEADER_KEY: str = "X-Request-ID"

    # 日志
    LOG_CID_DEFAULT_VALUE: str = "-"
    LOG_CID_UUID_LENGTH: int = 32  # 日志 correlation_id 长度，必须小于等于 32
    LOG_STD_LEVEL: str = "INFO"
    LOG_ACCESS_FILE_LEVEL: str = "INFO"
    LOG_ERROR_FILE_LEVEL: str = "ERROR"
    LOG_STD_FORMAT: str = (
        "<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</> | <lvl>{level: <8}</> | "
        "<cyan> {correlation_id} </> | <lvl>{message}</>"
    )
    LOG_FILE_FORMAT: str = (
        "<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</> | <lvl>{level: <8}</> | "
        "<cyan> {correlation_id} </> | <lvl>{message}</>"
    )
    LOG_ACCESS_FILENAME: str = "fba_access.log"
    LOG_ERROR_FILENAME: str = "fba_error.log"

    OPERA_LOG_PATH_EXCLUDE: list[str] = [
        "/favicon.ico",
        "/docs",
        "/redoc",
        "/openapi",
    ]

    @model_validator(mode="before")
    @classmethod
    def check_env(cls, values: Any) -> Any:
        if values.get("ENVIRONMENT") == "pro":
            values["FASTAPI_OPENAPI_URL"] = None
            values["FASTAPI_STATIC_FILES"] = False
        return values


@lru_cache
def get_settings() -> Settings:
    return Settings()


settings = get_settings()