X-Git-Url: http://git.veekun.com/zzz-floof.git/blobdiff_plain/dce049a4e2dd5fee51fe9d8ebafd4c9401c0c5dd..dc9ecbad36153af00773b4951120b4697b33193e:/floof/model/comments.py diff --git a/floof/model/comments.py b/floof/model/comments.py index bcfee9f..5e57c14 100644 --- a/floof/model/comments.py +++ b/floof/model/comments.py @@ -12,6 +12,8 @@ def indent_comments(comments): The first comment's indent level is 0. The comments must be a complete subtree, ordered by their `left` property. + + This function will also cache the `parent` property for each comment. """ last_comment = None @@ -103,3 +105,16 @@ class Comment(Entity): self.left = max_right + 1 self.right = max_right + 2 + @property + def parent(self): + """Returns this comment's parent. This is cached, hence its being a + property and not a method. + """ + if not hasattr(self, '_parent'): + self._parent = Comment.query \ + .filter(Comment.discussion_id == self.discussion_id) \ + .filter(Comment.left < self.left) \ + .filter(Comment.right > self.right) \ + .order_by(Comment.left.desc()) \ + .first() + return self._parent