File tree Expand file tree Collapse file tree 1 file changed +15
-6
lines changed Expand file tree Collapse file tree 1 file changed +15
-6
lines changed Original file line number Diff line number Diff line change 1
- from src .model import Experiment
2
- from src .router .base import BaseController
1
+ from litestar .dto import DataclassDTO , DTOConfig
2
+
3
+ from src .model .experiment import Experiment , ExperimentDataclass
4
+ from src .model .facility import Facility
5
+ from src .router .base import DataclassController
3
6
from src .router .utils .dto import DTOGenerator
4
7
5
8
__all__ = ("ExperimentController" ,)
6
9
7
10
8
- ExperimentDTO = DTOGenerator [Experiment ]()
11
+ class ExperimentDTO (DataclassDTO [ExperimentDataclass ]):
12
+ config = DTOConfig (rename_strategy = "camel" )
13
+
14
+
15
+ ExperimentReturnDTO = DTOGenerator [Experiment ]()
9
16
10
17
11
- class ExperimentController (BaseController [Experiment ]):
18
+ class ExperimentController (DataclassController [Experiment , ExperimentDataclass ]):
12
19
path = "/experiment"
13
- dto = ExperimentDTO .write_dto
14
- return_dto = ExperimentDTO .read_dto
20
+ dto = ExperimentDTO
21
+ return_dto = ExperimentDTO
22
+ attr_map = {"facilities" : ("facility_id" , Facility )}
23
+ selectinload_attrs = [Experiment .facilities ]
You can’t perform that action at this time.
0 commit comments