from sqlalchemy import Column, Date, Float, ForeignKey, Integer, String
from sqlalchemy.orm import relationship
from config.database import Base


class Maintenance(Base):
    __tablename__ = "maintenance"

    id = Column(Integer, primary_key=True, index=True)
    equipment_id = Column(Integer, ForeignKey("equipment.id"), nullable=False)
    scenario_id = Column(Integer, ForeignKey("scenario.id"), nullable=False)
    maintenance_date = Column(Date, nullable=False)
    description = Column(String, nullable=True)
    cost = Column(Float, nullable=False)

    equipment = relationship("Equipment")
    scenario = relationship("Scenario", back_populates="maintenance_items")

    def __repr__(self) -> str:
        return (
            f"<Maintenance id={self.id} equipment_id={self.equipment_id} "
            f"scenario_id={self.scenario_id} date={self.maintenance_date} cost={self.cost}>"
        )