121 lines
3.7 KiB
Python
121 lines
3.7 KiB
Python
from flask import Blueprint, jsonify, request
|
|
from app.extensions import db
|
|
# 导入模型
|
|
from app.models.inbound.buy import StockBuy
|
|
from app.models.inbound.stocktake import StocktakeDraft # 新增
|
|
|
|
# 尝试导入半成品和成品
|
|
try:
|
|
from app.models.inbound.semi import StockSemi
|
|
except ImportError:
|
|
StockSemi = None
|
|
|
|
try:
|
|
from app.models.inbound.product import StockProduct
|
|
except ImportError:
|
|
StockProduct = None
|
|
|
|
from app.services.print.network_print_service import NetworkPrintService
|
|
|
|
bp = Blueprint('stock_ops', __name__)
|
|
|
|
|
|
@bp.route('/all', methods=['GET'])
|
|
def get_all_stock():
|
|
"""
|
|
获取所有库存 > 0 的物品（无论状态是 在库 还是 部分出库）
|
|
"""
|
|
try:
|
|
# 1. 采购件 (核心修改：只看库存数量 > 0)
|
|
materials = []
|
|
if StockBuy:
|
|
materials = StockBuy.query.filter(StockBuy.stock_quantity > 0).all()
|
|
|
|
# 2. 半成品
|
|
semis = []
|
|
if StockSemi:
|
|
try:
|
|
semis = StockSemi.query.filter(StockSemi.stock_quantity > 0).all()
|
|
except Exception:
|
|
semis = []
|
|
|
|
# 3. 成品
|
|
products = []
|
|
if StockProduct:
|
|
try:
|
|
products = StockProduct.query.filter(StockProduct.stock_quantity > 0).all()
|
|
except Exception:
|
|
products = []
|
|
|
|
return jsonify({
|
|
"materials": [item.to_dict() for item in materials],
|
|
"semis": [item.to_dict() for item in semis],
|
|
"products": [item.to_dict() for item in products]
|
|
}), 200
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
return jsonify({"message": f"查询库存失败: {str(e)}"}), 500
|
|
|
|
|
|
# --- 草稿箱接口 (断点续传) ---
|
|
|
|
@bp.route('/draft/list', methods=['GET'])
|
|
def get_drafts():
|
|
"""获取当前用户的盘点进度"""
|
|
user_id = request.args.get('user_id', 'admin')
|
|
drafts = StocktakeDraft.query.filter_by(user_id=user_id).all()
|
|
return jsonify([d.to_dict() for d in drafts]), 200
|
|
|
|
|
|
@bp.route('/draft/add', methods=['POST'])
|
|
def add_draft():
|
|
"""扫码同步"""
|
|
data = request.json
|
|
user_id = data.get('user_id', 'admin')
|
|
uuid = data.get('uuid')
|
|
|
|
# 避免重复插入
|
|
exists = StocktakeDraft.query.filter_by(user_id=user_id, uuid=uuid).first()
|
|
if not exists:
|
|
draft = StocktakeDraft(user_id=user_id, uuid=uuid)
|
|
db.session.add(draft)
|
|
db.session.commit()
|
|
|
|
return jsonify({"message": "Saved"}), 200
|
|
|
|
|
|
@bp.route('/draft/clear', methods=['POST'])
|
|
def clear_draft():
|
|
"""清空进度 (开始新盘点或结束后)"""
|
|
data = request.json
|
|
user_id = data.get('user_id', 'admin')
|
|
|
|
StocktakeDraft.query.filter_by(user_id=user_id).delete()
|
|
db.session.commit()
|
|
return jsonify({"message": "Cleared"}), 200
|
|
|
|
|
|
# --- 打印接口 (保持不变) ---
|
|
|
|
@bp.route('/print/selection', methods=['POST'])
|
|
def print_selection():
|
|
try:
|
|
data = request.json
|
|
items = data.get('items', [])
|
|
if not items: return jsonify({"message": "未选择任何物品"}), 400
|
|
printer = NetworkPrintService()
|
|
success, msg = printer.print_outbound_selection(items)
|
|
return jsonify({"message": "打印指令已发送" if success else msg}), 200 if success else 500
|
|
except Exception as e:
|
|
return jsonify({"message": str(e)}), 500
|
|
|
|
|
|
@bp.route('/print/stocktake', methods=['POST'])
|
|
def print_stocktake():
|
|
try:
|
|
data = request.json
|
|
printer = NetworkPrintService()
|
|
success, msg = printer.print_stocktake_report(data)
|
|
return jsonify({"message": "盘点报告已发送" if success else msg}), 200 if success else 500
|
|
except Exception as e:
|
|
return jsonify({"message": str(e)}), 500 |