"""Migration: composite index on Nonconformities(assessment_id, severity).

AssessmentResults(assessment_id) already exists via SQLModel init.
Safe to run multiple times (idempotent via IF NOT EXISTS).
"""
from __future__ import annotations

import sqlite3
from pathlib import Path

DB_PATH = Path(__file__).resolve().parent.parent / "fmcg.db"


def run() -> None:
    con = sqlite3.connect(DB_PATH)
    con.execute(
        "CREATE INDEX IF NOT EXISTS ix_nonconformities_assessment_severity "
        "ON nonconformities(assessment_id, severity)"
    )
    con.commit()
    con.close()
    print("Index ix_nonconformities_assessment_severity created (or already exists).")


if __name__ == "__main__":
    run()