在 Python 中通过读取 .env 文件获得环境变量

Undefined443 / 2024-02-01 / 原文

在编写 Python 脚本时,我们会使用一些私密数据,如调用 API 时使用的 token。为了避免隐私泄露,这些私密数据一般不直接写入脚本文件中。而是写入一个文件,并通过读取文件的方式获取私密数据内容。这个文件就是 .env 文件。

.env 文件中以环境变量的形式存储着我们的私密数据。比如:

OPENAI_API_KEY=sk-sfdsfw9r9erthgsyorwtogsgjft
GOOGLE_API_KEY=jGHODSFljdsfsodhfsdjgshgoihsdgioh

引入所需的包:

pip install python-dotenv

在 Python 脚本中读取这些环境变量:

from os import environ  # 用于获取环境变量
from dotenv import load_dotenv  # 用于将 .env 文件的内容加载到环境变量中

load_dotenv()  # 从同级目录的 .env 文件中加载环境变量

OPENAI_API_KEY=environ.get("OPENAI_API_KEY")  # 获取环境变量