airflow_vertica_operator_dag_sql_execution_example.py

1import datetime
2
3from airflow import DAG
4from airflow.providers.vertica.operators.vertica import VerticaOperator
5
6with DAG(
7    dag_id="example_vertica",
8    start_date=datetime.datetime(2021, 1, 1),
9    template_searchpath="/usr/local/airflow/sql",
10    tags=["example"],
11    catchup=False,
12) as dag:
13    # [START howto_operator_vertica]
14    create_table = VerticaOperator(
15        task_id="create_table",
16        sql="""
17            CREATE TABLE IF NOT EXISTS partner (
18                partner_id INTEGER,
19                partner_name VARCHAR(100)
20            );
21        """,
22        vertica_conn_id="vertica_default",
23    )
24    # [END howto_operator_vertica]
25
26    # [START howto_operator_vertica_external_file]
27    insert_data = VerticaOperator(
28        task_id="insert_data",
29        sql="insert_data.sql",
30        vertica_conn_id="vertica_default",
31    )
32    # [END howto_operator_vertica_external_file]
33
34    create_table >> insert_data