Skip to content

add BatchIDQuery to tasks_resource #1330

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
tsmathis merged 2 commits into from
Oct 24, 2025
Merged

add BatchIDQuery to tasks_resource #1330

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
1f925aa
add BatchIDQuery to tasks_resource
tsmathis Oct 23, 2025
5fad695
add constructor for setting BatchIdQuery target field
tsmathis Oct 23, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion emmet-api/emmet/api/routes/materials/materials/query_operators.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down Expand Up @@ -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:
Expand Down
2 changes: 2 additions & 0 deletions emmet-api/emmet/api/routes/materials/tasks/resources.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from emmet.api.resource import ReadOnlyResource

from emmet.api.routes.materials.materials.query_operators import (
BatchIdQuery,
ChemsysQuery,
ElementsQuery,
FormulaQuery,
Expand All @@ -26,6 +27,7 @@ def task_resource(task_store):
task_store,
TaskDoc,
query_operators=[
BatchIdQuery(field="batch_id"),
FormulaQuery(),
ChemsysQuery(),
ElementsQuery(),
Expand Down
Loading