fix channel callbacks to show seq ids for channels that need them

s4w3d0ff · Jul 14, 2019 · e2887de · e2887de
1 parent 1992241
commit e2887de
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/poloniex/__init__.py b/poloniex/__init__.py
@@ -745,8 +745,15 @@ def on_message(self, data):
                 return False
         if 'callback' in self.channels[chan]:
             # activate chan callback
-            if not chan in ['account', 'heartbeat']:
+            if chan == 'heartbeat':
+                # show whole heartbeat
+                self.socket._callback(self.channels[chan]['callback'], message)
+            elif chan in ['ticker', '24hvolume']:
+                # ticker and 24hvolume dont need seq id
                 message = message[2]
+            else:
+                # show seq id for everything else
+                message = message[1]
             self.socket._callback(self.channels[chan]['callback'], message)
 
     def on_error(self, error):