Phase 1 of Migration. Running Scripts
This commit is contained in:
67
backend/migration/migrate_mfg_audit_log.py
Normal file
67
backend/migration/migrate_mfg_audit_log.py
Normal file
@@ -0,0 +1,67 @@
|
||||
"""
|
||||
Phase 1 — Step 1.3: mfg_audit_log (SQLite → Postgres)
|
||||
|
||||
Run on VPS:
|
||||
docker compose exec backend python -m migration.migrate_mfg_audit_log
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import sys
|
||||
|
||||
from sqlalchemy import text
|
||||
|
||||
from migration.utils import open_sqlite, AsyncPgSession, parse_dt, log_run, pg_count
|
||||
|
||||
SCRIPT = "migrate_mfg_audit_log"
|
||||
|
||||
|
||||
async def run() -> None:
|
||||
sqlite = await open_sqlite()
|
||||
rows = await sqlite.execute_fetchall("SELECT * FROM mfg_audit_log ORDER BY id")
|
||||
await sqlite.close()
|
||||
|
||||
source_count = len(rows)
|
||||
print(f"Source (SQLite): {source_count} mfg_audit_log rows")
|
||||
|
||||
if source_count == 0:
|
||||
print("Nothing to migrate.")
|
||||
await log_run(SCRIPT, 0, 0, notes="source empty")
|
||||
return
|
||||
|
||||
# mfg_audit_log uses a BIGSERIAL PK — we don't preserve SQLite integer IDs
|
||||
# because the Postgres sequence will assign new ones. We insert in the same
|
||||
# timestamp order so the audit trail remains coherent.
|
||||
records = [
|
||||
{
|
||||
"timestamp": parse_dt(r["timestamp"]),
|
||||
"admin_user": r["admin_user"],
|
||||
"action": r["action"],
|
||||
"serial_number": r["serial_number"],
|
||||
"detail": r["detail"],
|
||||
}
|
||||
for r in rows
|
||||
]
|
||||
|
||||
async with AsyncPgSession() as session:
|
||||
async with session.begin():
|
||||
await session.execute(
|
||||
text("""
|
||||
INSERT INTO mfg_audit_log (timestamp, admin_user, action, serial_number, detail)
|
||||
VALUES (:timestamp, :admin_user, :action, :serial_number, :detail)
|
||||
"""),
|
||||
records,
|
||||
)
|
||||
dest_count = await pg_count(session, "mfg_audit_log")
|
||||
|
||||
if dest_count < source_count:
|
||||
msg = f"Count mismatch: source={source_count} postgres={dest_count}"
|
||||
print(f"ERROR: {msg}", file=sys.stderr)
|
||||
await log_run(SCRIPT, source_count, dest_count, success=False, notes=msg)
|
||||
sys.exit(1)
|
||||
|
||||
print(f"Postgres: {dest_count} rows ✓")
|
||||
await log_run(SCRIPT, source_count, dest_count)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(run())
|
||||
Reference in New Issue
Block a user