Bug fix, forgot to return on_open

Author: jbonzo

polygon/__init__.py

@@ -0,0 +1,2 @@
+from polygon.websocket import WebSocketClient, STOCKS_CLUSTER, FOREX_CLUSTER, CRYPTO_CLUSTER
+from polygon.rest import RESTClient

polygon/rest/__init__.py

@@ -1,75 +1 @@
-# coding: utf-8
-
-# flake8: noqa
-
-"""
-    Polygon API
-
-    The future of fintech.  # noqa: E501
-
-    OpenAPI spec version: 1.0.1
-    
-    Generated by: https://github.com/swagger-api/swagger-codegen.git
-"""
-
-from __future__ import absolute_import
-
-# import apis into sdk package
-from polygon.rest.api import CryptoApi
-from polygon.rest.api import ForexCurrenciesApi
-from polygon.rest.api import ReferenceApi
-from polygon.rest.api import StocksEquitiesApi
-# import ApiClient
-from polygon.rest.api_client import ApiClient
-from polygon.rest.configuration import Configuration
-# import models into sdk package
-from polygon.rest.models import AggResponse
-from polygon.rest.models import Aggregate
-from polygon.rest.models import Aggv2
-from polygon.rest.models import AnalystRatings
-from polygon.rest.models import Company
-from polygon.rest.models import ConditionTypeMap
-from polygon.rest.models import Conflict
-from polygon.rest.models import CryptoExchange
-from polygon.rest.models import CryptoSnapshotAgg
-from polygon.rest.models import CryptoSnapshotBookItem
-from polygon.rest.models import CryptoSnapshotTicker
-from polygon.rest.models import CryptoSnapshotTickerBook
-from polygon.rest.models import CryptoTick
-from polygon.rest.models import CryptoTickJson
-from polygon.rest.models import Dividend
-from polygon.rest.models import Earning
-from polygon.rest.models import Error
-from polygon.rest.models import Exchange
-from polygon.rest.models import Financial
-from polygon.rest.models import Financials
-from polygon.rest.models import Forex
-from polygon.rest.models import ForexAggregate
-from polygon.rest.models import ForexSnapshotAgg
-from polygon.rest.models import ForexSnapshotTicker
-from polygon.rest.models import HistTrade
-from polygon.rest.models import LastForexQuote
-from polygon.rest.models import LastForexTrade
-from polygon.rest.models import LastQuote
-from polygon.rest.models import LastTrade
-from polygon.rest.models import MarketHoliday
-from polygon.rest.models import MarketStatus
-from polygon.rest.models import News
-from polygon.rest.models import NotFound
-from polygon.rest.models import Quote
-from polygon.rest.models import RatingSection
-from polygon.rest.models import Split
-from polygon.rest.models import StockSymbol
-from polygon.rest.models import StocksSnapshotAgg
-from polygon.rest.models import StocksSnapshotBookItem
-from polygon.rest.models import StocksSnapshotQuote
-from polygon.rest.models import StocksSnapshotTicker
-from polygon.rest.models import StocksSnapshotTickerBook
-from polygon.rest.models import StocksV2NBBO
-from polygon.rest.models import StocksV2Trade
-from polygon.rest.models import Symbol
-from polygon.rest.models import SymbolTypeMap
-from polygon.rest.models import Ticker
-from polygon.rest.models import TickerSymbol
-from polygon.rest.models import Trade
-from polygon.rest.models import Unauthorized
+from .rest_client import RESTClient

polygon/websocket/__init__.py

@@ -0,0 +1 @@
+from polygon.websocket.websocket_client import WebSocketClient, STOCKS_CLUSTER, FOREX_CLUSTER, CRYPTO_CLUSTER

polygon_client/websocket_client.py renamed to polygon/websocket/websocket_client.py

@@ -9,8 +9,8 @@
 CRYPTO_CLUSTER = "crypto"
 
 
-class WebSocketClient(object):
-    DEFAULT_HOST = 'socket.polygon.io'
+class WebSocketClient:
+    DEFAULT_HOST = "socket.polygon.io"
 
     # TODO: Either an instance of the client couples 1:1 with the cluster or an instance of the Client couples 1:3 with
     #  the 3 possible clusters (I think I like client per, but then a problem is the user can make multiple clients for
@@ -32,6 +32,9 @@ def __init__(self, cluster: str, auth_key: str, process_message: Optional[Callab
         # self._run_thread is only set if the client is run asynchronously
         self._run_thread: Optional[threading.Thread] = None
 
+        # TODO: this probably isn't great design.
+        #  If the user defines their own signal handler then this will gets overwritten.
+        #  We still need to make sure that killing, terminating, interrupting the program closes the connection
         signal.signal(signal.SIGINT, self._cleanup_signal_handler())
         signal.signal(signal.SIGTERM, self._cleanup_signal_handler())
 
@@ -92,6 +95,7 @@ def _default_process_message(message):
     def _default_on_open(self):
         def f(ws):
             self._authenticate(ws)
+
         return f
 
     @staticmethod

polygon_client/__init__.py

@@ -6,7 +6,9 @@ certifi==2019.9.11
 chardet==3.0.4
 idna==2.8
 multidict==4.5.2
+requests==2.22.0
 six==1.12.0
+urllib3==1.25.6
 websocket-client==0.56.0
 websockets==8.0.2
 yarl==1.3.0