import pathlib import logging from .stackgroup import StackGroup from .jinja import read_config_file from .exceptions import InvalidProjectDir logger = logging.getLogger(__name__) class CloudBender(object): """Config Class to handle recursive conf/* config tree""" def __init__(self, root_path): self.root = pathlib.Path(root_path) self.sg = None self.all_stacks = [] self.ctx = { "config_path": self.root.joinpath("config"), "template_path": self.root.joinpath("cloudformation"), "hooks_path": self.root.joinpath("hooks"), "docs_path": self.root.joinpath("docs"), "outputs_path": self.root.joinpath("outputs"), "artifact_paths": [self.root.joinpath("artifacts")], } if not self.ctx["config_path"].is_dir(): raise InvalidProjectDir( "Check '{0}' exists and is a valid CloudBender project folder.".format( self.ctx["config_path"] ) ) def read_config(self): """Load the /config.yaml, /*.yaml as stacks, sub-folders are sub-groups""" # Read top level config.yaml and extract CloudBender CTX _config = read_config_file(self.ctx["config_path"].joinpath("config.yaml")) # Legacy naming if _config and _config.get("CloudBender"): self.ctx.update(_config.get("CloudBender")) if _config and _config.get("cloudbender"): self.ctx.update(_config.get("cloudbender")) # Make sure all paths are abs for k, v in self.ctx.items(): if k in [ "config_path", "template_path", "hooks_path", "docs_path", "artifact_paths", "outputs_path", ]: if isinstance(v, list): new_list = [] for p in v: path = pathlib.Path(p) if not path.is_absolute(): new_list.append(self.root.joinpath(path)) else: new_list.append(path) self.ctx[k] = new_list elif isinstance(v, str): if not v.is_absolute(): self.ctx[k] = self.root.joinpath(v) self.sg = StackGroup(self.ctx["config_path"], self.ctx) self.sg.read_config() self.all_stacks = self.sg.get_stacks() def dump_config(self): logger.debug("".format(vars(self))) self.sg.dump_config() def clean(self): for s in self.all_stacks: s.delete_template_file() def resolve_stacks(self, token): stacks = [] # remove optional leading "config/" to allow bash path expansions if token.startswith("config/"): token = token[7:] # If path ends with yaml we look for stacks if token.endswith(".yaml"): stacks = self.sg.get_stacks(token, match_by="path") # otherwise assume we look for a group, if we find a group return all stacks below else: # Strip potential trailing slash token = token.rstrip("/") sg = self.sg.get_stackgroup(token, match_by="path") if sg: stacks = sg.get_stacks() return stacks def filter_stacks(self, filter_by, stacks=None): # filter_by is a dict { property, value } # if no group of stacks provided, look in all available if not stacks: stacks = self.all_stacks matching_stacks = [] for s in stacks: match = True for p, v in filter_by.items(): if not (hasattr(s, p) and getattr(s, p) == v): match = False break if match: matching_stacks.append(s) return matching_stacks