diff options
-rwxr-xr-x | clab | 9 |
1 files changed, 7 insertions, 2 deletions
@@ -69,8 +69,13 @@ class AddressBook(object): logging.info('Adding from file %s' % filename) with open(filename) as f: parsed_yaml = yaml.safe_load(f) - entry = Entry(parsed_yaml) - self.entries.append(entry) + if type(parsed_yaml) is list: + for parsed_entry in parsed_yaml: + entry = Entry(parsed_entry) + self.entries.append(entry) + else: + entry = Entry(parsed_yaml) + self.entries.append(entry) def find(self, patterns): return [e for e in self.entries if self.matches(e, patterns)] |