added additional error handling for search functions

pyaurorax/conjunctions/conjunctions.py

@@ -5,6 +5,7 @@
 import datetime
 import humanize
 from typing import Dict, List, Optional, Union
+from ..exceptions import AuroraXSearchException
 from .classes.search import Search
 from ..api import AuroraXRequest, urls
 from ..requests import STANDARD_POLLING_SLEEP_TIME
@@ -136,6 +137,11 @@ def search(start: datetime.datetime,
         print("[%s] Waiting for data ..." % (datetime.datetime.now()))
     s.wait(poll_interval=poll_interval, verbose=verbose)
 
+    # check if error condition encountered
+    if (s.status["search_result"]["error_condition"] is True):
+        # error encountered
+        raise AuroraXSearchException(s.logs[-1]["summary"])
+
     # get the data
     if (verbose is True):
         print("[%s] Retrieving data ..." % (datetime.datetime.now()))

pyaurorax/data_products/data_products.py

@@ -9,7 +9,8 @@
 from .classes.search import Search
 from ..sources import (DataSource,
                        list as sources_list)
-from ..exceptions import (AuroraXValidationException,
+from ..exceptions import (AuroraXSearchException,
+                          AuroraXValidationException,
                           AuroraXUploadException,
                           AuroraXBadParametersException,
                           AuroraXException)
@@ -137,6 +138,11 @@ def search(start: datetime.datetime,
         print("[%s] Waiting for data ..." % (datetime.datetime.now()))
     s.wait(poll_interval=poll_interval, verbose=verbose)
 
+    # check if error condition encountered
+    if (s.status["search_result"]["error_condition"] is True):
+        # error encountered
+        raise AuroraXSearchException(s.logs[-1]["summary"])
+
     # get the data
     if (verbose is True):
         print("[%s] Retrieving data ..." % (datetime.datetime.now()))

pyaurorax/ephemeris/ephemeris.py

@@ -9,7 +9,8 @@
 from .classes.search import Search
 from ..sources import (DataSource,
                        list as sources_list)
-from ..exceptions import (AuroraXValidationException,
+from ..exceptions import (AuroraXSearchException,
+                          AuroraXValidationException,
                           AuroraXUploadException,
                           AuroraXBadParametersException)
 from ..requests import STANDARD_POLLING_SLEEP_TIME
@@ -132,6 +133,11 @@ def search(start: datetime.datetime,
         print("[%s] Waiting for data ..." % (datetime.datetime.now()))
     s.wait(poll_interval=poll_interval, verbose=verbose)
 
+    # check if error condition encountered
+    if (s.status["search_result"]["error_condition"] is True):
+        # error encountered
+        raise AuroraXSearchException(s.logs[-1]["summary"])
+
     # get the data
     if (verbose is True):
         print("[%s] Retrieving data ..." % (datetime.datetime.now()))

pyaurorax/exceptions.py

@@ -96,3 +96,10 @@ class AuroraXTimeoutException(AuroraXException):
     A timeout was reached while communicating with the AuroraX API
     """
     pass
+
+
+class AuroraXSearchException(AuroraXException):
+    """
+    An error occured in the API while performing a search
+    """
+    pass