使用 FastAPI 从 get 请求添加更多数据

Question

我开始在 python 和 sqlalchemy 中使用 FastAPI 构建一个 api：这是模型的一部分：

class Game(Base):
    __tablename__ = "games"

    id = Column(Integer, primary_key=True, index=True)
    league_id = Column(Integer)
    radiant_score = Column(Integer)
    dire_score = Column(Integer)
    duration = Column(Integer)
    is_valid = Column(Boolean, default=True)

    playerstats = relationship("PlayerStat", back_populates="match")


class PlayerStat(Base):
    __tablename__ = "playerstats"

    match_id = Column(Integer, ForeignKey("games.id"), primary_key=True)
    slot = Column(Integer, primary_key=True)
    hero_id = Column(Integer, ForeignKey("heros.id"))
    num_kills = Column(Integer, default=None)
    isRadiant = Column(Boolean, default=None)

    match = relationship("Game", back_populates="playerstats")
    heros = relationship("Hero", back_populates="playerstats")

之后，我为 pydantic 创建模式/模型（请原谅长部分）：

    class PlayerStatBase(BaseModel):
    slot: int
    hero_id: int
    num_kills: int
    isRadiant: bool


class PlayerStatCreate(PlayerStatBase):
    pass


class PlayerStat(PlayerStatBase):
    slot: int
    hero_id: int
    num_kills: int
    isRadiant: bool

    class Config:
        orm_mode = True

class GameBase(BaseModel):
    id: int
    league_id: int
    radiant_score: int
    dire_score: int
    duration: int
    is_valid: bool


class GameCreate(GameBase):
    pass


class Game(GameBase):
    id: int
    league_id: int
    radiant_score: int
    dire_score: int
    duration: int
    is_valid: bool
    players: List[PlayerStat] = [{}]

    class Config:
        orm_mode = True

以及我与 api 一起使用的 crud 函数：

    def get_match(db: Session, match_id: int):
    print(db.query(models.Game).filter(models.Game.id == match_id))
    return db.query(models.Game).filter(models.Game.id == match_id).first()

api路线是：

@app.get("/matches/{match_id}", response_model=schemas.Game)
def read_game(match_id: int, db: Session = Depends(get_db)):
    db_game = crud.get_match(db, match_id=match_id)
    if db_game is None:
        raise HTTPException(status_code=404, detail="Game not found")
    return db_game

我得到的结果是下一个：

{
  "id": 1,
  "league_id": 10,
  "radiant_score": 41,
  "dire_score": 5,
  "duration": 3541,
  "is_valid": true,
  "players": [
    {}
  ]
}

我想用相应比赛的玩家统计列表（按插槽排序）填充“玩家”，如下所示：

"players" : [
  {
    "slot": 0,
    "hero_id": 14,
    "num_kills": 54,
    "isRadiant": true
  },
  {
    "slot": 1,
    "hero_id": 15,
    "num_kills": 1,
    "isRadiant": false
  }
]

我想我需要尝试模型/架构或 CRUD 函数之一，但真的不知道哪一个？另外，也许有一些无用或构建不良的 pydantic 模式

PS：我遵循 FastAPI 文档的指南（我建议阅读）。

谢谢你的帮助！

Answer 1

我相信你的问题不属于fastapi的范围，而是sqlalchemy的范围。当您查询具有关系的 orm 对象时，fastapi 的标准是在访问关系时延迟加载关系。由于您从未直接访问关系playerstats，因此它不会加载。有关信息，请参阅文档。

问题的解决方案应该是将 crud 函数更新为：

return db.query(models.Game).filter(models.Game.id == match_id)
                            .options(selectinload(models.Game.playerstats)).first()

“Select In Load”是一种预加载类型，它将在提交查询时加载关系。如果您希望每个查询都发生此行为，您可以将 orm 更新为：

playerstats = relationship("PlayerStat", back_populates="match", lazy="selectin")

我希望这有帮助。这是我在 stackoverflow 上的第一个答案:)

编辑：实际上还有另一件事。在您的 orm 中，这种关系称为“playerstats”，而您在 pydantic 模型中将属性命名为“players”。那是行不通的。将 pydantic 属性名称从“players”更改为“playerstats”，现在一切都应该可以工作了。

编辑2：正如你所猜测的，一切都还不能正常工作。我刚刚发现还缺少一件事。在 pydantic 模型中，您可以设置 orm 选项。这在使用 sqlalchemy 时非常重要。我向您所有的 pydantic 模型推荐这个。这必须在每个 pydantic 模型及其属性模型上设置

class OtherModel(BaseModel):
    value: str = None

    class Config:
        orm_mode = True


class SomePydanticModel(BaseModel):
    value: str = None
    some_other_model: OtherModel = None

    class Config:
         orm_mode = True

现在我们还可以再次修改你的 crud 方法的 return 语句：

return Game.from_orm(db.query(models.Game).options(selectinload(models.Game.playerstats)).filter(models.Game.id == match_id).first())

现在一切终于应该可以正常工作了:)