ramitkataria commented on code in PR #53201:
URL: https://github.com/apache/airflow/pull/53201#discussion_r2214576941
##########
airflow-core/src/airflow/models/trigger.py:
##########
@@ -192,6 +194,15 @@ def fetch_trigger_ids_with_asset(cls, session: Session =
NEW_SESSION) -> set[str
query = select(asset_trigger_association_table.columns.trigger_id)
return {trigger_id for trigger_id in session.scalars(query)}
+ @classmethod
+ @provide_session
+ def fetch_trigger_ids_with_deadline_callback(cls, session: Session =
NEW_SESSION) -> set[str]:
+ """Fetch all the trigger IDs associated with a Deadline Alert."""
+ from airflow.models.deadline import Deadline
+
+ query =
select(Deadline.trigger_id).where(Deadline.trigger_id.is_not(None))
+ return {trigger_id for trigger_id in session.scalars(query)}
Review Comment:
Makes sense, I've made it more general purpose now so anytime there's a new
special case similar to this, only a function in the trigger model class would
need to be modified
--
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]
