Fix wrong item when proxy is filtering

Ref #272
Alignak-monitoring-contrib · Mar 14, 2018 · 9dc0767 · 9dc0767
1 parent c8bb034
commit 9dc0767
Showing 1 changed file with 10 additions and 13 deletions.
diff --git a/alignak_app/qobjects/alignak/problems.py b/alignak_app/qobjects/alignak/problems.py
@@ -86,15 +86,13 @@ def update_action_buttons(self):
 
         """
 
-        # Get QStandardItem
-        standard_item = self.problems_model.item(
-            self.problem_table.selectionModel().currentIndex().row(),
-            self.problem_table.selectionModel().currentIndex().column()
+        # Get item by UserRole
+        item = self.problem_table.model().data(
+            self.problem_table.selectionModel().currentIndex(),
+            Qt.UserRole
         )
 
-        if standard_item:
-            item = standard_item.data(Qt.UserRole)
-
+        if item:
             # If the elements had been ack or downtimed, they would not be present
             self.actions_widget.acknowledge_btn.setEnabled(True)
             self.actions_widget.downtime_btn.setEnabled(True)
@@ -201,14 +199,13 @@ def add_spied_host(self):
 
         """
 
-        # Get QStandardItem
-        standard_item = self.problems_model.item(
-            self.problem_table.selectionModel().currentIndex().row(),
-            self.problem_table.selectionModel().currentIndex().column()
+        # Get item by UserRole
+        item = self.problem_table.model().data(
+            self.problem_table.selectionModel().currentIndex(),
+            Qt.UserRole
         )
 
-        if standard_item:
-            item = standard_item.data(Qt.UserRole)
+        if item:
             if 'service' in item.item_type:
                 item_id = item.data['host']
             else: