Spaces:
Running
Running
| #!/usr/bin/env python | |
| """Show or set a public VARIABLE on one of the DemoPrep HF Spaces. | |
| Variables are the per-Space switches that make ONE commit behave differently on | |
| test and prod — e.g. MUZE_ENGINE_MAIN_TABS=false as an emergency off on one Space | |
| (unset = ON everywhere since 2026-09-22). | |
| They are NOT secrets: on a public Space anyone can read them. Secrets go through | |
| scripts/setup_test_mcp_secrets.py. | |
| ./.venv/bin/python scripts/set_space_variable.py test-mcp --show | |
| ./.venv/bin/python scripts/set_space_variable.py test-mcp MUZE_ENGINE_MAIN_TABS true | |
| ./.venv/bin/python scripts/set_space_variable.py mcp MUZE_ENGINE_MAIN_TABS false --dry-run | |
| ./.venv/bin/python scripts/set_space_variable.py test-mcp MUZE_ENGINE_MAIN_TABS --delete | |
| Space may be a short name (test-mcp, mcp, demoprep, test-demoprep) or a full | |
| `owner/name` repo id. | |
| Setting or deleting a variable RESTARTS the Space and kills any build in flight. | |
| Check the MCP `status` tool for active_builds first. Set the variable BEFORE | |
| pushing code that reads it — old code ignores the variable, so the extra | |
| restart is harmless, and the new code comes up already in the intended mode. | |
| Token: HF_TOKEN from the environment, else from the .env named by --env | |
| (default: <repo root>/.env). Reading variables on a public Space needs no token. | |
| """ | |
| import argparse | |
| import os | |
| import sys | |
| from pathlib import Path | |
| SHORT_NAMES = { | |
| "test-mcp": "thoughtspot-demoprep/test-mcp", | |
| "mcp": "thoughtspot-demoprep/mcp", | |
| "demoprep": "thoughtspot-dp/demoprep", | |
| "test-demoprep": "thoughtspot-dp/test-demoprep", | |
| } | |
| DEFAULT_ENV = Path(__file__).resolve().parent.parent / ".env" | |
| def repo_id_of(space: str) -> str: | |
| if "/" in space: | |
| return space | |
| if space in SHORT_NAMES: | |
| return SHORT_NAMES[space] | |
| sys.exit(f"ERROR: unknown Space {space!r}. Use one of {sorted(SHORT_NAMES)} or a full owner/name.") | |
| def token_from(env_path: Path) -> str: | |
| tok = (os.getenv("HF_TOKEN") or "").strip() | |
| if tok: | |
| return tok | |
| if env_path.exists(): | |
| for raw in env_path.read_text().splitlines(): | |
| line = raw.strip() | |
| if line.startswith("HF_TOKEN="): | |
| return line.split("=", 1)[1].strip().strip('"').strip("'") | |
| sys.exit(f"ERROR: no HF_TOKEN in the environment or in {env_path}. " | |
| "A write needs one; pass --env to point at the .env that has it.") | |
| def show(api, repo_id: str) -> None: | |
| variables = api.get_space_variables(repo_id=repo_id) | |
| if not variables: | |
| print(f"{repo_id}: no variables set") | |
| return | |
| print(f"{repo_id}:") | |
| for key, var in sorted(variables.items()): | |
| updated = getattr(var, "updated_at", None) or getattr(var, "updatedAt", "") | |
| print(f" {key} = {var.value!r} (updated {updated})") | |
| def main() -> None: | |
| ap = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter) | |
| ap.add_argument("space", help="test-mcp | mcp | demoprep | test-demoprep | owner/name") | |
| ap.add_argument("key", nargs="?", help="variable name") | |
| ap.add_argument("value", nargs="?", help="value to set") | |
| ap.add_argument("--show", action="store_true", help="list the Space's variables and exit") | |
| ap.add_argument("--delete", action="store_true", help="delete KEY instead of setting it") | |
| ap.add_argument("--dry-run", action="store_true", help="say what would change, touch nothing") | |
| ap.add_argument("--env", type=Path, default=DEFAULT_ENV, help="the .env holding HF_TOKEN") | |
| args = ap.parse_args() | |
| from huggingface_hub import HfApi | |
| repo_id = repo_id_of(args.space) | |
| if args.show or not args.key: | |
| # Reads work without a token on a public Space; send one if we have it. | |
| tok = (os.getenv("HF_TOKEN") or "").strip() or None | |
| show(HfApi(token=tok), repo_id) | |
| return | |
| if args.delete and args.value is not None: | |
| sys.exit("ERROR: --delete takes KEY only, no value.") | |
| if not args.delete and args.value is None: | |
| sys.exit("ERROR: give a VALUE to set, or --delete, or --show.") | |
| action = f"DELETE {args.key}" if args.delete else f"SET {args.key}={args.value!r}" | |
| print(f"{action} on {repo_id} — this RESTARTS the Space and kills in-flight builds.") | |
| if args.dry_run: | |
| print("--dry-run: nothing was changed.") | |
| return | |
| if input("Proceed? [y/N] ").strip().lower() != "y": | |
| sys.exit("Aborted.") | |
| api = HfApi(token=token_from(args.env)) | |
| if args.delete: | |
| api.delete_space_variable(repo_id=repo_id, key=args.key) | |
| else: | |
| api.add_space_variable(repo_id=repo_id, key=args.key, value=args.value) | |
| print("done. Now:") | |
| show(api, repo_id) | |
| print("\nNEXT: after the restart, call the `config_check` tool and confirm `muze.main_tabs` " | |
| "(or whatever the variable drives) reads as intended. A booted container is not a verified one.") | |
| if __name__ == "__main__": | |
| main() | |