from sqlalchemy import Column, Integer, String, DateTime, Enum, JSON
from datetime import datetime
from app.core.database import Base


class VasSlotSnapshot(Base):
    __tablename__ = "vas_slot_snapshot"

    id = Column(Integer, primary_key=True, autoincrement=True)

    country = Column(String(100), nullable=False)
    city = Column(String(100), nullable=False)
    visa_type = Column(String(100), nullable=False)
    routing_key = Column(String(255), nullable=False)

    availability_status = Column(
        Enum('None','Available','Waitlist'),
        nullable=False
    )

    earliest_date = Column(DateTime)
    availability = Column(JSON, nullable=False)

    snapshot_source = Column(Enum('worker','manual','sync'), nullable=False)
    snapshot_at = Column(DateTime, nullable=False)