1234567891011121314151617 |
- import os
- from dotenv import load_dotenv
- # 支持 .env 文件自动加载
- dotenv_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), ".env")
- load_dotenv(dotenv_path)
- def get_env(key: str, default: str = "") -> str:
- """获取环境变量"""
- return os.getenv(key, default)
- def get_int_env(key: str, default: int = 0) -> int:
- """获取整数类型环境变量"""
- try:
- return int(os.getenv(key, default))
- except ValueError:
- return default
|