From 51f11b50e4f04c706452f73857b90fa3b538a7d9 Mon Sep 17 00:00:00 2001 From: Luis Henrique Date: Thu, 30 Nov 2023 14:38:32 -0300 Subject: [PATCH] refactor: update functions and gateways property --- src/cloudformation.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/src/cloudformation.py b/src/cloudformation.py index c9046651..6f25eadd 100644 --- a/src/cloudformation.py +++ b/src/cloudformation.py @@ -63,22 +63,20 @@ def construct_getatt(node: yaml.nodes.Node) -> List[Any]: class Template: def __init__(self, template: Optional[str] = None) -> None: self._template = self.load(template) - self._functions = None - self._gateways = None @property def template(self): return self._template @property - def functions(self): - if not self._functions: + def functions(self) -> List[Dict[str, Any]]: + if not hasattr(self, "_functions"): self._functions = self.find_nodes(self._template["Resources"], NodeType.LAMBDA) return self._functions @property - def gateways(self): - if not self._gateways: + def gateways(self) -> List[Dict[str, Any]]: + if not hasattr(self, "_gateways"): self._gateways = self.find_nodes(self._template["Resources"], NodeType.API_GATEWAY) return self._gateways