pythonasyncioapiasyncfastapiframeworkjsonjson-schemaopenapiopenapi3pydanticpython-typespython3redocreststarletteswaggerswagger-uiuvicornweb
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.1 KiB
38 lines
1.1 KiB
from typing import Union
|
|
|
|
from fastapi import FastAPI, Header, HTTPException
|
|
from pydantic import BaseModel
|
|
|
|
fake_secret_token = "coneofsilence"
|
|
|
|
fake_db = {
|
|
"foo": {"id": "foo", "title": "Foo", "description": "There goes my hero"},
|
|
"bar": {"id": "bar", "title": "Bar", "description": "The bartenders"},
|
|
}
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
class Item(BaseModel):
|
|
id: str
|
|
title: str
|
|
description: Union[str, None] = None
|
|
|
|
|
|
@app.get("/items/{item_id}", response_model=Item)
|
|
async def read_main(item_id: str, x_token: str = Header()):
|
|
if x_token != fake_secret_token:
|
|
raise HTTPException(status_code=400, detail="Invalid X-Token header")
|
|
if item_id not in fake_db:
|
|
raise HTTPException(status_code=404, detail="Item not found")
|
|
return fake_db[item_id]
|
|
|
|
|
|
@app.post("/items/", response_model=Item)
|
|
async def create_item(item: Item, x_token: str = Header()):
|
|
if x_token != fake_secret_token:
|
|
raise HTTPException(status_code=400, detail="Invalid X-Token header")
|
|
if item.id in fake_db:
|
|
raise HTTPException(status_code=400, detail="Item already exists")
|
|
fake_db[item.id] = item
|
|
return item
|
|
|