From 5b73881849733f773c1421b3cef42efea7dd36bd Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Mon, 4 Apr 2022 02:05:40 +0200 Subject: [PATCH] dfs --- files/routes/front.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/files/routes/front.py b/files/routes/front.py index 9418ff9560..2b3a209633 100644 --- a/files/routes/front.py +++ b/files/routes/front.py @@ -133,14 +133,12 @@ def notifications(v): if c.parent_submission: if not c.replies2: c.replies2 = c.child_comments.filter(or_(Comment.author_id == v.id, Comment.id.in_(all))).all() - print(c.replies2) - cids = cids | set(x.id for x in c.replies2) + cids.update(x.id for x in c.replies2) while c.parent_comment and (c.parent_comment.author_id == v.id or c.parent_comment in comments): c = c.parent_comment if not c.replies2: c.replies2 = c.child_comments.filter(or_(Comment.author_id == v.id, Comment.id.in_(all))).all() - print(c.replies2) - cids = cids | set(x.id for x in c.replies2) + cids.update(x.id for x in c.replies2) cids.add(c.id) else: while c.parent_comment: