phonon: method and other queries

Author: tschaume

emmet-api/emmet/api/routes/materials/phonon/query_operators.py

@@ -1,3 +1,4 @@
+from fastapi import Query
 from emmet.core.mpid import MPID
 
 from maggma.api.query_operator import QueryOperator
@@ -19,3 +20,22 @@ def query(
         ),
     ) -> STORE_PARAMS:
         return {"criteria": {"task_id": str(task_id)}}
+
+
+class PhononMethodQuery(QueryOperator):
+    """
+    Method to query phonon method
+    """
+
+    def query(
+        self,
+        phonon_method: str = Query(
+            None,
+            description="Phonon Method to search for",
+        ),
+    ) -> STORE_PARAMS:
+        crit = {}
+        if phonon_method in {"dfpt", "phonopy", "pheasy"}:
+            crit = {"phonon_method": phonon_method}
+
+        return {"criteria": crit}

emmet-api/emmet/api/routes/materials/phonon/resources.py

@@ -1,8 +1,15 @@
-from maggma.api.query_operator import PaginationQuery, SparseFieldsQuery
+from maggma.api.query_operator import PaginationQuery, SparseFieldsQuery, NumericQuery
 from maggma.api.resource import ReadOnlyResource
 
 from emmet.core.phonon import PhononBSDOSDoc
-from emmet.api.routes.materials.materials.query_operators import MultiMaterialIDQuery
+from emmet.api.routes.materials.materials.query_operators import (
+    MultiMaterialIDQuery,
+    ElementsQuery,
+    FormulaQuery,
+    ChemsysQuery,
+    SymmetryQuery,
+)
+from emmet.api.routes.materials.phonon.query_operators import PhononMethodQuery
 from emmet.api.core.global_header import GlobalHeaderProcessor
 from emmet.api.core.settings import MAPISettings
 
@@ -13,6 +20,12 @@ def phonon_bsdos_resource(phonon_bs_store):
         PhononBSDOSDoc,
         query_operators=[
             MultiMaterialIDQuery(),
+            FormulaQuery(),
+            ChemsysQuery(),
+            ElementsQuery(),
+            SymmetryQuery(),
+            PhononMethodQuery(),
+            NumericQuery(model=PhononBSDOSDoc, excluded_fields=["composition"]),
             PaginationQuery(),
             SparseFieldsQuery(
                 PhononBSDOSDoc, default_fields=["material_id", "last_updated"]