Improvement of the Flask redirection for the view mark_as_read().

1 files changed, 10 insertions, 1 deletions

diff --git a/pyaggr3g470r/views.py b/pyaggr3g470r/views.py
index ee163099..e44f0f28 100644
--- a/pyaggr3g470r/views.py
+++ b/pyaggr3g470r/views.py
@@ -89,6 +89,10 @@ def logout():
     flash("Logged out successfully.", 'success')
     return redirect(url_for('home'))
 
+def redirect_url(default='index'):
+    return request.args.get('next') or \
+            request.referrer or \
+            url_for(default)
 
 
 
@@ -192,9 +196,14 @@ def mark_as_read(feed_id=None):
                 for article in unread_articles:
                     article.readed = True
                     article.save()
+                flash('Articles of the feed "' + feed.title  + '" marked as read.', 'info')
+                break
+        else:
+            flash("This feed do not exist.", 'warning')
     else:
         models.Article.objects(readed=False).update(set__readed=True)
-    return redirect(url_for('home'))
+        flash("All articles marked as read", 'info')
+    return redirect(redirect_url())
 
 @app.route('/like/<article_id>', methods=['GET'])
 @login_required