airflow_dag_powerbi_dataset_refresh_operator_with_wait.py

1from datetime import datetime
2from airflow import DAG
3from airflow_powerbi_plugin.operators.powerbi import PowerBIDatasetRefreshOperator
4
5with DAG(
6    dag_id='example_powerbi_refresh',
7    schedule_interval=None,
8    start_date=datetime(2023, 1, 1),
9    catchup=False,
10) as dag:
11
12    refresh_dataset = PowerBIDatasetRefreshOperator(
13        task_id='refresh_dataset',
14        dataset_id='your_dataset_id_here',
15        workspace_id='your_workspace_id_here',
16        powerbi_conn_id='powerbi_default',
17        wait_for_termination=True,
18        timeout=3600,
19        check_interval=60
20    )
21
22    refresh_dataset