224 lines
7.7 KiB
Python
224 lines
7.7 KiB
Python
# inventory-backend/app/api/v1/stock/adjustment.py
|
|
from flask import Blueprint, request, jsonify
|
|
from flask_jwt_extended import jwt_required, get_jwt_identity, get_jwt
|
|
from app.utils.decorators import permission_required
|
|
from app.extensions import db
|
|
from app.models.stock.adjustment import StockAdjustment
|
|
from app.models.base import MaterialBase
|
|
from app.models.inbound.buy import StockBuy
|
|
from app.models.inbound.semi import StockSemi
|
|
from app.models.inbound.product import StockProduct
|
|
from datetime import datetime
|
|
import random
|
|
import string
|
|
|
|
adjustment_bp = Blueprint('adjustment', __name__, url_prefix='/stock/adjustment')
|
|
|
|
|
|
def generate_order_no():
|
|
"""生成单号 ADJ-YYYYMMDD-XXXX"""
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
suffix = ''.join(random.choices(string.digits, k=4))
|
|
return f'ADJ-{today}-{suffix}'
|
|
|
|
|
|
def get_stock_model(source_table):
|
|
"""根据source_table获取对应的库存模型"""
|
|
if source_table == 'stock_buy':
|
|
return StockBuy
|
|
elif source_table == 'stock_semi':
|
|
return StockSemi
|
|
elif source_table == 'stock_product':
|
|
return StockProduct
|
|
return None
|
|
|
|
|
|
# --------------------------------------------------------
|
|
# 1. 获取调整单列表
|
|
# GET /api/v1/stock/adjustment/list
|
|
# --------------------------------------------------------
|
|
@adjustment_bp.route('/list', methods=['GET'])
|
|
@jwt_required()
|
|
@permission_required('stock_adjustment:list')
|
|
def get_list():
|
|
page = int(request.args.get('page', 1))
|
|
limit = int(request.args.get('limit', 10))
|
|
keyword = request.args.get('keyword', '')
|
|
adjust_type = request.args.get('adjust_type', '')
|
|
status = request.args.get('status', '')
|
|
|
|
query = StockAdjustment.query
|
|
|
|
if keyword:
|
|
query = query.filter(
|
|
db.or_(
|
|
StockAdjustment.order_no.ilike(f'%{keyword}%'),
|
|
StockAdjustment.sku.ilike(f'%{keyword}%'),
|
|
StockAdjustment.material_name.ilike(f'%{keyword}%')
|
|
)
|
|
)
|
|
|
|
if adjust_type:
|
|
query = query.filter(StockAdjustment.adjust_type == adjust_type)
|
|
|
|
if status:
|
|
query = query.filter(StockAdjustment.status == status)
|
|
|
|
# 按创建时间降序
|
|
query = query.order_by(StockAdjustment.create_time.desc())
|
|
|
|
pagination = query.paginate(page=page, per_page=limit, error_out=False)
|
|
|
|
return jsonify({
|
|
'code': 200,
|
|
'data': {
|
|
'items': [item.to_dict() for item in pagination.items],
|
|
'total': pagination.total,
|
|
'page': page,
|
|
'limit': limit
|
|
}
|
|
})
|
|
|
|
|
|
# --------------------------------------------------------
|
|
# 2. 创建调整单
|
|
# POST /api/v1/stock/adjustment/create
|
|
# --------------------------------------------------------
|
|
@adjustment_bp.route('/create', methods=['POST'])
|
|
@jwt_required()
|
|
@permission_required('stock_adjustment:operation')
|
|
def create():
|
|
data = request.get_json()
|
|
if not data:
|
|
return jsonify({'code': 400, 'msg': '请求参数不能为空'}), 400
|
|
|
|
# 必填字段验证
|
|
required_fields = ['source_table', 'stock_id', 'adjust_type', 'adjust_quantity', 'reason']
|
|
for field in required_fields:
|
|
if field not in data or not data.get(field):
|
|
return jsonify({'code': 400, 'msg': f'{field} 为必填项'}), 400
|
|
|
|
source_table = data['source_table']
|
|
stock_id = int(data['stock_id'])
|
|
adjust_type = data['adjust_type'] # 'profit' or 'loss'
|
|
adjust_quantity = float(data['adjust_quantity'])
|
|
reason = data['reason']
|
|
operator = get_jwt_identity() or 'system'
|
|
|
|
# 获取库存记录
|
|
StockModel = get_stock_model(source_table)
|
|
if not StockModel:
|
|
return jsonify({'code': 400, 'msg': '无效的库存类型'}), 400
|
|
|
|
stock = StockModel.query.get(stock_id)
|
|
if not stock:
|
|
return jsonify({'code': 404, 'msg': '库存记录不存在'}), 404
|
|
|
|
# 获取物料信息
|
|
base_id = getattr(stock, 'base_id', None)
|
|
material = MaterialBase.query.get(base_id) if base_id else None
|
|
|
|
# 计算库存变动
|
|
if adjust_type == 'profit':
|
|
# 盘盈：增加库存
|
|
new_stock_qty = float(stock.stock_quantity or 0) + adjust_quantity
|
|
new_avail_qty = float(stock.available_quantity or 0) + adjust_quantity
|
|
elif adjust_type == 'loss':
|
|
# 盘亏：减少库存
|
|
new_stock_qty = float(stock.stock_quantity or 0) - adjust_quantity
|
|
new_avail_qty = float(stock.available_quantity or 0) - adjust_quantity
|
|
if new_stock_qty < 0 or new_avail_qty < 0:
|
|
return jsonify({'code': 400, 'msg': '库存不足，无法盘亏'}), 400
|
|
else:
|
|
return jsonify({'code': 400, 'msg': '无效的调整类型'}), 400
|
|
|
|
try:
|
|
# 创建调整单
|
|
adjustment = StockAdjustment(
|
|
order_no=generate_order_no(),
|
|
base_id=base_id,
|
|
stock_id=stock_id,
|
|
source_table=source_table,
|
|
sku=getattr(stock, 'sku', None) or getattr(stock, 'SKU', None),
|
|
material_name=material.name if material else getattr(stock, 'sku', '未知'),
|
|
spec_model=getattr(material, 'spec_model', None) if material else None,
|
|
warehouse_location=getattr(stock, 'warehouse_location', None),
|
|
adjust_type=adjust_type,
|
|
adjust_quantity=adjust_quantity,
|
|
reason=reason,
|
|
status='completed',
|
|
operator=operator
|
|
)
|
|
db.session.add(adjustment)
|
|
|
|
# 更新库存
|
|
stock.stock_quantity = new_stock_qty
|
|
stock.available_quantity = new_avail_qty
|
|
|
|
db.session.commit()
|
|
|
|
return jsonify({
|
|
'code': 200,
|
|
'msg': '调整成功',
|
|
'data': adjustment.to_dict()
|
|
})
|
|
except Exception as e:
|
|
db.session.rollback()
|
|
return jsonify({'code': 500, 'msg': f'调整失败: {str(e)}'}), 500
|
|
|
|
|
|
# --------------------------------------------------------
|
|
# 3. 获取库存列表（用于选择物料）
|
|
# GET /api/v1/stock/adjustment/stocks
|
|
# --------------------------------------------------------
|
|
@adjustment_bp.route('/stocks', methods=['GET'])
|
|
@jwt_required()
|
|
@permission_required('stock_adjustment:list')
|
|
def get_stocks():
|
|
"""获取可用于调整的库存列表"""
|
|
source_table = request.args.get('source_table', 'stock_buy')
|
|
keyword = request.args.get('keyword', '')
|
|
page = int(request.args.get('page', 1))
|
|
limit = int(request.args.get('limit', 20))
|
|
|
|
StockModel = get_stock_model(source_table)
|
|
if not StockModel:
|
|
return jsonify({'code': 400, 'msg': '无效的库存类型'}), 400
|
|
|
|
query = StockModel.query.filter(StockModel.stock_quantity > 0)
|
|
|
|
if keyword:
|
|
query = query.filter(
|
|
db.or_(
|
|
StockModel.sku.ilike(f'%{keyword}%'),
|
|
StockModel.barcode.ilike(f'%{keyword}%')
|
|
)
|
|
)
|
|
|
|
pagination = query.paginate(page=page, per_page=limit, error_out=False)
|
|
|
|
items = []
|
|
for stock in pagination.items:
|
|
base_id = getattr(stock, 'base_id', None)
|
|
material = MaterialBase.query.get(base_id) if base_id else None
|
|
items.append({
|
|
'stock_id': stock.id,
|
|
'source_table': source_table,
|
|
'sku': getattr(stock, 'sku', None) or getattr(stock, 'SKU', None),
|
|
'material_name': material.name if material else getattr(stock, 'sku', '未知'),
|
|
'spec_model': getattr(material, 'spec_model', None) if material else None,
|
|
'stock_quantity': float(stock.stock_quantity or 0),
|
|
'available_quantity': float(stock.available_quantity or 0),
|
|
'warehouse_location': getattr(stock, 'warehouse_location', None),
|
|
})
|
|
|
|
return jsonify({
|
|
'code': 200,
|
|
'data': {
|
|
'items': items,
|
|
'total': pagination.total,
|
|
'page': page,
|
|
'limit': limit
|
|
}
|
|
})
|