fix: Support config() method in jinja. (#962)

Author: crericha

sqlmesh/utils/__init__.py

@@ -76,6 +76,11 @@ def __deepcopy__(self, memo: t.Dict[t.Any, AttributeDict]) -> AttributeDict:
             copy[k] = deepcopy(v, memo)
         return copy
 
+    def __call__(self, **kwargs: t.Dict[str, t.Any]) -> str:
+        self.update(**kwargs)
+        # Return an empty string, so that this method can be used within Jinja
+        return ""
+
 
 class registry_decorator:
     """A decorator that registers itself."""

tests/dbt/test_transformation.py

@@ -1,3 +1,4 @@
+import typing as t
 from pathlib import Path
 
 import agate
@@ -11,6 +12,7 @@
     IncrementalByUniqueKeyKind,
     ModelKind,
     ModelKindName,
+    SqlModel,
     ViewKind,
 )
 from sqlmesh.dbt.column import (
@@ -211,6 +213,21 @@ def test_schema_jinja(sushi_test_project: Project):
     model_config.to_sqlmesh(context).render_query().sql() == "SELECT 1 AS one FROM sushi AS sushi"
 
 
+def test_config_jinja(sushi_test_project: Project):
+    hook = "{{ config(alias='bar') }} {{ config.alias }}"
+    model_config = ModelConfig(
+        name="model",
+        package_name="package",
+        schema="sushi",
+        sql="""SELECT 1 AS one FROM foo""",
+        **{"pre-hook": hook},
+    )
+    context = sushi_test_project.context
+    model = t.cast(SqlModel, model_config.to_sqlmesh(context))
+    assert model.pre_statements[0].sql() == hook
+    assert model.render_pre_statements()[0].sql() == "bar"
+
+
 def test_this(assert_exp_eq, sushi_test_project: Project):
     model_config = ModelConfig(
         name="model",