import pendulum
from airflow.datasets import Dataset
from airflow.decorators import dag
from airflow.models import Variable
from utils.jenkins import jenkins_job


@dag(
    schedule=None,
    start_date=pendulum.datetime(2021, 1, 1, tz="UTC"),
    tags=["example"],
)
def call_jenkins_job_and_wait():
    token = Variable.get("jenkins_token_zhoushuai")
    jenkins_job(
        group_id="tiqa_invite_case",
        jenkins_url="http://cicd.tidbcloud.com",
        username="zhoushuai@pingcap.com",
        password=token,
        job_name="platform-scene-online",
        input_fn=lambda ti: [
            {
                "URL": "https://dev.tidbcloud.com/",
                "REQUESTURL": "https://us-west-2.staging.shared.aws.tidbcloud.com",
                "ORGID": "60205",
                "PROJECTID": "3199728",
                "TENANTEMAIL": "zhoushuai@pingcap.com",
                "REDIRECTURL": "https://staging.tidbcloud.com/auth_redirect",
                "RUNUSER": "zhoushuai",
                "SCOPE": "Invite",
            }
        ],
        outlets=[
            Dataset("mysql://test/test/test"),
        ],
    )


call_jenkins_job_and_wait()