Introduce product details table
This commit is contained in:
112
alembic/versions/3b61a6fbc6b2_introduce_product_details.py
Normal file
112
alembic/versions/3b61a6fbc6b2_introduce_product_details.py
Normal file
@@ -0,0 +1,112 @@
|
||||
"""introduce product details
|
||||
|
||||
Revision ID: 3b61a6fbc6b2
|
||||
Revises: 958d7aee2b21
|
||||
Create Date: 2026-04-22 11:56:31.223186
|
||||
|
||||
"""
|
||||
|
||||
import datetime
|
||||
from typing import Sequence, Union
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision: str = "3b61a6fbc6b2"
|
||||
down_revision: Union[str, Sequence[str], None] = "958d7aee2b21"
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
"""Upgrade schema."""
|
||||
# ### commands auto generated by Alembic - please adjust! ###
|
||||
op.create_table(
|
||||
"aps_product_detail",
|
||||
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
|
||||
sa.Column("name", sa.String(), nullable=False),
|
||||
sa.Column("product_id", sa.Integer(), nullable=False),
|
||||
sa.Column("description", sa.String(), nullable=True),
|
||||
sa.Column("price", sa.Numeric(precision=10, scale=2), nullable=False),
|
||||
sa.Column(
|
||||
"unit_of_measure",
|
||||
sa.Enum("g", "kg", "l", "piece", native_enum=False),
|
||||
nullable=False,
|
||||
),
|
||||
sa.Column("allow_fractional", sa.Boolean(), nullable=False),
|
||||
sa.Column("temporarily_unavailable", sa.Boolean(), nullable=False),
|
||||
sa.Column("vat_rate", sa.Numeric(precision=10, scale=2), nullable=False),
|
||||
sa.Column("valid_from", sa.DateTime(), nullable=False),
|
||||
sa.ForeignKeyConstraint(
|
||||
["product_id"],
|
||||
["aps_product.id"],
|
||||
),
|
||||
sa.PrimaryKeyConstraint("id"),
|
||||
sa.UniqueConstraint("name"),
|
||||
)
|
||||
|
||||
# Step 2: Copy data from aps_product to aps_product_detail
|
||||
# Bind the current connection to a temporary session
|
||||
conn = op.get_bind()
|
||||
|
||||
# Fetch all products
|
||||
products = conn.execute(
|
||||
sa.text(
|
||||
"SELECT id, name, price, unit_of_measure, allow_fractional, vat_rate FROM aps_product"
|
||||
)
|
||||
).fetchall()
|
||||
|
||||
# Insert data into aps_product_detail
|
||||
for product in products:
|
||||
conn.execute(
|
||||
sa.text(
|
||||
"""
|
||||
INSERT INTO aps_product_detail
|
||||
(name, product_id, description, price, unit_of_measure, allow_fractional, temporarily_unavailable, vat_rate, valid_from)
|
||||
VALUES (:name, :product_id, :description, :price, :unit_of_measure, :allow_fractional, :temporarily_unavailable, :vat_rate, :valid_from)
|
||||
"""
|
||||
),
|
||||
{
|
||||
"name": product.name,
|
||||
"product_id": product.id,
|
||||
"description": "",
|
||||
"price": product.price,
|
||||
"unit_of_measure": product.unit_of_measure,
|
||||
"allow_fractional": product.allow_fractional,
|
||||
"temporarily_unavailable": False,
|
||||
"vat_rate": product.vat_rate,
|
||||
"valid_from": datetime.datetime(1970, 1, 1, 0, 0, 0),
|
||||
},
|
||||
)
|
||||
with op.batch_alter_table("aps_product") as batch_op:
|
||||
batch_op.drop_column("vat_rate")
|
||||
batch_op.drop_column("name")
|
||||
|
||||
batch_op.drop_column("allow_fractional")
|
||||
batch_op.drop_column("unit_of_measure")
|
||||
batch_op.drop_column("price")
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
"""Downgrade schema."""
|
||||
# ### commands auto generated by Alembic - please adjust! ###
|
||||
op.add_column(
|
||||
"aps_product",
|
||||
sa.Column("price", sa.NUMERIC(precision=10, scale=2), nullable=False),
|
||||
)
|
||||
op.add_column(
|
||||
"aps_product",
|
||||
sa.Column("unit_of_measure", sa.VARCHAR(length=5), nullable=False),
|
||||
)
|
||||
op.add_column(
|
||||
"aps_product", sa.Column("allow_fractional", sa.BOOLEAN(), nullable=False)
|
||||
)
|
||||
op.add_column(
|
||||
"aps_product",
|
||||
sa.Column("vat_rate", sa.NUMERIC(precision=10, scale=2), nullable=False),
|
||||
)
|
||||
op.add_column("aps_product", sa.Column("name", sa.VARCHAR(), nullable=False))
|
||||
op.drop_table("aps_product_detail")
|
||||
# ### end Alembic commands ###
|
||||
Reference in New Issue
Block a user