FastAPI 如何根据开发或生产环境设置环境变量

在本文中，我们将介绍如何在FastAPI中根据开发或生产环境设置环境变量的方法。

理解环境变量

首先，我们需要了解什么是环境变量。环境变量是操作系统或应用程序在运行时定义的一些值，可以在整个系统中使用。这些变量可以存储配置信息、API密钥、数据库连接等敏感信息。在不同的环境中，我们可能需要设置不同的环境变量。

创建配置文件

为了更好地管理环境变量，我们可以创建一个配置文件来存储不同环境的配置。在项目根目录下创建一个名为.env的文件，并在其中定义环境变量。例如：

# 开发环境配置
DB_HOST_DEV=localhost
DB_PORT_DEV=5432
DB_NAME_DEV=mydatabase

# 生产环境配置
DB_HOST_PROD=production_host
DB_PORT_PROD=5432
DB_NAME_PROD=mydatabase

在上面的示例中，我们定义了两组环境变量，分别用于开发环境和生产环境。

安装dotenv

要在FastAPI中加载.env文件中的环境变量，我们需要使用python-dotenv库。安装该库可以使用以下命令：

pip install python-dotenv

加载环境变量

下一步是在FastAPI应用程序中加载环境变量。我们可以在项目的主文件（例如main.py）中添加以下代码：

from dotenv import load_dotenv

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

现在，我们已经成功地加载了.env文件中定义的环境变量。

使用环境变量

一旦我们加载了环境变量，我们就可以在FastAPI应用程序的不同部分使用它们。假设我们要连接到一个数据库，我们可以在代码中使用环境变量来获取数据库连接的详细信息。以下是一个示例：

from fastapi import FastAPI
import os

app = FastAPI()

@app.get("/")
def read_root():
    db_host = os.getenv("DB_HOST_DEV")  # 获取开发环境的数据库主机
    db_port = os.getenv("DB_PORT_DEV")  # 获取开发环境的数据库端口
    db_name = os.getenv("DB_NAME_DEV")  # 获取开发环境的数据库名称

    # 在这里执行数据库连接和操作

    return {"message": "Hello FastAPI!"}

在上面的示例中，我们使用os.getenv()函数来获取环境变量的值，并将其用于数据库连接的配置。

切换环境

要在不同的环境之间切换，我们只需要修改.env文件中的环境变量。例如，要切换到生产环境，我们可以将.env文件中的DB_HOST_DEV、DB_PORT_DEV和DB_NAME_DEV的值修改为生产环境的配置：

# 开发环境配置
DB_HOST_DEV=development_host
DB_PORT_DEV=5432
DB_NAME_DEV=mydatabase

# 生产环境配置
DB_HOST_PROD=production_host
DB_PORT_PROD=5432
DB_NAME_PROD=mydatabase

总结

在本文中，我们学习了如何在FastAPI中根据开发或生产环境设置环境变量。首先，我们创建了一个.env文件来存储不同环境的配置。然后，我们使用python-dotenv库加载了.env文件中的环境变量，并在应用程序中使用它们。通过修改.env文件中的环境变量值，我们可以轻松地在不同的环境之间切换配置。

有了这种灵活的环境变量设置方式，我们可以更好地管理和保护应用程序中的敏感信息，并根据不同的环境需求进行配置。

希望本文对于你理解FastAPI中如何根据开发或生产环境设置环境变量有所帮助。谢谢阅读！