Merge pull request #25 from tamaskozak/terraform-output-json1

Terraform output json1

Author: akuzminsky

setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.8.3
+current_version = 0.8.4
 commit = True
 tag = False
 

setup.py

@@ -25,7 +25,7 @@ def parse_requirements(req_file):
 if __name__ == "__main__":
     setup(
         name="terraform-ci",
-        version="0.8.3",
+        version="0.8.4",
         description="Terraform CI runs terraform in Travis-CI",
         long_description=dedent(
             """

terraform_ci/__init__.py

@@ -9,7 +9,8 @@
 
 import boto3
 
-__version__ = "0.8.3"
+
+__version__ = "0.8.4"
 
 DEFAULT_TERRAFORM_VARS = ".env/tf_env.json"
 LOG = logging.getLogger(__name__)
@@ -390,8 +391,22 @@ def setup_logging(logger, debug=False):  # pragma: no cover
     logger.setLevel(logging.DEBUG)
 
 
+def terraform_output(path):
+    """
+    Run terraform output and return the json results as a dict
+    :param path: Path to directory with terraform module.
+    :type path: str
+    :return: dict from terraform output
+    """
+    cmd = "terraform output -json"
+    ret, cout, cerr = execute(cmd.split(), stdout=PIPE, stderr=None, cwd=path)
+    if ret:
+        raise CalledProcessError(returncode=ret, cmd=cmd, output=cout, stderr=cerr)
+    return json.loads(cout)
+
+
 @contextmanager
-def terraform_apply(path, destroy_after=True):
+def terraform_apply(path, destroy_after=True, json_output=False):
     """
     Run terraform init and apply, then return a generator.
     If destroy_after is True, run terraform destroy afterwards.
@@ -400,8 +415,10 @@ def terraform_apply(path, destroy_after=True):
     :type path: str
     :param destroy_after: Run terraform destroy after context it returned back.
     :type destroy_after: bool
-    :return: Nothing. The function just yields to use it in the ``with``
-        block.
+    :param json_output: Yield terraform output result as a dict (available in the context)
+    :type json_output: bool
+    :return: If json_output is true then yield the result from terraform_output otherwise nothing.
+        Use it in the ``with`` block.
     :raise CalledProcessError: if either of terraform commands (except ``terraform destroy``)
         exits with non-zero.
     """
@@ -420,7 +437,10 @@ def terraform_apply(path, destroy_after=True):
                 raise CalledProcessError(
                     returncode=ret, cmd=cmd, output=cout, stderr=cerr
                 )
-        yield
+        if json_output:
+            yield terraform_output(path)
+        else:
+            yield
 
     finally:
         if destroy_after: