config.py 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. # -*- coding: utf-8 -*-
  2. from typing import Any, Dict, Optional
  3. from pydantic import (
  4. AnyHttpUrl,
  5. BaseSettings,
  6. DirectoryPath,
  7. PostgresDsn,
  8. SecretStr,
  9. validator,
  10. )
  11. class Settings(BaseSettings):
  12. # SERVER_NAME: str
  13. # SERVER_HOST: AnyHttpUrl
  14. PLATFORM_HOST: AnyHttpUrl
  15. PLATFORM_SECRET: SecretStr
  16. TRANSFER_HOST: AnyHttpUrl
  17. # WEATHER_HOST: AnyHttpUrl
  18. # HTTPS_PROXY: AnyHttpUrl
  19. TENCENT_NLP_ENDPOINT: str
  20. TENCENT_SECRET_ID_V1: str
  21. TENCENT_SECRET_KEY_V1: str
  22. DUCKLING_HOST: AnyHttpUrl
  23. # PROJECT_DIR: DirectoryPath
  24. # LOGS_DIR: DirectoryPath
  25. ML_MODELS_DIR: DirectoryPath
  26. PROJECT_NAME: str
  27. REDIS_HOST: str
  28. REDIS_PORT: int
  29. REDIS_DB: int
  30. REDIS_PASSWORD: str
  31. MILVUS_HOST: str
  32. MILVUS_PORT: str
  33. POSTGRES_SERVER: str
  34. POSTGRES_USER: str
  35. POSTGRES_PASSWORD: str
  36. POSTGRES_DB: str
  37. SQLALCHEMY_DATABASE_URL: Optional[PostgresDsn] = None
  38. @validator("SQLALCHEMY_DATABASE_URL", pre=True)
  39. def assemble_db_connection(cls, v: Optional[str], values: Dict[str, Any]) -> Any:
  40. if isinstance(v, str):
  41. return v
  42. return PostgresDsn.build(
  43. scheme="postgresql",
  44. user=values.get("POSTGRES_USER"),
  45. password=values.get("POSTGRES_PASSWORD"),
  46. host=values.get("POSTGRES_SERVER"),
  47. path=f'/{values.get("POSTGRES_DB") or ""}',
  48. )
  49. class Config:
  50. case_sensitive = True
  51. env_file = ".env"
  52. settings = Settings()