diff options
author | Lars Wirzenius <liw@liw.fi> | 2013-04-20 12:41:21 +0100 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2013-04-20 12:41:21 +0100 |
commit | 20b39287d7c06b96926d6ef6ef723067f76da262 (patch) | |
tree | f53b4b5188cd6948a5bcf42560d1a4853a5989a2 | |
parent | 7ab405c8cf32a49ead868b9d5aa8b05b8c5942e5 (diff) | |
download | clab-20b39287d7c06b96926d6ef6ef723067f76da262.tar.gz |
Allow lists of entries in a YAML file
-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)] |