73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
# app/models/base.py
|
||
from app.extensions import db
|
||
import json
|
||
|
||
class MaterialBase(db.Model):
|
||
"""
|
||
基础信息表模型
|
||
对应数据库表: material_base
|
||
"""
|
||
__tablename__ = 'material_base'
|
||
|
||
# 1. 基础字段
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
name = db.Column(db.String(255), nullable=False, comment='名称')
|
||
common_name = db.Column(db.String(255), comment='俗名')
|
||
category = db.Column(db.String(100), comment='类别')
|
||
material_type = db.Column(db.String(100), comment='类型')
|
||
spec_model = db.Column(db.String(255), comment='规格型号')
|
||
unit = db.Column(db.String(50), comment='计量单位')
|
||
|
||
# 可见等级
|
||
visibility_level = db.Column(db.Integer, default=0, comment='信息可见等级')
|
||
|
||
# 链接与图片 (现在存储 JSON 字符串)
|
||
manual_link = db.Column(db.Text, comment='通用说明书')
|
||
product_image = db.Column(db.Text, comment='通用产品图')
|
||
|
||
# 启用状态
|
||
is_enabled = db.Column(db.Boolean, default=True, comment='是否启用')
|
||
|
||
# ============================================================
|
||
# 关联关系区域
|
||
# ============================================================
|
||
|
||
# 1. 关联采购库存 (StockBuy) - 修改 back_populates 为 'base'
|
||
stock_buys = db.relationship('StockBuy', back_populates='base', lazy='dynamic')
|
||
|
||
# 2. 关联半成品库存 (StockSemi) - 修改 back_populates 为 'base'
|
||
stock_semis = db.relationship('StockSemi', back_populates='base', lazy='dynamic')
|
||
|
||
# 3. 关联成品库存 (StockProduct) - 修改 back_populates 为 'base'
|
||
stock_products = db.relationship('StockProduct', back_populates='base', lazy='dynamic')
|
||
|
||
def to_dict(self):
|
||
"""
|
||
序列化方法
|
||
"""
|
||
# 辅助解析函数：将数据库存储的 JSON 字符串转为 List
|
||
def parse_list(json_str):
|
||
if not json_str:
|
||
return []
|
||
try:
|
||
# 兼容旧数据：如果不是 JSON 格式（比如是单个 URL），则包装成 list
|
||
if not json_str.startswith('['):
|
||
return [json_str]
|
||
return json.loads(json_str)
|
||
except:
|
||
return []
|
||
|
||
return {
|
||
'id': self.id,
|
||
'name': self.name,
|
||
'commonName': self.common_name,
|
||
'category': self.category,
|
||
'type': self.material_type,
|
||
'spec': self.spec_model,
|
||
'unit': self.unit,
|
||
'visibilityLevel': self.visibility_level,
|
||
# 修改：解析为列表返回
|
||
'generalManual': parse_list(self.manual_link),
|
||
'generalImage': parse_list(self.product_image),
|
||
'isEnabled': 1 if self.is_enabled else 0,
|
||
} |