Merge pull request #1148 from fishtown-analytics/feature/jinja-do-sta…

…tement Jinja expression statements (#1113)
dbt-labs · Nov 23, 2018 · f72e0a8 · f72e0a8
2 parents 029ef17 + 069bc3a
commit f72e0a8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/dbt/clients/jinja.py b/dbt/clients/jinja.py
@@ -246,7 +246,7 @@ def __call__(self, *args, **kwargs):
 
 def get_environment(node=None, capture_macros=False):
  args = {
- 'extensions': []
+ 'extensions': ['jinja2.ext.do']
  }
 
  if capture_macros:

diff --git a/test/unit/test_jinja.py b/test/unit/test_jinja.py
@@ -0,0 +1,11 @@
+import unittest
+
+from dbt.clients.jinja import get_template
+
+class TestJinja(unittest.TestCase):
+ def test_do(self):
+ s = '{% set my_dict = {} %}\n{% do my_dict.update(a=1) %}'
+
+ template = get_template(s, {})
+ mod = template.make_module()
+ self.assertEqual(mod.my_dict, {'a': 1})