From 82721f886537e392a145f9df9f47f90cf26c5d27 Mon Sep 17 00:00:00 2001
From: Zane Bitter <zbitter@redhat.com>
Date: Wed, 19 Jul 2017 17:35:39 -0400
Subject: [PATCH] Add Output definitions to StackDefinition

Change-Id: I8356980d8687b2f1ffb4abe198adec8e70f00f69
Partially-Implements: blueprint stack-definition
---
 heat/engine/stk_defn.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/heat/engine/stk_defn.py b/heat/engine/stk_defn.py
index c07039bdb..17aadd970 100644
--- a/heat/engine/stk_defn.py
+++ b/heat/engine/stk_defn.py
@@ -42,6 +42,7 @@ class StackDefinition(object):
                                               template.env.param_defaults)
         self._resource_defns = None
         self._resources = {}
+        self._output_defns = None
 
     def clone_with_new_template(self, new_template, stack_identifier,
                                 clear_resource_data=False):
@@ -75,6 +76,21 @@ class StackDefinition(object):
             self._load_rsrc_defns()
         return set(self._resource_defns)
 
+    def _load_output_defns(self):
+        self._output_defns = self._template.outputs(self)
+
+    def output_definition(self, output_name):
+        """Return the definition of the given output."""
+        if self._output_defns is None:
+            self._load_output_defns()
+        return self._output_defns[output_name]
+
+    def enabled_output_names(self):
+        """Return the set of names of all enabled outputs in the template."""
+        if self._output_defns is None:
+            self._load_output_defns()
+        return set(self._output_defns)
+
     def all_rsrc_names(self):
         """Return the set of names of all resources in the template.