From 6d6b1e9155096f60f8d6c5f02fe8491cd69fa653 Mon Sep 17 00:00:00 2001
From: wyzula-jan <133381102+wyzula-jan@users.noreply.github.com>
Date: Tue, 17 Oct 2023 10:17:41 +0200
Subject: [PATCH] refactor: bec_dispatcher.py changed to Ivan's version

---
 bec_widgets/bec_dispatcher.py | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/bec_widgets/bec_dispatcher.py b/bec_widgets/bec_dispatcher.py
index 9a6a5a42..53424469 100644
--- a/bec_widgets/bec_dispatcher.py
+++ b/bec_widgets/bec_dispatcher.py
@@ -1,7 +1,6 @@
 import argparse
 import itertools
 import os
-from dataclasses import dataclass
 from typing import Callable
 
 from bec_lib import BECClient
@@ -16,17 +15,15 @@ _signal_class_factory = (
 )
 
 
-@dataclass
 class _Connection:
     """Utility class to keep track of slots connected to a particular redis consumer"""
 
-    consumer: RedisConsumerThreaded
-    slots = set()
-    # keep a reference to a new signal class, so it is not gc'ed
-    _signal_container = next(_signal_class_factory)()
-
-    def __post_init__(self):
-        self.signal = self._signal_container.signal
+    def __init__(self, consumer) -> None:
+        self.consumer: RedisConsumerThreaded = consumer
+        self.slots = set()
+        # keep a reference to a new signal class, so it is not gc'ed
+        self._signal_container = next(_signal_class_factory)()
+        self.signal: pyqtSignal = self._signal_container.signal
 
 
 class _BECDispatcher(QObject):