Format qvm-shutdown with black

Author: ben-grande

qubesadmin/tools/qvm_shutdown.py

@@ -20,7 +20,7 @@
 # You should have received a copy of the GNU Lesser General Public License along
 # with this program; if not, see <http://www.gnu.org/licenses/>.
 
-''' Shutdown a qube '''
+"""Shutdown a qube"""
 
 from __future__ import print_function
 
@@ -33,38 +33,56 @@
 import qubesadmin.exc
 
 parser = qubesadmin.tools.QubesArgumentParser(
-    description=__doc__, vmname_nargs='+')
+    description=__doc__, vmname_nargs="+"
+)
 
-parser.add_argument('--wait',
-    action='store_true', default=False,
-    help='wait for the VMs to shut down')
+parser.add_argument(
+    "--wait",
+    action="store_true",
+    default=False,
+    help="wait for the VMs to shut down",
+)
 
-parser.add_argument('--timeout',
-    action='store', type=float,
+parser.add_argument(
+    "--timeout",
+    action="store",
+    type=float,
     default=60,
-    help='timeout after which domains are killed when using --wait'
-        ' (default: %(default)d)')
+    help="timeout after which domains are killed when using --wait"
+    " (default: %(default)d)",
+)
 
 parser.add_argument(
-    '--force',
-    action='store_true', default=False,
-    help='shut down even if other qubes depend on this one (e.g. as NetVM'
-         ' or AudioVM); does not affect how the qube itself is shut down;'
-         ' use with caution')
+    "--force",
+    action="store_true",
+    default=False,
+    help="shut down even if other qubes depend on this one (e.g. as NetVM"
+    " or AudioVM); does not affect how the qube itself is shut down;"
+    " use with caution",
+)
 
 parser.add_argument(
-    '--dry-run',
-    action='store_true', dest='dry_run', default=False,
-    help='don\'t really shutdown or kill the domains; useful with --wait')
+    "--dry-run",
+    action="store_true",
+    dest="dry_run",
+    default=False,
+    help="don't really shutdown or kill the domains; useful with --wait",
+)
 
 
 def failed_domains(vms):
-    '''Find the domains that have not successfully been shut down'''
+    """Find the domains that have not successfully been shut down"""
 
     # DispVM might have been deleted before we check them, so NA is acceptable.
-    return [vm for vm in vms
-            if not (vm.get_power_state() == 'Halted'
-                or (vm.klass == 'DispVM' and vm.get_power_state() == 'NA'))]
+    return [
+        vm
+        for vm in vms
+        if not (
+            vm.get_power_state() == "Halted"
+            or (vm.klass == "DispVM" and vm.get_power_state() == "NA")
+        )
+    ]
+
 
 def main(args=None, app=None):  # pylint: disable=missing-docstring
     args = parser.parse_args(args, app=app)
@@ -86,14 +104,15 @@ def main(args=None, app=None):  # pylint: disable=missing-docstring
                     pass
                 except qubesadmin.exc.QubesException as e:
                     if not args.wait:
-                        vm.log.error('Shutdown error: {}'.format(e))
+                        vm.log.error("Shutdown error: {}".format(e))
                     shutdown_failed.add(vm)
         if not args.wait:
             if shutdown_failed:
                 parser.error_runtime(
-                    'Failed to shut down: ' +
-                    ', '.join(vm.name for vm in shutdown_failed),
-                    len(shutdown_failed))
+                    "Failed to shut down: "
+                    + ", ".join(vm.name for vm in shutdown_failed),
+                    len(shutdown_failed),
+                )
             return
         awaiting = remaining_domains - shutdown_failed
         remaining_domains = shutdown_failed
@@ -103,16 +122,21 @@ def main(args=None, app=None):  # pylint: disable=missing-docstring
 
         try:
             # pylint: disable=no-member
-            loop.run_until_complete(asyncio.wait_for(
-                qubesadmin.events.utils.wait_for_domain_shutdown(
-                    awaiting), args.timeout))
+            loop.run_until_complete(
+                asyncio.wait_for(
+                    qubesadmin.events.utils.wait_for_domain_shutdown(awaiting),
+                    args.timeout,
+                )
+            )
         except (TimeoutError, asyncio.TimeoutError):
             if not args.dry_run:
                 current_vms = failed_domains(awaiting)
                 if current_vms:
                     args.app.log.info(
-                        'Killing remaining qubes: {}'
-                        .format(', '.join([str(vm) for vm in current_vms])))
+                        "Killing remaining qubes: {}".format(
+                            ", ".join([str(vm) for vm in current_vms])
+                        )
+                    )
                 for vm in current_vms:
                     try:
                         vm.kill()
@@ -126,10 +150,10 @@ def main(args=None, app=None):  # pylint: disable=missing-docstring
     failed = failed_domains(args.domains)
     if failed:
         parser.error_runtime(
-            'Failed to shut down: ' +
-            ', '.join(vm.name for vm in failed),
-            len(failed))
+            "Failed to shut down: " + ", ".join(vm.name for vm in failed),
+            len(failed),
+        )
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     sys.exit(main())