From b8ddb39752b2bb34315044666d59661cbbc0c4c2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marek=20Marczykowski-G=C3=B3recki?=
 <marmarek@invisiblethingslab.com>
Date: Tue, 25 Jun 2019 06:28:00 +0200
Subject: [PATCH] Allow app.domains[vm] where vm is a QubesVM object

Similar as in the core-admin API.
---
 qubesadmin/app.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/qubesadmin/app.py b/qubesadmin/app.py
index e397997..de16cd4 100644
--- a/qubesadmin/app.py
+++ b/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