欢迎访问宙启技术站
智能推送

使用Python的read_dotenv()函数从dotenv文件读取环境变量

发布时间:2024-01-19 22:00:15

在Python中,我们可以使用python-dotenv库中的read_dotenv()函数从.env文件中读取环境变量。

首先,我们需要安装python-dotenv库。可以使用以下命令来安装:

pip install python-dotenv

接下来,我们需要创建一个名为.env的文件,并在其中定义我们想要设置的环境变量。例如,在.env文件中,我们可以定义以下环境变量:

DB_HOST=localhost
DB_USER=myuser
DB_PASSWORD=mypassword

在我们的Python脚本中,我们需要导入dotenv库,并使用load_dotenv()函数来加载.env文件中的环境变量。然后,我们可以使用os.environ.get()函数来访问这些环境变量。代码如下所示:

import os
from dotenv import load_dotenv

# 加载.env文件中的环境变量
load_dotenv()

# 访问环境变量
db_host = os.environ.get('DB_HOST')
db_user = os.environ.get('DB_USER')
db_password = os.environ.get('DB_PASSWORD')

# 打印环境变量的值
print(f'DB_HOST: {db_host}')
print(f'DB_USER: {db_user}')
print(f'DB_PASSWORD: {db_password}')

运行以上代码,输出将会是:

DB_HOST: localhost
DB_USER: myuser
DB_PASSWORD: mypassword

这样,我们就成功地从.env文件中读取了环境变量并将它们存储到相应的变量中。

使用read_dotenv()函数从.env文件读取环境变量的好处是,我们可以将敏感信息(如数据库密码)保存在.env文件中,并将该文件添加到.gitignore中,以确保不会将这些敏感信息上传到代码仓库中。这可以增加代码的安全性。

需要注意的是,.env文件应该与Python脚本文件在同一个目录中,或者你可以使用绝对路径来指定.env的位置。

除了使用os.environ.get()函数,我们还可以使用os.getenv()函数来访问环境变量。它们之间的区别是,os.getenv()函数在环境变量不存在的情况下返回None,而os.environ.get()函数在环境变量不存在的情况下返回默认值(如果提供了默认值)。

希望这个例子能帮助你了解如何使用read_dotenv()函数从.env文件中读取环境变量。祝你编写Python代码时愉快!