diff --git a/app/main/views.py b/app/main/views.py index cbbadb0..bcd8367 100644 --- a/app/main/views.py +++ b/app/main/views.py @@ -114,7 +114,7 @@ def edit(id): def follow(username): user = User.query.filter_by(username=username).first() if user is None: - flash('Invalud user') + flash('Invalid user') return redirect(url_for('.index')) if current_user.is_following(user): flash('You are already follwoing this user.') diff --git a/app/models.py b/app/models.py index 26508d7..1424abc 100644 --- a/app/models.py +++ b/app/models.py @@ -226,6 +226,11 @@ class User(UserMixin, db.Model): return self.followers.filter_by( follower_id=user.id).first() is not None + @property + def followed_posts(self): + return Post.query.join(Follow, Follow.followed_id == Post.author_id) \ + .filter(Follow.follower_id == self.id) + def __repr__(self): return '' % self.username