Skip to content

Pre-populate a suggested partition key when manually triggering cron-partitioned Dags - #71392

Draft
Lee-W wants to merge 1 commit into
apache:mainfrom
astronomer:suggest-partition-key-on-trigger
Draft

Pre-populate a suggested partition key when manually triggering cron-partitioned Dags#71392
Lee-W wants to merge 1 commit into
apache:mainfrom
astronomer:suggest-partition-key-on-trigger

Conversation

@Lee-W

@Lee-W Lee-W commented Aug 11, 2026

Copy link
Copy Markdown
Member

Manually triggering a Dag whose timetable is CronPartitionTimetable leaves the Partition Key field blank. If the user submits without typing one, the Dag run is created unpartitioned and any downstream asset-partition-aware Dag can never be satisfied by it.

Add a new Timetable.suggest_partition_key(now) hook, overridden by CronPartitionTimetable to suggest the most recently elapsed partition tick. Thread the suggestion through a new suggested_partition_key field on DAGResponse to pre-populate the trigger form's Partition Key field. The field stays editable and still submits null if the user clears it.


Was generative AI tooling used to co-author this PR?
  • Yes (please specify the tool below)

  • Read the Pull Request Guidelines for more information. Note: commit author/co-author name and email in commits become permanently public when merged.
  • For fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
  • When adding dependency, check compliance with the ASF 3rd Party License Policy.
  • For significant user-facing changes create newsfragment: {pr_number}.significant.rst, in airflow-core/newsfragments. You can add this file in a follow-up commit after the PR is created so you know the PR number.

@boring-cyborg boring-cyborg Bot added area:airflow-ctl area:API Airflow's REST/HTTP API area:UI Related to UI/UX. For Frontend Developers. backport-to-airflow-ctl/v0-1-test labels Aug 11, 2026
@Lee-W
Lee-W force-pushed the suggest-partition-key-on-trigger branch 4 times, most recently from c74c887 to 589276a Compare August 12, 2026 11:18
…partitioned Dags

Manually triggering a Dag whose timetable is CronPartitionTimetable leaves
the Partition Key field blank. If the user submits without typing one, the
Dag run is created unpartitioned and any downstream asset-partition-aware
Dag can never be satisfied by it.

Add a new Timetable.suggest_partition_key(now) hook, overridden by
CronPartitionTimetable to suggest the most recently elapsed partition tick.
Thread the suggestion through a new suggested_partition_key field on
DAGResponse to pre-populate the trigger form's Partition Key field. The
field stays editable and still submits null if the user clears it.

The Dags list view builds its payload by reading every DAGResponse field off
a bare DagModel, so a field that only the single-Dag endpoints supply has to
be declared as such or that endpoint fails with AttributeError. The trigger
form applies the suggestion again whenever a newer one arrives, because a
response served from cache can name a partition that has since elapsed.
@Lee-W
Lee-W force-pushed the suggest-partition-key-on-trigger branch from 589276a to eed1aa6 Compare August 12, 2026 13:52
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:airflow-ctl area:API Airflow's REST/HTTP API area:UI Related to UI/UX. For Frontend Developers. backport-to-airflow-ctl/v0-1-test

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant