Fastapi pydantic 相关架构响应按列过滤

Question

我正在使用 fastapi pydantic 模式获取发票请求 api 及其相应项目。 API 工作正常，但考虑到带有发票 ID 的所有相关项目行。但我希望根据某些列过滤外键相关数据。与 is_active == True 类似，仅考虑那些发票项目行。

{
   'id': 1
   'invoice_number': 1
   'item_name': 'this is invoice one'
   'invoice_items': [
      {'invoice_id': 1, 'item_name': item one, deleted_at: None},
      {'invoice_id': 1, 'item_name': item two, deleted_at: None},
      {'invoice_id': 1, 'item_name': item three, deleted_at: None}
   ]
}

Answer 1

I just stumbled upon your question.

您不应该在 pydantic 模式中进行过滤。
您可以在将数据提供给响应之前过滤数据。
从数据库中获取信息后，您可以执行以下操作：

invoice = {
    'id': 1,
    'invoice_number': 1,
    'item_name': 'this is invoice one',
    'invoice_items': [
        {'invoice_id': 1, 'item_name': 'item one', 'is_active': True},
        {'invoice_id': 1, 'item_name': 'item two', 'is_active': False},
        {'invoice_id': 1, 'item_name': 'item three', 'is_active': True},
    ]
}

invoice_items = invoice.invoice_items

filtered_invoice_items = [
    inv_itm 
    for inv_itm in invoice_items 
    if inv_itm.is_active is True
]

invoice.invoice_items = filtered_invoice_items

然后您的发票将如下所示，您可以将该发票作为回复：

invoice = {
    'id': 1,
    'invoice_number': 1,
    'item_name': 'this is invoice one',
    'invoice_items': [
        {'invoice_id': 1, 'item_name': 'item one', 'is_active': True},
        {'invoice_id': 1, 'item_name': 'item three', 'is_active': True},
    ]
}

我希望此建议对您有所帮助。
祝你好运！