diff --git a/ks_includes/printer.py b/ks_includes/printer.py index 143a11fc..616f8567 100644 --- a/ks_includes/printer.py +++ b/ks_includes/printer.py @@ -153,11 +153,14 @@ class Printer: return [] def get_config_section(self, section): + return self.config[section] if section in self.config else False + + def get_macro(self, macro): return next( ( self.config[key] for key in self.config.keys() - if key.find(section) > -1 + if key.find(macro) > -1 ), False, )