47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
从 .env 文件加载环境变量
|
|||
|
|
"""
|
|||
|
|
import os
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
def load_env(env_path: Path | None = None) -> None:
|
|||
|
|
"""
|
|||
|
|
从 .env 文件加载环境变量
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
env_path: .env 文件路径（默认：当前目录的 .env）
|
|||
|
|
"""
|
|||
|
|
if env_path is None:
|
|||
|
|
# 查找 .env 文件
|
|||
|
|
possible_paths = [
|
|||
|
|
Path(".env"),
|
|||
|
|
Path(__file__).parent.parent / ".env",
|
|||
|
|
Path.cwd() / ".env",
|
|||
|
|
]
|
|||
|
|
for p in possible_paths:
|
|||
|
|
if p.exists():
|
|||
|
|
env_path = p
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
if not env_path or not env_path.exists():
|
|||
|
|
return # 没有 .env 文件，跳过
|
|||
|
|
|
|||
|
|
print(f"📄 加载环境变量：{env_path}")
|
|||
|
|
|
|||
|
|
with open(env_path, "r", encoding="utf-8") as f:
|
|||
|
|
for line in f:
|
|||
|
|
line = line.strip()
|
|||
|
|
# 跳过空行和注释
|
|||
|
|
if not line or line.startswith("#"):
|
|||
|
|
continue
|
|||
|
|
# 解析 KEY=VALUE
|
|||
|
|
if "=" in line:
|
|||
|
|
key, value = line.split("=", 1)
|
|||
|
|
key = key.strip()
|
|||
|
|
value = value.strip()
|
|||
|
|
# 只设置尚未存在的环境变量
|
|||
|
|
if key not in os.environ:
|
|||
|
|
os.environ[key] = value
|
|||
|
|
print(f" ✓ {key} = {value[:10]}...")
|