Code for chapter 3: Response models and Error handling

Author: Youngestdev

ch03/todos/api.py

@@ -0,0 +1,13 @@
+from fastapi import FastAPI
+from todo import todo_router
+
+app = FastAPI()
+
+
+@app.get("/")
+async def welcome():
+    return {
+        "message": "Hello World"
+    }
+
+app.include_router(todo_router)

ch03/todos/model.py

@@ -0,0 +1,45 @@
+from typing import List
+
+from pydantic import BaseModel
+
+
+class Todo(BaseModel):
+    id: int
+    item: str
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "id": 1,
+                "item": "Example schmea!"
+            }
+        }
+
+
+class TodoItem(BaseModel):
+    item: str
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "item": "Read the next chapter of the book"
+            }
+        }
+
+
+class TodoItems(BaseModel):
+    todos: List[TodoItem]
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "todos": [
+                    {
+                        "item": "Example schema 1!"
+                    },
+                    {
+                        "item": "Example schema 2!"
+                    }
+                ]
+            }
+        }

ch03/todos/requirements.txt

@@ -0,0 +1,2 @@
+fastapi==0.70.0
+uvicorn==0.15.0

ch03/todos/todo.py

@@ -0,0 +1,72 @@
+from fastapi import APIRouter, Path, HTTPException, status
+from model import Todo, TodoItem, TodoItems
+
+todo_router = APIRouter()
+
+todo_list = []
+
+
+@todo_router.post("/todo")
+async def add_todo(todo: Todo):
+    todo_list.append(todo)
+    return {
+        "message": "Todo added successfully."
+    }
+
+
+@todo_router.get("/todo", response_model=TodoItems)
+async def retrieve_todo():
+    return {
+        "todos": todo_list
+    }
+
+
+@todo_router.get("/todo/{todo_id}")
+async def get_single_todo(todo_id: int = Path(..., title="The ID of the todo to retrieve.")):
+    for todo in todo_list:
+        if todo.id == todo_id:
+            return {
+                "todo": todo
+            }
+    raise HTTPException(
+        status_code=status.HTTP_404_NOT_FOUND,
+        detail="Todo with supplied ID doesn't exist",
+    )
+
+
+@todo_router.put("/todo/{todo_id}")
+async def update_todo(todo_data: TodoItem, todo_id: int = Path(..., title="The ID of the todo to be updated.")):
+    for todo in todo_list:
+        if todo.id == todo_id:
+            todo.item = todo_data.item
+            return {
+                "message": "Todo updated successfully."
+            }
+
+    raise HTTPException(
+        status_code=status.HTTP_404_NOT_FOUND,
+        detail="Todo with supplied ID doesn't exist",
+    )
+
+
+@todo_router.delete("/todo/{todo_id}")
+async def delete_single_todo(todo_id: int):
+    for index in range(len(todo_list)):
+        todo = todo_list[index]
+        if todo.id == todo_id:
+            todo_list.pop(index)
+            return {
+                "message": "Todo deleted successfully."
+            }
+    raise HTTPException(
+        status_code=status.HTTP_404_NOT_FOUND,
+        detail="Todo with supplied ID doesn't exist",
+    )
+
+
+@todo_router.delete("/todo")
+async def delete_all_todo():
+    todo_list.clear()
+    return {
+        "message": "Todos deleted successfully."
+    }