import uuid
from datetime import datetime
from sqlalchemy import Column, String, Integer, Boolean, DateTime, JSON
from sqlalchemy.orm import relationship
from utils.utils import Base


class Bus(Base):
    __tablename__ = "buses"

    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    bus_number = Column(String(50), unique=True, nullable=False, index=True)
    bus_type = Column(String(50), nullable=False)
    total_seats = Column(Integer, nullable=False)
    seat_layout_config = Column(JSON, nullable=False)
    amenities = Column(JSON, nullable=True)
    is_active = Column(Boolean, nullable=False, default=True)
    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)

    schedules = relationship("Schedule", back_populates="bus")