debug supported in sockets, logs all deltas and tx received in socket.log file

Author: Insmael

.gitignore

@@ -123,5 +123,6 @@ cryptomarket/exchange/client.pyc
 cryptomarket/.DS_Store
 
 test.py
+socket.log
 
 */__pycache__/*

cryptomarket/exchange/client.py

@@ -5,7 +5,6 @@
 from __future__ import unicode_literals
 
 import json
-import logging
 import requests
 import time
 import warnings
@@ -29,7 +28,7 @@ class Client(object):
     BASE_API_URI = 'https://api.cryptomkt.com/'
     API_VERSION = 'v2'
 
-    def __init__(self, api_key, api_secret, base_api_uri=None, api_version=None):
+    def __init__(self, api_key, api_secret, base_api_uri=None, api_version=None, debug=False):
         if not api_key:
             raise ValueError('Missing `api_key`.')
         if not api_secret:
@@ -528,11 +527,11 @@ def get_auth_socket(self):
         return self._make_api_object(response, APIObject)
 
 
-    def get_socket(self):
+    def get_socket(self, debug=False):
         """returns a socket connection with cryptomkt.
         """
         if self.socket is None:
             auth = self.get_auth_socket()
             del auth['verify']
-            self.socket = Socket(auth)
+            self.socket = Socket(auth, debug=debug)
         return self.socket

cryptomarket/exchange/socket.py

@@ -3,14 +3,27 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 
+import logging 
+
 import socketio
 
 from pubsub import pub
 
 from .patch_json import patch
 
 class Socket(object):
-    def __init__(self, socketids):
+    def __init__(self, socketids, debug=False):
+        logger = logging.getLogger(__name__)
+        logger.setLevel(logging.DEBUG)
+        f_handler = logging.FileHandler('socket.log')
+        if debug:
+            f_handler.setLevel(logging.DEBUG)
+        f_format = logging.Formatter('%(asctime)s - %(message)s')
+        #f_format.datefmt = '%d-%b-%y %H:%M:%S'
+        f_handler.setFormatter(f_format)
+        logger.addHandler(f_handler)
+        self.logger = logger
+
         self.url_worker = 'https://worker.cryptomkt.com'
         self.currencies_data = dict()
         self.balance_data = dict()
@@ -33,16 +46,21 @@ def connect():
         def disconnect():
             print('disconnected from server')
 
+        self.logger.debug('connecting to cryptomarket')
         sio.connect(self.url_worker)
+
+        self.logger.debug('authenticating')
         sio.emit('user-auth', socketids)
 
 
         @sio.on('currencies')
         def currencies_handler(data):
+            self.logger.debug('currencies recieved: {}'.format(data))
             self.currencies_data = data
 
         @sio.on('currencies-delta')
         def currencies_delta_handler(delta):
+            self.logger.debug('delta-currencies recieved: {}'.format(delta))
             if self.currencies_data['to_tx'] != delta['from_tx']:
                 sio.emit('currencies')
                 return
@@ -53,6 +71,7 @@ def currencies_delta_handler(delta):
 
         @sio.on('balance')
         def balance_handler(data):
+            self.logger.debug('balance recieved: {}'.format(data))
             for _, value in data['data'].items():
                 currency = value['currency']
                 currency_data = self.currencies_data['data'][currency]
@@ -73,6 +92,7 @@ def balance_handler(data):
 
         @sio.on('balance-delta')
         def balance_delta_handler(deltas):
+            self.logger.debug('balance-delta recieved: {}'.format(deltas))
             for delta in deltas:
                 if self.balance_data['to_tx'] != delta['from_tx']:
                     sio.emit('balance')
@@ -99,11 +119,13 @@ def balance_delta_handler(deltas):
 
         @sio.on('open-orders')
         def open_orders_handler(data):
+            self.logger.debug('open-orders recieved: {}'.format(data))
             self.open_orders_data = data
             pub.sendMessage('open-orders', data=data['data'].copy())
 
         @sio.on('open-orders-delta')
         def open_orders_delta_handler(deltas):
+            self.logger.debug('open-orders-delta recieved: {}'.format(deltas))
             for delta in deltas:
                 if self.open_orders_data['to_tx'] != delta['from_tx']:
                     sio.emit('open-orders')
@@ -119,11 +141,13 @@ def open_orders_delta_handler(deltas):
 
         @sio.on('historical-orders')
         def historical_orders_handler(data):
+            self.logger.debug('historical-orders recieved: {}'.format(data))
             self.historical_orders_data = data
             pub.sendMessage('historical-orders', data=data['data'].copy())
 
         @sio.on('historical-orders-delta')
         def historical_orders_delta_handler(deltas):
+            self.logger.debug('historical-orders-delta recieved: {}'.format(deltas))
             for delta in deltas:
                 if self.historical_orders_data['to_tx'] != delta['from_tx']:
                     sio.emit('historical-orders')
@@ -139,11 +163,13 @@ def historical_orders_delta_handler(deltas):
 
         @sio.on('operated')
         def operated_handler(data):
+            self.logger.debug('operated recieved: {}'.format(data))
             self.operated_data = data
             pub.sendMessage('operated', data=data['data'].copy())
 
         @sio.on('operated-delta')
         def operated_delta_handler(deltas):
+            self.logger.debug('operated-delta recieved: {}'.format(deltas))
             for delta in deltas:
                 if self.operated_data['to_tx'] != delta['from_tx']:
                     sio.emit('operated')
@@ -159,6 +185,7 @@ def operated_delta_handler(deltas):
 
         @sio.on('open-book')
         def open_book_handler(data):
+            self.logger.debug('open-book recieved: {}'.format(data))
             stock_id = data['stock_id']
             self.open_book_data.update({stock_id:data})
             pub.sendMessage(
@@ -172,6 +199,7 @@ def open_book_handler(data):
 
         @sio.on('open-book-delta')
         def open_book_delta_handler(delta):
+            self.logger.debug('open-book-delta recieved: {}'.format(delta))
             stock_id = delta['stock_id']
             if (stock_id not in self.open_book_data
                 or self.open_book_data[stock_id]['to_tx'] != delta['from_tx']):
@@ -194,6 +222,7 @@ def open_book_delta_handler(delta):
 
         @sio.on('historical-book')
         def historical_book_handler(data):
+            self.logger.debug('historical-book recieved: {}'.format(data))
             stock_id = data['stock_id']
             self.historical_book_data.update({stock_id:data})
             pub.sendMessage(
@@ -203,6 +232,7 @@ def historical_book_handler(data):
 
         @sio.on('historical-book-delta')
         def historical_book_delta_handler(delta):
+            self.logger.debug('historical-book-delta recieved: {}'.format(delta))
             stock_id = delta['stock_id']
             if (stock_id not in self.historical_book_data
                or self.historical_book_data[stock_id]['to_tx'] != delta['from_tx']):
@@ -221,6 +251,7 @@ def historical_book_delta_handler(delta):
 
         @sio.on('candles')
         def candles_handler(data):
+            self.logger.debug('candles recieved: {}'.format(data))
             stock_id = data['stock_id']
             self.candles_data.update({stock_id:data})
 
@@ -236,6 +267,7 @@ def candles_handler(data):
 
         @sio.on('candles-delta')
         def candles_delta_handler(delta):
+            self.logger.debug('candles-delta recieved: {}'.format(delta))
             stock_id = delta['stock_id']
             if (stock_id not in self.candles_data
                 or self.candles_data[stock_id]['to_tx'] != delta['from_tx']):
@@ -257,11 +289,13 @@ def candles_delta_handler(delta):
 
         @sio.on('board')
         def board_handler(data):
+            self.logger.debug('board recieved: {}'.format(data))
             self.board_data = data
             pub.sendMessage('ticker', data=data.copy())
 
         @sio.on('board-delta')
         def board_delta_handler(delta):
+            self.logger.debug('board-delta recieved: {}'.format(delta))
             if self.board_data['to_tx'] != delta['from_tx']:
                 sio.emit('board')
                 return
@@ -273,12 +307,15 @@ def board_delta_handler(delta):
         self.sio = sio
 
     def subscribe(self, *market_pairs):
+        self.logger.debug('starting subscriptions: {}'.format(market_pairs))
         for pair in market_pairs:
             self.sio.emit('subscribe', pair)
 
     def unsubscribe(self, *market_pairs):
+        self.logger.debug('ending subscriptions: {}'.format(market_pairs))
         for pair in market_pairs:
             self.sio.emit('unsubscribe', pair)
 
     def on(self, event, handler):
+        self.logger.debug('subscribing to event: {}'.format(event))
         pub.subscribe(handler, event)