skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[themotte/rDrama#451] Add a test for comment.descendant_count accuracy

Browse source
This commit is contained in:
faul_sname 2023-01-03 01:45:37 -08:00
parent 8d24fc3a48
commit 2e29b468ec
2 changed files with 62 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

15
files/helpers/comments.py
View file

@ -8,10 +8,17 @@ def update_stateful_counters(comment, delta):
When a comment changes publish status, we need to update all affected stateful
comment counters (e.g. author comment count, post comment count)
"""
update_post_comment_count(comment, delta)
update_author_comment_count(comment, delta)
update_ancestor_descendant_counts(comment, delta)
def update_post_comment_count(comment, delta):
author = comment.author
comment.post.comment_count += delta
g.db.add(comment.post)
def update_author_comment_count(comment, delta):
author = comment.author
comment.author.comment_count = g.db.query(Comment).filter(
Comment.author_id == comment.author_id,
Comment.parent_submission != None,
@ -20,6 +27,14 @@ def update_stateful_counters(comment, delta):
).count()
g.db.add(comment.author)
def update_ancestor_descendant_counts(comment, delta):
parent = comment.parent_comment_writable
if parent is None:
return
parent.descendant_count += delta
g.db.add(parent)
update_ancestor_descendant_counts(parent, delta)
def comment_on_publish(comment:Comment):
"""
Run when comment becomes visible: immediately for non-filtered comments,