在FastAPI（Pydantic）中展平或重新映射模型

Question

我在FastAPI中设置了一个简单的API。所以在我的例子中，我有一个用户表和一个项目表。每个用户都可以被分配到多个项目。没有限制。这意味着，两个表都通过多对多字段链接在一起。

当我打印出所有用户时，它将显示如下：

[
  {
    "id": 1,
    "username": "test1",
    "articles": [
      {
        "id": 1,
        "title": "foo",
      },
      {
        "id": 2,
        "title": "bar",
      }
    ],
  },
  {
    "id": 2,
    "username": "test2",
    "articles": [
      {
        "id": 1,
        "title": "foo",
      },
    ],
  },
]

但是我希望将articles属性展平。只有文章的ID应该是可见的，周围没有不需要的字典。

应该是这样的：

[
  {
    "id": 1,
    "username": "test1",
    "articles": [
      1,
      2,
    ],
  },
  {
    "id": 2,
    "username": "test2",
    "articles": [
      1,
    ],
  },
]

模式。py:

class ArticleBase(BaseModel):
    id: int
    title: str

class UserBase(BaseModel):
    id: int
    username: str
    articles: List[ArticleBase]

主要的py:

@app.get("/users/", response_model=List[schemas.UserBase])
def get_users(db: Session = Depends(crud.get_db)):
    return crud.get_users(db)

积垢。py:

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

def get_users(db: Session):
    return db.query(models.User).all()

models.py：

user_article_assoc = Table('user_article_assoc', Base.metadata,
                   Column('user_id', Integer, ForeignKey('user.id')),
                   Column('article_id', Integer, ForeignKey('article.id')))

class User(Base):
    __tablename__ = "user"
    id = Column(Integer, primary_key=True, index=True)
    username = Column(String)
    articles = relationship("Article", secondary=user_article_assoc, backref="users")

class Article(Base):
    __tablename__ = "article"
    id = Column(Integer, primary_key=True, index=True)
    title = Column(String)

Answer 1

这段代码将把你的用户整理成一个字典列表，你可以很容易地修改它来转换成其他模型：

flatternedUsers = list(map(lambda u: { 'id' :u.id, 'username': u.username, 'articles': list(map(operator.attrgetter('id'), u.articles)) }, users))

您还可以为此使用对象映射器库，例如对象映射器或odin。对象-对象映射器的好处是，如果您的用户现在有一个新属性，例如生日，它将自动映射到您想要返回的扁平化用户模型中的相同属性，假设您没有忘记添加生日归因于他们。