amoghrajesh commented on code in PR #43783:
URL: https://github.com/apache/airflow/pull/43783#discussion_r1833958404
##########
airflow/api_fastapi/common/parameters.py:
##########
@@ -406,6 +407,37 @@ def depends(self, dag_id: str | None = None) ->
_DagIdFilter:
return self.set_value(dag_id)
+class _UriPatternSearch(_SearchParam):
+ """Search on uri."""
+
+ def __init__(self, skip_none: bool = True) -> None:
+ super().__init__(AssetModel.uri, skip_none)
+
+ def depends(self, uri_pattern: str | None = None) -> _UriPatternSearch:
+ return self.set_value(uri_pattern)
+
+
+class _DagIdAssetReferencePatternSearch(_SearchParam):
+ """Search on dag_id."""
+
+ def __init__(self, skip_none: bool = True) -> None:
+ super().__init__(AssetModel.consuming_dags, skip_none)
+ self.task_attribute = AssetModel.producing_tasks
+
+ def depends(self, dag_ids: str | None = None) ->
_DagIdAssetReferencePatternSearch:
+ return self.set_value(dag_ids)
+
+ def to_orm(self, select: Select) -> Select:
+ if self.value is None and self.skip_none:
+ return select
+ if self.value is not None:
+ dags_list = self.value.split(",")
+ return select.filter(
Review Comment:
I got this portion from the legacy API:
```
query = query.filter(
(AssetModel.consuming_dags.any(DagScheduleAssetReference.dag_id.in_(dags_list)))
|
(AssetModel.producing_tasks.any(TaskOutletAssetReference.dag_id.in_(dags_list)))
)
```
Do i have to make this change @ephraimbuddy ?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
