env_loader.py 486 B

1234567891011121314151617
  1. import os
  2. from dotenv import load_dotenv
  3. # 支持 .env 文件自动加载
  4. dotenv_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), ".env")
  5. load_dotenv(dotenv_path)
  6. def get_env(key: str, default: str = "") -> str:
  7. """获取环境变量"""
  8. return os.getenv(key, default)
  9. def get_int_env(key: str, default: int = 0) -> int:
  10. """获取整数类型环境变量"""
  11. try:
  12. return int(os.getenv(key, default))
  13. except ValueError:
  14. return default