From 020763318859d436da1a2a8797659c6fa70a26d5 Mon Sep 17 00:00:00 2001
From: Sven Augustin <sven.augustin@psi.ch>
Date: Wed, 21 Dec 2022 13:16:54 +0100
Subject: [PATCH] more generic logic

---
 grum/dictlist/dictlistwidget.py | 5 ++---
 grum/mainwin.py                 | 3 ++-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/grum/dictlist/dictlistwidget.py b/grum/dictlist/dictlistwidget.py
index f9859f1..39f2a8d 100644
--- a/grum/dictlist/dictlistwidget.py
+++ b/grum/dictlist/dictlistwidget.py
@@ -32,9 +32,8 @@ class DictListWidget(QListWidget):
     def prependItem(self, itm):
         self.insertItem(0, itm)
 
-    def set_alarm_by_key(self, key, state):
-        itm = self.findItems(key, Qt.MatchExactly)[0]
-        itm.set_alarm(state)
+    def get(self, key):
+        return self.findItems(key, Qt.MatchExactly)[0]
 
 
 
diff --git a/grum/mainwin.py b/grum/mainwin.py
index 9610051..d026925 100644
--- a/grum/mainwin.py
+++ b/grum/mainwin.py
@@ -54,7 +54,8 @@ class MainWindow(QMainWindow):
             desc = self.lst.data[name]
             sub.plot.setData(*desc.data)
         else:
-            self.lst.set_alarm_by_key(name, True)
+            item = self.lst.get(name)
+            item.set_alarm(True)
 
 
     def on_select_list_item(self, item):