18 lines
615 B
Python
18 lines
615 B
Python
# 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()
|