Merge pull request carbonblack#242 from crahan/alertquery_len_fix

abowersox-cb · web-flow · commit f06a1bceb6f1 · 2020-07-13T17:09:03.000-06:00
Add missing __len__() implementation for IterableQueryMixin to fix len() on Alert query
diff --git a/src/cbapi/psc/alerts_query.py b/src/cbapi/psc/alerts_query.py
@@ -24,6 +24,8 @@ def __init__(self, doc_class, cb):
         self._time_filter = {}
         self._sortcriteria = {}
         self._bulkupdate_url = "/appservices/v6/orgs/{0}/alerts/workflow/_criteria"
+        self._count_valid = False
+        self._total_results = 0
 
     def _update_criteria(self, key, newlist):
         """
diff --git a/src/cbapi/psc/base_query.py b/src/cbapi/psc/base_query.py
@@ -261,7 +261,7 @@ def one(self):
         return res[0]
 
     def __len__(self):
-        return 0
+        return self._count()
 
     def __getitem__(self, item):
         return None
