Исходный код raito.plugins.pagination.util

from .enums import PaginationMode
from .paginators import (
    BasePaginator,
    InlinePaginator,
    ListPaginator,
    PhotoPaginator,
    RichPaginator,
    TextPaginator,
)

__all__ = ("get_paginator",)




def get_paginator(mode: PaginationMode) -> type[BasePaginator]:
    """Get paginator class by mode.

    :param mode: pagination mode
    :type mode: PaginationMode
    :return: paginator class
    :rtype: type[BasePaginator]
    :raises ValueError: if mode is not supported
    """
    if mode == PaginationMode.INLINE:
        return InlinePaginator
    if mode == PaginationMode.TEXT:
        return TextPaginator
    if mode == PaginationMode.PHOTO:
        return PhotoPaginator
    if mode == PaginationMode.LIST:
        return ListPaginator
    if mode == PaginationMode.RICH:
        return RichPaginator

    raise ValueError(f"Unsupported pagination mode: {mode}")