Fixed contingency issues

SanPen · Nov 16, 2023 · cde8448 · cde8448
1 parent d1c9e0d
commit cde8448
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 18 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/src/GridCal/Gui/Main/SubClasses/Model/diagrams.py b/src/GridCal/Gui/Main/SubClasses/Model/diagrams.py
@@ -1322,11 +1322,15 @@ def get_selected_contingency_devices(self) -> List[dev.EditableDevice]:
         Get the selected buses
         :return:
         """
-        lst: List[dev.EditableDevice] = list()
-        for k, elm in enumerate(self.circuit.get_contingency_devices()):
-            if elm.graphic_obj is not None:
-                if elm.graphic_obj.isSelected():
-                    lst.append(elm)
+        diagram = self.get_selected_diagram_widget()
+
+        if isinstance(diagram, BusBranchEditorWidget):
+            lst = diagram.get_selection_api_objects()
+        elif isinstance(diagram, BusViewerWidget):
+            lst = diagram.get_selection_api_objects()
+        else:
+            lst = list()
+
         return lst
 
     def get_selected_investment_devices(self) -> List[dev.EditableDevice]: