kitos
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
"""Rename data_classification values to match the org's real scheme.
|
|
|
|
public -> public_release, internal -> general_use, sensitive -> confidential.
|
|
restricted is unchanged. Applies to tests, campaigns, and evidences tables
|
|
(all plain String(20) columns — no native enum type to alter).
|
|
|
|
Revision ID: b059
|
|
Revises: b058
|
|
Create Date: 2026-07-07
|
|
"""
|
|
|
|
from alembic import op
|
|
|
|
revision = "b059"
|
|
down_revision = "b058"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
_TABLES = ("tests", "campaigns", "evidences")
|
|
|
|
_RENAMES_UP = {
|
|
"public": "public_release",
|
|
"internal": "general_use",
|
|
"sensitive": "confidential",
|
|
}
|
|
|
|
_RENAMES_DOWN = {v: k for k, v in _RENAMES_UP.items()}
|
|
|
|
|
|
def upgrade() -> None:
|
|
for table in _TABLES:
|
|
for old, new in _RENAMES_UP.items():
|
|
op.execute(f"UPDATE {table} SET data_classification = '{new}' WHERE data_classification = '{old}'")
|
|
op.alter_column(table, "data_classification", server_default="confidential")
|
|
|
|
|
|
def downgrade() -> None:
|
|
for table in _TABLES:
|
|
for new, old in _RENAMES_DOWN.items():
|
|
op.execute(f"UPDATE {table} SET data_classification = '{old}' WHERE data_classification = '{new}'")
|
|
op.alter_column(table, "data_classification", server_default="internal")
|