如何在 Python 脚本中设置环境变量

在开发和运行 Python 脚本时，设置环境变量是一种常见的技术手段。环境变量可以为脚本提供配置信息、路径和其他系统级参数，使得脚本能够在不同的环境中正常运行。本文将详细介绍如何在 Python 脚本中设置环境变量，并提供一些示例和注意事项，帮助读者更好地理解和应用这一技术。

一、使用 os 模块设置环境变量

Python 提供了 os 模块，其中包含了设置环境变量的函数。下面是一个简单的示例，演示如何使用 os 模块设置环境变量：

import os

# 设置环境变量
os.environ['MY_VARIABLE'] = 'my_value'

# 获取环境变量
value = os.environ.get('MY_VARIABLE')
print(value)  # 输出：my_value

在上述示例中，我们使用 os.environ 字典来设置环境变量。通过将键值对添加到该字典中，我们可以在脚本中创建和修改环境变量。使用 os.environ.get() 方法可以获取特定环境变量的值。

二、示例：使用环境变量配置数据库连接信息

让我们以一个实际的示例来说明如何在 Python 脚本中使用环境变量来配置数据库连接信息。假设我们正在开发一个 Web 应用程序，需要连接到一个数据库。我们可以使用环境变量来存储数据库的主机名、用户名和密码，以便在不同环境中轻松配置数据库连接。

import os
import psycopg2

# 从环境变量获取数据库连接信息
db_host = os.environ.get('DB_HOST')
db_user = os.environ.get('DB_USER')
db_password = os.environ.get('DB_PASSWORD')

# 连接到数据库
conn = psycopg2.connect(host=db_host, user=db_user, password=db_password)

在上述示例中，我们使用了 os.environ.get() 方法从环境变量中获取数据库连接信息。通过在不同的环境中设置相应的环境变量，我们可以轻松地切换数据库连接配置，而无需修改脚本代码。

注意事项：

1. 在设置环境变量时，避免使用敏感信息或密码作为明文存储。可以使用加密或其他安全措施来保护敏感信息。
2. 环境变量的命名应具有描述性，以便在脚本中易于理解和使用。
3. 在使用环境变量之前，应该先检查其是否存在。可以使用 os.environ.get() 方法，并提供默认值以防环境变量不存在。
4. 当脚本需要使用多个环境变量时，建议将这些变量存储在一个配置文件中，并在脚本中读取该配置文件来设置环境变量。

结论：

在 Python 脚本中设置环境变量是一种强大的技术手段，可以使脚本在不同环境中灵活配置和运行。通过使用 os.environ 字典和 os.environ.get() 方法，我们可以方便地设置和获取环境变量的值。在实际应用中，我们可以利用环境变量来配置数据库连接、API 密钥、路径和其他系统级参数，从而增加脚本的可配置性和可移植性。在使用环境变量时，务必注意安全性和良好的命名规范，以确保脚本的可靠性和可维护性。

火焰兔

海报 链接