From fd9e8e8326fd3dc98ac3bf5e6e058f0a14737649 Mon Sep 17 00:00:00 2001 From: Kirk Byers Date: Mon, 22 Jun 2020 02:56:54 -0700 Subject: [PATCH] Allow empty defaults.yaml and groups.yaml (#7) --- nornir_utils/plugins/inventory/yaml_inventory.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nornir_utils/plugins/inventory/yaml_inventory.py b/nornir_utils/plugins/inventory/yaml_inventory.py index 381d97b..cc3be82 100644 --- a/nornir_utils/plugins/inventory/yaml_inventory.py +++ b/nornir_utils/plugins/inventory/yaml_inventory.py @@ -93,7 +93,7 @@ def load(self) -> Inventory: if self.defaults_file.exists(): with open(self.defaults_file, "r") as f: - defaults_dict = yml.load(f) + defaults_dict = yml.load(f) or {} defaults = _get_defaults(defaults_dict) else: defaults = Defaults() @@ -108,7 +108,7 @@ def load(self) -> Inventory: groups = Groups() if self.group_file.exists(): with open(self.group_file, "r") as f: - groups_dict = yml.load(f) + groups_dict = yml.load(f) or {} for n, g in groups_dict.items(): groups[n] = _get_inventory_element(Group, g, n, defaults)