PUDNIC模型中的对象属性列表 [英] List of object attributes in pydantic model

问题描述

我使用Fast API创建Web服务。

有以下sqlAlChemy模型：

class User(Base):
    __tablename__ = 'user'
    account_name = Column(String, primary_key=True, index=True, unique=True)
    email = Column(String, unique=True, index=True, nullable=False)
    roles = relationship("UserRole", back_populates="users", lazy=False, uselist=True)


class UserRole(Base):
    __tablename__ = 'user_role'
    __table_args__ = (UniqueConstraint('role_name', 'user_name', name='user_role_uc'),)
    role_name = Column(String, ForeignKey('role.name'), primary_key=True)
    user_name = Column(String, ForeignKey('user.account_name'), primary_key=True)
    users = relationship("User", back_populates="roles")

僵化架构如下：

class UserRole(BaseModel):
    role_name: str

    class Config:
        orm_mode = True


class UserBase(BaseModel):
    account_name: str
    email: EmailStr
    roles: List[UserRole] = []

    class Config:
        orm_mode = True

我现在拥有的是：

{
  "account_name": "Test.Test",
  "email": "[e-mail]",
  "roles": [
    {
      "role_name": "all:admin"
    },
    {
      "role_name": "all:read"
    }
  ]
}

我想实现的是从API中获取用户，结构如下：

{
  "account_name": "Test.Test",
  "email": "[e-mail]",
  "roles": [
    "all:admin",
    "all:read"
  ]
}

这可能吗？我应该如何更改架构才能实现此目的？

推荐答案

如果您没事 处理How to&Quot；Get User from API&Quot；问题语句 通过修改FastAPI路径定义，请参见下文。

您是否可以更改FastAPI路径定义使用的响应模型 以便处理所需的输出格式？

二进制响应模型定义示例：

class UserResponse(BaseModel):
    account_name: str
    email: EmailStr
    roles: List[str]

示例SQL查询+序列化函数：

def get_user_response(user_id) -> UserResponse:
    user = User.query.get(user_id)
    user_roles = UserRole.query.filter(user=user_id).all()
    role_names = [r.role_name for r in user_roles]
    response = UserResponse(
        account_name=user.account_name,
        email=user.email,
        roles=role_names
    }
    return response

FastAPI路径定义示例：

@app.get("/users/{user_id}", response_model=UserResponse)
async def read_item(user_id):
    return get_user_response(user_id)

注意事项：

• 我对用户查询使用的是user_id，但这可以替换为您最终用作该表的主键的任何内容。
• 响应模型与UserBase非常相似(您可以 子类UserBase而不是模型，以避免重新定义 account_name和email，代价是必须重写 类Config)。
• 可能有方法覆盖UserBase的序列化格式 从数据库中查询模型时自动序列化的SQLalChemy模型对象 并允许您消除或减少上述示例定义中的get_user_response()函数中的代码。

这篇关于PUDNIC模型中的对象属性列表的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！