#!/usr/bin/env python3
"""
Migración v4: bultos y numero_documento en notas.
Ejecutar: python3 bascunan_db_migrate_v4.py /home/embalado/bascunan.db
"""
import sqlite3, sys

DB_PATH = sys.argv[1] if len(sys.argv) > 1 else "bascunan.db"

MIGRATIONS = [
    "ALTER TABLE estados_notas ADD COLUMN bultos INTEGER DEFAULT 1",
    "ALTER TABLE estados_notas ADD COLUMN numero_documento TEXT",
    "ALTER TABLE estados_notas ADD COLUMN nombre_cliente TEXT",
    "ALTER TABLE estados_notas ADD COLUMN total_nota REAL",
]

conn = sqlite3.connect(DB_PATH)
for sql in MIGRATIONS:
    try:
        conn.execute(sql)
        col = sql.split("ADD COLUMN ")[1].split()[0]
        print(f"  + {col}")
    except sqlite3.OperationalError as e:
        print(f"  = ya existe: {e}")
conn.commit()
conn.close()
print("✓ Migración v4 completa.")