from sqlalchemy import Column, DateTime, Float, Integer, String, UUID, ForeignKey
from src.database.core import Base
from src.models import DefaultMixin, IdentityMixin, TimeStampMixin
from sqlalchemy.orm import relationship


class OverhaulJob(Base, DefaultMixin):
    __tablename__ = "oh_tr_overhaul_job"

    overhaul_activity_id = Column(UUID(as_uuid=True), ForeignKey(
        "oh_ms_overhaul_activity.id"), nullable=False)
    job_id = Column(UUID(as_uuid=True), ForeignKey(
        "oh_ms_job.id", ondelete="cascade"))
    notes = Column(String, nullable=True)
    status = Column(String, nullable=True, default="pending")

    job = relationship(
        "MasterActivity", lazy="raise", back_populates="overhaul_jobs"
    )