From 4e6633c6fe2bac585d2d19b7f5fe70c6eb283ece Mon Sep 17 00:00:00 2001 From: Thomas Schneider Date: Fri, 9 Sep 2016 13:35:14 +0200 Subject: [PATCH] check if a component exists before accessing it - added output when an expected key is missing --- pymegacli/components.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pymegacli/components.py b/pymegacli/components.py index aa05ce0..c35b7a7 100644 --- a/pymegacli/components.py +++ b/pymegacli/components.py @@ -88,7 +88,10 @@ def __setitem__(self, key, value): self.props[key] = value def __getitem__(self, key): - return self.props[key] + if key in self.props: + return self.props[key] + else: + print key+" doesn't exist" def get(self, key, default=None): return self.props.get(key, default)