Merge pull request #65 from certego/develop

0.7.8
certego · Nov 30, 2023 · 0aa0bf4 · 0aa0bf4
2 parents 6c26138 + fb99814
commit 0aa0bf4
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 7 deletions.
diff --git a/certego_saas/ext/upload/__init__.py b/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
diff --git a/certego_saas/ext/upload/elastic.py b/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):

diff --git a/certego_saas/version.py b/certego_saas/version.py
@@ -1 +1 @@
-VERSION = "0.7.7"
+VERSION = "0.7.8"