31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
|
"""SBOM ecosystem enum expansion: add terraform, ansible, tool
|
||
|
|
|
||
|
|
Revision ID: d6e7f8a9b0c1
|
||
|
|
Revises: c5d6e7f8a9b0
|
||
|
|
Create Date: 2026-03-12 00:00:00.000000
|
||
|
|
"""
|
||
|
|
from typing import Sequence, Union
|
||
|
|
|
||
|
|
from alembic import op
|
||
|
|
|
||
|
|
revision: str = "d6e7f8a9b0c1"
|
||
|
|
down_revision: Union[str, None] = "c5d6e7f8a9b0"
|
||
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
||
|
|
depends_on: Union[str, Sequence[str], None] = None
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade() -> None:
|
||
|
|
# PostgreSQL requires each ADD VALUE in its own statement and cannot be
|
||
|
|
# run inside a transaction that also modifies data. ADD VALUE is
|
||
|
|
# transactional in PG 12+ (no COMMIT needed).
|
||
|
|
op.execute("ALTER TYPE ecosystem ADD VALUE IF NOT EXISTS 'terraform'")
|
||
|
|
op.execute("ALTER TYPE ecosystem ADD VALUE IF NOT EXISTS 'ansible'")
|
||
|
|
op.execute("ALTER TYPE ecosystem ADD VALUE IF NOT EXISTS 'tool'")
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade() -> None:
|
||
|
|
# PostgreSQL does not support removing enum values without recreating the
|
||
|
|
# type. Document the limitation and do nothing — reverting this migration
|
||
|
|
# requires a full type recreation if needed.
|
||
|
|
pass
|