airflow_dag_opsgenie_create_alert_operator_example.py

1import os
2from datetime import datetime
3
4from airflow import DAG
5from airflow.providers.opsgenie.operators.opsgenie import OpsgenieCreateAlertOperator
6
7ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID")
8DAG_ID = "example_opsgenie_alert"
9
10with DAG(
11    dag_id=DAG_ID,
12    schedule=None,
13    start_date=datetime(2021, 1, 1),
14    catchup=False,
15    tags=["example"],
16) as dag:
17
18    # [START howto_operator_opsgenie_create_alert]
19    create_alert = OpsgenieCreateAlertOperator(
20        task_id="create_alert_task",
21        message="An example alert from Airflow",
22        alias="airflow_example_alert",
23        description="This alert was created by the OpsgenieCreateAlertOperator in Airflow.",
24        priority="P3",
25        tags=["airflow", "test"],
26        opsgenie_conn_id="opsgenie_default",
27    )
28    # [END howto_operator_opsgenie_create_alert]
29
30    create_alert