FjpdZddlmZddlmZmZddlmZmZddl m Z ddl m Z erddl mZGdd Zy ) z.Provide CommentForest for submission comments.) annotations)heappopheappush) TYPE_CHECKINGcast)DuplicateReplaceException) MoreComments)modelsceZdZdZedd ddZddZ d ddZddZddZ dd Z dd Z d d d ddZ y) CommentForestzyA forest of comments starts with multiple top-level comments. Each of these comments can be a tree of replies. N parent_treec>g}|Dcgc]}d|f}}|r|jd\}ttr6t||r|jj _n0|xs|_n$|jfdjD|r|Scc}w)zCReturn a list of :class:`.MoreComments` objects obtained from tree.Nrc3&K|]}|f ywN).0itemcomments X/root/workspace/ytshorts/venv/lib/python3.12/site-packages/praw/models/comment_forest.py z6CommentForest._gather_more_comments..(sIgt_Is)pop isinstancer rreplies _comments _remove_fromextend)treer more_commentsxqueueparentrs @r_gather_more_commentsz#CommentForest._gather_more_commentss-/ nrZsij\`bc[dZsZs#iilOFG'<00+1>>+C+CG(+6+>$G( III[ts Bc |j|S)aReturn the comment at position ``index`` in the list. This method is to be used like an array access, such as: .. code-block:: python first_comment = submission.comments[0] Alternatively, the presence of this method enables one to iterate over all top level comments, like so: .. code-block:: python for comment in submission.comments: print(comment.body) )r)selfindexs r __getitem__zCommentForest.__getitem__+s$~~e$$c(||ng|_||_y)aInitialize a :class:`.CommentForest` instance. :param submission: An instance of :class:`.Submission` that is the parent of the comments. :param comments: Initialize the forest with a list of comments (default: ``None``). N)r _submission)r% submissioncommentss r__init__zCommentForest.__init__?sRZQeXkm%r(c,t|jS)z6Return the number of top-level comments in the forest.)lenr)r%s r__len__zCommentForest.__len__Os4>>""r(c|j|jjvrt|j|_t |t s |jr|jj|y|j|jjvsJd|jj|j}|jjj|y)NzYPRAW Error occurred. Please file a bug report and include the code that caused the error.) namer*_comments_by_idrr+rr is_rootrappend parent_idr)r%rr"s r_insert_commentzCommentForest._insert_commentSs <<4++;; ;+ +!-- g| , NN ! !' *$$(8(8(H(HH k H%%55g6G6GHF NN $ $ + +G 4r(cB||_|D]}|j|_yr)rr*r+)r%r,rs r_updatezCommentForest._update`s&! 2G!%!1!1G  2r(cg}t|j}|rZ|jd}|j|t |t s%|j |jj|rZ|S)zReturn a flattened list of all comments. This list may contain :class:`.MoreComments` instances if :meth:`.replace_more` was not called first. r)listrrr5rr rr)r%r,r!rs rr;zCommentForest.listesa@BT^^$iilG OOG $g|4 W__667  r( r)limit thresholdc|}|j|j}g}|rt|}||dks|j|kr-|j ||j j |Ptd|jd}||dz}|j||jD]}|j|_ t||!|D]} |j| |j j ||r||zS)aeUpdate the comment forest by resolving instances of :class:`.MoreComments`. :param limit: The maximum number of :class:`.MoreComments` instances to replace. Each replacement requires 1 API request. Set to ``None`` to have no limit, or to ``0`` to remove all :class:`.MoreComments` instances without additional requests (default: ``32``). :param threshold: The minimum number of children comments a :class:`.MoreComments` instance must have in order to be replaced. :class:`.MoreComments` instances that represent "continue this thread" links unfortunately appear to have 0 children (default: ``0``). :returns: A list of :class:`.MoreComments` instances that were not replaced. :raises: ``prawcore.TooManyRequests`` when used concurrently. For example, to replace up to 32 :class:`.MoreComments` instances of a submission try: .. code-block:: python submission = reddit.submission("3hahrw") submission.comments.replace_more() Alternatively, to replace :class:`.MoreComments` instances within the replies of a single comment try: .. code-block:: python comment = reddit.comment("d8r4im1") comment.refresh() comment.replies.replace_more() .. note:: This method can take a long time as each replacement will discover at most 100 new :class:`.Comment` instances. As a result, consider looping and handling exceptions until the method returns successfully. For example: .. code-block:: python while True: try: submission.comments.replace_more() break except PossibleExceptions: print("Handling replace_more exception") sleep(1) .. warning:: If this method is called, and the comments are refreshed, calling this method again will result in a :class:`.DuplicateReplaceException`. r*list[models.Comment | models.MoreComments]F)updater ) r#rrcountr5rremoverr,r*r+rr7) r%r=r> remainingrskippedr new_commentsmorers r replace_morezCommentForest.replace_morewsn 224>>B =)D%)q.TZZ)=St$!!((.< U +L$Q 22rLrKzlist[models.MoreComments]) __name__ __module__ __qualname____doc__ staticmethodr#r'r-r0r7r9r;rIrr(rr r s JN8G  (%.GK&%&D&  & # 52  3$35qU'r(r N)rR __future__rheapqrrtypingrrpraw.exceptionsrpraw.models.reddit.morer prawr r rr(rrZs)4"#&50}'}'r(