Source code for recipient.collection

from __future__ import annotations

from onegov.core.collection import GenericCollection
from onegov.recipient.model import GenericRecipient

from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from sqlalchemy.orm import Query, Session



[docs]
class GenericRecipientCollection[T: GenericRecipient](GenericCollection[T]):

    def __init__(self, session: Session, type: str):
        super().__init__(session)

[docs]
        self.type = type


    @property

[docs]
    def model_class(self) -> type[T]:
        return GenericRecipient.get_polymorphic_class(  # type:ignore
            self.type, GenericRecipient)  # type:ignore[arg-type]



[docs]
    def query(self) -> Query[T]:
        return super().query().order_by(self.model_class.order)