1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
from dataclasses import dataclass
from typing import List, Optional
from litestar import Litestar, get, post, put
from litestar.exceptions import NotFoundException
@dataclass
class TodoItem:
title: str
done: bool
TODO_LIST: List[TodoItem] = [
TodoItem(title="Start writing TODO list", done=True),
TodoItem(title="???", done=False),
TodoItem(title="Profit", done=False),
]
def get_todo_by_title(todo_name) -> TodoItem:
for item in TODO_LIST:
if item.title == todo_name:
return item
raise NotFoundException(detail=f"TODO {todo_name!r} not found")
@get("/")
async def get_list(done: Optional[bool] = None) -> List[TodoItem]:
if done is None:
return TODO_LIST
return [item for item in TODO_LIST if item.done == done]
@post("/")
async def add_item(data: TodoItem) -> List[TodoItem]:
TODO_LIST.append(data)
return TODO_LIST
@put("/{item_title:str}")
async def update_item(item_title: str, data: TodoItem) -> List[TodoItem]:
todo_item = get_todo_by_title(item_title)
todo_item.title = data.title
todo_item.done = data.done
return TODO_LIST
app = Litestar([get_list, add_item, update_item])
|
