Automatic PEP8 style update with Black and isort

https://github.com/psf/black
https://github.com/pycqa/isort

Author: bkus

tasks.py

@@ -1,33 +1,32 @@
 # pylint: disable=missing-module-docstring,missing-function-docstring
-import os
-import sys
+import distutils.version
+import itertools
 import json
+import os
 import pathlib
+import sys
 import types
-import itertools
-import distutils.version
-import requests
+
 import invoke
 import invoke.program
 import PyInquirer
+import requests
 
 script_dir = os.path.dirname(os.path.realpath(__file__))
 
+
 def ask_for_confirmation(prompt, batch_mode, default):
     if batch_mode:
         print(prompt)
-        print("Batch mode is on. Autoselecting default option ({})".format(
-            {True: "yes", False: "no"}[default]
-        ))
+        print(
+            "Batch mode is on. Autoselecting default option ({})".format(
+                {True: "yes", False: "no"}[default]
+            )
+        )
         confirmed = default
     else:
         conf_questions = [
-            {
-                "type": "confirm",
-                "name": "continue",
-                "message": prompt,
-                "default": True
-            }
+            {"type": "confirm", "name": "continue", "message": prompt, "default": True}
         ]
         conf_answers = PyInquirer.prompt(conf_questions)
         confirmed = bool(conf_answers) and conf_answers["continue"]
@@ -37,14 +36,17 @@ def ask_for_confirmation(prompt, batch_mode, default):
 
 def get_plugin_file_path():
     with open(
-        os.path.join(script_dir, "vagrant-plugins-routeros/vagrant_routeros_plugin_version.json")
+        os.path.join(
+            script_dir, "vagrant-plugins-routeros/vagrant_routeros_plugin_version.json"
+        )
     ) as ver_f:
         plugin_version = json.load(ver_f)["vagrant_routeros_plugin_version"]
     return os.path.join(
         script_dir,
-        "vagrant-plugins-routeros/pkg/vagrant-routeros-{}.gem".format(plugin_version)
+        "vagrant-plugins-routeros/pkg/vagrant-routeros-{}.gem".format(plugin_version),
     )
 
+
 def build_routeros(context, routeros_branch):
     if routeros_branch == "routeros-long-term":
         branch_name = "6 (long-term)"
@@ -78,18 +80,16 @@ def build_routeros(context, routeros_branch):
     box_file_name = "build/boxes/{}_{}.box".format(routeros_branch, ros_version)
     if os.path.isfile(box_file_name):
         print("'{}' has alredy been built".format(box_file_name))
-        ask_for_confirmation(
-            "Do you want to rebuild it?", context.routeros.batch, True
-        )
+        ask_for_confirmation("Do you want to rebuild it?", context.routeros.batch, True)
 
     packer_error_action = "cleanup" if context.routeros.batch else "ask"
     print("Building the box...")
     context.run(
-        f"packer build -var \"ros_ver={ros_version}\" "
-        f"-var \"box_file_name={box_file_name}\" "
+        f'packer build -var "ros_ver={ros_version}" '
+        f'-var "box_file_name={box_file_name}" '
         "-var-file vagrant-plugins-routeros/vagrant_routeros_plugin_version.json "
         f"-on-error={packer_error_action} -force routeros.pkr.hcl",
-        echo=True
+        echo=True,
     )
 
     description_md = pathlib.Path(box_file_name).with_suffix(".md")
@@ -107,9 +107,7 @@ def build_plugin(context):
     print(f"Using helper VM in '{vm_dir}'")
     current_vm_state = ""
     with context.cd(vm_dir):
-        run_result = context.run(
-            command="vagrant status --machine-readable", hide=True
-        )
+        run_result = context.run(command="vagrant status --machine-readable", hide=True)
     for line in run_result.stdout.splitlines():
         values = line.split(",")
         if len(values) > 3:
@@ -143,10 +141,15 @@ def build_plugin(context):
         with context.cd(vm_dir):
             context.run("vagrant halt")
 
+
 def remove_test_boxes(context):
     for line in context.run("vagrant box list", hide=True).stdout.splitlines():
         box_name = line.split(" ")[0]
-        if box_name in ("packer_test_routeros", "packer_test_routeros-long-term", "packer_test_routeros7"):
+        if box_name in (
+            "packer_test_routeros",
+            "packer_test_routeros-long-term",
+            "packer_test_routeros7",
+        ):
             context.run(f"vagrant box remove -f {box_name}", pty=True)
 
 
@@ -157,39 +160,41 @@ def do_cleanup(context):
 
     files_2del = (pathlib.Path(script_dir) / "build" / "boxes").glob("*.box")
     files_2del = itertools.chain(
-        files_2del,
-        (pathlib.Path(script_dir) / "build" / "boxes").glob("*.md")
+        files_2del, (pathlib.Path(script_dir) / "build" / "boxes").glob("*.md")
     )
     files_2del = itertools.chain(
         files_2del,
-        (pathlib.Path(script_dir) / "vagrant-plugins-routeros" / "pkg").glob("*.gem")
+        (pathlib.Path(script_dir) / "vagrant-plugins-routeros" / "pkg").glob("*.gem"),
     )
     files_2del = itertools.chain(
-        files_2del,
-        (pathlib.Path(script_dir) / "packer_cache").rglob("*")
+        files_2del, (pathlib.Path(script_dir) / "packer_cache").rglob("*")
     )
 
     for f_2del in files_2del:
         if f_2del.is_file():
             print(f"  Deleting {f_2del}")
             f_2del.unlink()
 
+
 def register_test_box(context, routeros_branch):
     boxes_dir = pathlib.Path(script_dir) / "build" / "boxes"
-    box_versions = [item.stem.split("_")[1] for item in boxes_dir.glob(f"{routeros_branch}_*.box")]
+    box_versions = [
+        item.stem.split("_")[1] for item in boxes_dir.glob(f"{routeros_branch}_*.box")
+    ]
     if len(box_versions) == 0:
         sys.exit(
             f"Couldn't find files matching pattern 'build/boxes/{routeros_branch}"
             f"_*.box'. Use 'inv {routeros_branch}' to build a box"
         )
     box_file = (
-        f"{routeros_branch}_" +
-        max(box_versions, key=distutils.version.LooseVersion) +
-        ".box"
+        f"{routeros_branch}_"
+        + max(box_versions, key=distutils.version.LooseVersion)
+        + ".box"
     )
     box_file = str(boxes_dir / box_file)
     context.run(f"vagrant box add packer_test_{routeros_branch} {box_file}", pty=True)
 
+
 def test_ping(context, vm_name, ping_target):
     print(f"Pinging {ping_target} from {vm_name}")
     ping_output = context.run(
@@ -198,22 +203,25 @@ def test_ping(context, vm_name, ping_target):
     assert "received=3" in ping_output
     assert "packet-loss=0%" in ping_output
 
+
 @invoke.task(default=True)
 def show_help(context):
     """This help message"""
-    context.run('invoke --list')
+    context.run("invoke --list")
     print("Use --help parameter to view task's options")
     print("Examples:")
     print("  inv build --help")
     print("  inv build --batch")
     print("  inv routeros")
     print("  inv plugin --batch")
 
+
 @invoke.task()
 def cleanup(context):
     """Delete build artefacts and temporary files"""
     do_cleanup(context)
 
+
 @invoke.task()
 def test(context):
     """Register temporary vagrant boxes and run some tests against them"""
@@ -244,6 +252,7 @@ def test(context):
     print("Removing test boxes...")
     remove_test_boxes(context)
 
+
 @invoke.task(help={"batch": "Batch mode (disables interactive prompts)"})
 def build(context, batch=False):
     """Build all"""
@@ -256,34 +265,39 @@ def build(context, batch=False):
     build_routeros(context, routeros_branch="routeros7")
     test(context)
 
+
 @invoke.task(help={"batch": "Batch mode (disables interactive prompts)"})
 def routeros_long_term(context, batch=False):
     """Build RouterOS (long-term)"""
 
     context.routeros.batch = batch
     build_routeros(context, routeros_branch="routeros-long-term")
 
+
 @invoke.task(help={"batch": "Batch mode (disables interactive prompts)"})
 def routeros(context, batch=False):
     """Build RouterOS (stable)"""
 
     context.routeros.batch = batch
     build_routeros(context, routeros_branch="routeros")
 
+
 @invoke.task(help={"batch": "Batch mode (disables interactive prompts)"})
 def routeros7(context, batch=False):
     """Build RouterOS 7 (stable)"""
 
     context.routeros.batch = batch
     build_routeros(context, routeros_branch="routeros7")
 
+
 @invoke.task(help={"batch": "Batch mode (disables interactive prompts)"})
 def plugin(context, batch=False):
     """Build 'vagrant-routeros' plugin"""
 
     context.routeros.batch = batch
     build_plugin(context)
 
+
 @invoke.task()
 def outdated(context):  # pylint: disable=unused-argument
     """Check if currently published box versions are up to date"""
@@ -293,34 +307,36 @@ def outdated(context):  # pylint: disable=unused-argument
             branch_name="6 (long-term)",
             version_url="http://upgrade.mikrotik.com/routeros/LATEST.6fix",
             box_name="cheretbe/routeros-long-term",
-            box_url="https://app.vagrantup.com/api/v1/box/cheretbe/routeros-long-term"
+            box_url="https://app.vagrantup.com/api/v1/box/cheretbe/routeros-long-term",
         ),
         types.SimpleNamespace(
             branch_name="6 (stable)",
             version_url="http://upgrade.mikrotik.com/routeros/LATEST.6",
             box_name="cheretbe/routeros",
-            box_url="https://app.vagrantup.com/api/v1/box/cheretbe/routeros"
+            box_url="https://app.vagrantup.com/api/v1/box/cheretbe/routeros",
         ),
         types.SimpleNamespace(
             branch_name="7 (stable)",
             version_url="http://upgrade.mikrotik.com/routeros/NEWEST7.stable",
             box_name="cheretbe/routeros7",
-            box_url="https://app.vagrantup.com/api/v1/box/cheretbe/routeros7"
-        )
+            box_url="https://app.vagrantup.com/api/v1/box/cheretbe/routeros7",
+        ),
     ]
 
     for ros_version in ros_version_info:
         print(f"Checking RouterOS {ros_version.branch_name} version")
         current_version = distutils.version.LooseVersion(
             requests.get(ros_version.version_url).text.split(" ")[0]
         )
-        box_version = (
-            requests.get(ros_version.box_url).json()["current_version"]["version"]
-        )
+        box_version = requests.get(ros_version.box_url).json()["current_version"][
+            "version"
+        ]
         box_os_version = distutils.version.LooseVersion(box_version.split("-")[0])
 
         if box_os_version == current_version:
-            print(f"Published version {box_version} of '{ros_version.box_name}' is up to date")
+            print(
+                f"Published version {box_version} of '{ros_version.box_name}' is up to date"
+            )
         elif current_version > box_os_version:
             print(
                 f"[!] '{ros_version.box_name}' box version {box_version} needs "
@@ -332,10 +348,5 @@ def outdated(context):  # pylint: disable=unused-argument
                 f"is greater than currently published version {current_version}"
             )
 
-invoke.main.program.config.update(
-    {
-        "routeros": {
-            "batch": False
-        }
-    }
-)
+
+invoke.main.program.config.update({"routeros": {"batch": False}})