KCGL/inventory-backend/app/api/v1/inbound/base.py

# 文件路径: app/api/v1/inbound/base.py

from flask import Blueprint, request, jsonify, send_file
from app.services.inbound.base_service import MaterialBaseService
import traceback
import datetime

inbound_base_bp = Blueprint('stock_base', __name__)


# ==============================================================================
# 1. 搜索接口 (GET /api/v1/inbound/base/search)
# ==============================================================================
@inbound_base_bp.route('/search', methods=['GET'])
def search_base():
    try:
        keyword = request.args.get('keyword', '')
        data = MaterialBaseService.search_material(keyword)
        return jsonify({"code": 200, "msg": "success", "data": data})
    except Exception as e:
        traceback.print_exc()
        return jsonify({"code": 500, "msg": str(e)}), 500


# ==============================================================================
# 2. 列表接口 (GET /api/v1/inbound/base/list)
# ==============================================================================
@inbound_base_bp.route('/list', methods=['GET'])
def get_list():
    try:
        page = request.args.get('pageNum', 1, type=int)
        limit = request.args.get('pageSize', 10, type=int)

        # 构造筛选条件
        filters = {
            'keyword': request.args.get('keyword', ''),
            'company': request.args.get('company', ''),
            'category': request.args.get('category', ''),
            'type': request.args.get('type', ''),
            'isEnabled': request.args.get('isEnabled', None)
        }

        result = MaterialBaseService.get_list(page, limit, filters)
        return jsonify({"code": 200, "msg": "success", "data": result})
    except Exception as e:
        traceback.print_exc()
        return jsonify({"code": 500, "msg": str(e)}), 500


# ==============================================================================
# 2.1 选项接口 (GET /api/v1/inbound/base/options)
# ==============================================================================
@inbound_base_bp.route('/options', methods=['GET'])
def get_options():
    try:
        data = MaterialBaseService.get_distinct_options()
        return jsonify({"code": 200, "msg": "success", "data": data})
    except Exception as e:
        traceback.print_exc()
        return jsonify({"code": 500, "msg": str(e)}), 500


# ==============================================================================
# 2.2 导出接口 (GET /api/v1/inbound/base/export)
# ==============================================================================
@inbound_base_bp.route('/export', methods=['GET'])
def export_data():
    try:
        # 获取筛选条件
        filters = {
            'keyword': request.args.get('keyword', ''),
            'company': request.args.get('company', ''),
            'category': request.args.get('category', ''),
            'type': request.args.get('type', ''),
            'isEnabled': request.args.get('isEnabled', None)
        }

        # 生成 Excel 文件流
        file_stream = MaterialBaseService.export_excel(filters)

        # 生成文件名：库存统计+年月日+时分秒 (北京时间 UTC+8)
        # 简单处理：UTC时间 + 8小时
        beijing_time = datetime.datetime.utcnow() + datetime.timedelta(hours=8)
        filename = f"库存统计_{beijing_time.strftime('%Y%m%d_%H%M%S')}.xlsx"

        # 发送文件
        # 注意：download_name 仅在较新 Flask 版本有效，旧版本可能需要手动 header，
        # 但通常浏览器下载名由前端 Blob 处理或 Content-Disposition 决定。
        return send_file(
            file_stream,
            mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
            as_attachment=True,
            download_name=filename
        )
    except Exception as e:
        traceback.print_exc()
        return jsonify({"code": 500, "msg": f"导出失败: {str(e)}"}), 500


# ==============================================================================
# 3. 新增接口 (POST /api/v1/inbound/base/)
# ==============================================================================
@inbound_base_bp.route('/', methods=['POST'])
def create():
    try:
        data = request.get_json()
        if not data:
            return jsonify({"code": 400, "msg": "No data provided"}), 400

        MaterialBaseService.create_material(data)
        return jsonify({"code": 200, "msg": "新增成功"})
    except ValueError as e:
        # 捕获业务逻辑验证错误 (如名称为空)
        return jsonify({"code": 400, "msg": str(e)}), 400
    except Exception as e:
        # 捕获系统错误
        traceback.print_exc()
        return jsonify({"code": 500, "msg": f"系统错误: {str(e)}"}), 500


# ==============================================================================
# 4. 修改接口 (PUT /api/v1/inbound/base/<id>)
# ==============================================================================
@inbound_base_bp.route('/<int:id>', methods=['PUT'])
def update(id):
    try:
        data = request.get_json()
        MaterialBaseService.update_material(id, data)
        return jsonify({"code": 200, "msg": "修改成功"})
    except Exception as e:
        traceback.print_exc()
        return jsonify({"code": 500, "msg": str(e)}), 500


# ==============================================================================
# 5. 删除接口 (DELETE /api/v1/inbound/base/<id>)
# ==============================================================================
@inbound_base_bp.route('/<int:id>', methods=['DELETE'])
def delete(id):
    try:
        MaterialBaseService.delete_material(id)
        return jsonify({"code": 200, "msg": "删除成功"})
    except Exception as e:
        traceback.print_exc()
        return jsonify({"code": 500, "msg": str(e)}), 500