summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLars Wirzenius <liw@liw.fi>2013-04-20 12:41:21 +0100
committerLars Wirzenius <liw@liw.fi>2013-04-20 12:41:21 +0100
commit20b39287d7c06b96926d6ef6ef723067f76da262 (patch)
treef53b4b5188cd6948a5bcf42560d1a4853a5989a2
parent7ab405c8cf32a49ead868b9d5aa8b05b8c5942e5 (diff)
downloadclab-20b39287d7c06b96926d6ef6ef723067f76da262.tar.gz
Allow lists of entries in a YAML file
-rwxr-xr-xclab9
1 files changed, 7 insertions, 2 deletions
diff --git a/clab b/clab
index 9b93ab4..daa2dbc 100755
--- a/clab
+++ b/clab
@@ -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)]