Use pattern-matching for send_smoother_event

equinor · Nov 20, 2024 · d8182ac · d8182ac
1 parent fec9389
commit d8182ac
Showing 1 changed file with 34 additions and 30 deletions.
diff --git a/src/ert/run_models/base_run_model.py b/src/ert/run_models/base_run_model.py
@@ -230,40 +230,44 @@ def send_event(self, event: StatusEvents) -> None:
     def send_smoother_event(
         self, iteration: int, run_id: uuid.UUID, event: AnalysisEvent
     ) -> None:
-        if isinstance(event, AnalysisStatusEvent):
-            self.send_event(
-                RunModelStatusEvent(iteration=iteration, run_id=run_id, msg=event.msg)
-            )
-        elif isinstance(event, AnalysisTimeEvent):
-            self.send_event(
-                RunModelTimeEvent(
-                    iteration=iteration,
-                    run_id=run_id,
-                    elapsed_time=event.elapsed_time,
-                    remaining_time=event.remaining_time,
+        match event:
+            case AnalysisStatusEvent(msg=msg):
+                self.send_event(
+                    RunModelStatusEvent(iteration=iteration, run_id=run_id, msg=msg)
                 )
-            )
-        elif isinstance(event, AnalysisErrorEvent):
-            self.send_event(
-                RunModelErrorEvent(
-                    iteration=iteration,
-                    run_id=run_id,
-                    error_msg=event.error_msg,
-                    data=event.data,
+            case AnalysisTimeEvent():
+                self.send_event(
+                    RunModelTimeEvent(
+                        iteration=iteration,
+                        run_id=run_id,
+                        elapsed_time=event.elapsed_time,
+                        remaining_time=event.remaining_time,
+                    )
                 )
-            )
-        elif isinstance(event, AnalysisDataEvent):
-            self.send_event(
-                RunModelDataEvent(
-                    iteration=iteration, run_id=run_id, name=event.name, data=event.data
+            case AnalysisErrorEvent():
+                self.send_event(
+                    RunModelErrorEvent(
+                        iteration=iteration,
+                        run_id=run_id,
+                        error_msg=event.error_msg,
+                        data=event.data,
+                    )
                 )
-            )
-        elif isinstance(event, AnalysisCompleteEvent):
-            self.send_event(
-                RunModelUpdateEndEvent(
-                    iteration=iteration, run_id=run_id, data=event.data
+            case AnalysisDataEvent():
+                self.send_event(
+                    RunModelDataEvent(
+                        iteration=iteration,
+                        run_id=run_id,
+                        name=event.name,
+                        data=event.data,
+                    )
+                )
+            case AnalysisCompleteEvent():
+                self.send_event(
+                    RunModelUpdateEndEvent(
+                        iteration=iteration, run_id=run_id, data=event.data
+                    )
                 )
-            )
 
     @property
     def queue_system(self) -> QueueSystem: