FastAPI FastApi/Pydantic 从主表中访问多对一关系

在本文中，我们将介绍如何使用FastAPI和Pydantic从父表中访问多对一关系。我们将学习如何定义模型、创建路由和使用数据库进行查询操作。

阅读更多：FastAPI 教程

简介

在关系数据库中，多对一关系指的是一个父表中的多个记录对应于子表中的一个记录。在FastAPI中，我们可以使用Pydantic模型定义这种关系，并在路由中使用这些模型进行查询操作。接下来，我们将通过一个示例来演示如何实现这种多对一关系检索。

示例

假设我们有两个表，分别为User和Address。每个用户可以有多个地址，但每个地址只能属于一个用户。下面是这两个表的定义示例：

from sqlalchemy import Column, ForeignKey, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship

Base = declarative_base()

class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, index=True)
    addresses = relationship("Address", back_populates="user")

class Address(Base):
    __tablename__ = "addresses"

    id = Column(Integer, primary_key=True, index=True)
    street = Column(String, index=True)
    user_id = Column(Integer, ForeignKey("users.id"))

    user = relationship("User", back_populates="addresses")

在上面的代码中，我们使用了SQLAlchemy来定义表和关系。User表包含了一个与Address表的一对多关系，并使用了relationship属性进行映射。同样地，Address表包含了一个与User表的多对一关系映射。

接下来，我们可以使用FastAPI来定义路由，并从父表中访问多对一关系。以下是一个简单的示例：

from fastapi import FastAPI
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

from models import Base, User, Address

app = FastAPI()

# 创建数据库引擎和Session
SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"
engine = create_engine(SQLALCHEMY_DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

# 创建表
Base.metadata.create_all(bind=engine)

@app.get("/users/{user_id}/addresses")
def read_user_addresses(user_id: int):
    db = SessionLocal()

    # 通过user_id查询用户信息
    user = db.query(User).filter(User.id == user_id).first()

    if not user:
        return {"message": "User not found"}

    # 获取用户的地址信息
    addresses = user.addresses

    # 将地址信息转换为字典列表
    addresses_dict = [{"id": address.id, "street": address.street} for address in addresses]

    return {"addresses": addresses_dict}

在上面的代码中，我们使用read_user_addresses路由来查询指定用户的所有地址信息。该路由从数据库中查询用户信息，并返回地址信息的JSON响应。

现在我们可以运行FastAPI应用程序，并使用以下请求来测试路由：

GET http://localhost:8000/users/1/addresses

我们将在响应中获得用户ID为1的所有地址信息的JSON列表。

总结

在本文中，我们学习了如何使用FastAPI和Pydantic实现从父表中访问多对一关系。我们定义了模型并创建了路由进行查询操作。通过这个例子，我们演示了如何通过路由从父表中检索多对一关系的信息。希望这篇文章能够帮助您更好地理解和应用FastAPI和Pydantic中的多对一关系访问。