gql

dagster-io · Aug 22, 2024 · d14e094 · d14e094
1 parent c87e7f2
commit d14e094
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 11 deletions.
diff --git a/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql b/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql
diff --git a/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts b/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts
diff --git a/python_modules/dagster-graphql/dagster_graphql/schema/inputs.py b/python_modules/dagster-graphql/dagster_graphql/schema/inputs.py
@@ -379,7 +379,9 @@ class Meta:
 
 
 class GrapheneBulkActionsFilter(graphene.InputObjectType):
-    status = graphene.InputField("dagster_graphql.schema.backfill.GrapheneBulkActionStatus")
+    statuses = graphene.List(
+        graphene.NonNull("dagster_graphql.schema.backfill.GrapheneBulkActionStatus")
+    )
     createdBefore = graphene.InputField(graphene.Float)
     createdAfter = graphene.InputField(graphene.Float)
 
@@ -388,12 +390,14 @@ class Meta:
         name = "BulkActionsFilter"
 
     def to_selector(self):
-        status = BulkActionStatus[self.status.value] if self.status else None
+        statuses = (
+            [BulkActionStatus[status.value] for status in self.statuses] if self.statuses else None
+        )
         created_before = datetime_from_timestamp(self.createdBefore) if self.createdBefore else None
         created_after = datetime_from_timestamp(self.createdAfter) if self.createdAfter else None
 
         return BulkActionsFilter(
-            status=status,
+            statuses=statuses,
             created_before=created_before,
             created_after=created_after,
         )

diff --git a/python_modules/dagster-graphql/dagster_graphql_tests/graphql/test_asset_backfill.py b/python_modules/dagster-graphql/dagster_graphql_tests/graphql/test_asset_backfill.py
@@ -1203,5 +1203,5 @@ def test_get_backfills_with_filters():
             get_backfills_result = execute_dagster_graphql(
                 context,
                 BACKFILLS_WITH_FILTERS_QUERY,
-                variables={"filters": {"status": "REQUESTED"}},
+                variables={"filters": {"statuses": ["REQUESTED"]}},
             )
diff --git a/python_modules/dagster/dagster/_core/storage/runs/sql_run_storage.py b/python_modules/dagster/dagster/_core/storage/runs/sql_run_storage.py
@@ -846,7 +846,7 @@ def get_backfills(
                 raise DagsterInvariantViolationError(
                     "Conflicting status filters provided to get_backfills. Choose one of status or BulkActionsFilter.statuses."
                 )
-            statuses = [status] or (filters.statuses if filters else None)
+            statuses = [status] if status else (filters.statuses if filters else None)
             assert statuses
             query = query.where(
                 BulkActionsTable.c.status.in_([status.value for status in statuses])