diff --git a/qubesadmin/features.py b/qubesadmin/features.py index 54d6437..cd33d59 100644 --- a/qubesadmin/features.py +++ b/qubesadmin/features.py @@ -67,6 +67,15 @@ class Features(object): _NO_DEFAULT = object() + def get(self, item, default=_NO_DEFAULT): + '''Get a feature, return default value if missing.''' + try: + return self[item] + except KeyError: + if default is self._NO_DEFAULT: + raise + return default + def check_with_template(self, feature, default=_NO_DEFAULT): ''' Check if the vm's template has the specified feature. ''' try: