from nexios.pagination import SyncDataHandler
from sqlalchemy import select
from models import User

class UserDataHandler(SyncDataHandler):
    def __init__(self, session):
        self.session = session
    
    def get_total_items(self):
        result = self.session.execute(select(User))
        return len(result.scalars().all())
    
    def get_items(self, offset, limit):
        result = self.session.execute(
            select(User).offset(offset).limit(limit)
        )
        return result.scalars().all()