Use hasattr to determine if handler is callable

bmoscon · Feb 24, 2024 · 17ddcd8 · 17ddcd8
1 parent df692b6
commit 17ddcd8
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/cryptofeed/backends/aggregate.py b/cryptofeed/backends/aggregate.py
@@ -13,9 +13,13 @@
 class AggregateCallback:
     def __init__(self, handler):
         self.handler = handler
-        if not callable(self.handler):
-            setattr(self, 'start', self.handler.start)
-            setattr(self, 'stop', self.handler.stop)
+        if (
+            hasattr(self.handler, "__class__")
+            and hasattr(self.handler, "start")
+            and hasattr(self.handler, "stop")
+        ):
+            setattr(self, "start", self.handler.start)
+            setattr(self, "stop", self.handler.stop)
             self.__name__ = self.handler.__class__