lint

pinheadmz · pinheadmz · commit e9a4632896b3 · 2025-05-21T13:16:24.000-04:00
diff --git a/src/warnet/constants.py b/src/warnet/constants.py
@@ -48,13 +48,13 @@ class AnnexMember(Enum):
 
 PLUGIN_ANNEX = "annex"
 
-DEFAULT_IMAGE_REPO="bitcoindevproject/bitcoin"
+DEFAULT_IMAGE_REPO = "bitcoindevproject/bitcoin"
 
 # Bitcoin Core config
-FORK_OBSERVER_RPCAUTH="forkobserver:1418183465eecbd407010cf60811c6a0$d4e5f0647a63429c218da1302d7f19fe627302aeb0a71a74de55346a25d8057c"
+FORK_OBSERVER_RPCAUTH = "forkobserver:1418183465eecbd407010cf60811c6a0$d4e5f0647a63429c218da1302d7f19fe627302aeb0a71a74de55346a25d8057c"
 # Fork Observer config
-FORK_OBSERVER_RPC_USER="forkobserver"
-FORK_OBSERVER_RPC_PASSWORD="tabconf2024"
+FORK_OBSERVER_RPC_USER = "forkobserver"
+FORK_OBSERVER_RPC_PASSWORD = "tabconf2024"
 
 # Directories and files for non-python assets, e.g., helm charts, example scenarios, default configs
 SRC_DIR = files("warnet")
diff --git a/src/warnet/deploy.py b/src/warnet/deploy.py
@@ -15,6 +15,8 @@
     DEFAULTS_FILE,
     DEFAULTS_NAMESPACE_FILE,
     FORK_OBSERVER_CHART,
+    FORK_OBSERVER_RPC_PASSWORD,
+    FORK_OBSERVER_RPC_USER,
     HELM_COMMAND,
     INGRESS_HELM_COMMANDS,
     LOGGING_CRD_COMMANDS,
@@ -29,8 +31,6 @@
     AnnexMember,
     HookValue,
     WarnetContent,
-    FORK_OBSERVER_RPC_USER,
-    FORK_OBSERVER_RPC_PASSWORD,
 )
 from .control import _logs, _run
 from .k8s import (
diff --git a/src/warnet/graph.py b/src/warnet/graph.py
@@ -4,21 +4,20 @@
 import sys
 from pathlib import Path
 
-from rich import print
-from rich.console import Console
-from rich.table import Table
-
 import click
 import inquirer
 import yaml
+from rich import print
+from rich.console import Console
+from rich.table import Table
 
 from resources.scenarios.ln_framework.ln import Policy
 
 from .constants import (
+    DEFAULT_IMAGE_REPO,
     DEFAULT_TAG,
-    SUPPORTED_TAGS,
     FORK_OBSERVER_RPCAUTH,
-    DEFAULT_IMAGE_REPO,
+    SUPPORTED_TAGS,
 )
 
 
@@ -49,7 +48,7 @@ def custom_graph(
     index = 0
 
     for entry in tanks:
-        for i in range(int(entry["count"])):
+        for _ in range(int(entry["count"])):
             if ":" in entry["version"] and "/" in entry["version"]:
                 repo, tag = entry["version"].split(":")
                 image = {"repository": repo, "tag": tag}
@@ -98,11 +97,10 @@ def custom_graph(
             "repository": DEFAULT_IMAGE_REPO,
             "pullPolicy": "IfNotPresent",
         },
-        "defaultConfig":
-            f"rpcauth={FORK_OBSERVER_RPCAUTH}\n" +
-            "rpcwhitelist=forkobserver:getchaintips,getblockheader,getblockhash,getblock,getnetworkinfo\n" +
-            "rpcwhitelistdefault=0\n" +
-            "debug=rpc\n"
+        "defaultConfig": f"rpcauth={FORK_OBSERVER_RPCAUTH}\n"
+        + "rpcwhitelist=forkobserver:getchaintips,getblockheader,getblockhash,getblock,getnetworkinfo\n"
+        + "rpcwhitelistdefault=0\n"
+        + "debug=rpc\n",
     }
 
     # Configure logging
@@ -118,12 +116,15 @@ def custom_graph(
 
 
 def inquirer_create_network(project_path: Path):
-    network_name_prompt = inquirer.prompt([
-        inquirer.Text(
-            "network_name",
-            message=click.style("Enter your network name", fg="blue", bold=True),
-            validate=lambda _, x: len(x) > 0,
-        )])
+    network_name_prompt = inquirer.prompt(
+        [
+            inquirer.Text(
+                "network_name",
+                message=click.style("Enter your network name", fg="blue", bold=True),
+                validate=lambda _, x: len(x) > 0,
+            )
+        ]
+    )
     if not network_name_prompt:
         click.secho("Setup cancelled by user.", fg="yellow")
         return False
@@ -140,81 +141,108 @@ def inquirer_create_network(project_path: Path):
 
         Console().print(table)
 
-        add_more_prompt = inquirer.prompt([
-            inquirer.List(
-                "add_more",
-                message=click.style(f"How many nodes to add? (0 = done)", fg="blue", bold=True),
-                choices=["0", "4", "8", "12", "20", "50", "other"],
-                default="12")])
+        add_more_prompt = inquirer.prompt(
+            [
+                inquirer.List(
+                    "add_more",
+                    message=click.style("How many nodes to add? (0 = done)", fg="blue", bold=True),
+                    choices=["0", "4", "8", "12", "20", "50", "other"],
+                    default="12",
+                )
+            ]
+        )
         if not add_more_prompt:
             click.secho("Setup cancelled by user.", fg="yellow")
             return False
         if add_more_prompt["add_more"].startswith("0"):
             break
 
         if add_more_prompt["add_more"] == "other":
-            how_many_prompt = inquirer.prompt([
-                inquirer.Text(
-                    "how_many",
-                    message=click.style("Enter the number of nodes", fg="blue", bold=True),
-                    validate=lambda _, x: int(x) > 0)])
+            how_many_prompt = inquirer.prompt(
+                [
+                    inquirer.Text(
+                        "how_many",
+                        message=click.style("Enter the number of nodes", fg="blue", bold=True),
+                        validate=lambda _, x: int(x) > 0,
+                    )
+                ]
+            )
             if not how_many_prompt:
                 click.secho("Setup cancelled by user.", fg="yellow")
                 return False
             how_many = how_many_prompt["how_many"]
         else:
             how_many = add_more_prompt["add_more"]
 
-        tank_details_prompt = inquirer.prompt([
-            inquirer.List(
-                "version",
-                message=click.style("Which version would you like to add to network?", fg="blue", bold=True),
-                choices=["other"] + SUPPORTED_TAGS,
-                default=DEFAULT_TAG,
+        tank_details_prompt = inquirer.prompt(
+            [
+                inquirer.List(
+                    "version",
+                    message=click.style(
+                        "Which version would you like to add to network?", fg="blue", bold=True
+                    ),
+                    choices=["other"] + SUPPORTED_TAGS,
+                    default=DEFAULT_TAG,
                 ),
-            inquirer.List(
-                "connections",
-                message=click.style(
-                    "How many connections would you like each of these nodes to have?",
-                    fg="blue",
-                    bold=True,
+                inquirer.List(
+                    "connections",
+                    message=click.style(
+                        "How many connections would you like each of these nodes to have?",
+                        fg="blue",
+                        bold=True,
+                    ),
+                    choices=["0", "1", "2", "8", "12", "other"],
+                    default="8",
                 ),
-                choices=["0", "1", "2", "8", "12", "other"],
-                default="8",
-            )])
+            ]
+        )
         if not tank_details_prompt:
             click.secho("Setup cancelled by user.", fg="yellow")
             return False
             break
         if tank_details_prompt["version"] == "other":
-            custom_version_prompt = inquirer.prompt([
-                inquirer.Text(
-                    "version",
-                    message=click.style("Provide dockerhub repository/image:tag", fg="blue", bold=True),
-                    validate=lambda _, x: "/" in x and ":" in x)])
+            custom_version_prompt = inquirer.prompt(
+                [
+                    inquirer.Text(
+                        "version",
+                        message=click.style(
+                            "Provide dockerhub repository/image:tag", fg="blue", bold=True
+                        ),
+                        validate=lambda _, x: "/" in x and ":" in x,
+                    )
+                ]
+            )
             if not custom_version_prompt:
                 click.secho("Setup cancelled by user.", fg="yellow")
                 return False
             tank_details_prompt["version"] = custom_version_prompt["version"]
 
         if tank_details_prompt["connections"] == "other":
-            how_many_conn_prompt = inquirer.prompt([
-                inquirer.Text(
-                    "how_many_conn",
-                    message=click.style("Enter the number of connections", fg="blue", bold=True),
-                    validate=lambda _, x: int(x) > 0)])
+            how_many_conn_prompt = inquirer.prompt(
+                [
+                    inquirer.Text(
+                        "how_many_conn",
+                        message=click.style(
+                            "Enter the number of connections", fg="blue", bold=True
+                        ),
+                        validate=lambda _, x: int(x) > 0,
+                    )
+                ]
+            )
             if not how_many_conn_prompt:
                 click.secho("Setup cancelled by user.", fg="yellow")
                 return False
             how_many_conn = how_many_conn_prompt["how_many_conn"]
         else:
             how_many_conn = tank_details_prompt["connections"]
 
-        tanks.append({
-            "version": tank_details_prompt["version"],
-            "count": how_many,
-            "connections": how_many_conn
-        })
+        tanks.append(
+            {
+                "version": tank_details_prompt["version"],
+                "count": how_many,
+                "connections": how_many_conn,
+            }
+        )
 
     fork_observer = click.prompt(
         click.style(
diff --git a/test/plugin_test.py b/test/plugin_test.py
@@ -14,9 +14,7 @@
 class PluginTest(TestBase):
     def __init__(self):
         super().__init__()
-        self.network_dir = (
-            Path(os.path.dirname(__file__)) / "data" / "network_with_plugins"
-        )
+        self.network_dir = Path(os.path.dirname(__file__)) / "data" / "network_with_plugins"
         self.plugins_dir = Path(os.path.dirname(__file__)).parent / "resources" / "plugins"
         self.simln_exec = self.plugins_dir / "simln" / "plugin.py"
 
