shanxia2.0/backend/app/services/pl_mobile_stop_meta.py

"""Optional independent stop metadata. No old-table columns or default-path queries."""
from decimal import Decimal
from fastapi import HTTPException

META_FIELDS = ('alias_name', 'brand_mode', 'brand_id', 'brand_name')
IDENTITY_FIELDS = ('contract_id', 'contract_product_id', 'product_id', 'package_spec', 'model', 'unit')
SOURCE_FIELDS = ('product_name', 'brand_mode', 'brand_id', 'brand_name')


def _value(row, field):
    return str(getattr(row, field, None) or '').strip()


def _saved_meta_values(row):
    values = {field: _value(row, field) for field in META_FIELDS}
    # NULL asks the mobile reader to infer the current Web name; empty means explicitly cleared.
    if getattr(row, 'alias_name', None) is None:
        values['alias_name'] = None
    return values


def assert_mobile_meta_enabled(enabled, requested):
    if requested and not enabled:
        raise HTTPException(503, '移动运输产品独立快照尚未启用，请联系管理员完成迁移并启用后重试；本次未保存')


def validate_mobile_meta_request(payload, enabled):
    requested = bool(getattr(payload, 'mobile_product_meta', False))
    stops = ([payload.load] if payload.load else []) + list(payload.deliveries or [])
    rows = [p for stop in stops for p in stop.products]
    if not requested and any(getattr(p, 'mobile_product_meta', None) is not None or getattr(p, 'unlinked_draft', False) for p in rows):
        raise HTTPException(422, '段产品快照必须显式启用移动保存')
    assert_mobile_meta_enabled(enabled, requested)
    if requested and any(getattr(p, 'mobile_product_meta', None) is None for p in rows):
        raise HTTPException(422, '移动段产品快照不完整，请重新加载')

    if requested:
        if any(not _value(p, 'client_key') for p in payload.products or []) or any(not _value(p, 'plan_product_client_key') and not (getattr(payload, 'save_mode', '') == 'draft' and getattr(p, 'unlinked_draft', False)) for p in rows):
            raise HTTPException(422, '移动产品关联键不完整，请重新加载')
        values = [getattr(p, 'plan_qty', None) for p in rows] + [getattr(p, 'quantity', None) for p in payload.products or []]
        for value in values:
            if value is not None:
                qty = Decimal(str(value))
                if not qty.is_finite() or qty < 0:
                    raise HTTPException(422, '移动产品数量须为非负有限数')


def resolve_stop_meta(snapshots, used, tenant_no, plan_id, stop_type, stop_id, product, plan_product):
    explicit = getattr(product, 'mobile_product_meta', None)
    if explicit is not None:
        return {field: _value(explicit, field) for field in META_FIELDS}
    matches = [row for row in snapshots if row.id not in used
               and row.tenant_no == tenant_no and row.plan_id == plan_id
               and row.stop_type == stop_type and row.stop_id == stop_id
               and all(_value(row, field) == _value(product, field) for field in IDENTITY_FIELDS)]
    source_id = _value(product, 'id')
    if source_id:
        matches = [row for row in matches if row.stop_product_id == source_id]
    if len(matches) > 1:
        raise HTTPException(422, '该段有无法唯一关联的产品快照，请在移动端重新加载后保存')
    if not matches:
        return None
    row = matches[0]
    used.add(row.id)
    result = _saved_meta_values(row)
    # Existing Web legal fields take precedence only when actually changed.
    if (_value(product, 'product_name') != _value(row, 'product_name')
            or _value(plan_product, 'product_name') != _value(row, 'source_product_name')):
        result['alias_name'] = None
    if any(_value(plan_product, f) != _value(row, 'source_' + f) for f in SOURCE_FIELDS[1:]):
        result.update({f: _value(plan_product, f) for f in SOURCE_FIELDS[1:]})
    return result


def load_stop_meta(db, tenant_no, plan_id, enabled):
    if not enabled:
        return []
    from app.models.pl_mobile_stop_meta import PlPlanStopProductMeta
    from sqlalchemy.exc import OperationalError, ProgrammingError
    try:
        return db.query(PlPlanStopProductMeta).filter(
            PlPlanStopProductMeta.tenant_no == tenant_no,
            PlPlanStopProductMeta.plan_id == plan_id,
            PlPlanStopProductMeta.del_flag == 0,
        ).all()
    except (OperationalError, ProgrammingError) as exc:
        raise HTTPException(503, '移动运输产品快照不可用，请联系管理员核查迁移后重试；本次未保存') from exc


def write_stop_meta(db, snapshots, used, tenant_no, plan_id, stop_type, stop_id,
                    stop_product_id, product, plan_product, user_id, dept_id, now):
    values = resolve_stop_meta(snapshots, used, tenant_no, plan_id, stop_type, stop_id, product, plan_product)
    if values is None:
        return
    from app.models.pl_mobile_stop_meta import PlPlanStopProductMeta
    from app.core.utils import generate_uuid
    db.add(PlPlanStopProductMeta(
        id=generate_uuid(), tenant_no=tenant_no, plan_id=plan_id, stop_type=stop_type,
        stop_id=stop_id, stop_product_id=stop_product_id,
        **{f: _value(product, f) for f in IDENTITY_FIELDS},
        product_name=_value(product, 'product_name'),
        **{'source_' + f: _value(plan_product, f) for f in SOURCE_FIELDS},
        **values, order_num=0, create_by=user_id, create_dept_id=dept_id or '',
        create_time=now, update_by=user_id, update_time=now, del_flag=0,
    ))


def attach_stop_meta(result, snapshots):
    from app.schemas.pl.plan import PlanStopProductMeta
    by_id = {(row.stop_type, row.stop_product_id): row for row in snapshots}
    stops = [('load', result.load)] + [('delivery', row) for row in result.deliveries or []]
    for stop_type, stop in stops:
        if stop is None:
            continue
        for product in stop.products:
            row = by_id.get((stop_type, product.id))
            if row:
                product.mobile_product_meta = PlanStopProductMeta(**_saved_meta_values(row))