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

dotenv模块及其dotenv_values()函数在Python中的使用指南

发布时间:2024-01-19 17:58:51

dotenv模块是一个用于加载.env文件中的环境变量的Python库。.env文件通常用于存储项目中使用的敏感信息,如密钥、数据库连接字符串等。dotenv模块提供了dotenv_values()函数,通过该函数可以将.env文件中的变量加载到Python的环境变量中,使得它们可以在代码中轻松访问。

下面是dotenv_values()函数的使用指南,包括安装模块、导入模块、使用dotenv_values()函数以及一个完整的使用示例。

1. 安装dotenv模块

使用pip命令安装dotenv模块:

   $ pip install python-dotenv
   

2. 导入模块

在Python脚本中导入dotenv模块:

   from dotenv import dotenv_values
   

3. 使用dotenv_values()函数

dotenv_values()函数可以接受一个参数,用于指定.env文件的路径,默认情况下,它会在当前目录中查找.env文件。函数将返回一个包含.env文件中所有变量的字典。可以使用该函数获取.env文件中的变量并将其加载到Python的环境变量中。

   env = dotenv_values()
   

4. 完整使用示例

假设我们有一个.env文件,其中包含以下内容:

   API_KEY=my_api_key
   DB_HOST=localhost
   DB_USER=my_user
   DB_PASSWORD=my_password
   

在Python脚本中使用dotenv_values()函数加载.env文件中的变量:

   from dotenv import dotenv_values

   # 使用默认的.env文件路径
   env = dotenv_values()

   # 访问.env文件中的变量
   api_key = env["API_KEY"]
   db_host = env["DB_HOST"]
   db_user = env["DB_USER"]
   db_password = env["DB_PASSWORD"]

   # 使用变量
   print(f"API key: {api_key}")
   print(f"DB host: {db_host}")
   print(f"DB user: {db_user}")
   print(f"DB password: {db_password}")
   

运行上述代码将输出以下结果:

   API key: my_api_key
   DB host: localhost
   DB user: my_user
   DB password: my_password
   

以上就是dotenv模块及其dotenv_values()函数在Python中的使用指南,以及一个完整的使用示例。使用dotenv模块可以方便地将敏感信息从.env文件加载到Python的环境变量中,提高代码的安全性。