feature: new train schedules exceptions

editoast/editoast_models/src/train_schedule_exception.rs

@@ -1,7 +1,14 @@
+use std::collections::HashMap;
+
+use database::DbConnection;
 use editoast_derive::Model;
+use itertools::Itertools;
+use schemas::paced_train::PacedTrainException;
 use schemas::train_schedule_exception::TrainScheduleExceptionChangeGroups;
 
 use crate as editoast_models;
+use crate::prelude::List;
+use crate::prelude::SelectionSettings;
 
 #[derive(Debug, Clone, Model)]
 #[cfg_attr(test, derive(serde::Deserialize))]
@@ -18,6 +25,33 @@ pub struct TrainScheduleException {
     pub change_groups: TrainScheduleExceptionChangeGroups,
 }
 
+#[cfg(any(test, feature = "testing"))]
+impl TrainScheduleException {
+    pub fn fixture_created(key: &str, occurrence_index: Option<i64>) -> Self {
+        Self {
+            id: 1,
+            disabled: false,
+            key: Some(key.into()),
+            occurrence_index,
+            timetable_id: 1,
+            train_schedule_id: 1,
+            change_groups: TrainScheduleExceptionChangeGroups::fixture_created(),
+        }
+    }
+
+    pub fn fixture_modified(key: &str, occurrence_index: i64) -> Self {
+        Self {
+            id: 1,
+            disabled: false,
+            key: Some(key.into()),
+            occurrence_index: Some(occurrence_index),
+            timetable_id: 1,
+            train_schedule_id: 1,
+            change_groups: TrainScheduleExceptionChangeGroups::fixture_modified(),
+        }
+    }
+}
+
 impl From<TrainScheduleException> for schemas::TrainScheduleException {
     fn from(train_schedule_exception: TrainScheduleException) -> Self {
         Self {
@@ -31,3 +65,52 @@ impl From<TrainScheduleException> for schemas::TrainScheduleException {
         }
     }
 }
+
+impl TrainScheduleException {
+    pub async fn retrieve_exceptions_by_train_schedules(
+        conn: &mut DbConnection,
+        timetable_id: i64,
+        train_schedules_ids: Vec<i64>,
+    ) -> Result<Vec<TrainScheduleException>, editoast_models::Error> {
+        let train_schedules_ids_for_settings = train_schedules_ids.clone();
+        let exceptions_settings = SelectionSettings::new()
+            .filter(move || editoast_models::TrainScheduleException::TIMETABLE_ID.eq(timetable_id))
+            .filter(move || {
+                editoast_models::TrainScheduleException::TRAIN_SCHEDULE_ID
+                    .eq_any(train_schedules_ids_for_settings.clone())
+            });
+
+        let mut exceptions_map: HashMap<i64, Vec<TrainScheduleException>> =
+            Self::list(conn, exceptions_settings)
+                .await?
+                .into_iter()
+                .into_group_map_by(|e| e.train_schedule_id);
+
+        let exceptions = train_schedules_ids
+            .into_iter()
+            .flat_map(|id| exceptions_map.remove(&id).unwrap_or_default())
+            .collect();
+
+        Ok(exceptions)
+    }
+}
+
+impl From<TrainScheduleException> for PacedTrainException {
+    fn from(train_schedule_exception: TrainScheduleException) -> Self {
+        let exception_type = match train_schedule_exception.occurrence_index {
+            Some(occurrence_index) => schemas::paced_train::ExceptionType::Modified {
+                occurrence_index: occurrence_index as usize,
+            },
+            None => schemas::paced_train::ExceptionType::Created {},
+        };
+        Self {
+            id: Some(train_schedule_exception.id),
+            key: train_schedule_exception
+                .key
+                .unwrap_or_else(|| train_schedule_exception.id.to_string()),
+            exception_type,
+            disabled: train_schedule_exception.disabled,
+            change_groups: train_schedule_exception.change_groups,
+        }
+    }
+}

editoast/schemas/src/fixtures.rs

@@ -109,6 +109,7 @@ pub fn rolling_stock_with_invalid_effort_curves_json() -> &'static str {
 
 pub fn simple_created_exception_with_change_groups(key: &str) -> PacedTrainException {
     PacedTrainException {
+        id: None,
         key: key.into(),
         exception_type: ExceptionType::Created {},
         disabled: false,

editoast/schemas/src/paced_train.rs

@@ -129,6 +129,8 @@ impl Serialize for TrainSchedule {
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq, ToSchema)]
 #[cfg_attr(feature = "testing", derive(Default))]
 pub struct PacedTrainException {
+    /// TODO The new exception table id (for incremental migration)
+    pub id: Option<i64>,
     /// Unique key for the exception within the paced train, required and generated by the frontend.
     pub key: String,
     #[serde(flatten)]