Преглед изворни кода

Allow app.domains[vm] where vm is a QubesVM object

Similar as in the core-admin API.
Marek Marczykowski-Górecki пре 4 година
родитељ
комит
b8ddb39752
1 измењених фајлова са 4 додато и 0 уклоњено
  1. 4 0
      qubesadmin/app.py

+ 4 - 0
qubesadmin/app.py

@@ -83,6 +83,8 @@ class VMCollection(object):
                 del self._vm_objects[name]
 
     def __getitem__(self, item):
+        if isinstance(item, qubesadmin.vm.QubesVM):
+            item = item.name
         if not self.app.blind_mode and item not in self:
             raise KeyError(item)
         return self.get_blind(item)
@@ -104,6 +106,8 @@ class VMCollection(object):
         return self._vm_objects[item]
 
     def __contains__(self, item):
+        if isinstance(item, qubesadmin.vm.QubesVM):
+            item = item.name
         self.refresh_cache()
         return item in self._vm_list