airflow_salesforce_operator_soql_query_example_dag.py

1import os
2from datetime import datetime
3
4from airflow import DAG
5from airflow.providers.salesforce.operators.salesforce import SalesforceOperator
6
7ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID")
8DAG_ID = "example_salesforce_operator"
9
10with DAG(
11    dag_id=DAG_ID,
12    start_date=datetime(2021, 1, 1),
13    catchup=False,
14    tags=["example"],
15) as dag:
16    # [START howto_operator_salesforce]
17    sql_query = "SELECT Id, Name FROM Account LIMIT 5"
18    salesforce_task = SalesforceOperator(
19        task_id="run_salesforce_query",
20        sql=sql_query,
21        salesforce_conn_id="salesforce_default",
22    )
23    # [END howto_operator_salesforce]
24
25    salesforce_task