threathunter: Enforce org_key presence

woodruffw · woodruffw · commit 7ba0f0716b87 · 2019-05-20T14:21:28.000-04:00
diff --git a/src/cbapi/psc/threathunter/rest_api.py b/src/cbapi/psc/threathunter/rest_api.py
@@ -1,6 +1,7 @@
 from cbapi.psc.threathunter.query import Query
 from cbapi.connection import BaseAPI
 from cbapi.psc.threathunter.models import ReportSeverity
+from cbapi.errors import CredentialError
 import logging
 
 log = logging.getLogger(__name__)
@@ -21,6 +22,9 @@ def __init__(self, *args, **kwargs):
         super(CbThreatHunterAPI, self).__init__(product_name="psc", *args, **kwargs)
         self._lr_scheduler = None
 
+        if not self.credentials.get("org_key", None):
+            raise CredentialError("No organization key specified")
+
     def _perform_query(self, cls, **kwargs):
         if hasattr(cls, "_query_implementation"):
             return cls._query_implementation(self)
