gabriel / musehub public
backfill_structured_delta.py python
123 lines 4.2 KB
Raw
sha256:34035d72cef530c1ab9d6a6f53be18d803dde705fc3157617d70352a96d0747b Merge branch 'fix/wire-push-external-parent-manifest' into dev Human 8 days ago
1 """Backfill structured_delta from local Muse history into musehub_commits column.
2
3 Migration 0021 added structured_delta as a first-class column. Historical rows
4 are NULL because the data was in the now-dropped commit_meta blob. This script
5 reads structured_delta for each commit directly from the local Muse object
6 store (which has the full history) and writes it back to the DB.
7
8 Usage (run from the host, not inside Docker — needs the local muse CLI):
9 python3 deploy/backfill_structured_delta.py --repo ~/ecosystem/musehub --dry-run
10 python3 deploy/backfill_structured_delta.py --repo ~/ecosystem/musehub
11 """
12 from __future__ import annotations
13
14 import argparse
15 import asyncio
16 import json
17 import subprocess
18 import sys
19 import time
20
21 import sqlalchemy as sa
22 from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
23 from sqlalchemy.orm import sessionmaker
24
25 from musehub.db.database import get_database_url
26 from musehub.db.musehub_repo_models import MusehubCommit
27
28
29 def muse(repo: str, *args: str) -> dict:
30 result = subprocess.run(
31 ["muse", "-C", repo, *args, "--json"],
32 capture_output=True, text=True, check=True,
33 )
34 return json.loads(result.stdout)
35
36
37 async def run(repo_path: str, dry_run: bool, batch_size: int) -> None:
38 engine = create_async_engine(get_database_url(), echo=False)
39 async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
40
41 print(f"Reading commit log from {repo_path}…")
42 log = muse(repo_path, "log")
43 commits = log["commits"]
44 print(f"Found {len(commits)} commits in local history.")
45
46 updated = 0
47 skipped = 0
48 missing = 0
49 errors = 0
50 t0 = time.monotonic()
51
52 async with async_session() as session:
53 for i, entry in enumerate(commits, 1):
54 commit_id: str = entry["commit_id"]
55
56 # Check whether this commit exists in the DB and needs backfilling
57 row = (await session.execute(
58 sa.select(MusehubCommit.commit_id, MusehubCommit.structured_delta)
59 .where(MusehubCommit.commit_id == commit_id)
60 )).one_or_none()
61
62 if row is None:
63 missing += 1
64 continue
65
66 if row.structured_delta is not None:
67 skipped += 1
68 continue
69
70 # Read structured_delta from local Muse object store
71 try:
72 commit_data = muse(repo_path, "read", commit_id)
73 delta = commit_data.get("structured_delta")
74 except subprocess.CalledProcessError as exc:
75 print(f" ✗ {commit_id[:16]}… muse read failed: {exc.stderr.strip()}", file=sys.stderr)
76 errors += 1
77 continue
78
79 if not isinstance(delta, dict):
80 skipped += 1
81 continue
82
83 if not dry_run:
84 await session.execute(
85 sa.update(MusehubCommit)
86 .where(MusehubCommit.commit_id == commit_id)
87 .values(structured_delta=delta)
88 )
89
90 updated += 1
91
92 if updated % batch_size == 0:
93 if not dry_run:
94 await session.commit()
95 elapsed = time.monotonic() - t0
96 print(f" [{i}/{len(commits)}] {updated} updated, {skipped} skipped, {missing} not-in-db — {elapsed:.1f}s")
97
98 if not dry_run:
99 await session.commit()
100
101 elapsed = time.monotonic() - t0
102 tag = "[DRY RUN] " if dry_run else ""
103 print(f"\n{tag}Done in {elapsed:.1f}s:")
104 print(f" updated : {updated}")
105 print(f" skipped : {skipped} (already had delta or no delta in local store)")
106 print(f" not-in-db: {missing} (local commits not yet pushed to server)")
107 print(f" errors : {errors}")
108
109 await engine.dispose()
110
111
112 def main() -> None:
113 parser = argparse.ArgumentParser(description=__doc__)
114 parser.add_argument("--repo", required=True, help="Path to local Muse repo (e.g. ~/ecosystem/musehub)")
115 parser.add_argument("--dry-run", action="store_true")
116 parser.add_argument("--batch-size", type=int, default=100)
117 args = parser.parse_args()
118
119 asyncio.run(run(args.repo, args.dry_run, args.batch_size))
120
121
122 if __name__ == "__main__":
123 main()
File History 14 commits
sha256:34035d72cef530c1ab9d6a6f53be18d803dde705fc3157617d70352a96d0747b Merge branch 'fix/wire-push-external-parent-manifest' into dev Human 8 days ago
sha256:fc04e4cae9e1774d6a21b65c45daeed0e6787eb581d13aa1b03bfe9384a34226 Merge branch 'fix/two-column-scroll-layout' into dev Human 8 days ago
sha256:408916fc5973ba59c6e4eebaa80ebdcc801c0a63205651e25009d11548f79454 chore: bump version to 0.2.0.dev2 — nightly.2, matching muse Sonnet 4.6 patch 11 days ago
sha256:d035733f21ccff27735fddebfbbe0ed24565a32a22db8de5885402262671ecd2 chore: bump version to 0.2.0rc15 for musehub#113 fix release Sonnet 4.6 patch 14 days ago
sha256:0032d6cfa33bc3c8367436ad768e7dd0e339b4332153160247da8266cb5fa352 Merge branch 'task/version-tags-phase3-server' into dev Human 17 days ago
sha256:4669620efda9ff41c55bdefd1f7bfe1c239d468428744c84ead9957e5a003a53 merge: rescue snapshot-recovery hardening (c00aa21d) into d… Opus 4.8 minor 29 days ago
sha256:a59da49c4611b970fc4b6ae48678ce4943261c213a07ddbd73ce9201df869b4a fix: remove false-positive proposal_comments index drop fro… Sonnet 4.6 patch 33 days ago
sha256:0a240d6dbff234f07d98a28a4a9a68db702f3f9ff9260196f24219bdb1c0b6f3 feat: render markdown mists as HTML with heading anchor links Sonnet 4.6 patch 34 days ago
sha256:24a7d47486ebc4ebd1832830580e177ec6f877b48dced8c000e198cdec4ce9d6 Merge 'task/bump-version-rc12' into 'dev' — proposal: Bump … Human 35 days ago
sha256:b9ff931d147e0114a1f17060f415b89ed551c170a91ff226c70437aa5c85f9ee Merge 'task/bump-version-rc12' into 'dev' — proposal: Bump … Human 35 days ago
sha256:d1122d21e73471879b460037b22c0b50fded7c423444a176f248428f75dac39c Merge 'task/fix-issue-pagination-cursor' into 'dev' — propo… Human 35 days ago
sha256:01e18975e73d2b3cd5b6db7929c895bef9aa6e0d4391dc5b2adfc548b41318dd Merge 'feat/adding-debug-logs-to-staging' into 'dev' — prop… Human 35 days ago
sha256:6b1949fc2797ca4c1936a637a4cbfec828ef56cf52398a2e74ca3c4f494e728f fix: use wire_bytes not mpack_bytes_raw in compute_object_b… Sonnet 4.6 patch 48 days ago
sha256:b99f2455dc346966d040133f5203297e6e3ef5803a93728a2c30568d0a0f7583 rename: delta_add → delta_upsert across wire format, models… Sonnet 4.6 patch 50 days ago