Agent Zero
134 lines
5.2 KiB
Python
134 lines
5.2 KiB
Python
"""create_equipment_stocklocation_equipmentgroup_tables
|
|
|
|
Revision ID: da13d44bd483
|
|
Revises: a5b1e93690ff
|
|
Create Date: 2026-05-31 11:31:54.796382
|
|
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "da13d44bd483"
|
|
down_revision: Union[str, None] = "a5b1e93690ff"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table(
|
|
"stock_locations",
|
|
sa.Column("id", sa.String(length=36), nullable=False),
|
|
sa.Column("account_id", sa.String(length=36), nullable=False),
|
|
sa.Column("name", sa.String(length=255), nullable=False),
|
|
sa.Column("address", sa.String(length=500), nullable=True),
|
|
sa.Column("is_default", sa.Boolean(), nullable=False),
|
|
sa.Column(
|
|
"created_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("(CURRENT_TIMESTAMP)"),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("(CURRENT_TIMESTAMP)"),
|
|
nullable=False,
|
|
),
|
|
sa.ForeignKeyConstraint(["account_id"], ["accounts.id"], ondelete="CASCADE"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_table(
|
|
"equipment",
|
|
sa.Column("id", sa.String(length=36), nullable=False),
|
|
sa.Column("account_id", sa.String(length=36), nullable=False),
|
|
sa.Column("name", sa.String(length=255), nullable=False),
|
|
sa.Column("category", sa.String(length=100), nullable=True),
|
|
sa.Column("brand", sa.String(length=255), nullable=True),
|
|
sa.Column("serial_number", sa.String(length=100), nullable=True),
|
|
sa.Column("barcode", sa.String(length=100), nullable=True),
|
|
sa.Column("qr_code", sa.String(length=500), nullable=True),
|
|
sa.Column("status", sa.String(length=20), nullable=False),
|
|
sa.Column("purchase_price", sa.Float(), nullable=True),
|
|
sa.Column("current_value", sa.Float(), nullable=True),
|
|
sa.Column("weight_kg", sa.Float(), nullable=True),
|
|
sa.Column("dimensions", sa.String(length=255), nullable=True),
|
|
sa.Column("power_watt", sa.Float(), nullable=True),
|
|
sa.Column("notes", sa.Text(), nullable=True),
|
|
sa.Column("custom_fields", sa.Text(), nullable=True),
|
|
sa.Column("location_id", sa.String(length=36), nullable=True),
|
|
sa.Column("supplier_id", sa.String(length=36), nullable=True),
|
|
sa.Column(
|
|
"created_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("(CURRENT_TIMESTAMP)"),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("(CURRENT_TIMESTAMP)"),
|
|
nullable=False,
|
|
),
|
|
sa.ForeignKeyConstraint(["account_id"], ["accounts.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(
|
|
["location_id"], ["stock_locations.id"], ondelete="SET NULL"
|
|
),
|
|
sa.ForeignKeyConstraint(["supplier_id"], ["contacts.id"], ondelete="SET NULL"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("barcode"),
|
|
sa.UniqueConstraint("serial_number"),
|
|
)
|
|
op.create_table(
|
|
"equipment_groups",
|
|
sa.Column("id", sa.String(length=36), nullable=False),
|
|
sa.Column("account_id", sa.String(length=36), nullable=False),
|
|
sa.Column("name", sa.String(length=255), nullable=False),
|
|
sa.Column("description", sa.Text(), nullable=True),
|
|
sa.Column("daily_rate", sa.Float(), nullable=True),
|
|
sa.Column("default_location_id", sa.String(length=36), nullable=True),
|
|
sa.Column(
|
|
"created_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("(CURRENT_TIMESTAMP)"),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("(CURRENT_TIMESTAMP)"),
|
|
nullable=False,
|
|
),
|
|
sa.ForeignKeyConstraint(["account_id"], ["accounts.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(
|
|
["default_location_id"], ["stock_locations.id"], ondelete="SET NULL"
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_table(
|
|
"equipment_group_items",
|
|
sa.Column("group_id", sa.String(length=36), nullable=False),
|
|
sa.Column("equipment_id", sa.String(length=36), nullable=False),
|
|
sa.Column("quantity", sa.Float(), nullable=False),
|
|
sa.ForeignKeyConstraint(["equipment_id"], ["equipment.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(
|
|
["group_id"], ["equipment_groups.id"], ondelete="CASCADE"
|
|
),
|
|
sa.PrimaryKeyConstraint("group_id", "equipment_id"),
|
|
)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_table("equipment_group_items")
|
|
op.drop_table("equipment_groups")
|
|
op.drop_table("equipment")
|
|
op.drop_table("stock_locations")
|
|
# ### end Alembic commands ###
|