Source code for translator_directory.collections.time_report

from __future__ import annotations

from onegov.core.collection import GenericCollection, Pagination
from onegov.translator_directory.models.time_report import (
    TranslatorTimeReport,
)
from sqlalchemy import desc


from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from onegov.translator_directory.app import TranslatorDirectoryApp
    from sqlalchemy.orm import Query
    from typing import Self



[docs]
class TimeReportCollection(
    GenericCollection[TranslatorTimeReport], Pagination[TranslatorTimeReport]
):


[docs]
    batch_size = 20


    def __init__(
        self,
        app: TranslatorDirectoryApp,
        page: int = 0,
        archive: bool = False,
    ) -> None:
        super().__init__(app.session())

[docs]
        self.app = app


[docs]
        self.page = page


[docs]
        self.archive = archive



[docs]
    def __eq__(self, other: object) -> bool:
        return (
            isinstance(other, self.__class__)
            and self.page == other.page
            and self.archive == other.archive
        )



[docs]
    def query(self) -> Query[TranslatorTimeReport]:
        q = self.session.query(TranslatorTimeReport)
        if self.archive:
            q = q.filter(TranslatorTimeReport.exported == True)
        else:
            q = q.filter(TranslatorTimeReport.exported == False)
        return q.order_by(desc(TranslatorTimeReport.assignment_date))


    @property

[docs]
    def page_index(self) -> int:
        return self.page



[docs]
    def subset(self) -> Query[TranslatorTimeReport]:
        return self.query()



[docs]
    def page_by_index(self, index: int) -> Self:
        return self.__class__(self.app, index, self.archive)



[docs]
    def for_accounting_export(self) -> Query[TranslatorTimeReport]:
        """Query confirmed but not yet exported time reports."""
        return (
            self.session.query(TranslatorTimeReport)
            .filter(TranslatorTimeReport.status == 'confirmed')
            .filter(TranslatorTimeReport.exported == False)
        )