6

我使用 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")

Pydantic模式如下：

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": "Test.Test@test.com",
  "roles": [
    {
      "role_name": "all:admin"
    },
    {
      "role_name": "all:read"
    }
  ]
}

我想要实现的是从以下结构中的 api 获取用户：

{
  "account_name": "Test.Test",
  "email": "Test.Test@test.com",
  "roles": [
    "all:admin",
    "all:read"
  ]
}

那可能吗？我应该如何更改架构来获得这个？

4

2 回答 2

4

如果您可以通过修改 fastapi 路径定义来处理如何“从 api 获取用户”问题陈述，请参见下文。

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

pydantic 响应模型定义示例：

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

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

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用户查询，但这可以替换为您最终用作该表的主键的任何内容。
  • UserResponse响应模型非常类似于UserBase（您可能会使用子类而UserBase不是模型来避免重新定义 account_nameand email，但必须权衡必须覆盖类' Config）。
  • 可能有一种方法可以覆盖UserBase sqlalchemy 模型对象的序列化格式，当您从数据库中查询模型时，该模型对象会自动序列化，并允许您消除或减少get_user_response()上面示例定义中函数中的代码。
于 2021-04-27T23:18:01.497 回答
0

由于 UserRole 是一个类，它被表示为一个对象（使用字典）。如果要将其表示为字符串列表，则必须转换数据（并更改 Pydantic 模型的字段声明）。有几种方法可以解决这个问题，但是 pydantic 模型文档是一个很好的起点。请注意，ORM 模型用作数据表示层，而 pydantic 模型是验证（可能还有序列化）层，因此有很多地方可以“挂钩”。

于 2020-06-25T07:04:33.357 回答