[Draft] Add endpoint_body_model arg to pipeline models

forloop_common_structures/core/pipeline.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass, field
-from typing import Optional
+from typing import Any, Optional
 
 
 @dataclass
@@ -12,17 +12,20 @@ class Pipeline:
     remaining_nodes_uids: list = field(default_factory=list)
 
     project_uid: str = "0"
+    endpoint_body_model: dict[str, Any] = field(default_factory=dict)
     uid: Optional[str] = None
 
     def update(self, **kwargs) -> None:
         for key, value in kwargs.items():
             if key in vars(self).keys():
                 setattr(self, key, value)
             else:
-                raise AttributeError(f"Attribute '{key}' cannot be updated, as it does not exist")
+                raise AttributeError(
+                    f"Attribute '{key}' cannot be updated, as it does not exist"
+                )
 
     def run(self):
         self.is_active = True
 
-    def stop(self):  #todo: pause
+    def stop(self):  # todo: pause
         self.is_active = False

forloop_common_structures/database_models.py

@@ -106,24 +106,25 @@ class DBPipeline(dh.AbstractModel):
 
     project_uid: int  # Foreign Key Many-to-1
     system_reactivation_status: str = None
+    endpoint_body_model: dict[Any] = {}
 
 
 def cast_pipeline_types_to_app(pipelines_df: pd.DataFrame) -> pd.DataFrame:
     """Cast DB datatypes to in-app python datatypes."""
     pipelines_df = pipelines_df.astype(
         {"uid": str, "project_uid": str, "is_active": bool}
     )
-    pipelines_df[["active_nodes_uids", "remaining_nodes_uids"]
-                ] = pipelines_df[["active_nodes_uids", "remaining_nodes_uids"]].map(json.loads)
+    pipelines_df[["active_nodes_uids", "remaining_nodes_uids", "endpoint_body_model"]
+                ] = pipelines_df[["active_nodes_uids", "remaining_nodes_uids", "endpoint_body_model"]].map(json.loads)
     return pipelines_df
 
 
 def cast_pipeline_types_to_db(pipelines_df: pd.DataFrame) -> pd.DataFrame:
     """Cast in-app python datatypes to DB datatypes."""
     pipelines_df = pipelines_df.drop("uid", axis=1)
     pipelines_df = pipelines_df.astype({"project_uid": int, "is_active": int})
-    pipelines_df[["active_nodes_uids", "remaining_nodes_uids"]
-                ] = pipelines_df[["active_nodes_uids", "remaining_nodes_uids"]].map(json.dumps)
+    pipelines_df[["active_nodes_uids", "remaining_nodes_uids", "endpoint_body_model"]
+                ] = pipelines_df[["active_nodes_uids", "remaining_nodes_uids", "endpoint_body_model"]].map(json.dumps)
     pipelines_df = pipelines_df.map(escape_if_string)
     return pipelines_df