config.py 1.9 KB

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