# scraper_cursor.py
from typing import Optional
from sqlalchemy.orm import Session
from database import InstaCursor

def get_cursor(db: Session, username: str) -> Optional[str]:
    row = db.query(InstaCursor).filter_by(username=username).first()
    return row.next_max_id if row else None

def set_cursor(db: Session, username: str, next_max_id: Optional[str]) -> None:
    row = db.query(InstaCursor).filter_by(username=username).first()
    if row is None:
        row = InstaCursor(username=username, next_max_id=next_max_id)
        db.add(row)
    else:
        row.next_max_id = next_max_id
    db.commit()