KCGL/inventory-backend/app/api/v1/scan_draft.py

# inventory-backend/app/api/v1/scan_draft.py
"""
扫码草稿接口（出库 / 借库共用）。

场景
----
扫码作业可能很长（一张单几十项），工人常需中途暂停去处理更紧急的单据。
改造前切换单据会清空已扫内容，刷新/退出页面则全部丢失。

由于库存在申请审批通过时已**预占**，暂停期间货不会被他人抢走 ——
因此草稿只记录「扫到哪了」，**不涉及任何库存操作**。即使草稿丢失也只是
需要重扫，不会造成库存错乱。

隔离
----
按 (user_id, biz_type, request_id) 隔离：一人一单，互不影响。
user_id 一律取自 JWT，**不接受入参覆盖**，因此不可能读写他人的草稿。
"""
from flask import Blueprint, request, jsonify
from flask_jwt_extended import jwt_required, get_jwt_identity
import traceback

from app.extensions import db
from app.models.scan_draft import ScanDraft

scan_draft_bp = Blueprint('scan_draft', __name__)

BIZ_TYPES = {'outbound', 'borrow'}


def _parse_identity():
    identity = get_jwt_identity()
    if not identity:
        return None, None, None, '用户未登录'

    biz_type = (request.args.get('biz_type') or '').strip()
    if biz_type not in BIZ_TYPES:
        return None, None, None, f"biz_type 无效（仅支持 {', '.join(sorted(BIZ_TYPES))}）"

    request_id = request.args.get('request_id', type=int)
    if not request_id:
        return None, None, None, 'request_id 不能为空'

    return int(identity), biz_type, request_id, None


def _find(user_id, biz_type, request_id):
    return ScanDraft.query.filter_by(
        user_id=user_id, biz_type=biz_type, request_id=request_id
    ).first()


# --------------------------------------------------------
# 读取草稿：GET /api/v1/scan-draft?biz_type=outbound&request_id=123
# --------------------------------------------------------
@scan_draft_bp.route('', methods=['GET'])
@jwt_required()
def get_scan_draft():
    """
    读取当前用户在某张单据上的扫码草稿。

    返回 { items: [...完整购物车快照...], item_count, total_qty, updated_at }
    无草稿时 items 为空数组（非 404，便于前端直接使用）。
    """
    try:
        user_id, biz_type, request_id, err = _parse_identity()
        if err:
            return jsonify({'code': 400, 'msg': err}), 400

        draft = _find(user_id, biz_type, request_id)
        items = draft.get_items() if draft else []

        return jsonify({
            'code': 200, 'msg': 'success',
            'data': {
                'items': items,
                'item_count': len(items),
                'total_qty': round(sum(float(i.get('out_quantity') or 0) for i in items), 4),
                'updated_at': (
                    draft.updated_at.strftime('%Y-%m-%d %H:%M:%S')
                    if draft and draft.updated_at else None
                ),
            }
        }), 200

    except Exception as e:
        traceback.print_exc()
        return jsonify({'code': 500, 'msg': f'读取草稿失败: {str(e)}'}), 500


# --------------------------------------------------------
# 保存草稿：POST /api/v1/scan-draft
#   Body: { biz_type, request_id, request_no?, items: [...完整购物车快照...] }
#
# ★ 全量覆盖语义：提交的 items 即当前完整清单。
#   前端把 cartItems 整体发过来即可，不必逐条比对增删 —— 逻辑单一，
#   也不会出现「已移除的物料在草稿里阴魂不散」。
#
# ★ 注意：items 为空时会**删除**该单据的草稿（而非存一个空草稿）。
#   这符合「清空列表 = 放弃这次作业」的直觉，也避免残留空记录影响徽标。
# --------------------------------------------------------
@scan_draft_bp.route('', methods=['POST'])
@jwt_required()
def save_scan_draft():
    try:
        identity = get_jwt_identity()
        if not identity:
            return jsonify({'code': 401, 'msg': '用户未登录'}), 401
        user_id = int(identity)

        data = request.get_json(silent=True) or {}
        biz_type = (data.get('biz_type') or '').strip()
        if biz_type not in BIZ_TYPES:
            return jsonify({'code': 400, 'msg': f"biz_type 无效（仅支持 {', '.join(sorted(BIZ_TYPES))}）"}), 400

        try:
            request_id = int(data.get('request_id'))
        except (TypeError, ValueError):
            return jsonify({'code': 400, 'msg': 'request_id 无效'}), 400

        request_no = (data.get('request_no') or '')[:100]
        items = data.get('items') or []

        draft = _find(user_id, biz_type, request_id)

        # 空清单 → 删除草稿
        if not items:
            if draft:
                db.session.delete(draft)
                db.session.commit()
            return jsonify({'code': 200, 'msg': 'success',
                            'data': {'saved': 0, 'cleared': True}}), 200

        if draft is None:
            draft = ScanDraft(user_id=user_id, biz_type=biz_type, request_id=request_id)
            db.session.add(draft)

        draft.request_no = request_no
        draft.set_items(items)

        db.session.commit()
        return jsonify({
            'code': 200, 'msg': 'success',
            'data': {'saved': len(items)}
        }), 200

    except Exception as e:
        db.session.rollback()
        traceback.print_exc()
        return jsonify({'code': 500, 'msg': f'保存草稿失败: {str(e)}'}), 500


# --------------------------------------------------------
# 清除草稿：DELETE /api/v1/scan-draft?biz_type=outbound&request_id=123
#   提交成功后调用，避免下次打开该单据时恢复出已提交的内容。
# --------------------------------------------------------
@scan_draft_bp.route('', methods=['DELETE'])
@jwt_required()
def clear_scan_draft():
    try:
        user_id, biz_type, request_id, err = _parse_identity()
        if err:
            return jsonify({'code': 400, 'msg': err}), 400

        draft = _find(user_id, biz_type, request_id)
        deleted = 0
        if draft:
            db.session.delete(draft)
            db.session.commit()
            deleted = 1

        return jsonify({'code': 200, 'msg': 'success', 'data': {'deleted': deleted}}), 200

    except Exception as e:
        db.session.rollback()
        traceback.print_exc()
        return jsonify({'code': 500, 'msg': f'清除草稿失败: {str(e)}'}), 500


# --------------------------------------------------------
# 草稿概览：GET /api/v1/scan-draft/overview?biz_type=outbound
#
# 供单据下拉显示「已扫 N 项」进度徽标 —— 工人扫开页面就能看到哪张单
# 之前扫到一半，不必逐个点开试。
# --------------------------------------------------------
@scan_draft_bp.route('/overview', methods=['GET'])
@jwt_required()
def get_draft_overview():
    try:
        identity = get_jwt_identity()
        if not identity:
            return jsonify({'code': 401, 'msg': '用户未登录'}), 401

        biz_type = (request.args.get('biz_type') or '').strip()
        if biz_type not in BIZ_TYPES:
            return jsonify({'code': 400, 'msg': 'biz_type 无效'}), 400

        rows = ScanDraft.query.filter_by(
            user_id=int(identity), biz_type=biz_type
        ).all()

        drafts = []
        for r in rows:
            items = r.get_items()
            if not items:
                continue          # 防御：残留的空草稿不参与展示
            drafts.append({
                'request_id': r.request_id,
                'request_no': r.request_no or '',
                'item_count': len(items),
                'total_qty': round(sum(float(i.get('out_quantity') or 0) for i in items), 4),
                'updated_at': r.updated_at.strftime('%Y-%m-%d %H:%M:%S') if r.updated_at else None,
            })

        return jsonify({'code': 200, 'msg': 'success', 'data': {'drafts': drafts}}), 200

    except Exception as e:
        traceback.print_exc()
        return jsonify({'code': 500, 'msg': f'获取草稿概览失败: {str(e)}'}), 500