Python导入环境变量

在进行Python编程时，我们经常需要使用环境变量来存储一些配置信息，比如数据库连接信息、API密钥等等。但是，在Python中默认情况下是无法读取环境变量的，需要手动进行导入。本文将介绍如何在Python中导入环境变量。

1. 了解环境变量

在Linux和Unix系统中，环境变量是一些系统级别的变量，用于存储一些配置信息，比如系统路径，用户信息等等。在Windows操作系统中也有类似的机制。

在Python中，可以通过os模块来读取和设置环境变量。

2. 读取环境变量

要读取环境变量，可以使用os.environ.get()方法。该方法接受一个字符串参数，表示要获取的环境变量的名称。如果该环境变量不存在，则返回None。

import os

db_host = os.environ.get('DB_HOST')
db_port = os.environ.get('DB_PORT')
db_user = os.environ.get('DB_USER')
db_password = os.environ.get('DB_PASSWORD')

上述代码中，我们使用os.environ.get()方法分别获取了四个环境变量的值。接下来就可以使用这些变量来连接数据库了。

3. 设置环境变量

要设置环境变量，可以使用os.environ字典。该字典的每个元素对应一个环境变量的名称和值。例如，要设置名为API_KEY的环境变量，可以使用以下代码：

import os

os.environ['API_KEY'] = 'your_api_key_here'

上述代码中，我们将API_KEY环境变量的值设置为your_api_key_here。

4. 从文件中导入环境变量

在实际项目中，我们通常会将环境变量的值存储在一个配置文件中，然后在代码中读取该文件并导入环境变量。

假设我们有一个名为.env的配置文件，包含以下内容：

DB_HOST=localhost
DB_PORT=5432
DB_USER=your_db_user
DB_PASSWORD=your_db_password

要从该文件中读取环境变量并导入，在Python中可以使用python-dotenv模块。要使用该模块，可以先使用pip命令进行安装：

pip install python-dotenv

然后，在代码中可以使用以下代码来从文件中读取环境变量并导入：

from dotenv import load_dotenv
import os

load_dotenv('.env')

db_host = os.environ.get('DB_HOST')
db_port = os.environ.get('DB_PORT')
db_user = os.environ.get('DB_USER')
db_password = os.environ.get('DB_PASSWORD')

上述代码中，我们首先使用load_dotenv()方法载入.env文件中的环境变量。然后，就可以像之前一样使用os.environ.get()方法来获取环境变量的值了。

5. 总结

本文介绍了如何在Python中导入环境变量。具体来说，我们了解了环境变量的概念，以及如何使用os模块读取和设置环境变量。我们还介绍了如何使用python-dotenv模块从配置文件中导入环境变量。希望本文能够帮助你更好地应用环境变量来存储配置信息。