From 111dcb043d848954a3f08e37412aa463fe938a02 Mon Sep 17 00:00:00 2001 From: mohamedemad4 Date: Sun, 6 Sep 2020 15:47:26 +0200 Subject: [PATCH] ammends to the last commit --- tiingo/wsclient.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tiingo/wsclient.py b/tiingo/wsclient.py index 5d564190..2cb585c5 100644 --- a/tiingo/wsclient.py +++ b/tiingo/wsclient.py @@ -1,9 +1,12 @@ import os import websocket -import thread +try: + import thread +except ImportError: + import _thread as thread import time import json -from exceptions import MissingRequiredArgument +from tiingo.exceptions import MissingRequiredArgumentError GLOB_config=None GLOB_on_msg_cb=None @@ -93,7 +96,7 @@ def __init__(self,config={},endpoint=None,on_msg_cb=None): self.on_msg_cb = on_msg_cb if not self.on_msg_cb: - raise MissingRequiredArgument("please define on_msg_cb It's a callback that gets called when new messages arrive " + raise MissingRequiredArgumentError("please define on_msg_cb It's a callback that gets called when new messages arrive " "Example:" "def cb_fn(msg):" " print(msg)") @@ -108,5 +111,4 @@ def __init__(self,config={},endpoint=None,on_msg_cb=None): on_error = genericWebsocketClient.on_error, on_close = genericWebsocketClient.on_close, on_open = genericWebsocketClient.on_open) - ws.run_forever() - + ws.run_forever() \ No newline at end of file