Allow force running of wizard

Author: Drummersbrother

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "open-edison"
-version = "0.1.57"
+version = "0.1.58"
 description = "Open-source MCP security, aggregation, and monitoring. Single-user, self-hosted MCP proxy."
 readme = "README.md"
 authors = [

src/cli.py

@@ -47,6 +47,11 @@ def _parse_args(argv: list[str] | None = None) -> argparse.Namespace:
         action="store_true",
         help="(For the setup wizard) Skip OAuth for remote servers (they will be omitted from import)",
     )
+    parser.add_argument(
+        "--wizard-force",
+        action="store_true",
+        help="(For the setup wizard) Force running the setup wizard even if it has already been run",
+    )
     # Website runs from packaged assets by default; no extra website flags
 
     # Subcommands (extensible)
@@ -132,7 +137,7 @@ def main(argv: list[str] | None = None) -> NoReturn:  # noqa: C901
         raise SystemExit(result_code)
 
     # Run import tui if necessary
-    run_import_tui(args)
+    run_import_tui(args, force=args.wizard_force)
 
     try:
         asyncio.run(_run_server(args))

src/setup_tui/main.py

@@ -242,14 +242,14 @@ def error(self, msg: object, *args: object, **kwargs: object) -> None:
 
 
 # Triggered from cli.py
-def run_import_tui(args: argparse.Namespace) -> None:
+def run_import_tui(args: argparse.Namespace, force: bool = False) -> None:
     """Run the import TUI, if necessary."""
     # Find config dir, check if ".setup_tui_run" exists
     config_dir = get_config_dir()
     config_dir.mkdir(parents=True, exist_ok=True)
 
     setup_tui_run_file = config_dir / ".setup_tui_run"
-    if not setup_tui_run_file.exists():
+    if not setup_tui_run_file.exists() or force:
         run(dry_run=args.wizard_dry_run, skip_oauth=args.wizard_skip_oauth)
 
     setup_tui_run_file.touch()