Reconnection bug.

Author: slazarov

README.md

@@ -55,6 +55,16 @@ pip install bittrex-websocket-aio
 ```
 
 # Methods
+#### Custom URL
+Custom URLs can be passed to the client upon instantiating.
+```python
+# 'https://socket.bittrex.com/signalr' is currently Cloudflare protected
+# 'https://beta.bittrex.com/signalr' (DEFAULT) is not
+
+# Create the socket instance
+ws = MySocket(url=None)
+# rest of your code
+```
 #### Subscribe Methods
 ```python
 def subscribe_to_exchange_deltas(self, tickers):
@@ -153,6 +163,10 @@ def on_error(self, error):
 Check the examples folder.
 
 # Change log
+0.0.0.2.5 - 22/04/2018
+* Custom urls can now be passed to the client
+* If `cfscrape` is installed, the client will automatically use it
+
 0.0.2 - 14/04/2018
 * Implemented reconnection
 * Implemented private account-level methods. Check `authenticate(self, api_key, api_secret)`.

bittrex_websocket/websocket_client.py

@@ -15,18 +15,24 @@
 from ._exceptions import *
 from signalr_aio import Connection
 
+try:
+    from cfscrape import create_scraper as Session
+except ImportError:
+    from requests import Session
+
 logger = logging.getLogger(__name__)
 
 
 class BittrexSocket(WebSocket):
 
-    def __init__(self):
+    def __init__(self, url=None):
         self.control_queue = None
         self.invokes = []
         self.tickers = None
         self.connection = None
         self.threads = []
         self.credentials = None
+        self.url = BittrexParameters.URL if url is None else url
         self._start_main_thread()
 
     def _start_main_thread(self):
@@ -50,7 +56,7 @@ def control_queue_handler(self):
                 self.control_queue.task_done()
 
     def _handle_connect(self):
-        connection = Connection(BittrexParameters.URL)
+        connection = Connection(BittrexParameters.URL, Session())
         hub = connection.register_hub(BittrexParameters.HUB)
         connection.received += self._on_debug
         connection.error += self.on_error
@@ -66,7 +72,7 @@ def _handle_connect(self):
 
     def _connection_handler(self):
         try:
-            logger.info('Establishing connection to Bittrex.')
+            logger.info('Establishing connection to Bittrex through {}.'.format(self.url))
             self.connection.conn.start()
         except ConnectionClosed as e:
             if e.code == 1000:

examples/ticker_updates.py

@@ -17,8 +17,8 @@
 
 def main():
     class MySocket(BittrexSocket):
-        def __init__(self):
-            super().__init__()
+        def __init__(self, url=None):
+            super().__init__(url)
             self.ticker_updates_container = {}
 
         async def on_public(self, msg):

requirements.txt

@@ -1,3 +1,2 @@
-uvloop>=0.9.1
 websockets>=4.0.1
 signalr-client-aio>=0.0.1.6.1

setup.py

@@ -4,14 +4,13 @@
 
 install_requires = \
     [
-        'uvloop>=0.9.1',
         'websockets>=4.0.1',
         'signalr-client-aio>=0.0.1.6.1'
     ]
 
 setup(
     name='bittrex-websocket-aio',
-    version='0.0.0.2.3',
+    version='0.0.0.2.5',
     author='Stanislav Lazarov',
     author_email='[email protected]',
     license='MIT',