Skip to content
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.

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

Add MwaaTriggerDagRunOperator and MwaaHook to Amazon Provider Package #46579

Open
wants to merge 11 commits into
base: main
Choose a base branch
from

Conversation

ramitkataria
Copy link

This PR adds AWS Managed Workflows for Apache Airflow (MWAA) integration to the Amazon Provider Package. The integration includes:

  1. MwaaTriggerDagRunOperator: Allows triggering DAGs in an MWAA environment
  2. MwaaHook: A generic hook to interact with MWAA environments via REST API

To run the example_mwaa system test, there needs to be an existing environment already created. In the future, when we add a CreateEnvironmentOperator, there will be a separate system test just for that operator. This approach was chosen due to the significant time required for MWAA environment creation.


^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in newsfragments.

Copy link

boring-cyborg bot commented Feb 8, 2025

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our pre-commits will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@ramitkataria ramitkataria marked this pull request as ready for review February 8, 2025 02:40
@ramitkataria ramitkataria changed the title Ramitkataria/triggerdag op Add MwaaTriggerDagRunOperator and MwaaHook Feb 8, 2025
@ramitkataria ramitkataria changed the title Add MwaaTriggerDagRunOperator and MwaaHook Add MwaaTriggerDagRunOperator and MwaaHook to Amazon Provider Package Feb 8, 2025
Comment on lines +65 to +69
trigger_dag_run = MwaaTriggerDagRunOperator(
task_id="trigger_dag_run",
env_name=env_name,
trigger_dag_id=trigger_dag_id,
)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is the use case here that you running the dag in environment A yet want to trigger another DAG in environment B?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants