DFIQ 1.1 changes (#1122)

yeti-platform · Aug 22, 2024 · b1f8787 · b1f8787
1 parent 3fff0d5
commit b1f8787
Show file tree

Hide file tree

Showing 17 changed files with 542 additions and 537 deletions.
diff --git a/core/database_arango.py b/core/database_arango.py
@@ -105,6 +105,15 @@ def connect(
                 ],
             },
         )
+
+        for collection_data in self.db.collections():
+            if collection_data["system"]:
+                continue
+            collection = self.db.collection(collection_data["name"])
+            for index in collection.indexes():
+                if index["type"] == "persistent":
+                    collection.delete_index(index["id"])
+
         self.db.collection("observables").add_persistent_index(
             fields=["value", "type"], unique=True
         )
@@ -116,7 +125,7 @@ def connect(
             fields=["name", "type"], unique=True
         )
         self.db.collection("dfiq").add_persistent_index(
-            fields=["dfiq_id", "type"], unique=True
+            fields=["uuid"], unique=True, sparse=True
         )
 
     def clear(self, truncate=True):