x

src/database/init_db.py

@@ -1,6 +1,7 @@
 import asyncio
 import sys
 from pathlib import Path
+from sqlalchemy import text
 
 project_root = Path(__file__).parent.parent.parent
 src_root = Path(__file__).parent.parent
@@ -122,6 +123,7 @@ async def init_database():
     try:
         await db_manager.connect()
         await db_manager.create_tables()
+        await ensure_schema_updates()
         
         async with db_manager.session() as session:
             perm_map = await init_permissions(session)
@@ -156,5 +158,30 @@ async def init_database():
         await db_manager.disconnect()
 
 
+async def ensure_schema_updates():
+    async with db_manager.engine.begin() as conn:
+        await conn.execute(text("ALTER TABLE products ADD COLUMN IF NOT EXISTS item_type VARCHAR(20) DEFAULT 'finished'"))
+        await conn.execute(text("UPDATE products SET item_type = 'finished' WHERE item_type IS NULL"))
+        await conn.execute(text("ALTER TABLE sales_orders ADD COLUMN IF NOT EXISTS production_status VARCHAR(20) DEFAULT 'not_started'"))
+        await conn.execute(text("ALTER TABLE sales_orders ADD COLUMN IF NOT EXISTS production_no VARCHAR(50)"))
+        await conn.execute(text("ALTER TABLE sales_orders ADD COLUMN IF NOT EXISTS planned_material_cost DOUBLE PRECISION DEFAULT 0"))
+        await conn.execute(text("ALTER TABLE sales_orders ADD COLUMN IF NOT EXISTS actual_material_cost DOUBLE PRECISION DEFAULT 0"))
+        await conn.execute(text("""
+            CREATE TABLE IF NOT EXISTS product_materials (
+                id SERIAL PRIMARY KEY,
+                finished_product_id INTEGER NOT NULL REFERENCES products(id),
+                material_product_id INTEGER NOT NULL REFERENCES products(id),
+                quantity DOUBLE PRECISION NOT NULL,
+                loss_rate DOUBLE PRECISION DEFAULT 0,
+                created_at TIMESTAMP DEFAULT NOW(),
+                updated_at TIMESTAMP DEFAULT NOW()
+            )
+        """))
+        await conn.execute(text("""
+            CREATE UNIQUE INDEX IF NOT EXISTS uq_product_material_unique
+            ON product_materials (finished_product_id, material_product_id)
+        """))
+
+
 if __name__ == "__main__":
     asyncio.run(init_database())