Python如何设置和读取环境变量

在Python中使用环境变量可以通过os模块的os.environ字典实现。1）获取环境变量使用os.environ.get()方法，2）设置环境变量通过字典赋值实现，3）结合配置文件可增强安全性，4）缓存环境变量可优化性能。

在Python中使用环境变量是一个非常实用的技能，尤其在处理配置管理和跨环境部署时。环境变量允许我们将配置信息从代码中分离出来，这样可以更灵活地管理应用程序的设置。让我们深入探讨一下如何在Python中使用环境变量，并分享一些我在实际项目中的经验和见解。

首先，Python通过os模块提供了访问环境变量的功能。我们可以通过os.environ字典来读取、设置和删除环境变量。下面是一个简单的例子，展示如何获取和设置一个环境变量：

import os

# 获取环境变量
db_url = os.environ.get('DATABASE_URL')
if db_url is None:
    print("DATABASE_URL 环境变量未设置")
else:
    print(f"DATABASE_URL: {db_url}")

# 设置环境变量
os.environ['MY_VAR'] = 'my_value'
print(f"MY_VAR: {os.environ['MY_VAR']}")

在这个例子中，我们使用os.environ.get()方法来安全地获取环境变量，如果变量不存在，它会返回None而不是抛出异常。设置环境变量则直接通过字典赋值来实现。

在实际项目中，我发现使用环境变量的一个常见场景是配置数据库连接。我曾在一个项目中使用Flask框架，通过环境变量来配置数据库URL，这样可以轻松地在开发、测试和生产环境之间切换数据库连接：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('DATABASE_URL', 'sqlite:///default.db')
db = SQLAlchemy(app)

这里的DATABASE_URL环境变量可以根据不同的部署环境来设置，而代码本身不需要修改。这大大提高了代码的灵活性和可维护性。

然而，使用环境变量也有一些需要注意的地方。首先，环境变量是全局的，如果不小心修改了某个环境变量，可能会影响到其他依赖它的程序。其次，环境变量通常是明文存储的，这在处理敏感信息（如数据库密码）时可能存在安全隐患。

为了解决安全问题，我通常会结合使用环境变量和配置文件。配置文件可以加密存储敏感信息，而环境变量则用于指示配置文件的位置或其他非敏感配置：

import os
import json

# 假设有一个配置文件 config.json
config_file = os.environ.get('CONFIG_FILE', 'config.json')

with open(config_file, 'r') as f:
    config = json.load(f)

# 使用配置文件中的数据库URL
app.config['SQLALCHEMY_DATABASE_URI'] = config['database_url']

这种方法既保持了灵活性，又增强了安全性。

在性能优化方面，使用环境变量不会对Python程序的运行速度产生显著影响，因为读取环境变量是一个非常快的操作。然而，如果你需要频繁地读取环境变量，可能考虑将它们缓存到一个全局变量中，以避免重复的I/O操作：

import os

# 缓存环境变量
CACHED_DB_URL = os.environ.get('DATABASE_URL')

def get_db_url():
    return CACHED_DB_URL

总的来说，使用环境变量在Python开发中是一个强大的工具，可以帮助我们更灵活地管理配置，提高代码的可维护性和安全性。通过结合环境变量和配置文件，我们可以在不同环境中轻松部署应用程序，同时保护敏感信息。希望这些经验和见解能对你在使用环境变量时有所帮助。