23 lines
580 B
Python
23 lines
580 B
Python
|
from dataclasses import dataclass
|
||
|
|
from typing import TypeVar
|
||
|
|
|
||
|
|
from sqlalchemy.sql import Select
|
||
|
|
|
||
|
|
SelectT = TypeVar("SelectT", bound=Select)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(frozen=True)
|
||
|
|
class PageParams:
|
||
|
|
limit: int = 100
|
||
|
|
offset: int = 0
|
||
|
|
|
||
|
|
def __post_init__(self) -> None:
|
||
|
|
if self.limit < 1 or self.limit > 1000:
|
||
|
|
raise ValueError("limit must be between 1 and 1000")
|
||
|
|
if self.offset < 0:
|
||
|
|
raise ValueError("offset must be >= 0")
|
||
|
|
|
||
|
|
|
||
|
|
def apply_pagination(query: SelectT, page: PageParams) -> SelectT:
|
||
|
|
return query.offset(page.offset).limit(page.limit)
|