Source code for redditwarp.model_loaders.comment_tree_ASYNC

from __future__ import annotations
from typing import TYPE_CHECKING, Any, Mapping
if TYPE_CHECKING:
    from ..client_ASYNC import Client
    from ..models.more_comments_ASYNC import MoreComments
    from ..models.comment_ASYNC import Comment

from ..models.comment_tree_ASYNC import MoreCommentsTreeNode, CommentTreeNode, SubmissionTreeNode
from ..models.more_comments_ASYNC import ContinueThisThread, LoadMoreComments
from .comment_ASYNC import load_comment
from .submission_ASYNC import load_submission

def load_more_comments(
    d: Mapping[str, Any],
    client: Client,
    submission_id36: str,
    sort: str,
) -> MoreComments:
    _, _, comment_id36 = d['parent_id'].partition('_')
    go_deeper = d['id'] == '_'
    if go_deeper:
        return ContinueThisThread(
            submission_id36=submission_id36,
            comment_id36=comment_id36,
            sort=sort,
            d=d,
            client=client,
        )
    return LoadMoreComments(
        submission_id36=submission_id36,
        comment_id36=comment_id36,
        child_id36s=d['children'],
        sort=sort,
        count=d['count'],
        d=d,
        client=client,
    )


def load_submission_tree_node(d: Any, client: Client, sort: str) -> SubmissionTreeNode:
    def load_comment_tree_node(d: Any, client: Client, submission_id36: str, sort: str) -> CommentTreeNode:
        value = load_comment(d, client)
        children = []
        more = None

        replies_data = d['replies']
        if replies_data:
            elements = list(replies_data['data']['children'])
            if elements:
                m = elements[-1]
                if m['kind'] == 'more':
                    elements.pop()
                    more = load_more_comments(m['data'], client, submission_id36, sort)

                for m in elements:
                    node = load_comment_tree_node(m['data'], client, submission_id36, sort)
                    children.append(node)

        return CommentTreeNode(value, children, more)

    submission_data = d[0]['data']['children'][0]['data']
    submission_id36: str = submission_data['id']

    value = load_submission(submission_data, client)
    children = []
    more = None

    elements = list(d[1]['data']['children'])
    if elements:
        m = elements[-1]
        if m['kind'] == 'more':
            elements.pop()
            more = load_more_comments(m['data'], client, submission_id36, sort)

        for m in elements:
            node = load_comment_tree_node(m['data'], client, submission_id36, sort)
            children.append(node)

    return SubmissionTreeNode(value, children, more)


def load_more_comments_tree_node(d: Any, client: Client, submission_id36: str, sort: str) -> MoreCommentsTreeNode:
    value_lookup: dict[object, Comment] = {}
    children_lookup: dict[object, list[CommentTreeNode]] = {}
    more_lookup: dict[object, MoreComments] = {}

    elements = d['json']['data']['things']
    for m in elements:
        data = m['data']
        if m['kind'] == 'more':
            more_lookup[data['parent_id']] = load_more_comments(data, client, submission_id36, sort)
        else:
            idt = data['name']
            value_lookup[idt] = load_comment(data, client)
            children_lookup[idt] = []

    root_children: list[CommentTreeNode] = []
    for m in elements:
        data = m['data']
        if m['kind'] != 'more':
            idt = data['name']
            children_lookup.get(data['parent_id'], root_children).append(
                CommentTreeNode(
                    value_lookup[idt],
                    children_lookup[idt],
                    more_lookup.pop(idt, None),
                ),
            )

    root_more = None
    if more_lookup:
        root_more = more_lookup.popitem()[1]
    if more_lookup:
        raise Exception

    return MoreCommentsTreeNode(None, root_children, root_more)