airflow_mssql_operator_dag_create_table_insert_data.py

1from __future__ import annotations
2
3import datetime
4
5from airflow import DAG
6from airflow.providers.microsoft.mssql.operators.mssql import MsSqlOperator
7
8with DAG(
9    dag_id="example_mssql",
10    schedule=None,
11    start_date=datetime.datetime(2021, 1, 1),
12    catchup=False,
13    tags=["example"],
14) as dag:
15
16    # [START howto_operator_mssql]
17    create_table_mssql_task = MsSqlOperator(
18        task_id="create_mssql_table",
19        mssql_conn_id="mssql_default",
20        sql=r"""
21            CREATE TABLE dbo.Employees (
22                EmployeeID INT PRIMARY KEY,
23                FirstName NVARCHAR(50),
24                LastName NVARCHAR(50),
25                Department NVARCHAR(50)
26            );
27        """,
28    )
29
30    insert_data_mssql_task = MsSqlOperator(
31        task_id="insert_mssql_data",
32        mssql_conn_id="mssql_default",
33        sql=r"""
34            INSERT INTO dbo.Employees (EmployeeID, FirstName, LastName, Department)
35            VALUES (1, 'John', 'Doe', 'IT');
36            INSERT INTO dbo.Employees (EmployeeID, FirstName, LastName, Department)
37            VALUES (2, 'Jane', 'Smith', 'HR');
38        """,
39    )
40    # [END howto_operator_mssql]
41
42    create_table_mssql_task >> insert_data_mssql_task