From ac4d089417f1b38c97a00d4a96604ca7f12c7ad6 Mon Sep 17 00:00:00 2001 From: Thomas Wilson Date: Tue, 31 Dec 2019 23:07:41 +0800 Subject: [PATCH] Add check for strings. Closes #12 --- configspec/manager.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/configspec/manager.py b/configspec/manager.py index 6214f75..6420b23 100644 --- a/configspec/manager.py +++ b/configspec/manager.py @@ -179,7 +179,10 @@ class ConfigManager(): if isinstance(bundle_names, dict): for name, conf_spec in bundle_names.items(): config_values[name] = self.validate_bundle(name, conf_spec) - config_values[name] = self.get_config_bundle(name, conf_spec) + elif isinstance(bundle_names, str): + # Protection against a single name being passed anyway and otherwise + # being parsed as letters + config_values[bundle_names] = self.validate_bundle(bundle_names) else: for name in bundle_names: config_values[name] = self.validate_bundle(name) @@ -219,7 +222,12 @@ class ConfigManager(): """ config_values = {} - for name in bundle_names: + if isinstance(bundle_names, str): + # Protection against a single name being passed anyway and otherwise + # being parsed as letters + config_values[bundle_names] = self.get_config_bundle(bundle_names) + else: + for name in bundle_names: config_values[name] = self.get_config_bundle(name) return config_values