KCGL/inventory-backend/app/models/base.py

# 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,
        }