# app/services/slot_snapshot_service.py
from sqlalchemy.orm import Session
from app.models.slot_snapshot import VasSlotSnapshot
from app.schemas.slot_snapshot import SlotSnapshotCreate
from datetime import datetime

class SlotSnapshotService:

    def create(db: Session, data: SlotSnapshotCreate):
        rec = VasSlotSnapshot(**data.dict())
        db.add(rec)
        db.commit()
        db.refresh(rec)
        return rec

    def latest_for(db: Session, country:str, city:str, visa_type:str):
        return db.query(VasSlotSnapshot).filter_by(country=country, city=city, visa_type=visa_type).order_by(VasSlotSnapshot.snapshot_at.desc()).first()