Ensures only adds network local to commands which utilise that arg.

Author: thewhaleking

tests/e2e_tests/utils.py

@@ -1,3 +1,4 @@
+import inspect
 import os
 import re
 import shutil
@@ -31,20 +32,28 @@ def exec_command(
         inputs: list[str] = None,
     ):
         extra_args = extra_args or []
-        # Ensure if we forget to add `--network ws://127.0.0.1:9945` that it will run still using the local chain
-        if not any(
-            (
-                x in extra_args
-                for x in (
-                    "--network",
-                    "--chain",
-                    "--subtensor.network",
-                    "--subtensor.chain_endpoint",
-                )
-            )
-        ):
-            extra_args.extend(["--network", "ws://127.0.0.1:9945"])
         cli_manager = CLIManager()
+        for group in cli_manager.app.registered_groups:
+            if group.name == command:
+                for command_ in group.typer_instance.registered_commands:
+                    if command_.name == sub_command:
+                        if "network" in inspect.getcallargs(
+                            command_.callback
+                        ).keys() and not any(
+                            (
+                                x in extra_args
+                                for x in (
+                                    "--network",
+                                    "--chain",
+                                    "--subtensor.network",
+                                    "--subtensor.chain_endpoint",
+                                )
+                            )
+                        ):
+                            # Ensure if we forget to add `--network ws://127.0.0.1:9945` that it will run still
+                            # using the local chain
+                            extra_args.extend(["--network", "ws://127.0.0.1:9945"])
+
         # Capture stderr separately from stdout
         runner = CliRunner(mix_stderr=False)
         # Prepare the command arguments