from iti.applications.extensions import db
from iti.applications.common.crud import IdModelMixin, TimeModelMixin
from iti.applications.common.utils import BaseSchema
from apiflask.fields import String, Integer, DateTime
import uuid

class IotDevice(db.Model, IdModelMixin, TimeModelMixin):
    """
    设备信息表
    """

    __tablename__ = "iot_device"
    id = db.Column(
        db.String(36),
        primary_key=True,
        default=lambda: str(uuid.uuid4().hex),
        comment="标识",
    )
    workshop_id = db.Column(db.String(36), nullable=False, unique=True, comment="车间ID")
    device_name = db.Column(db.String(255), nullable=False, unique=True, comment="设备名称")
    device_number = db.Column(db.String(20), nullable=False, comment="设备编号")
    description = db.Column(db.Text, nullable=True, comment="设备描述")
    brand_name = db.Column(db.String(255), nullable=False, comment="品牌名称")
    specification_model = db.Column(db.String(255), nullable=False, comment="规格型号")
    status = db.Column(db.Integer, nullable=False, default=0, comment="状态 0:已停机 1:生产中 2:维修中")
    

class IotDeviceSchema(BaseSchema):
    """
    设备信息表响应结构
    """
    class Meta:
        name = "IotDevice"

    id = String()
    workshop_id = String()
    device_name = String()
    device_number = String()
    description = String()
    brand_name = String()
    specification_model = String()
    status = Integer()
    created_at = DateTime(format="%Y-%m-%d %H:%M:%S")
    updated_at = DateTime(format="%Y-%m-%d %H:%M:%S")