diff --git a/emmet-api/emmet/api/routes/materials/materials/query_operators.py b/emmet-api/emmet/api/routes/materials/materials/query_operators.py index ded52e97e7..e59ca4a764 100644 --- a/emmet-api/emmet/api/routes/materials/materials/query_operators.py +++ b/emmet-api/emmet/api/routes/materials/materials/query_operators.py @@ -474,6 +474,9 @@ def query( class BatchIdQuery(QueryOperator): """Method to generate a query on batch_id""" + def __init__(self, field="builder_meta.batch_id"): + self._field = field + def query( self, batch_id: str | None = Query( @@ -502,7 +505,7 @@ def query( ) crit = {} # type: dict - k = "builder_meta.batch_id" + k = self._field if batch_id: crit[k] = batch_id elif batch_id_not_eq: diff --git a/emmet-api/emmet/api/routes/materials/tasks/resources.py b/emmet-api/emmet/api/routes/materials/tasks/resources.py index 605103ecdb..e9ae152cfe 100644 --- a/emmet-api/emmet/api/routes/materials/tasks/resources.py +++ b/emmet-api/emmet/api/routes/materials/tasks/resources.py @@ -2,6 +2,7 @@ from emmet.api.resource import ReadOnlyResource from emmet.api.routes.materials.materials.query_operators import ( + BatchIdQuery, ChemsysQuery, ElementsQuery, FormulaQuery, @@ -26,6 +27,7 @@ def task_resource(task_store): task_store, TaskDoc, query_operators=[ + BatchIdQuery(field="batch_id"), FormulaQuery(), ChemsysQuery(), ElementsQuery(),