#!/usr/bin/env python3
"""Fix run_case queue lookup for the new aa043- RID scheme."""
src = open('/opt/struktur/obsidian-graphiti-import-staging-p2i/p2i_e2e.py').read()
old = '    rows = queue_rows(kc, "legacy-queue-" + str(fid))'
new = ('    rows = kc.execute("SELECT * FROM graphiti_import_queue '
       "WHERE reconciliation_record_id LIKE 'aa043-%' AND obsidian_path LIKE "
       "'%e2e-note%' AND action=?\", (expect_action or \"CREATE\",)).fetchall()")
assert old in src, 'pattern not found'
open('/opt/struktur/obsidian-graphiti-import-staging-p2i/p2i_e2e.py', 'w').write(src.replace(old, new))
import py_compile
py_compile.compile('/opt/struktur/obsidian-graphiti-import-staging-p2i/p2i_e2e.py', doraise=True)
print('run_case query updated + SYNTAX_OK')