浏览代码

features: implement get() method

Marek Marczykowski-Górecki 7 年之前
父节点
当前提交
64f7eecf58
共有 1 个文件被更改,包括 9 次插入0 次删除
  1. 9 0
      qubesadmin/features.py

+ 9 - 0
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: