from app.db.db_base import DbCommon
from sqlalchemy import String, Integer
from sqlalchemy.orm import Mapped, mapped_column


class ProjectType(DbCommon):
    """
    项目类别表 - 标识项目的类型目前存在的（目标识别，OCR识别，瑕疵检测，图像分类）
    """
    __tablename__ = "project_type"
    type_code = Mapped[str] = mapped_column(String(20), unique=True, nullable=False)
    type_name = Mapped[str] = mapped_column(String(20))
    icon_path = Mapped[str] = mapped_column(String(255))
    description = Mapped[str] = mapped_column(String(255))
    type_status = Mapped[str] = mapped_column(String(10))


class ProjectInfo(DbCommon):
    """项目信息表"""
    __tablename__ = "project_info"
    project_no = Mapped[str] = mapped_column(String(32), unique=True, nullable=False)
    project_name = Mapped[str] = mapped_column(String(32), unique=True, nullable=False)
    type_code = Mapped[str] = mapped_column(String(10))
    description = Mapped[str] = mapped_column(String(255))
    project_status = Mapped[str] = mapped_column(String(10))
    user_id = Mapped[int] = mapped_column(Integer)
    train_version = Mapped[int] = mapped_column(Integer)


class ProjectLabel(DbCommon):
    """项目标签表"""
    __tablename__ = "project_label"
    label_name = Mapped[str] = mapped_column(String(32), unique=True, nullable=False)
    project_id = Mapped[int] = mapped_column(Integer, nullable=False)


class ProjectImage(DbCommon):
    """项目图片表"""
    __tablename__ = "project_image"
    image_url = Mapped[str] = mapped_column(String(255), nullable=False)
    thumb_image_url = Mapped[str] = mapped_column(String(255), nullable=False)
    project_id = Mapped[int] = mapped_column(Integer)