Kitos
55dba1e00a
- Add UniqueConstraint(test_id, detection_rule_id) named uq_tdr_test_rule to TestDetectionResult model - Alembic b025: safely deduplicate existing rows before creating constraint
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
"""add_unique_test_detection_result
|
|
|
|
Enforce one evaluation per (test, detection_rule) pair. Before creating
|
|
the constraint, duplicate rows (if any) are collapsed so the migration
|
|
never fails on an existing database.
|
|
|
|
Revision ID: b025uqtdr
|
|
Revises: b024critidx
|
|
Create Date: 2026-02-18 14:00:00.000000
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
|
|
revision: str = "b025uqtdr"
|
|
down_revision: Union[str, None] = "b024critidx"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Remove duplicates keeping the most recently evaluated row
|
|
op.execute("""
|
|
DELETE FROM test_detection_results
|
|
WHERE id NOT IN (
|
|
SELECT DISTINCT ON (test_id, detection_rule_id) id
|
|
FROM test_detection_results
|
|
ORDER BY test_id, detection_rule_id, evaluated_at DESC NULLS LAST
|
|
)
|
|
""")
|
|
|
|
op.create_unique_constraint(
|
|
"uq_tdr_test_rule",
|
|
"test_detection_results",
|
|
["test_id", "detection_rule_id"],
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_constraint("uq_tdr_test_rule", "test_detection_results", type_="unique")
|