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
50
51
|
from __future__ import annotations
from dataclasses import dataclass
from litestar import Controller, Litestar, patch, post, put
from litestar.dto import DataclassDTO, DTOConfig, DTOData
@dataclass
class Person:
name: str
age: int
email: str
id: int
class ReadDTO(DataclassDTO[Person]):
config = DTOConfig(exclude={"email"})
class WriteDTO(DataclassDTO[Person]):
config = DTOConfig(exclude={"id"})
class PatchDTO(DataclassDTO[Person]):
config = DTOConfig(exclude={"id"}, partial=True)
class PersonController(Controller):
dto = WriteDTO
return_dto = ReadDTO
@post("/person", sync_to_thread=False)
def create_person(self, data: DTOData[Person]) -> Person:
# Logic for persisting the person goes here
return data.create_instance(id=1)
@put("/person/{person_id:int}", sync_to_thread=False)
def update_person(self, person_id: int, data: DTOData[Person]) -> Person:
# Usually the Person would be retrieved from a database
person = Person(id=person_id, name="John", age=50, email="email_of_john@example.com")
return data.update_instance(person)
@patch("/person/{person_id:int}", dto=PatchDTO, sync_to_thread=False)
def patch_person(self, person_id: int, data: DTOData[Person]) -> Person:
# Usually the Person would be retrieved from a database
person = Person(id=person_id, name="John", age=50, email="email_of_john@example.com")
return data.update_instance(person)
app = Litestar(route_handlers=[PersonController])
|
