modified: pybit/__init__.py

verata-veritatis · Jun 16, 2021 · fb6d54f · fb6d54f
1 parent 2423554
commit fb6d54f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pybit/__init__.py b/pybit/__init__.py
@@ -1816,7 +1816,11 @@ def _on_message(self, message):
 
                 # Record the initial snapshot.
                 elif 'snapshot' in msg_json['type']:
-                    self.data[topic] = msg_json['data']
+                    if 'order_book' in msg_json['data']:
+                        self.data[topic] = msg_json['data']['order_book'] if self.trim else msg_json
+                    else:
+                        self.data[topic] = msg_json['data'] if self.trim else msg_json
+                    #self.data[topic] = msg_json['data']
 
             # For incoming 'order' and 'stop_order' data.
             elif any(i in topic for i in ['order', 'stop_order']):