x

src/database/init_db.py

@@ -197,6 +197,59 @@ async def ensure_schema_updates():
             CREATE INDEX IF NOT EXISTS idx_mold_cavity_is_fallback
             ON mold_cavity_data (is_fallback)
         """))
+        await conn.execute(text("""
+            DO $$
+            BEGIN
+                IF NOT EXISTS (
+                    SELECT 1 FROM pg_constraint
+                    WHERE conname = 'uq_inventory_product_warehouse'
+                ) THEN
+                    ALTER TABLE inventory
+                    ADD CONSTRAINT uq_inventory_product_warehouse UNIQUE (product_id, warehouse_id);
+                END IF;
+            END $$;
+        """))
+        await conn.execute(text("""
+            DO $$
+            BEGIN
+                IF NOT EXISTS (
+                    SELECT 1 FROM pg_constraint
+                    WHERE conname = 'ck_inventory_qty_nonnegative'
+                ) THEN
+                    ALTER TABLE inventory
+                    ADD CONSTRAINT ck_inventory_qty_nonnegative
+                    CHECK (quantity >= 0 AND locked_quantity >= 0 AND locked_quantity <= quantity);
+                END IF;
+            END $$;
+        """))
+        await conn.execute(text("""
+            DO $$
+            BEGIN
+                IF NOT EXISTS (
+                    SELECT 1 FROM pg_constraint
+                    WHERE conname = 'ck_purchase_order_items_qty'
+                ) THEN
+                    ALTER TABLE purchase_order_items
+                    ADD CONSTRAINT ck_purchase_order_items_qty
+                    CHECK (quantity > 0 AND received_quantity >= 0 AND received_quantity <= quantity);
+                END IF;
+            END $$;
+        """))
+        await conn.execute(text("""
+            DO $$
+            BEGIN
+                IF NOT EXISTS (
+                    SELECT 1 FROM pg_constraint
+                    WHERE conname = 'ck_sales_order_items_qty'
+                ) THEN
+                    ALTER TABLE sales_order_items
+                    ADD CONSTRAINT ck_sales_order_items_qty
+                    CHECK (quantity > 0 AND delivered_quantity >= 0 AND delivered_quantity <= quantity);
+                END IF;
+            END $$;
+        """))
+        await conn.execute(text("ALTER TABLE sales_orders ADD COLUMN IF NOT EXISTS manufacturing_date TIMESTAMP WITHOUT TIME ZONE"))
+        await conn.execute(text("ALTER TABLE sales_orders ALTER COLUMN manufacturing_date TYPE TIMESTAMP WITHOUT TIME ZONE"))
 
 
 if __name__ == "__main__":