fix: Set template fields for each operator

DAGs do not render properly when template fields do not exist. Since we were defining `template_fields` at the base class, these may not be present in all subclasses. We moved `template_fields` to each subclass to fix this.

Author: tomasfarias

airflow_dbt_python/operators/dbt.py

@@ -53,9 +53,6 @@ class DbtBaseOperator(BaseOperator):
         "profiles_dir",
         "profile",
         "target",
-        "select",
-        "models",
-        "exclude",
         "state",
     ]
 
@@ -315,6 +312,12 @@ class DbtRunOperator(DbtBaseOperator):
     https://docs.getdbt.com/reference/commands/run.
     """
 
+    template_fields = DbtBaseOperator.template_fields + [
+        "select",
+        "models",
+        "exclude",
+    ]
+
     def __init__(
         self,
         full_refresh: Optional[bool] = None,
@@ -344,6 +347,11 @@ class DbtSeedOperator(DbtBaseOperator):
     https://docs.getdbt.com/reference/commands/seed.
     """
 
+    template_fields = DbtBaseOperator.template_fields + [
+        "select",
+        "exclude",
+    ]
+
     def __init__(
         self,
         full_refresh: Optional[bool] = None,
@@ -374,6 +382,12 @@ class DbtTestOperator(DbtBaseOperator):
     https://docs.getdbt.com/reference/commands/test.
     """
 
+    template_fields = DbtBaseOperator.template_fields + [
+        "select",
+        "models",
+        "exclude",
+    ]
+
     def __init__(
         self,
         data: Optional[bool] = None,
@@ -405,6 +419,12 @@ class DbtCompileOperator(DbtBaseOperator):
     https://docs.getdbt.com/reference/commands/compile.
     """
 
+    template_fields = DbtBaseOperator.template_fields + [
+        "select",
+        "models",
+        "exclude",
+    ]
+
     def __init__(
         self,
         parse_only: Optional[bool] = None,
@@ -490,6 +510,11 @@ class DbtSnapshotOperator(DbtBaseOperator):
     https://docs.getdbt.com/reference/commands/snapshot.
     """
 
+    template_fields = DbtBaseOperator.template_fields + [
+        "select",
+        "exclude",
+    ]
+
     def __init__(
         self,
         select: Optional[list[str]] = None,
@@ -515,6 +540,12 @@ class DbtLsOperator(DbtBaseOperator):
     https://docs.getdbt.com/reference/commands/list.
     """
 
+    template_fields = DbtBaseOperator.template_fields + [
+        "select",
+        "exclude",
+        "resource_types",
+    ]
+
     def __init__(
         self,
         resource_types: Optional[list[str]] = None,
@@ -558,6 +589,11 @@ class DbtRunOperationOperator(DbtBaseOperator):
     https://docs.getdbt.com/reference/commands/run-operation.
     """
 
+    template_fields = DbtBaseOperator.template_fields + [
+        "macro",
+        "args",
+    ]
+
     def __init__(
         self,
         macro: str,
@@ -628,6 +664,11 @@ class DbtBuildOperator(DbtBaseOperator):
     https://docs.getdbt.com/reference/commands/build.
     """
 
+    template_fields = DbtBaseOperator.template_fields + [
+        "select",
+        "exclude",
+    ]
+
     def __init__(
         self,
         full_refresh: Optional[bool] = None,

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "airflow-dbt-python"
-version = "0.9.1"
+version = "0.9.2"
 description = "A dbt operator for Airflow that uses the dbt Python package"
 authors = ["Tomás Farías Santana <tomas@tomasfarias.dev>"]
 license = "MIT"