rag-gateway/app/models/label_model.py

from datetime import datetime
from typing import Optional

from pydantic import BaseModel, constr
from sqlalchemy import Column, Integer, String, DateTime, Table, ForeignKey, UniqueConstraint
from sqlalchemy.orm import relationship, backref

from app.models.base_model import Base


class LabelModel(Base):
    __tablename__ = 'label'
    id = Column(Integer, primary_key=True, index=True)
    created_at = Column(DateTime, default=datetime.now())
    updated_at = Column(DateTime, default=datetime.now(), onupdate=datetime.now())
    name = Column(String(128), unique=True, nullable=False, index=True)
    status = Column(String(10), default="1")
    creator = Column(Integer)
    label_type = Column(Integer, default=1)

    def to_json(self):
        return {
            'labelId': self.id,
            'labelName': self.name,
            'labelType': self.label_type
        }


class LabelWorkerModel(Base):
    __tablename__ = 'label_worker'
    __table_args__ = (UniqueConstraint('label_id', 'object_id', name='label_object_id_ix'),)
    id = Column(Integer, primary_key=True, index=True)
    label_id = Column(Integer)
    object_id = Column(String(36))



class LabelData(BaseModel):
    labelName: str
    labelType: Optional[int] = 1


class SignLabelData(BaseModel):
    labelIdList: list
    objectId:str