tonym
8681f7aee7
Some checks failed
CI / style (push) Has been cancelled
CI / build-client (push) Has been cancelled
CI / build-amd64 (push) Has been cancelled
CI / build-tester (push) Has been cancelled
CI / test (cockroachdb) (push) Has been cancelled
CI / test (mariadb) (push) Has been cancelled
CI / test (postgres) (push) Has been cancelled
CI / test (sqlite) (push) Has been cancelled
CI / build-arm64 (push) Has been cancelled
CI / build-armv7 (push) Has been cancelled
CI / publish-images (push) Has been cancelled
CI / publish-release (push) Has been cancelled
Issue Manager / issue-manager (push) Has been cancelled
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
"""Add hierarchical locations.
|
|
|
|
Revision ID: e5f6g7h8i9j0
|
|
Revises: d4e5f6g7h8i9
|
|
Create Date: 2025-01-21 01:00:00.000000
|
|
|
|
"""
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "e5f6g7h8i9j0"
|
|
down_revision = "d4e5f6g7h8i9"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Create location table and add location_id to spool."""
|
|
# Create location table
|
|
op.create_table(
|
|
"location",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("name", sa.String(length=64), nullable=False),
|
|
sa.Column("parent_id", sa.Integer(), nullable=True),
|
|
sa.Column("location_type", sa.String(length=32), nullable=True),
|
|
sa.Column("description", sa.String(length=256), nullable=True),
|
|
sa.ForeignKeyConstraint(
|
|
["parent_id"],
|
|
["location.id"],
|
|
ondelete="SET NULL",
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_location_id"), "location", ["id"], unique=False)
|
|
op.create_index(op.f("ix_location_parent_id"), "location", ["parent_id"], unique=False)
|
|
|
|
# Add location_id column to spool table (SQLite compatible - no FK constraint)
|
|
op.add_column(
|
|
"spool",
|
|
sa.Column("location_id", sa.Integer(), nullable=True),
|
|
)
|
|
op.create_index(op.f("ix_spool_location_id"), "spool", ["location_id"], unique=False)
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Drop location table and remove location_id from spool."""
|
|
op.drop_index(op.f("ix_spool_location_id"), table_name="spool")
|
|
op.drop_column("spool", "location_id")
|
|
op.drop_index(op.f("ix_location_parent_id"), table_name="location")
|
|
op.drop_index(op.f("ix_location_id"), table_name="location")
|
|
op.drop_table("location")
|