kitchen/backend/migrations/add_gross_sell_price.py
jtricerolph bc874ac9ff Fix migration transaction poisoning: ADD COLUMN IF NOT EXISTS
Migrations used try/except around ADD COLUMN inside a single engine.begin()
block. When a 'column already exists' error was caught, asyncpg left the
transaction in aborted state, causing all subsequent DDL in the block to fail
with InFailedSQLTransactionError. Replace with IF NOT EXISTS to prevent the
error entirely.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-07-12 19:53:26 +00:00

22 lines
767 B
Python

"""Add gross_sell_price column to recipes table."""
import logging
from sqlalchemy import text
from database import engine
logger = logging.getLogger(__name__)
async def migrate():
async with engine.begin() as conn:
result = await conn.execute(text(
"SELECT column_name FROM information_schema.columns "
"WHERE table_name = 'recipes' AND column_name = 'gross_sell_price'"
))
if result.scalar_one_or_none():
logger.info("recipes.gross_sell_price already exists, skipping")
return
await conn.execute(text(
"ALTER TABLE recipes ADD COLUMN IF NOT EXISTS gross_sell_price NUMERIC(10,2) DEFAULT NULL"
))
logger.info("Added recipes.gross_sell_price column")