first add
This commit is contained in:
96
backend/app/core/config.py
Normal file
96
backend/app/core/config.py
Normal file
@@ -0,0 +1,96 @@
|
||||
"""应用配置"""
|
||||
from typing import Optional
|
||||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||||
|
||||
|
||||
class Settings(BaseSettings):
|
||||
"""应用配置"""
|
||||
|
||||
model_config = SettingsConfigDict(
|
||||
env_file=".env",
|
||||
env_file_encoding="utf-8",
|
||||
case_sensitive=False,
|
||||
)
|
||||
|
||||
# ============== 应用基础配置 ==============
|
||||
APP_NAME: str = "BtToxin Pipeline"
|
||||
APP_VERSION: str = "1.0.0"
|
||||
DEBUG: bool = False
|
||||
|
||||
# API 配置
|
||||
API_V1_PREFIX: str = "/api/v1"
|
||||
|
||||
# ============== 数据库配置 ==============
|
||||
POSTGRES_HOST: str = "localhost"
|
||||
POSTGRES_PORT: int = 5432
|
||||
POSTGRES_USER: str = "bttoxin"
|
||||
POSTGRES_PASSWORD: str = "bttoxin_password"
|
||||
POSTGRES_DB: str = "bttoxin_db"
|
||||
|
||||
@property
|
||||
def DATABASE_URL(self) -> str:
|
||||
"""数据库连接 URL"""
|
||||
return (
|
||||
f"postgresql://{self.POSTGRES_USER}:{self.POSTGRES_PASSWORD}"
|
||||
f"@{self.POSTGRES_HOST}:{self.POSTGRES_PORT}/{self.POSTGRES_DB}"
|
||||
)
|
||||
|
||||
# ============== Redis 配置 ==============
|
||||
REDIS_HOST: str = "localhost"
|
||||
REDIS_PORT: int = 6379
|
||||
REDIS_DB: int = 0
|
||||
REDIS_PASSWORD: Optional[str] = None
|
||||
|
||||
@property
|
||||
def REDIS_URL(self) -> str:
|
||||
"""Redis 连接 URL"""
|
||||
if self.REDIS_PASSWORD:
|
||||
return f"redis://:{self.REDIS_PASSWORD}@{self.REDIS_HOST}:{self.REDIS_PORT}/{self.REDIS_DB}"
|
||||
return f"redis://{self.REDIS_HOST}:{self.REDIS_PORT}/{self.REDIS_DB}"
|
||||
|
||||
# ============== Celery 配置 ==============
|
||||
CELERY_BROKER_URL: Optional[str] = None
|
||||
CELERY_RESULT_BACKEND: Optional[str] = None
|
||||
|
||||
def get_celery_broker_url(self) -> str:
|
||||
"""获取 Celery Broker URL"""
|
||||
return self.CELERY_BROKER_URL or self.REDIS_URL
|
||||
|
||||
def get_celery_result_backend(self) -> str:
|
||||
"""获取 Celery Result Backend URL"""
|
||||
return self.CELERY_RESULT_BACKEND or self.REDIS_URL
|
||||
|
||||
# ============== 工作空间配置 ==============
|
||||
WORKSPACE_BASE_PATH: str = "/data/jobs"
|
||||
TEMP_BASE_PATH: str = "/data/temp"
|
||||
MAX_UPLOAD_SIZE_MB: int = 500
|
||||
|
||||
# ============== Docker 配置 ==============
|
||||
DOCKER_IMAGE: str = "quay.io/biocontainers/bttoxin_digger:1.0.10--hdfd78af_0"
|
||||
DOCKER_PLATFORM: str = "linux/amd64"
|
||||
|
||||
# ============== S3 配置 ==============
|
||||
S3_ENDPOINT: Optional[str] = None
|
||||
S3_ACCESS_KEY: Optional[str] = None
|
||||
S3_SECRET_KEY: Optional[str] = None
|
||||
S3_BUCKET: str = "bttoxin-results"
|
||||
S3_REGION: str = "us-east-1"
|
||||
|
||||
# ============== 任务配置 ==============
|
||||
DEFAULT_THREADS: int = 4
|
||||
MAX_THREADS: int = 16
|
||||
TASK_TIMEOUT_SECONDS: int = 7200 # 2小时
|
||||
JOB_RETENTION_DAYS: int = 30
|
||||
|
||||
# ============== CORS 配置 ==============
|
||||
CORS_ORIGINS: list = ["http://localhost:3000", "http://localhost:5173"]
|
||||
|
||||
# ============== 日志配置 ==============
|
||||
LOG_LEVEL: str = "INFO"
|
||||
LOG_FORMAT: str = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||||
|
||||
|
||||
# 创建全局配置实例
|
||||
settings = Settings()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user