3
3
from django .core .exceptions import ObjectDoesNotExist , ValidationError
4
4
from django .db .models import Model
5
5
6
- from ara .domain .ara_entity import NewAraEntity , NewAraEntityCreateInput
6
+ from ara .domain .ara_entity import AraEntity , AraEntityCreateInput
7
7
8
8
T = TypeVar ("T" , bound = Model )
9
9
10
10
11
- class NewAraDjangoInfra (Generic [T ]):
11
+ class AraDjangoInfra (Generic [T ]):
12
12
def __init__ (self , model : Type [T ]) -> None :
13
13
self .model = model
14
14
@@ -105,22 +105,20 @@ def get_by(self, **kwargs) -> T | None:
105
105
except ObjectDoesNotExist :
106
106
return None
107
107
108
- def _to_model (self , entity : NewAraEntity ) -> Model :
108
+ def _to_model (self , entity : AraEntity ) -> Model :
109
109
raise NotImplementedError ()
110
110
111
111
@staticmethod
112
- def convert_model_to_entity (model : T ) -> NewAraEntity :
112
+ def convert_model_to_entity (model : T ) -> AraEntity :
113
113
raise NotImplementedError ()
114
114
115
- def _convert_entity_to_model (self , entity : NewAraEntity ) -> Model :
115
+ def _convert_entity_to_model (self , entity : AraEntity ) -> Model :
116
116
raise NotImplementedError ()
117
117
118
- def _convert_create_input_to_model (
119
- self , create_input : NewAraEntityCreateInput
120
- ) -> T :
118
+ def _convert_create_input_to_model (self , create_input : AraEntityCreateInput ) -> T :
121
119
raise NotImplementedError ()
122
120
123
- def bulk_update_entity (self , entities : list [NewAraEntity ]):
121
+ def bulk_update_entity (self , entities : list [AraEntity ]):
124
122
if len (entities ) == 0 :
125
123
return
126
124
@@ -137,7 +135,7 @@ def bulk_update_entity(self, entities: list[NewAraEntity]):
137
135
def bulk_update (self , instances : list [T ], fields : list [str ]):
138
136
return self .model .objects .bulk_update (instances , fields )
139
137
140
- def bulk_create (self , inputs : list [NewAraEntityCreateInput ]) -> list [NewAraEntity ]:
138
+ def bulk_create (self , inputs : list [AraEntityCreateInput ]) -> list [AraEntity ]:
141
139
instances = [self ._convert_create_input_to_model (input ) for input in inputs ]
142
140
created_instances = self .model .objects .bulk_create (instances )
143
141
entities = [
@@ -146,7 +144,7 @@ def bulk_create(self, inputs: list[NewAraEntityCreateInput]) -> list[NewAraEntit
146
144
]
147
145
return entities
148
146
149
- def save_entity (self , entity : NewAraEntity ):
147
+ def save_entity (self , entity : AraEntity ):
150
148
model = self ._convert_entity_to_model (entity )
151
149
model .save ()
152
150
return entity
0 commit comments