348 lines
14 KiB
Python
348 lines
14 KiB
Python
from app.extensions import db
|
||
from app.models.inbound.buy import StockBuy
|
||
from app.models.base import MaterialBase
|
||
|
||
# 尝试导入出库模型，如果不存在则忽略
|
||
try:
|
||
from app.models.outbound import TransOutbound
|
||
except ImportError:
|
||
TransOutbound = None
|
||
|
||
from datetime import datetime, timedelta, timezone
|
||
from sqlalchemy import or_, func, text, and_
|
||
import traceback
|
||
import json
|
||
|
||
|
||
class BuyInboundService:
|
||
|
||
# ============================================================
|
||
# 0. 辅助：唯一性校验 (核心修复)
|
||
# ============================================================
|
||
@staticmethod
|
||
def _check_unique(base_id, serial_number, batch_number, exclude_id=None):
|
||
"""
|
||
校验序列号和批号的唯一性逻辑
|
||
:param base_id: 当前物料的基础ID
|
||
:param serial_number: 序列号
|
||
:param batch_number: 批号
|
||
:param exclude_id: 排除的ID (用于编辑模式)
|
||
"""
|
||
# 1. 序列号 (SN) 全局唯一校验
|
||
# 解释: 不同规格的物料通常也不应该有相同的SN，防止扫码混淆
|
||
if serial_number:
|
||
query = StockBuy.query.filter(StockBuy.serial_number == serial_number)
|
||
if exclude_id:
|
||
query = query.filter(StockBuy.id != exclude_id)
|
||
|
||
exists = query.first()
|
||
if exists:
|
||
# [修改] 获取占用该SN的物料名称 (material -> base)
|
||
occupied_name = exists.base.name if exists.base else "未知物料"
|
||
raise ValueError(f"序列号【{serial_number}】已存在！被物料 [{occupied_name}] 占用，请核查。")
|
||
|
||
# 2. 批号 (BN) 同物料唯一校验
|
||
# 解释: 不同规格的物料可以有相同的批号(如都有 001 批次)，但同一个物料不能重复建单
|
||
if batch_number and base_id:
|
||
query = StockBuy.query.filter(
|
||
StockBuy.base_id == base_id,
|
||
StockBuy.batch_number == batch_number
|
||
)
|
||
if exclude_id:
|
||
query = query.filter(StockBuy.id != exclude_id)
|
||
|
||
if query.first():
|
||
raise ValueError(f"该物料已存在批号【{batch_number}】，请勿重复录入，可直接在该批次下追加库存。")
|
||
|
||
# ============================================================
|
||
# 1. 基础物料搜索
|
||
# ============================================================
|
||
@staticmethod
|
||
def search_base_material(keyword):
|
||
try:
|
||
query = MaterialBase.query.filter(MaterialBase.is_enabled == True)
|
||
if keyword:
|
||
query = query.filter(
|
||
or_(
|
||
MaterialBase.name.ilike(f'%{keyword}%'),
|
||
MaterialBase.spec_model.ilike(f'%{keyword}%'),
|
||
MaterialBase.pinyin.ilike(f'%{keyword}%') # 假设有拼音搜索
|
||
)
|
||
)
|
||
query = query.order_by(MaterialBase.id.desc()).limit(20)
|
||
results = []
|
||
for item in query.all():
|
||
results.append({
|
||
'id': item.id,
|
||
'name': item.name,
|
||
'spec': item.spec_model, # 确保这里字段对应正确
|
||
'category': item.category,
|
||
'unit': item.unit,
|
||
'type': item.material_type,
|
||
'status': '启用'
|
||
})
|
||
return results
|
||
except Exception as e:
|
||
traceback.print_exc()
|
||
return []
|
||
|
||
# ============================================================
|
||
# 2. 新增入库逻辑
|
||
# ============================================================
|
||
@staticmethod
|
||
def handle_inbound(data):
|
||
try:
|
||
base_id = data.get('base_id')
|
||
if not base_id:
|
||
raise ValueError("必须选择基础物料")
|
||
|
||
material = MaterialBase.query.get(base_id)
|
||
if not material:
|
||
raise ValueError("所选物料不存在")
|
||
|
||
# --- [修复点] 执行唯一性校验 ---
|
||
BuyInboundService._check_unique(
|
||
base_id=base_id,
|
||
serial_number=data.get('serial_number'),
|
||
batch_number=data.get('batch_number')
|
||
)
|
||
|
||
# 时间处理 (强制北京时间)
|
||
beijing_tz = timezone(timedelta(hours=8))
|
||
current_time = datetime.now(beijing_tz).replace(tzinfo=None)
|
||
in_date_val = current_time
|
||
|
||
if data.get('in_date'):
|
||
try:
|
||
date_str = str(data['in_date'])
|
||
if len(date_str) > 10:
|
||
in_date_val = datetime.strptime(date_str, '%Y-%m-%d %H:%M:%S')
|
||
else:
|
||
d_temp = datetime.strptime(date_str, '%Y-%m-%d')
|
||
in_date_val = datetime(d_temp.year, d_temp.month, d_temp.day,
|
||
current_time.hour, current_time.minute, current_time.second)
|
||
except:
|
||
in_date_val = current_time
|
||
|
||
in_qty = float(data.get('in_quantity') or 0)
|
||
u_price = float(data.get('unit_price') or 0)
|
||
|
||
# 获取全局打印ID
|
||
try:
|
||
seq_sql = text("SELECT nextval('global_print_seq')")
|
||
result = db.session.execute(seq_sql)
|
||
next_global_id = result.scalar()
|
||
except:
|
||
next_global_id = None
|
||
|
||
# SKU 生成
|
||
if next_global_id:
|
||
generated_sku = str(next_global_id).zfill(10)
|
||
else:
|
||
generated_sku = datetime.now().strftime('%Y%m%d%H%M%S')
|
||
|
||
final_barcode = data.get('barcode') or generated_sku
|
||
|
||
arrival_list = data.get('arrival_photo', [])
|
||
report_list = data.get('inspection_report', [])
|
||
|
||
new_stock = StockBuy(
|
||
base_id=material.id,
|
||
global_print_id=next_global_id,
|
||
sku=generated_sku,
|
||
barcode=final_barcode,
|
||
in_date=in_date_val,
|
||
serial_number=data.get('serial_number'),
|
||
batch_number=data.get('batch_number'),
|
||
status=data.get('status', '在库'),
|
||
in_quantity=in_qty,
|
||
stock_quantity=in_qty, # 初始库存等于入库数
|
||
available_quantity=in_qty,
|
||
inspection_status=data.get('inspection_status', '未检'),
|
||
warehouse_location=data.get('warehouse_location'),
|
||
unit_price=u_price,
|
||
total_price=in_qty * u_price,
|
||
currency=data.get('currency', 'CNY'),
|
||
exchange_rate=data.get('exchange_rate', 1.0),
|
||
supplier_name=data.get('supplier_name'),
|
||
buyer_name=data.get('purchaser'),
|
||
buyer_email=data.get('purchaser_email'),
|
||
original_link=data.get('source_link'),
|
||
detail_link=data.get('detail_link'),
|
||
arrival_photo=json.dumps(arrival_list),
|
||
inspection_report=json.dumps(report_list)
|
||
)
|
||
db.session.add(new_stock)
|
||
db.session.commit()
|
||
return new_stock
|
||
except Exception as e:
|
||
db.session.rollback()
|
||
raise e
|
||
|
||
# ============================================================
|
||
# 3. 更新入库逻辑
|
||
# ============================================================
|
||
@staticmethod
|
||
def update_inbound(stock_id, data):
|
||
try:
|
||
stock = StockBuy.query.get(stock_id)
|
||
if not stock:
|
||
raise ValueError("记录不存在")
|
||
|
||
# --- [修复点] 编辑时也要校验唯一性 (排除自身ID) ---
|
||
# 如果修改了物料(base_id)，或者修改了SN/BN，都需要校验
|
||
new_base_id = data.get('base_id', stock.base_id)
|
||
new_sn = data.get('serial_number', stock.serial_number)
|
||
new_bn = data.get('batch_number', stock.batch_number)
|
||
|
||
BuyInboundService._check_unique(
|
||
base_id=new_base_id,
|
||
serial_number=new_sn,
|
||
batch_number=new_bn,
|
||
exclude_id=stock_id
|
||
)
|
||
|
||
# 更新字段
|
||
field_mapping = {
|
||
'sku': 'sku', 'barcode': 'barcode', 'base_id': 'base_id',
|
||
'warehouse_location': 'warehouse_location',
|
||
'serial_number': 'serial_number', 'batch_number': 'batch_number',
|
||
'status': 'status', 'inspection_status': 'inspection_status',
|
||
'supplier_name': 'supplier_name', 'detail_link': 'detail_link',
|
||
'currency': 'currency', 'exchange_rate': 'exchange_rate',
|
||
'purchaser': 'buyer_name', 'purchaser_email': 'buyer_email',
|
||
'source_link': 'original_link'
|
||
}
|
||
for k, v in field_mapping.items():
|
||
if k in data: setattr(stock, v, data[k])
|
||
|
||
if 'arrival_photo' in data and isinstance(data['arrival_photo'], list):
|
||
stock.arrival_photo = json.dumps(data['arrival_photo'])
|
||
if 'inspection_report' in data and isinstance(data['inspection_report'], list):
|
||
stock.inspection_report = json.dumps(data['inspection_report'])
|
||
|
||
# 库存数量变更逻辑
|
||
if 'in_quantity' in data:
|
||
new_qty = float(data['in_quantity'])
|
||
diff = new_qty - float(stock.in_quantity)
|
||
if diff != 0:
|
||
stock.in_quantity = new_qty
|
||
stock.stock_quantity = float(stock.stock_quantity) + diff
|
||
stock.available_quantity = float(stock.available_quantity) + diff
|
||
|
||
if 'unit_price' in data:
|
||
stock.unit_price = float(data['unit_price'])
|
||
|
||
stock.total_price = float(stock.in_quantity) * float(stock.unit_price)
|
||
db.session.commit()
|
||
return stock
|
||
except Exception as e:
|
||
db.session.rollback()
|
||
raise e
|
||
|
||
# ============================================================
|
||
# 4. 删除逻辑
|
||
# ============================================================
|
||
@staticmethod
|
||
def delete_inbound(stock_id):
|
||
try:
|
||
stock = StockBuy.query.get(stock_id)
|
||
if not stock: raise ValueError("记录不存在")
|
||
db.session.delete(stock)
|
||
db.session.commit()
|
||
return True
|
||
except Exception as e:
|
||
db.session.rollback()
|
||
raise e
|
||
|
||
# ============================================================
|
||
# 5. 获取列表
|
||
# ============================================================
|
||
@staticmethod
|
||
def get_list(page, limit, keyword=None, statuses=None):
|
||
try:
|
||
query = db.session.query(StockBuy).outerjoin(MaterialBase, StockBuy.base_id == MaterialBase.id)
|
||
|
||
if keyword:
|
||
kw = f'%{keyword}%'
|
||
query = query.filter(
|
||
or_(
|
||
MaterialBase.name.ilike(kw),
|
||
MaterialBase.spec_model.ilike(kw),
|
||
StockBuy.batch_number.ilike(kw),
|
||
StockBuy.serial_number.ilike(kw),
|
||
StockBuy.sku.ilike(kw),
|
||
StockBuy.supplier_name.ilike(kw)
|
||
)
|
||
)
|
||
|
||
if not statuses:
|
||
statuses = ['在库', '借库']
|
||
|
||
if '已出库' in statuses:
|
||
query = query.filter(StockBuy.status.in_(statuses))
|
||
else:
|
||
query = query.filter(and_(StockBuy.status.in_(statuses), StockBuy.stock_quantity > 0))
|
||
|
||
pagination = query.order_by(StockBuy.in_date.desc()).paginate(page=page, per_page=limit, error_out=False)
|
||
current_items = pagination.items
|
||
|
||
def parse_img(json_str):
|
||
if not json_str: return []
|
||
try:
|
||
return json.loads(json_str) if json_str.startswith('[') else [json_str]
|
||
except:
|
||
return []
|
||
|
||
items = []
|
||
for item in current_items:
|
||
qty_stock = float(item.stock_quantity or 0)
|
||
qty_avail = float(item.available_quantity or 0)
|
||
|
||
date_display = ''
|
||
if item.in_date:
|
||
try:
|
||
date_display = item.in_date.strftime('%Y-%m-%d')
|
||
except:
|
||
date_display = str(item.in_date)[:10]
|
||
|
||
d = {
|
||
'id': item.id,
|
||
'base_id': item.base_id,
|
||
# [核心修改] 确保这里从关联的 .base 获取信息
|
||
'material_name': item.base.name if item.base else '',
|
||
'spec_model': item.base.spec_model if item.base else '',
|
||
'category': item.base.category if item.base else '',
|
||
'unit': item.base.unit if item.base else '',
|
||
'material_type': item.base.material_type if item.base else '',
|
||
|
||
'sku': item.sku,
|
||
'inbound_date': date_display,
|
||
'barcode': item.barcode,
|
||
'serial_number': item.serial_number,
|
||
'batch_number': item.batch_number,
|
||
'status': item.status,
|
||
'inspection_status': item.inspection_status,
|
||
'qty_inbound': float(item.in_quantity or 0),
|
||
'qty_stock': qty_stock,
|
||
'qty_available': qty_avail,
|
||
'warehouse_loc': item.warehouse_location,
|
||
'unit_price': float(item.unit_price or 0),
|
||
'total_price': float(item.total_price or 0),
|
||
'currency': item.currency,
|
||
'exchange_rate': float(item.exchange_rate or 1),
|
||
'supplier_name': item.supplier_name,
|
||
'purchaser': item.buyer_name,
|
||
'purchaser_email': item.buyer_email,
|
||
'source_link': item.original_link,
|
||
'detail_link': item.detail_link,
|
||
'arrival_photo': parse_img(item.arrival_photo),
|
||
'inspection_report': parse_img(item.inspection_report),
|
||
'global_print_id': item.global_print_id
|
||
}
|
||
items.append(d)
|
||
|
||
return {"total": pagination.total, "items": items}
|
||
except Exception as e:
|
||
traceback.print_exc()
|
||
return {"total": 0, "items": []} |