97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
"""应用配置"""
|
|
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()
|
|
|
|
|