when there is macros in the sql, make sure it can be replaced with va…

…lue (#4)
wherobots · May 28, 2024 · d512652 · d512652
1 parent 02cf257
commit d512652
Show file tree

Hide file tree

Showing 3 changed files with 96 additions and 86 deletions.
diff --git a/airflow_providers_wherobots/operators/sql.py b/airflow_providers_wherobots/operators/sql.py
@@ -4,6 +4,8 @@
 
 from __future__ import annotations
 
+from typing import Sequence
+
 from airflow.providers.common.sql.hooks.sql import DbApiHook
 from airflow.providers.common.sql.operators.sql import SQLExecuteQueryOperator
 from wherobots.db import Runtime
@@ -30,6 +32,10 @@ def wherobots_default_handler(cursor: WDbCursor) -> DataFrame | None:
 
 class WherobotsSqlOperator(SQLExecuteQueryOperator):  # type: ignore[misc]
 
+    template_fields: Sequence[str] = SQLExecuteQueryOperator.template_fields
+    template_fields_renderers = {"sql": "sql"}
+    conn_id_field = "wherobots_conn_id"
+
     def __init__(  # type: ignore[no-untyped-def]
         self,
         *,