Add like counter to user profile

- Also fixed some issue with get_post_count() method? Weird.
This commit is contained in:
Xevion
2022-03-30 01:41:32 -05:00
parent 11394bfb7e
commit 765e786231
2 changed files with 7 additions and 4 deletions

View File

@@ -53,11 +53,11 @@ class User(UserMixin, db.Model):
def get_post_count(self) -> int:
"""Returns the number of posts this user has made."""
return Post.query.filter_by(user_id=self.id).count()
return Post.query.filter_by(author=self).count()
def get_comment_count(self) -> int:
"""Returns the number of comments this user has made."""
return Comment.query.filter_by(user_id=self.id).count()
return Comment.query.filter_by(author=self).count()
def get_post_likes(self) -> int:
"""Returns the number of likes this user's posts have accumulated."""