Source code for redditwarp.pagination.paginators.listing.submission_and_comment_listing_paginator


from __future__ import annotations
from typing import Sequence, Optional

from ..listing.listing_paginator import ListingPaginator
from ....model_loaders.comment_SYNC import load_loose_comment
from ....model_loaders.submission_SYNC import load_submission
from ....exceptions import UnexpectedResultException

[docs]class SubmissionAndLooseCommentListingPaginator(ListingPaginator[object]):
[docs]    def fetch(self) -> Sequence[object]:
        data = self._fetch_data()
        l = []
        for child in data['children']:
            kind = child['kind']
            data = child['data']
            obj: Optional[object] = None
            if kind == 't1':
                obj = load_loose_comment(data, self.client)
            elif kind == 't3':
                obj = load_submission(data, self.client)
            if obj is None:
                raise UnexpectedResultException(f'unexpected kind {kind!r}')
            l.append(obj)
        return l