diff options
Diffstat (limited to 'pyaggr3g470r/views.py')
| -rw-r--r-- | pyaggr3g470r/views.py | 22 |
1 files changed, 14 insertions, 8 deletions
diff --git a/pyaggr3g470r/views.py b/pyaggr3g470r/views.py index 3f05c199..ec330e68 100644 --- a/pyaggr3g470r/views.py +++ b/pyaggr3g470r/views.py @@ -316,14 +316,17 @@ def favorites(): """ List favorites articles. """ - user = models.User.objects(email=g.user.email).first() + user = User.query.filter(User.id == g.user.id).first() result = [] nb_favorites = 0 for feed in user.feeds: - feed.articles = [article for article in feed.articles if article.like] - length = len(feed.articles) + new_feed = Feed() + new_feed.id = feed.id + new_feed.title = feed.title + new_feed.articles = [article for article in feed.articles if article.like] + length = len(new_feed.articles.all()) if length != 0: - result.append(feed) + result.append(new_feed) nb_favorites += length return render_template('favorites.html', feeds=result, nb_favorites=nb_favorites) @@ -333,14 +336,17 @@ def unread(): """ List unread articles. """ - user = models.User.objects(email=g.user.email).first() + user = User.query.filter(User.id == g.user.id).first() result = [] nb_unread = 0 for feed in user.feeds: - feed.articles = [article for article in feed.articles if not article.readed] - length = len(feed.articles) + new_feed = Feed() + new_feed.id = feed.id + new_feed.title = feed.title + new_feed.articles = [article for article in feed.articles if not article.readed] + length = len(new_feed.articles.all()) if length != 0: - result.append(feed) + result.append(new_feed) nb_unread += length return render_template('unread.html', feeds=result, nb_unread=nb_unread) |