airflow_mysql_operator_dag_create_table_insert_data.py

1import datetime
2
3from airflow import DAG
4from airflow.providers.mysql.operators.mysql import MySqlOperator
5
6with DAG(
7    dag_id="example_mysql_operator",
8    start_date=datetime.datetime(2023, 1, 1),
9    schedule="@daily",
10    catchup=False,
11) as dag:
12
13    # [START howto_operator_mysql]
14    create_table_mysql_task = MySqlOperator(
15        task_id="create_table_mysql",
16        mysql_conn_id="mysql_default",
17        sql="""
18            CREATE TABLE IF NOT EXISTS employees (
19                id INT AUTO_INCREMENT PRIMARY KEY,
20                name VARCHAR(50),
21                dept VARCHAR(50)
22            );
23        """,
24    )
25
26    insert_data_mysql_task = MySqlOperator(
27        task_id="insert_data_mysql",
28        mysql_conn_id="mysql_default",
29        sql="""
30            INSERT INTO employees (name, dept) 
31            VALUES ('John Doe', 'Engineering'), ('Jane Smith', 'Sales');
32        """,
33    )
34
35    create_table_mysql_task >> insert_data_mysql_task
36    # [END howto_operator_mysql]