Autoparts-DB/pos/services/inventory_vehicle_compat.py

"""Inventory Vehicle Compatibility Engine.

Links local inventory items to model_year_engine_ids so they appear
when browsing the Local catalog by vehicle.

Features:
- Auto-match by part_number against parts/aftermarket_parts/cross_refs
- Manual add/remove MYE compatibility
- Batch operations
"""

from typing import List, Dict, Optional


def get_compat_source(tenant_id):
    """Return the configured compatibility source: 'tecdoc', 'qwen', or 'both'.

    Reads from tenant_config table. Defaults to 'both'.
    """
    from tenant_db import get_tenant_conn
    try:
        conn = get_tenant_conn(tenant_id)
        cur = conn.cursor()
        cur.execute(
            "SELECT value FROM tenant_config WHERE key = 'vehicle_compat_source'"
        )
        row = cur.fetchone()
        cur.close()
        conn.close()
        source = row[0] if row else 'both'
        if source in ('tecdoc', 'qwen', 'both'):
            return source
    except Exception:
        pass
    return 'both'


def _norm_pn(pn):
    """Normalize a part number: uppercase, strip spaces and separators."""
    if not pn:
        return ''
    return (str(pn).replace(' ', '').replace('-', '').replace('/', '')
            .replace('.', '').replace('_', '').upper())


def auto_match_vehicle_compatibility(master_conn, tenant_conn, inventory_id, part_number,
                                     brand=None, name=None):
    """Find vehicle compatibility for an inventory item via the supplier catalog.

    Legacy note: the original implementation queried the removed TecDoc tables
    (parts / aftermarket_parts / part_cross_references / vehicle_parts) and
    always failed. This version uses the LIVE master tables:
      1. supplier_catalog.sku exact-normalized
      2. supplier_catalog_interchange.part_number exact-normalized (cross-refs)
      3. prefix fallback on catalog SKU (only for >=5 char numbers, few matches)
    then collects DISTINCT model_year_engine_id from supplier_catalog_compat
    and inserts them into the tenant inventory_vehicle_compat table.

    Returns dict {'matched': bool, 'matches': [...], 'myes': [...], 'inserted': int}
    (same shape as the legacy implementation).
    """
    cur = master_conn.cursor()
    clean_pn = _norm_pn(part_number)
    catalog_ids = set()

    if clean_pn:
        # 1. Exact match on catalog SKU (with and without inner spaces)
        cur.execute("""
            SELECT id FROM supplier_catalog
            WHERE REPLACE(UPPER(sku), '-', '') = %s
               OR REPLACE(UPPER(REPLACE(sku, ' ', '')), '-', '') = %s
            LIMIT 50
        """, (clean_pn, clean_pn))
        for r in cur.fetchall():
            catalog_ids.add(r[0])

        # 2. Exact match on interchange part numbers (cross references)
        if not catalog_ids:
            cur.execute("""
                SELECT DISTINCT catalog_id FROM supplier_catalog_interchange
                WHERE REPLACE(UPPER(REPLACE(part_number, '-', '')), ' ', '') = %s
                LIMIT 50
            """, (clean_pn,))
            for r in cur.fetchall():
                catalog_ids.add(r[0])

        # 3. Prefix fallback for longer numbers when exact found little
        if len(clean_pn) >= 5 and len(catalog_ids) < 3:
            cur.execute("""
                SELECT id FROM supplier_catalog
                WHERE REPLACE(UPPER(sku), '-', '') LIKE %s
                LIMIT 10
            """, (clean_pn[:8] + '%',))
            for r in cur.fetchall():
                catalog_ids.add(r[0])

    mye_ids = []
    if catalog_ids:
        cur.execute("""
            SELECT DISTINCT model_year_engine_id FROM supplier_catalog_compat
            WHERE catalog_id = ANY(%s) AND model_year_engine_id IS NOT NULL
            LIMIT 500
        """, (list(catalog_ids),))
        mye_ids = [r[0] for r in cur.fetchall()]
    cur.close()

    inserted = 0
    if mye_ids:
        cur2 = tenant_conn.cursor()
        for mye_id in mye_ids:
            cur2.execute("""
                INSERT INTO inventory_vehicle_compat
                    (inventory_id, model_year_engine_id, source, confidence)
                VALUES (%s, %s, 'supplier_catalog', 0.95)
                ON CONFLICT (inventory_id, model_year_engine_id) DO NOTHING
            """, (inventory_id, mye_id))
            if cur2.rowcount > 0:
                inserted += 1
        tenant_conn.commit()
        cur2.close()

    return {
        'matched': len(mye_ids) > 0,
        'matches': sorted(catalog_ids),
        'myes': mye_ids,
        'inserted': inserted,
    }


def add_compatibility(tenant_conn, inventory_id, model_year_engine_id, source='manual'):
    """Manually add a vehicle compatibility."""
    cur = tenant_conn.cursor()
    cur.execute("""
        INSERT INTO inventory_vehicle_compat
            (inventory_id, model_year_engine_id, source, confidence)
        VALUES (%s, %s, %s, 1.0)
        ON CONFLICT (inventory_id, model_year_engine_id) DO NOTHING
        RETURNING id
    """, (inventory_id, model_year_engine_id, source))
    row = cur.fetchone()
    tenant_conn.commit()
    cur.close()
    return row[0] if row else None


def remove_compatibility(tenant_conn, inventory_id, model_year_engine_id):
    cur = tenant_conn.cursor()
    cur.execute("""
        DELETE FROM inventory_vehicle_compat
        WHERE inventory_id = %s AND model_year_engine_id = %s
    """, (inventory_id, model_year_engine_id))
    deleted = cur.rowcount
    tenant_conn.commit()
    cur.close()
    return deleted


def remove_compatibility_by_id(tenant_conn, compat_id):
    """Remove a compatibility by its primary key (works for both MYE-linked
    and text-only QWEN records)."""
    cur = tenant_conn.cursor()
    cur.execute("""
        DELETE FROM inventory_vehicle_compat
        WHERE id = %s
    """, (compat_id,))
    deleted = cur.rowcount
    tenant_conn.commit()
    cur.close()
    return deleted


def remove_all_compatibility(tenant_conn, inventory_id):
    cur = tenant_conn.cursor()
    cur.execute("""
        DELETE FROM inventory_vehicle_compat WHERE inventory_id = %s
    """, (inventory_id,))
    deleted = cur.rowcount
    tenant_conn.commit()
    cur.close()
    return deleted


def get_compatibility(tenant_conn, master_conn, inventory_id):
    """Get all MYE compatibilities for an inventory item.

    Queries inventory_vehicle_compat from the tenant DB, then resolves
    vehicle details (brand/model/year/engine) from the master DB.

    Vehicles with model_year_engine_id IS NULL are text-only QWEN records
    (master DB lacks the vehicle) and are returned using their stored text.
    """
    # 1. Get all rows from tenant
    cur_t = tenant_conn.cursor()
    cur_t.execute("""
        SELECT id, model_year_engine_id, make, model, year, engine, engine_code,
               source, confidence, created_at
        FROM inventory_vehicle_compat
        WHERE inventory_id = %s
        ORDER BY COALESCE(make, ''), COALESCE(model, ''), COALESCE(year, 0)
    """, (inventory_id,))
    rows = cur_t.fetchall()
    cur_t.close()

    if not rows:
        return []

    # 2. Resolve MYE-linked vehicles from master DB
    mye_ids = [r[0] for r in rows if r[0] is not None]
    details = {}
    if mye_ids:
        cur_m = master_conn.cursor()
        cur_m.execute("""
            SELECT mye.id_mye, b.name_brand, m.name_model, y.year_car, e.name_engine
            FROM model_year_engine mye
            JOIN models m ON m.id_model = mye.model_id
            JOIN brands b ON b.id_brand = m.brand_id
            JOIN years y ON y.id_year = mye.year_id
            JOIN engines e ON e.id_engine = mye.engine_id
            WHERE mye.id_mye = ANY(%s)
            ORDER BY b.name_brand, m.name_model, y.year_car
        """, (mye_ids,))
        details = {r[0]: r for r in cur_m.fetchall()}
        cur_m.close()

    # 3. Merge
    result = []
    for (compat_id, mye_id, make, model, year, engine, engine_code,
         source, confidence, created_at) in rows:
        if mye_id is not None and mye_id in details:
            d = details[mye_id]
            result.append({
                'id': compat_id,
                'model_year_engine_id': mye_id,
                'brand': d[1],
                'model': d[2],
                'year': d[3],
                'engine': d[4],
                'engine_code': '',
                'source': source,
                'confidence': float(confidence),
                'created_at': str(created_at),
            })
        else:
            # Text-only QWEN record
            result.append({
                'id': compat_id,
                'model_year_engine_id': None,
                'brand': make or '',
                'model': model or '',
                'year': year,
                'engine': engine or '',
                'engine_code': engine_code or '',
                'source': source,
                'confidence': float(confidence),
                'created_at': str(created_at),
            })
    return result


def search_mye(master_conn, brand_id=None, model_id=None, year_id=None, engine_id=None):
    """Search MYE IDs by vehicle criteria."""
    cur = master_conn.cursor()
    clauses = []
    params = []
    if brand_id:
        clauses.append("mye.model_id IN (SELECT id_model FROM models WHERE brand_id = %s)")
        params.append(brand_id)
    if model_id:
        clauses.append("mye.model_id = %s")
        params.append(model_id)
    if year_id:
        clauses.append("mye.year_id = %s")
        params.append(year_id)
    if engine_id:
        clauses.append("mye.engine_id = %s")
        params.append(engine_id)

    where = " AND ".join(clauses) if clauses else "TRUE"
    cur.execute(f"""
        SELECT mye.id_mye, b.name_brand, m.name_model, y.year_car, e.name_engine
        FROM model_year_engine mye
        JOIN models m ON m.id_model = mye.model_id
        JOIN brands b ON b.id_brand = m.brand_id
        JOIN years y ON y.id_year = mye.year_id
        JOIN engines e ON e.id_engine = mye.engine_id
        WHERE {where}
        ORDER BY b.name_brand, m.name_model, y.year_car
        LIMIT 100
    """, tuple(params))
    rows = cur.fetchall()
    cur.close()
    return [
        {
            'id_mye': r[0],
            'brand': r[1],
            'model': r[2],
            'year': r[3],
            'engine': r[4],
        }
        for r in rows
    ]


def batch_add_compatibilities(tenant_conn, inventory_id, mye_ids, source='manual'):
    """Add multiple MYE compatibilities at once."""
    cur = tenant_conn.cursor()
    inserted = 0
    for mye_id in mye_ids:
        cur.execute("""
            INSERT INTO inventory_vehicle_compat
                (inventory_id, model_year_engine_id, source, confidence)
            VALUES (%s, %s, %s, 1.0)
            ON CONFLICT (inventory_id, model_year_engine_id) DO NOTHING
        """, (inventory_id, mye_id, source))
        if cur.rowcount > 0:
            inserted += 1
    tenant_conn.commit()
    cur.close()
    return inserted


def save_qwen_fitment(tenant_conn, inventory_id, fitment_result):
    """Save QWEN fitment results into inventory_vehicle_compat.

    Supports both TecDoc-linked vehicles (mye_id present) and text-only
    QWEN vehicles (mye_id=None) when the master DB lacks the vehicle.

    Args:
        tenant_conn: Connection to tenant DB.
        inventory_id: The inventory item ID.
        fitment_result: Dict from get_vehicle_fitment() with 'vehicles' list.

    Returns:
        int: Number of compatibilities inserted.
    """
    vehicles = fitment_result.get('vehicles', [])
    if not vehicles:
        return 0

    inserted = 0
    cur = tenant_conn.cursor()
    for v in vehicles:
        mye_id = v.get('mye_id')
        if mye_id is not None and mye_id:
            # TecDoc-linked vehicle
            cur.execute("""
                INSERT INTO inventory_vehicle_compat
                    (inventory_id, model_year_engine_id, source, confidence, created_at)
                VALUES (%s, %s, 'qwen_ai', %s, NOW())
                ON CONFLICT (inventory_id, model_year_engine_id, make, model, year) DO NOTHING
            """, (inventory_id, mye_id, fitment_result.get('confidence', 0)))
        else:
            # Text-only QWEN vehicle (master DB doesn't have this vehicle)
            cur.execute("""
                INSERT INTO inventory_vehicle_compat
                    (inventory_id, model_year_engine_id, make, model, year, engine, engine_code, source, confidence, created_at)
                VALUES (%s, NULL, %s, %s, %s, %s, %s, 'qwen_ai', %s, NOW())
                ON CONFLICT (inventory_id, model_year_engine_id, make, model, year) DO NOTHING
            """, (
                inventory_id,
                v.get('make', '') or '',
                v.get('model', '') or '',
                v.get('year', 0) or 0,
                v.get('engine', '') or '',
                v.get('engine_code', '') or '',
                fitment_result.get('confidence', 0),
            ))
        if cur.rowcount > 0:
            inserted += 1
    tenant_conn.commit()
    cur.close()
    return inserted


def get_inventory_by_vehicle(tenant_conn, master_conn, mye_id, branch_id=None):
    """Return local inventory items compatible with a given vehicle (MYE).

    Args:
        tenant_conn: Connection to tenant DB.
        master_conn: Connection to master DB (kept for API consistency).
        mye_id: model_year_engine_id.
        branch_id: Optional branch filter for stock.

    Returns:
        List of tuples: (id, part_number, name, brand, price_1, price_2, price_3,
                          image_url, description, stock)
    """
    cur = tenant_conn.cursor()

    if branch_id:
        # Stock for specific branch
        cur.execute("""
            SELECT i.id, i.part_number, i.name, i.brand,
                   i.price_1, i.price_2, i.price_3,
                   i.image_url, i.description,
                   COALESCE(s.stock, 0) as stock
            FROM inventory i
            JOIN inventory_vehicle_compat ivc ON ivc.inventory_id = i.id
            LEFT JOIN inventory_stock_summary s
                ON s.inventory_id = i.id AND s.branch_id = %s
            WHERE ivc.model_year_engine_id = %s
              AND i.is_active = true
            ORDER BY i.name
        """, (branch_id, mye_id))
    else:
        # Total stock across all branches
        cur.execute("""
            SELECT i.id, i.part_number, i.name, i.brand,
                   i.price_1, i.price_2, i.price_3,
                   i.image_url, i.description,
                   COALESCE(SUM(s.stock), 0) as stock
            FROM inventory i
            JOIN inventory_vehicle_compat ivc ON ivc.inventory_id = i.id
            LEFT JOIN inventory_stock_summary s ON s.inventory_id = i.id
            WHERE ivc.model_year_engine_id = %s
              AND i.is_active = true
            GROUP BY i.id, i.part_number, i.name, i.brand,
                     i.price_1, i.price_2, i.price_3,
                     i.image_url, i.description
            ORDER BY i.name
        """, (mye_id,))

    rows = cur.fetchall()
    cur.close()
    return rows