163 lines
6.4 KiB
Python
163 lines
6.4 KiB
Python
"""Initial database structure
|
|
|
|
Revision ID: 0001
|
|
Revises:
|
|
Create Date: 2023-11-18
|
|
|
|
"""
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "0001"
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Create users table
|
|
op.create_table(
|
|
"users",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("email", sa.String(), nullable=False),
|
|
sa.Column("hashed_password", sa.String(), nullable=False),
|
|
sa.Column("full_name", sa.String(), nullable=True),
|
|
sa.Column("is_active", sa.Boolean(), nullable=True, default=True),
|
|
sa.Column("is_superuser", sa.Boolean(), nullable=True, default=False),
|
|
sa.Column(
|
|
"created_at", sa.DateTime(timezone=True), server_default=sa.text("(CURRENT_TIMESTAMP)")
|
|
),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_users_email"), "users", ["email"], unique=True)
|
|
op.create_index(op.f("ix_users_id"), "users", ["id"], unique=False)
|
|
|
|
# Create categories table
|
|
op.create_table(
|
|
"categories",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("description", sa.Text(), nullable=True),
|
|
sa.Column(
|
|
"created_at", sa.DateTime(timezone=True), server_default=sa.text("(CURRENT_TIMESTAMP)")
|
|
),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_categories_id"), "categories", ["id"], unique=False)
|
|
op.create_index(op.f("ix_categories_name"), "categories", ["name"], unique=True)
|
|
|
|
# Create suppliers table
|
|
op.create_table(
|
|
"suppliers",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("contact_name", sa.String(), nullable=True),
|
|
sa.Column("email", sa.String(), nullable=True),
|
|
sa.Column("phone", sa.String(), nullable=True),
|
|
sa.Column("address", sa.Text(), nullable=True),
|
|
sa.Column(
|
|
"created_at", sa.DateTime(timezone=True), server_default=sa.text("(CURRENT_TIMESTAMP)")
|
|
),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_suppliers_id"), "suppliers", ["id"], unique=False)
|
|
op.create_index(op.f("ix_suppliers_name"), "suppliers", ["name"], unique=False)
|
|
|
|
# Create products table
|
|
op.create_table(
|
|
"products",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("name", sa.String(), nullable=False),
|
|
sa.Column("description", sa.Text(), nullable=True),
|
|
sa.Column("sku", sa.String(), nullable=True),
|
|
sa.Column("barcode", sa.String(), nullable=True),
|
|
sa.Column("price", sa.Float(), nullable=False, default=0.0),
|
|
sa.Column("cost_price", sa.Float(), nullable=True),
|
|
sa.Column("current_stock", sa.Integer(), nullable=False, default=0),
|
|
sa.Column("min_stock_level", sa.Integer(), nullable=True),
|
|
sa.Column("max_stock_level", sa.Integer(), nullable=True),
|
|
sa.Column("is_active", sa.Boolean(), nullable=True, default=True),
|
|
sa.Column("category_id", sa.String(), nullable=True),
|
|
sa.Column("supplier_id", sa.String(), nullable=True),
|
|
sa.Column(
|
|
"created_at", sa.DateTime(timezone=True), server_default=sa.text("(CURRENT_TIMESTAMP)")
|
|
),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(
|
|
["category_id"],
|
|
["categories.id"],
|
|
),
|
|
sa.ForeignKeyConstraint(
|
|
["supplier_id"],
|
|
["suppliers.id"],
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_products_barcode"), "products", ["barcode"], unique=True)
|
|
op.create_index(op.f("ix_products_id"), "products", ["id"], unique=False)
|
|
op.create_index(op.f("ix_products_name"), "products", ["name"], unique=False)
|
|
op.create_index(op.f("ix_products_sku"), "products", ["sku"], unique=True)
|
|
|
|
# Create inventory_movements table
|
|
op.create_table(
|
|
"inventory_movements",
|
|
sa.Column("id", sa.String(), nullable=False),
|
|
sa.Column("product_id", sa.String(), nullable=False),
|
|
sa.Column("quantity", sa.Integer(), nullable=False),
|
|
sa.Column(
|
|
"type",
|
|
sa.Enum("STOCK_IN", "STOCK_OUT", "ADJUSTMENT", "RETURN", name="movementtype"),
|
|
nullable=False,
|
|
),
|
|
sa.Column("reference", sa.String(), nullable=True),
|
|
sa.Column("notes", sa.Text(), nullable=True),
|
|
sa.Column("unit_price", sa.Float(), nullable=True),
|
|
sa.Column("created_by", sa.String(), nullable=True),
|
|
sa.Column(
|
|
"created_at", sa.DateTime(timezone=True), server_default=sa.text("(CURRENT_TIMESTAMP)")
|
|
),
|
|
sa.ForeignKeyConstraint(
|
|
["created_by"],
|
|
["users.id"],
|
|
),
|
|
sa.ForeignKeyConstraint(
|
|
["product_id"],
|
|
["products.id"],
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_inventory_movements_id"), "inventory_movements", ["id"], unique=False)
|
|
|
|
|
|
def downgrade() -> None:
|
|
# Drop inventory_movements table
|
|
op.drop_index(op.f("ix_inventory_movements_id"), table_name="inventory_movements")
|
|
op.drop_table("inventory_movements")
|
|
|
|
# Drop products table
|
|
op.drop_index(op.f("ix_products_sku"), table_name="products")
|
|
op.drop_index(op.f("ix_products_name"), table_name="products")
|
|
op.drop_index(op.f("ix_products_id"), table_name="products")
|
|
op.drop_index(op.f("ix_products_barcode"), table_name="products")
|
|
op.drop_table("products")
|
|
|
|
# Drop suppliers table
|
|
op.drop_index(op.f("ix_suppliers_name"), table_name="suppliers")
|
|
op.drop_index(op.f("ix_suppliers_id"), table_name="suppliers")
|
|
op.drop_table("suppliers")
|
|
|
|
# Drop categories table
|
|
op.drop_index(op.f("ix_categories_name"), table_name="categories")
|
|
op.drop_index(op.f("ix_categories_id"), table_name="categories")
|
|
op.drop_table("categories")
|
|
|
|
# Drop users table
|
|
op.drop_index(op.f("ix_users_id"), table_name="users")
|
|
op.drop_index(op.f("ix_users_email"), table_name="users")
|
|
op.drop_table("users")
|