Source code for org.models.atoz

from collections import OrderedDict
from itertools import groupby
from unidecode import unidecode


from typing import Generic, TypeVar, TYPE_CHECKING
if TYPE_CHECKING:
    from collections.abc import Sequence
    from onegov.org.request import OrgRequest



[docs]
_T = TypeVar('_T')




[docs]
class AtoZ(Generic[_T]):

    def __init__(self, request: 'OrgRequest') -> None:

[docs]
        self.request = request



[docs]
    def sortkey(self, item: _T) -> str:
        return unidecode(self.get_title(item)[0].upper())



[docs]
    def get_items_by_letter(self) -> dict[str, tuple[_T, ...]]:
        items_by_letter = OrderedDict()

        for letter, items in groupby(self.get_items(), self.sortkey):
            items_by_letter[unidecode(letter)] = tuple(
                sorted(items, key=lambda i: unidecode(self.get_title(i)))
            )

        return items_by_letter



[docs]
    def get_title(self, item: _T) -> str:
        raise NotImplementedError



[docs]
    def get_items(self) -> 'Sequence[_T]':
        raise NotImplementedError