refactor: ♻️ Improve CRUD functions

thermal_events/crud/base.py

@@ -59,20 +59,18 @@ def get(self, id: Any) -> Optional[ModelType]:
         with session_scope() as session:
             return session.query(self.model).filter(self.model.id == id).first()
 
-    def get_multi(self, *, skip: int = 0, limit: int = 100) -> List[ModelType]:
-        """Get multiple objects with optional pagination.
+    def get_multi(self, ids) -> List[ModelType]:
+        """Get multiple objects given their ids.
 
         Args:
-            skip (int, optional): The number of objects to skip. Defaults to 0.
-            limit (int, optional): The maximum number of objects to retrieve. Defaults
-            to 100.
+            ids (list): The list of ids of objects to query.
 
         Returns:
             List[ModelType]: The list of retrieved objects.
 
         """
         with session_scope() as session:
-            return session.query(self.model).offset(skip).limit(limit).all()
+            return session.query(self.model).filter(self.model.id.in_(ids)).all()
 
     def create(self, obj_in: Union[list, ModelType]) -> None:
         """Create a new object or a list of objects.

thermal_events/crud/thermal_event.py

@@ -378,6 +378,22 @@ def change_analysis_status(self, event_id: int, new_status: str):
             )
             session.commit()
 
+    def change_severity(self, event_id: int, new_severity: str):
+        """Change the severity of a ThermalEvent.
+
+        Args:
+            event_id (int):
+                id of the ThermalEvent to update.
+            new_severity (str):
+                New severity value.
+
+        """
+        with session_scope() as session:
+            session.query(ThermalEvent).filter_by(id=event_id).update(
+                {"severity": new_severity}
+            )
+            session.commit()
+
     def update(self, obj_in: Union[ThermalEvent, List[ThermalEvent]]) -> None:
         """Update an existing object or a list of ThermalEvent objects.