editoast: make budget nullable in project and study

Author: hamz2a

editoast/migrations/2024-03-12-143107_make_budget_nullable/down.sql

@@ -0,0 +1,2 @@
+ALTER TABLE study ALTER COLUMN budget SET NOT NULL;
+ALTER TABLE project ALTER COLUMN budget SET NOT NULL;

editoast/migrations/2024-03-12-143107_make_budget_nullable/up.sql

@@ -0,0 +1,2 @@
+ALTER TABLE study ALTER COLUMN budget DROP NOT NULL;
+ALTER TABLE project ALTER COLUMN budget DROP NOT NULL;

editoast/openapi.yaml

@@ -4475,6 +4475,7 @@ components:
       properties:
         budget:
           format: int32
+          nullable: true
           type: integer
         creation_date:
           format: date-time
@@ -4505,7 +4506,6 @@ components:
       - objectives
       - description
       - funders
-      - budget
       - creation_date
       - last_modification
       - tags
@@ -4515,6 +4515,7 @@ components:
       properties:
         budget:
           format: int32
+          nullable: true
           type: integer
         description:
           maxLength: 1024
@@ -5967,6 +5968,7 @@ components:
           type: string
         budget:
           format: int32
+          nullable: true
           type: integer
         business_code:
           type: string
@@ -6010,7 +6012,6 @@ components:
       - service_code
       - creation_date
       - last_modification
-      - budget
       - tags
       - state
       - study_type
@@ -6025,6 +6026,7 @@ components:
           type: string
         budget:
           format: int32
+          nullable: true
           type: integer
         business_code:
           type: string

editoast/src/fixtures.rs

@@ -295,7 +295,7 @@ pub mod tests {
             .service_code("BBB".into())
             .creation_date(Utc::now().naive_utc())
             .last_modification(Utc::now().naive_utc())
-            .budget(0)
+            .budget(Some(0))
             .tags(Tags::default())
             .state("some_state".into())
             .study_type("some_type".into())
@@ -313,7 +313,7 @@ pub mod tests {
             .objectives("".to_owned())
             .description("".to_owned())
             .funders("".to_owned())
-            .budget(0)
+            .budget(Some(0))
             .creation_date(Utc::now().naive_utc())
             .last_modification(Utc::now().naive_utc())
             .tags(Tags::default());

editoast/src/modelsv2/projects.rs

@@ -28,7 +28,7 @@ pub struct Project {
     pub objectives: String,
     pub description: String,
     pub funders: String,
-    pub budget: i32,
+    pub budget: Option<i32>,
     pub creation_date: NaiveDateTime,
     pub last_modification: NaiveDateTime,
     #[model(remote = "Vec<Option<String>>")]
@@ -239,7 +239,7 @@ pub mod test {
         // Patch a project
         let mut project = project_fixture.model.clone();
         project.name = "update_name".into();
-        project.budget = 1000;
+        project.budget = Some(1000);
         project.save(conn).await.unwrap();
         let project = Project::retrieve(conn, project.id).await.unwrap().unwrap();
         assert_eq!(project.name, String::from("update_name"));

editoast/src/modelsv2/study.rs

@@ -31,7 +31,7 @@ pub struct Study {
     pub start_date: Option<NaiveDate>,
     pub expected_end_date: Option<NaiveDate>,
     pub actual_end_date: Option<NaiveDate>,
-    pub budget: i32,
+    pub budget: Option<i32>,
     #[model(remote = "Vec<Option<String>>")]
     pub tags: Tags,
     pub state: String,

editoast/src/tables.rs

@@ -402,7 +402,7 @@ diesel::table! {
         description -> Varchar,
         #[max_length = 255]
         funders -> Varchar,
-        budget -> Int4,
+        budget -> Nullable<Int4>,
         creation_date -> Timestamptz,
         last_modification -> Timestamptz,
         tags -> Array<Nullable<Text>>,
@@ -612,7 +612,7 @@ diesel::table! {
         start_date -> Nullable<Date>,
         expected_end_date -> Nullable<Date>,
         actual_end_date -> Nullable<Date>,
-        budget -> Int4,
+        budget -> Nullable<Int4>,
         tags -> Array<Nullable<Text>>,
         #[max_length = 16]
         state -> Varchar,

editoast/src/views/projects.rs

@@ -76,8 +76,7 @@ struct ProjectCreateForm {
     #[serde(default)]
     #[schema(max_length = 1024)]
     pub funders: String,
-    #[serde(default)]
-    pub budget: i32,
+    pub budget: Option<i32>,
     /// The id of the image document
     pub image: Option<i64>,
     #[serde(default)]
@@ -269,7 +268,7 @@ impl From<ProjectPatchForm> for Changeset<Project> {
             .flat_description(project.description)
             .flat_objectives(project.objectives)
             .flat_funders(project.funders)
-            .flat_budget(project.budget)
+            .flat_budget(Some(project.budget))
             .flat_image(Some(project.image))
             .flat_tags(project.tags)
     }

editoast/src/views/study.rs

@@ -74,8 +74,7 @@ struct StudyCreateForm {
     pub business_code: String,
     #[serde(default)]
     pub service_code: String,
-    #[serde(default)]
-    pub budget: i32,
+    pub budget: Option<i32>,
     #[serde(default)]
     pub tags: Tags,
     pub state: String,
@@ -328,7 +327,7 @@ impl StudyPatchForm {
             .flat_start_date(Some(self.start_date))
             .flat_expected_end_date(Some(self.expected_end_date))
             .flat_actual_end_date(Some(self.actual_end_date))
-            .flat_budget(self.budget)
+            .flat_budget(Some(self.budget))
             .flat_tags(self.tags)
             .flat_state(self.state)
             .flat_study_type(self.study_type);

front/src/common/api/osrdEditoastApi.ts

@@ -1926,7 +1926,7 @@ export type ElectrificationsOnPathResponse = {
 };
 export type Tags = string[];
 export type Project = {
-  budget: number;
+  budget?: number | null;
   creation_date: string;
   description: string;
   funders: string;
@@ -1958,7 +1958,7 @@ export type Ordering =
   | 'LastModifiedDesc'
   | 'LastModifiedAsc';
 export type ProjectCreateForm = {
-  budget?: number;
+  budget?: number | null;
   description?: string;
   funders?: string;
   /** The id of the image document */
@@ -1979,7 +1979,7 @@ export type ProjectPatchForm = {
 };
 export type Study = {
   actual_end_date?: string | null;
-  budget: number;
+  budget?: number | null;
   business_code: string;
   creation_date: string;
   description: string;
@@ -2013,7 +2013,7 @@ export type StudyResponse = Study & {
 };
 export type StudyCreateForm = {
   actual_end_date?: string | null;
-  budget?: number;
+  budget?: number | null;
   business_code?: string;
   description?: string;
   expected_end_date?: string | null;