jtricerolph
bc874ac9ff
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>
22 lines
789 B
Python
22 lines
789 B
Python
"""Add unit column to recipe_ingredients table for display unit override."""
|
|
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 = 'recipe_ingredients' AND column_name = 'unit'"
|
|
))
|
|
if result.scalar_one_or_none():
|
|
logger.info("recipe_ingredients.unit already exists, skipping")
|
|
return
|
|
|
|
await conn.execute(text(
|
|
"ALTER TABLE recipe_ingredients ADD COLUMN IF NOT EXISTS unit VARCHAR(10) DEFAULT NULL"
|
|
))
|
|
logger.info("Added unit column to recipe_ingredients")
|