support python 2

ericsomdahl · Nov 28, 2017 · 6c8982d · 6c8982d
1 parent 2eef503
commit 6c8982d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/bittrex_websocket/order_book.py b/bittrex_websocket/order_book.py
@@ -4,7 +4,11 @@
 # bittrex_websocket/order_book.py
 # Stanislav Lazarov
 
-import queue
+try:
+    import queue as queue
+except ImportError:
+    import Queue as queue
+
 from copy import deepcopy
 from threading import Thread
 from time import sleep, time
@@ -86,7 +90,7 @@ def _retrieve(q, result):
                 q.task_done()
 
         print('Retrieving order book snapshot...')
-        for j in range(num_threads):
+        for j in list(range(num_threads)):
             worker = Thread(target=_retrieve, args=(order_queue, order_books))
             worker.setDaemon(True)
             worker.start()

diff --git a/bittrex_websocket/summary_state.py b/bittrex_websocket/summary_state.py
@@ -23,6 +23,6 @@ def on_message(self, *args, **kwargs):
     tickers = ['BTC-ETH', 'ETH-1ST', 'BTC-1ST', 'BTC-NEO', 'ETH-NEO']
     ws = MyBittrexSocket(tickers)
     ws.run()
-    for i in range(10):
+    for i in list(range(10)):
         sleep(1)
     ws.stop()
diff --git a/setup.py b/setup.py
@@ -29,6 +29,9 @@
         'Intended Audience :: Information Technology',
         'Topic :: Software Development :: Libraries :: Python Modules',
         'License :: OSI Approved :: MIT License',
+        'Programming Language :: Python',
+        'Programming Language :: Python :: 2',
+        'Programming Language :: Python :: 2.7',
         'Programming Language :: Python :: 3',
         'Programming Language :: Python :: 3.3',
         'Programming Language :: Python :: 3.4',