2

我最近开始了解 fastAPI 并通过教程和其他文档进行了工作。尽管 fastAPI 有很好的文档记录，但我找不到有关在使用数据库时如何处理嵌套输入的信息。

为了测试，我写了一个非常小的家庭API，有两个模型：

class Member(Base):
    __tablename__ = 'members'

    id = Column(Integer, primary_key=True, server_default=text("nextval('members_id_seq'::regclass)"))
    name = Column(String(128), nullable=False)
    age = Column(Integer, nullable=True)
    family_id = Column(Integer, ForeignKey('families.id', deferrable=True, initially='DEFERRED'), nullable=False, index=True)

    family = relationship("Family", back_populates="members")


class Family(Base):
    __tablename__ = 'families'

    id = Column(Integer, primary_key=True, server_default=text("nextval('families_id_seq'::regclass)"))
    family_name = Column(String(128), nullable=False)

    members = relationship("Member", back_populates="family")

我创建了一个包含两个表和此处描述的关系的 Postgres 数据库。使用 fastAPI 教程中的模式定义和 crud 文件，我可以创建单独的族和成员，并使用 get 请求以嵌套方式查看它们。这是嵌套模式：

class Family(FamilyBase):
    id: int
    members: List[Member]

    class Config:
        orm_mode = True

到目前为止，一切都很好。现在，我想添加一个帖子视图，它接受嵌套结构作为输入并相应地填充数据库。https://fastapi.tiangolo.com/tutorial/body-nested-models/上的文档原则上展示了如何做到这一点，但它错过了数据库（即 crud）部分。

由于输入将没有id字段并且显然不需要指定family_id，因此我有一个MemberStub架构和NestedFamilyCreate架构如下：

class MemberStub(BaseModel):
    name: str
    age: int

class NestedFamilyCreate(BaseModel):
    family_name: str
    members: List[MemberStub]

在我的路由例程families.py中，我有：

@app.post('/nested-families/', response_model=schemas.Family)
def create_family(family: schemas.NestedFamilyCreate, db: Session = Depends(get_db)):
    # no check for previous existence as names can be duplicates
    return crud.create_nested_family(db=db, family=family)

（response_model 指向包含所有 id 在内的所有成员的家庭的嵌套视图；见上文）。

我无法弄清楚的是如何编写crud.create_nested_family例程。基于教程中的简单创建，这看起来像：

def create_nested_family(db: Session, family: schemas.NestedFamilyCreate):
    # split information in family and members
    members = family.members
    core_family = None # ??? This is where I get stuck
    db_family = models.Family(**family.dict())  # This fails
    db.add(db_family)
    db.commit()
    db.refresh(db_family)
    return db_family

所以，我可以提取成员并遍历它们，但我首先需要创建一个db_family不能包含成员的新记录。然后，使用db.refresh，我将获得新的 family_id，我可以将其添加到 的每条记录中members。但是我该怎么做呢？如果我了解这里需要什么，我需要将我的嵌套模式映射到 FamilyCreate 的普通模式（它自己工作）和 MemberCreate 的一个普通模式（它也可以自己工作）。但是我该怎么做呢？

4

1 回答 1

7

在重新阅读 Pydantic 模型及其与 dict 的映射后，我找到了解决方案。

在 crud.py 中：

def create_nested_family(db: Session, family: schemas.NestedFamilyCreate):
    # split information in family and members
    family_data = family.dict()
    member_data = family_data.pop('members', None)   # ToDo: handle error if no members
    db_family = models.Family(**family_data)
    db.add(db_family)
    db.commit()
    db.refresh(db_family)
    # get family_id
    family_id = db_family.id
    # add members
    for m in member_data:
        m['family_id'] = family_id
        db_member = models.Member(**m)
        db.add(db_member)
        db.commit()
        db.refresh(db_member)
    return db_family

希望，这可能对其他人有用。

于 2020-04-14T17:30:26.537 回答