FastAPI（46）- JSONResponse

小菠蘿測試筆記發表於2021-10-03
ASTAPIJSON

背景

  • 建立 FastAPI 路徑操作函式時，通常可以從中返回任何資料：字典、列表、Pydantic 模型、資料庫模型等
  • 預設情況下，FastAPI 會使用 jsonable_encoder 自動將該返回值轉換為 JSON 字串
  • 然後，FastAPI 會將與 JSON 相容的資料（例如 dict）放在 JSONResponse 中，然後將 JSONResponse 返回給客戶端
  • 總結：預設情況下，FastAPI 將使用 JSONResponse 返回響應
  • 但是可以直接從路徑操作函式中返回自定義的 JSONResponse

 

返回響應資料的常見方式（基礎版）

https://www.cnblogs.com/poloyy/p/15364635.html

 

最簡單的栗子

路徑操作函式返回一個 Pydantic Model

#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
# author: 小菠蘿測試筆記
# blog:  https://www.cnblogs.com/poloyy/
# time: 2021/10/3 3:26 下午
# file: 38_responses.py
"""
from typing import Optional

import uvicorn
from fastapi import FastAPI
from fastapi.encoders import jsonable_encoder
from fastapi.responses import JSONResponse

from pydantic import BaseModel

app = FastAPI()


class Item(BaseModel):
    id: str
    name: str
    title: Optional[str] = None


@app.post("/item")
async def get_item(item: Item):
    # 列印看看傳進來的資料是什麼
    print(item, type(item))

    # 直接返回傳進來的資料
    return item

if __name__ == '__main__':
    uvicorn.run(app="38_responses:app", reload=True, host="127.0.0.1", port=8080)

 

正常傳參的請求結果

Response Header 的顯示 content-type 是 JSON 

 

console 列印結果

id='string' name='string' title='string' <class '38_responses.Item'>
INFO:     127.0.0.1:51856 - "POST /item HTTP/1.1" 200 OK 
  • item 型別的確是 Pydantic Model 類
  • 但最終返回給客戶端的是一個 JSON 資料

 

等價寫法

@app.post("/item")
async def get_item(item: Item):
    return item

這樣寫也能返回 JSON 資料，是因為FastAPI 是自動幫忙做了轉換的

等價寫法如下

from fastapi.encoders import jsonable_encoder

@app.post("/item")
async def get_item(item: Item):
    json_body = jsonable_encoder(item)
    return JSONResponse(content=json_body)

 

列印資料，來看看細節

@app.post("/item2")
async def get_item(item: Item):
    json_body = jsonable_encoder(item)
    
    print(json_body, type(json_body))
    
    return JSONResponse(content=json_body) 

console 列印結果

{'id': 'string', 'name': 'string', 'title': 'string'} <class 'dict'>
INFO:     127.0.0.1:52880 - "POST /item2 HTTP/1.1" 200 OK

 

假設將 item Pydantic Model 型別直接傳給 JSONResponse 呢？

@app.post("/item3")
async def get_item(item: Item):
    return JSONResponse(content=item)

 

訪問該介面就會報錯

    raise TypeError(f'Object of type {o.__class__.__name__} '
TypeError: Object of type Item is not JSON serializable
  • 型別錯誤：專案型別的物件不是 JSON 可序列化的
  • 因為它無法轉換為 JSON 資料，所以報錯了

 

看看 JSONResponse 原始碼

會呼叫 json.dumps() 方法

 

看看 Response 原始碼

看到其實可以自定義 status_code、headers、media_type 哦

headers 後面再用單獨的篇幅來講

 

修改 status_code 響應碼

@app.post("/item2")
async def get_item(item: Item):
    json_item = jsonable_encoder(item)
    return JSONResponse(content=json_item, status_code=status.HTTP_201_CREATED)

 

正確傳參的請求結果

 

更多自定義響應型別

 

相關文章