no breaking changes

clickingbuttons · clickingbuttons · commit 3cb61ae13a5d · 2022-05-24T15:19:04.000-04:00
diff --git a/polygon/exceptions.py b/polygon/exceptions.py
@@ -1,17 +1,17 @@
 
-class PolyBadResponse(Exception):
+class BadResponse(Exception):
     """
     Non-200 response from API
     """
     pass
 
-class PolyAuthError(Exception):
+class AuthError(Exception):
     """"
     Empty or invalid API key
     """
     pass
 
-class PolyMissingResults(Exception):
+class MissingResults(Exception):
     """
     Missing results key
     """
diff --git a/polygon/rest/base.py b/polygon/rest/base.py
@@ -8,7 +8,7 @@
 import pkg_resources  # part of setuptools
 from ..logging import get_logger
 import logging
-from ..exceptions import PolyAuthError, PolyBadResponse, PolyMissingResults
+from ..exceptions import AuthError, BadResponse, MissingResults
 
 logger = get_logger("RESTClient")
 version = "unknown"
@@ -29,7 +29,7 @@ def __init__(
         verbose: bool,
     ):
         if api_key is None:
-            raise PolyAuthError(
+            raise AuthError(
                 f"Must specify env var POLYGON_API_KEY or pass api_key in constructor"
             )
         self.API_KEY = api_key
@@ -74,7 +74,7 @@ def _get(
         )
 
         if resp.status != 200:
-            raise PolyBadResponse(resp.data.decode("utf-8"))
+            raise BadResponse(resp.data.decode("utf-8"))
 
         if raw:
             return resp
@@ -83,7 +83,7 @@ def _get(
 
         if result_key:
             if result_key not in obj:
-                raise PolyMissingResults(
+                raise MissingResults(
                     f'Expected key "{result_key}" in response {obj}.'
                     + "Make sure you have sufficient permissions and your request parameters are valid."
                     + f"This is the url that returned no results: {resp.geturl()}"
diff --git a/polygon/websocket/__init__.py b/polygon/websocket/__init__.py
@@ -11,7 +11,7 @@
 from websockets.exceptions import ConnectionClosedOK, ConnectionClosedError
 from ..logging import get_logger
 import logging
-from ..exceptions import PolyAuthError
+from ..exceptions import AuthError
 
 env_key = "POLYGON_API_KEY"
 logger = get_logger("WebSocketClient")
@@ -42,7 +42,7 @@ def __init__(
         :return: A client.
         """
         if api_key is None:
-            raise PolyAuthError(
+            raise AuthError(
                 f"Must specify env var {env_key} or pass api_key in constructor"
             )
         self.api_key = api_key
@@ -104,7 +104,7 @@ async def connect(
                 auth_msg_parsed = json.loads(auth_msg)
                 logger.debug("authed: %s", auth_msg)
                 if auth_msg_parsed[0]["status"] == "auth_failed":
-                    raise PolyAuthError(auth_msg_parsed[0]["message"])
+                    raise AuthError(auth_msg_parsed[0]["message"])
                 while True:
                     if self.schedule_resub:
                         logger.debug(
