feat: allow configuring the runner class [SBK-233] (#29)

.pre-commit-config.yaml

@@ -10,7 +10,7 @@ repos:
       - id: isort
 
 -   repo: https://github.com/psf/black
-    rev: 22.12.0
+    rev: 23.7.0
     hooks:
       - id: black
         name: black
@@ -21,7 +21,7 @@ repos:
     -   id: flake8
 
 -   repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v0.991
+    rev: v1.4.1
     hooks:
     -   id: mypy
         additional_dependencies: [types-setuptools, pydantic]

setup.py

@@ -10,8 +10,8 @@
         "hypothesis>=6.2.0,<7.0",  # Strategy-based fuzzer
     ],
     "lint": [
-        "black>=22.12.0",  # auto-formatter and linter
-        "mypy>=0.991",  # Static type analyzer
+        "black>=23.7.0",  # auto-formatter and linter
+        "mypy>=1.4.1,<2",  # Static type analyzer
         "types-setuptools",  # Needed for mypy type shed
         "flake8>=5.0.4",  # Style linter
         "isort>=5.10.1",  # Import sorting linter

silverback/_cli.py

@@ -13,23 +13,47 @@ def cli():
     """Work with SilverBack applications in local context (using Ape)."""
 
 
+def _runner_callback(ctx, param, val):
+    if not val:
+        return LiveRunner
+
+    elif runner := import_from_string(val):
+        return runner
+
+    raise ValueError(f"Failed to import runner '{val}'.")
+
+
+def _account_callback(ctx, param, val):
+    if val:
+        val = val.alias.replace("dev_", "TEST::")
+        os.environ["SILVERBACK_SIGNER_ALIAS"] = val
+
+    return val
+
+
+def _network_callback(ctx, param, val):
+    if val:
+        os.environ["SILVERBACK_NETWORK_CHOICE"] = val
+    else:
+        val = os.environ.get("SILVERBACK_NETWORK_CHOICE", "")
+
+    return val
+
+
 @cli.command()
 @ape_cli_context()
 @verbosity_option()
-@network_option(default=None)
-@click.option("--account", type=AccountAliasPromptChoice(), default=None)
+@network_option(default=None, callback=_network_callback)
+@click.option("--account", type=AccountAliasPromptChoice(), callback=_account_callback)
+@click.option(
+    "--runner",
+    help="An import str in format '<module>:<CustomRunner>'",
+    callback=_runner_callback,
+)
 @click.option("-x", "--max-exceptions", type=int, default=3)
 @click.argument("path")
-def run(cli_ctx, network, account, max_exceptions, path):
-    if network:
-        os.environ["SILVERBACK_NETWORK_CHOICE"] = network
-    else:
-        network = os.environ.get("SILVERBACK_NETWORK_CHOICE", "")
-
-    if account:
-        os.environ["SILVERBACK_SIGNER_ALIAS"] = account.alias.replace("dev_", "TEST::")
-
+def run(cli_ctx, network, account, runner, max_exceptions, path):
     with cli_ctx.network_manager.parse_network_choice(network):
         app = import_from_string(path)
-        runner = LiveRunner(app, max_exceptions=max_exceptions)
+        runner = runner(app, max_exceptions=max_exceptions)
         asyncio.run(runner.run())

silverback/runner.py

@@ -58,13 +58,9 @@ async def run(self):
         if len(tasks) == 0:
             raise SilverBackException("No tasks to execute")
 
-        try:
-            await asyncio.gather(*tasks)
-        except Exception as e:
-            logger.error(f"Critical exception, {type(e).__name__}: {e}")
+        await asyncio.gather(*tasks)
 
-        finally:
-            await self.app.broker.shutdown()
+        await self.app.broker.shutdown()
 
 
 class LiveRunner(BaseRunner):