Restart qube when retrying shutdown

Author: ben-grande

qui/tray/domains.py

@@ -326,11 +326,7 @@ def set_force(self, force):
         else:
             self.label.set_text(_("Restart"))
 
-    async def perform_action(self, *_args, **_kwargs):
-        try:
-            await asyncio.to_thread(self.vm.shutdown, force=self.force, wait=True)
-        except exc.QubesException as ex:
-            self.show_shutdown_dialog(ex)
+    async def start(self):
         if self.give_up:
             return
         try:
@@ -344,6 +340,14 @@ async def perform_action(self, *_args, **_kwargs):
                 ).format(self.vm.name, str(ex)),
             )
 
+    async def perform_action(self, *_args, **_kwargs):
+        try:
+            await asyncio.to_thread(self.vm.shutdown, force=self.force, wait=True)
+        except exc.QubesException as ex:
+            self.show_shutdown_dialog(ex)
+        else:
+            await self.start()
+
     def react_to_question(self, widget, response, action):
         asyncio.create_task(self.react_to_question_async(widget, response, action))
 
@@ -356,7 +360,8 @@ async def react_to_question_async(self, widget, response, action):
             await self.shutdown_from_response(response, action)
         except exc.QubesException as ex:
             self.show_shutdown_dialog(ex)
-            self.give_up = True
+        else:
+            await self.start()
         widget.destroy()