from sqlalchemy import Column, String, DateTime
from datetime import datetime
from database import Base


class Emitter(Base):
    """Emitter model representing a seismograph unit in the fleet"""
    __tablename__ = "emitters"

    id = Column(String, primary_key=True, index=True)
    unit_type = Column(String, nullable=False)
    last_seen = Column(DateTime, default=datetime.utcnow)
    last_file = Column(String, nullable=False)
    status = Column(String, nullable=False)  # OK, Pending, Missing
    notes = Column(String, nullable=True)

    def __repr__(self):
        return f"<Emitter(id={self.id}, type={self.unit_type}, status={self.status})>"