"""add augmented_parts

Revision ID: 8bc411b5814e
Revises: 0c3d1a5cff1f
Create Date: 2026-07-02 12:37:59.457033
"""

from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision: str = "8bc411b5814e"
down_revision: Union[str, None] = "0c3d1a5cff1f"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        "augmented_parts",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("sku", sa.String(), nullable=True),
        sa.Column("description", sa.String(), nullable=False),
        sa.Column("colour_id", sa.Integer(), nullable=False),
        sa.Column("manufacturer_id", sa.Integer(), nullable=False),
        sa.Column("scope_model_id", sa.Integer(), nullable=True),
        sa.Column("part_type_id", sa.Integer(), nullable=False),
        sa.Column("sub_part_type_id", sa.Integer(), nullable=True),
        sa.ForeignKeyConstraint(
            ["colour_id"], ["part_colours.id"], ondelete="RESTRICT"
        ),
        sa.ForeignKeyConstraint(
            ["manufacturer_id"], ["manufacturers.id"], ondelete="RESTRICT"
        ),
        sa.ForeignKeyConstraint(
            ["scope_model_id"], ["scope_models.id"], ondelete="SET NULL"
        ),
        sa.ForeignKeyConstraint(
            ["part_type_id"], ["part_types.id"], ondelete="RESTRICT"
        ),
        sa.ForeignKeyConstraint(
            ["sub_part_type_id"], ["sub_part_types.id"], ondelete="SET NULL"
        ),
        sa.PrimaryKeyConstraint("id"),
    )


def downgrade() -> None:
    op.drop_table("augmented_parts")