Merge pull request #59 from certego/develop

0.7.4

certego_saas/ext/upload/elastic.py

@@ -51,7 +51,8 @@ def upload(
         logger.info(f"Uploading {docs.count()} documents")
         jsons = map(lambda x: x.to_bulk(), docs)
         success, errors = bulk(client, jsons, request_timeout=timeout)
-        docs.delete()
+        for doc in docs.iterator():
+            doc.delete()
         return success, errors
 
     def clean(self):
@@ -68,7 +69,7 @@ def __repr__(self):
     from mongoengine import fields as mongo_fields
 
 except ImportError:
-    from django.db.models import JSONField, Model
+    from django.db.models import Index, JSONField, Model
     from django.db.models import fields as django_fields
 
     class BIDocument(__BIDocumentInterface, Model):
@@ -78,6 +79,16 @@ class BIDocument(__BIDocumentInterface, Model):
         count = django_fields.PositiveIntegerField()
         kwargs = JSONField()
 
+        class Meta:
+            indexes = [
+                Index(
+                    fields=[
+                        "index",
+                        "time",
+                    ]
+                )
+            ]
+
 else:
 
     class BIDocument(__BIDocumentInterface, Document):
@@ -88,3 +99,4 @@ class BIDocument(__BIDocumentInterface, Document):
         category = mongo_fields.StringField(required=True)
         count = mongo_fields.IntField(required=True, min_value=0)
         kwargs = mongo_fields.DictField(required=False)
+        meta = {"indexes": ["index", "time"]}

certego_saas/version.py

@@ -1 +1 @@
-VERSION = "0.7.3"
+VERSION = "0.7.4"