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")