allow in fastapi to return related models with only pk populated
This commit is contained in:
@ -64,6 +64,12 @@ async def get_items():
|
||||
return items
|
||||
|
||||
|
||||
@app.get("/items/raw/", response_model=List[Item])
|
||||
async def get_raw_items():
|
||||
items = await Item.objects.all()
|
||||
return items
|
||||
|
||||
|
||||
@app.post("/items/", response_model=Item)
|
||||
async def create_item(item: Item):
|
||||
await item.save()
|
||||
@ -76,6 +82,12 @@ async def create_category(category: Category):
|
||||
return category
|
||||
|
||||
|
||||
@app.get("/items/{item_id}")
|
||||
async def get_item(item_id: int):
|
||||
item = await Item.objects.get(pk=item_id)
|
||||
return item
|
||||
|
||||
|
||||
@app.put("/items/{item_id}")
|
||||
async def update_item(item_id: int, item: Item):
|
||||
item_db = await Item.objects.get(pk=item_id)
|
||||
@ -113,6 +125,14 @@ def test_all_endpoints():
|
||||
items = [Item(**item) for item in response.json()]
|
||||
assert items[0].name == "New name"
|
||||
|
||||
response = client.get("/items/raw/")
|
||||
items = [Item(**item) for item in response.json()]
|
||||
assert items[0].name == "New name"
|
||||
|
||||
response = client.get(f"/items/{item.pk}")
|
||||
new_item = Item(**response.json())
|
||||
assert new_item == item
|
||||
|
||||
response = client.delete(f"/items/{item.pk}")
|
||||
assert response.json().get("deleted_rows", "__UNDEFINED__") != "__UNDEFINED__"
|
||||
response = client.get("/items/")
|
||||
|
||||
Reference in New Issue
Block a user