0.7.8

certego_saas/ext/upload/__init__.py

@@ -1,3 +1,4 @@
 # flake8: noqa
+from .elastic import BIDocument, __BIDocumentInterface
 from .slack import Slack
 from .twitter import Twitter

certego_saas/ext/upload/elastic.py

@@ -47,15 +47,17 @@ def upload(
             qs.filter(index=index)
         docs = qs.order_by("+creation_date")
         if max_number:
+            if max_number > 10000:
+                return False, [
+                    "max_number can't be higher than 10000 for performance reasons."
+                ]
             docs = docs[:max_number]
-        num_docs = docs.count()
-        logger.info(f"Uploading {num_docs} documents")
+        logger.info(f"Uploading {docs.count()} documents")
         jsons = map(lambda x: x.to_bulk(), docs)
         success, errors = bulk(client, jsons, request_timeout=timeout)
-        logger.info(f"Finished Upload. Deleting {num_docs} documents")
-        for doc in docs.iterator():
-            doc.delete()
-        logger.info(f"Finished deleting {num_docs} documents")
+        logger.info("Finished Upload. Starting deletion documents")
+        docs.delete()
+        logger.info("Finished deleting documents")
         return success, errors
 
     def clean(self):

certego_saas/version.py

@@ -1 +1 @@
-VERSION = "0.7.7"
+VERSION = "0.7.8"