[feature] added support to pass a complex dict (#69)

* [feature] added support to pass a complex dict to the tf_vars parameter * [feature] added some tests * [cleanup] * fix linting Co-authored-by: Nikolai Shmatenkov <[email protected]> Co-authored-by: Ludo <[email protected]>
GoogleCloudPlatform · Jan 26, 2023 · 707a604 · 707a604
1 parent 6804937
commit 707a604
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/test/test_args.py b/test/test_args.py
@@ -47,6 +47,15 @@
     ({'tf_var_file': 'foo.tfvar'}, ['-var-file=foo.tfvar']),
     ({'tf_var_file': ['foo.tfvar', 'bar.tfvar']}, [
      '-var-file=foo.tfvar', '-var-file=bar.tfvar']),
+    ({'tf_vars': {'text': 'text'}}, ['-var', 'text=text']),
+    ({'tf_vars': {'number': 0}}, ['-var', 'number=0']),
+    ({'tf_vars': {'bool': False}}, ['-var', 'bool=False']),
+    ({'tf_vars': {'dict': {'text': 'text'}}},
+     ['-var', 'dict={"text": "text"}']),
+    ({'tf_vars': {'list': ['item1', 'item2']}},
+     ['-var', 'list=["item1", "item2"]']),
+    ({'tf_vars': {'dict': {'list': ['item1', 'item2']}}},
+     ['-var', 'dict={"list": ["item1", "item2"]}']),
 )
 
 

diff --git a/tftest.py b/tftest.py
@@ -146,7 +146,10 @@ def parse_args(init_vars=None, tf_vars=None, targets=None, **kw):
   if tf_vars:
     cmd_args += list(
         itertools.chain.from_iterable(
-            ("-var", "{}={}".format(k, v)) for k, v in tf_vars.items()))
+            ("-var",
+             "{}={}".format(k, json.dumps(v) if isinstance(v, (dict, list)) else v))
+            for k, v in tf_vars.items()
+        ))
   if targets:
     cmd_args += [("-target={}".format(t)) for t in targets]
   if kw.get('tf_var_file'):