Remove redundant fields and changed data payload to accept new fields

Author: vladbagmet

src/corva/models/merge/merge.py

@@ -8,8 +8,6 @@ class PartialRerunMergeEvent(CorvaBaseEvent):
     Attributes:
         event_type: EventType.PARTIAL_WELL_RERUN_MERGE
         partial_well_rerun_id: partial well rerun id
-        partition: partition
-        rerun_partition: rerun partition
         rerun_mode: rerun mode
         start: start
         end: end
@@ -29,8 +27,6 @@ class PartialRerunMergeEvent(CorvaBaseEvent):
 
     event_type: EventType
     partial_well_rerun_id: int
-    partition: int
-    rerun_partition: int
     rerun_mode: RerunMode
     start: int
     end: int
@@ -46,3 +42,6 @@ class PartialRerunMergeEvent(CorvaBaseEvent):
     source_type: SourceType
     log_type: str
     run_until: int
+
+    class Config:
+        extra = "allow"

src/corva/models/merge/raw.py

@@ -10,8 +10,6 @@
 
 class RawPartialMergeEventData(pydantic.BaseModel):
     partial_well_rerun_id: int
-    partition: int
-    rerun_partition: int
     rerun_mode: RerunMode
     start: int
     end: int
@@ -30,6 +28,9 @@ class RawPartialMergeEventData(pydantic.BaseModel):
     log_type: str
     run_until: int
 
+    class Config:
+        extra = "allow"
+
 
 class RawPartialRerunMergeEvent(CorvaBaseEvent, RawBaseEvent):
     event_type: EventType

tests/unit/test_docs/test_app_types.py

@@ -73,8 +73,6 @@ def test_tutorial007(app_runner):
     partial_rerun_merge_event = PartialRerunMergeEvent(
         event_type="partial-well-rerun-merge",
         partial_well_rerun_id=123,
-        partition=95,
-        rerun_partition=2,
         rerun_mode="realtime",
         start=1543847760,
         end=1543847760,

tests/unit/test_partial_rerun_merge_app.py

@@ -13,8 +13,6 @@
     "event_type": "partial-well-rerun-merge",
     "data": {
         "partial_well_rerun_id": 123,
-        "partition": 95,
-        "rerun_partition": 2,
         "rerun_mode": "realtime",
         "start": 1543847760,
         "end": 1543847760,