PostgreSQL 连接 PostgreSQL 数据库到 FastAPI

在本文中，我们将介绍如何在 FastAPI 框架中连接 PostgreSQL 数据库。我们将介绍必要的步骤和示例代码，以帮助您成功地连接 PostgreSQL 数据库到 FastAPI 应用程序。

准备工作

在开始之前，确保您已经安装了以下软件和工具：

Python 3.7 或更高版本
FastAPI 框架
psycopg2 包（用于连接 PostgreSQL 数据库）

您可以使用以下命令来安装 FastAPI 和 psycopg2：

pip install fastapi
pip install psycopg2

连接到 PostgreSQL 数据库

要连接到 PostgreSQL 数据库，您需要提供数据库的连接详细信息，如主机名，端口，用户名，密码和数据库名称。您可以在连接到数据库之前将这些信息保存在环境变量中，也可以直接在代码中硬编码这些信息。

以下是一个示例代码片段，展示了如何使用 psycopg2 包连接到 PostgreSQL 数据库：

import psycopg2


def connect_to_database():
    try:
        connection = psycopg2.connect(
            host="localhost",
            port="5432",
            user="your_username",
            password="your_password",
            database="your_database_name"
        )
        # 在此处执行您的数据库操作
        cursor = connection.cursor()
        cursor.execute("SELECT * FROM your_table_name")
        result = cursor.fetchall()
        print(result)
        # 关闭数据库连接
        cursor.close()
        connection.close()
    except (Exception, psycopg2.Error) as error:
        print("连接到数据库失败:", error)


# 调用连接数据库的函数
connect_to_database()

请确保将上述代码中的 your_username，your_password，your_database_name 和 your_table_name 替换为实际的值。

使用 FastAPI 构建 Web 服务

一旦成功连接到 PostgreSQL 数据库，您可以开始使用 FastAPI 构建 Web 服务，提供对数据库的访问和操作。

以下是一个使用 FastAPI 构建 Web 服务的示例代码：

from fastapi import FastAPI
import psycopg2

app = FastAPI()

@app.get("/data")
async def get_data():
    try:
        connection = psycopg2.connect(
            host="localhost",
            port="5432",
            user="your_username",
            password="your_password",
            database="your_database_name"
        )
        cursor = connection.cursor()
        cursor.execute("SELECT * FROM your_table_name")
        result = cursor.fetchall()
        return {"data": result}
    except (Exception, psycopg2.Error) as error:
        return {"error": str(error)}
    finally:
        cursor.close()
        connection.close()