在 Python 中通过读取 .env 文件获得环境变量
在编写 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") # 获取环境变量