import pendulum
from airflow.datasets import Dataset
from airflow.decorators import dag
from utils.github import github_action


@dag(
    schedule=None,
    start_date=pendulum.datetime(2021, 1, 1, tz="UTC"),
    tags=["example"],
)
def call_github_action_and_wait():
    github_action(
        group_id="dummy_deploy_aws_region",
        repo="tidbcloud/autonomous-airflow",
        workflow="deploy-aws-region.yml",
        input_fn=lambda ti: [
            {
                "tidb-cloud-instance": "dev",
                "region-name": "dummy",
                "aws-region": "dummy",
                "meta-uri": "dummy",
                "advance-options": "dummy",
            }
        ],
        outlets=[
            Dataset("mysql://test/test/test"),
        ],
    )


call_github_action_and_wait()