slot_snapshot_service.py 652 B

1234567891011121314151617
  1. # app/services/slot_snapshot_service.py
  2. from sqlalchemy.orm import Session
  3. from app.models.slot_snapshot import VasSlotSnapshot
  4. from app.schemas.slot_snapshot import SlotSnapshotCreate
  5. from datetime import datetime
  6. class SlotSnapshotService:
  7. def create(db: Session, data: SlotSnapshotCreate):
  8. rec = VasSlotSnapshot(**data.dict())
  9. db.add(rec)
  10. db.commit()
  11. db.refresh(rec)
  12. return rec
  13. def latest_for(db: Session, country:str, city:str, visa_type:str):
  14. return db.query(VasSlotSnapshot).filter_by(country=country, city=city, visa_type=visa_type).order_by(VasSlotSnapshot.snapshot_at.desc()).first()