base.py 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. import os
  2. from dotenv import load_dotenv
  3. from core.utils.path_utils import project_root,log_dir
  4. from pydantic import BaseSettings, Field, AnyUrl
  5. # 在 Settings 类之前加载 .env 文件
  6. load_dotenv(os.path.join(project_root,".env"))
  7. class Settings(BaseSettings):
  8. # 环境标识
  9. ENV: str = "prod" # prod/dev
  10. ENABLE_ALIYUN_LOG: bool = True
  11. # 日志配置
  12. LOG_DIR: str = log_dir
  13. LOG_LEVEL: str = "INFO"
  14. # 阿里云数据库配置 (RDS)
  15. DB_HOST: str = Field(..., env="DB_HOST")
  16. DB_PORT: int = Field(3306, env="DB_PORT")
  17. DB_USER: str = Field(..., env="DB_USER")
  18. DB_PASSWORD: str = Field(..., env="DB_PASSWORD")
  19. DB_NAME: str = Field(..., env="DB_NAME")
  20. DB_CHARSET: str = Field(..., env="DB_CHARSET")
  21. DB_POOL_SIZE: int = 20
  22. DB_POOL_RECYCLE: int = 3600
  23. # 阿里云RocketMQ配置
  24. ROCKETMQ_ENDPOINT: AnyUrl = Field(..., env="ROCKETMQ_ENDPOINT")
  25. ROCKETMQ_ACCESS_KEY_ID: str = Field(..., env="ROCKETMQ_ACCESS_KEY_ID")
  26. ROCKETMQ_ACCESS_KEY_SECRET: str = Field(..., env="ROCKETMQ_ACCESS_KEY_SECRET")
  27. ROCKETMQ_INSTANCE_ID: str = Field(..., env="ROCKETMQ_INSTANCE_ID")
  28. ROCKETMQ_WAIT_SECONDS: int = 10
  29. ROCKETMQ_BATCH: int = 1
  30. # 飞书配置
  31. FEISHU_APPID: str = Field(..., env="FEISHU_APPID")
  32. FEISHU_APPSECRET: str = Field(..., env="FEISHU_APPSECRET")
  33. # 连接池配置
  34. CONNECTION_TIMEOUT: int = 10
  35. REQUEST_TIMEOUT: int = 30
  36. # 阿里云日志
  37. ALIYUN_ACCESS_KEY_ID: str = Field(..., env="ALIYUN_ACCESS_KEY_ID")
  38. ALIYUN_ACCESS_KEY_SECRET: str = Field(..., env="ALIYUN_ACCESS_KEY_SECRET")
  39. class Config:
  40. env_file = ".env"
  41. env_file_encoding = 'utf-8'
  42. case_sensitive = False
  43. # @property
  44. # def database_url(self) -> str:
  45. # """生成安全的数据库连接字符串"""
  46. # return f"mysql+asyncmy://{self.DB_USER}:{self.DB_PASSWORD}@{self.DB_HOST}:{self.DB_PORT}/{self.DB_NAME}?charset={self.DB_CHARSET}"
  47. settings = Settings()