Ensure index added to es query ops

src/mp_api/routes/electronic_structure/query_operators.py

@@ -65,6 +65,12 @@ def query(
 
         return {"criteria": crit}
 
+    def ensure_indexes(self):
+
+        keys = ["band_gap", "efermi", "magnetic_ordering", "is_gap_direct", "is_metal"]
+
+        return [(key, False) for key in keys]
+
 
 class BSDataQuery(QueryOperator):
     """
@@ -131,6 +137,16 @@ def query(
 
         return {"criteria": crit}
 
+    def ensure_indexes(self):
+
+        keys = ["bandstructure"]
+
+        for bs_type in BSPathType:
+            for field in ["band_gap", "efermi"]:
+                keys.append(f"bandstructure.{bs_type.value}.{field}")
+
+        return [(key, False) for key in keys]
+
 
 class DOSDataQuery(QueryOperator):
     """
@@ -225,3 +241,12 @@ def query(
             crit.update({"dos.magnetic_ordering": magnetic_ordering.value})
 
         return {"criteria": crit}
+
+    def ensure_indexes(self):
+
+        keys = ["dos", "dos.magnetic_ordering"]
+
+        for proj_type in DOSProjectionType:
+            keys.append(f"dos.{proj_type.value}.$**")
+
+        return [(key, False) for key in keys]