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>
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
"""
|
|
Migration: Add budget settings columns to kitchen_settings table
|
|
|
|
Adds:
|
|
- forecast_api_url: URL for external forecasting API
|
|
- forecast_api_key: API key for forecast authentication
|
|
- budget_gp_target: Target GP percentage (default 65%)
|
|
- budget_lookback_weeks: Number of weeks for supplier % calculation (default 4)
|
|
"""
|
|
import asyncio
|
|
from sqlalchemy import text
|
|
from database import engine
|
|
|
|
|
|
async def migrate():
|
|
async with engine.begin() as conn:
|
|
# Add forecast_api_url column
|
|
try:
|
|
await conn.execute(text(
|
|
"""
|
|
ALTER TABLE kitchen_settings
|
|
ADD COLUMN IF NOT EXISTS forecast_api_url VARCHAR(500)
|
|
"""
|
|
))
|
|
print("+ Added forecast_api_url column")
|
|
except Exception as e:
|
|
if "already exists" in str(e).lower() or "duplicate column" in str(e).lower():
|
|
print("- forecast_api_url column already exists, skipping")
|
|
else:
|
|
raise
|
|
|
|
# Add forecast_api_key column
|
|
try:
|
|
await conn.execute(text(
|
|
"""
|
|
ALTER TABLE kitchen_settings
|
|
ADD COLUMN IF NOT EXISTS forecast_api_key VARCHAR(500)
|
|
"""
|
|
))
|
|
print("+ Added forecast_api_key column")
|
|
except Exception as e:
|
|
if "already exists" in str(e).lower() or "duplicate column" in str(e).lower():
|
|
print("- forecast_api_key column already exists, skipping")
|
|
else:
|
|
raise
|
|
|
|
# Add budget_gp_target column
|
|
try:
|
|
await conn.execute(text(
|
|
"""
|
|
ALTER TABLE kitchen_settings
|
|
ADD COLUMN IF NOT EXISTS budget_gp_target NUMERIC(5,2) DEFAULT 65.00
|
|
"""
|
|
))
|
|
print("+ Added budget_gp_target column")
|
|
except Exception as e:
|
|
if "already exists" in str(e).lower() or "duplicate column" in str(e).lower():
|
|
print("- budget_gp_target column already exists, skipping")
|
|
else:
|
|
raise
|
|
|
|
# Add budget_lookback_weeks column
|
|
try:
|
|
await conn.execute(text(
|
|
"""
|
|
ALTER TABLE kitchen_settings
|
|
ADD COLUMN IF NOT EXISTS budget_lookback_weeks INTEGER DEFAULT 4
|
|
"""
|
|
))
|
|
print("+ Added budget_lookback_weeks column")
|
|
except Exception as e:
|
|
if "already exists" in str(e).lower() or "duplicate column" in str(e).lower():
|
|
print("- budget_lookback_weeks column already exists, skipping")
|
|
else:
|
|
raise
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("Running migration: add_budget_settings")
|
|
asyncio.run(migrate())
|
|
print("Migration complete!")
|