Explorar o código

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

Similar as in the core-admin API.
Marek Marczykowski-Górecki %!s(int64=5) %!d(string=hai) anos
pai
achega
b8ddb39752
Modificáronse 1 ficheiros con 4 adicións e 0 borrados
  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