diff options
| -rw-r--r-- | pyaggr3g470r/decorators.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/pyaggr3g470r/decorators.py b/pyaggr3g470r/decorators.py index f4192605..565d25a6 100644 --- a/pyaggr3g470r/decorators.py +++ b/pyaggr3g470r/decorators.py @@ -4,7 +4,7 @@ from threading import Thread from functools import wraps -from flask import g +from flask import g, redirect, url_for, flash from pyaggr3g470r.models import Feed @@ -23,8 +23,10 @@ def feed_access_required(func): def decorated(*args, **kwargs): if kwargs.get('feed_id', None) != None: feed = Feed.query.filter(Feed.id == kwargs.get('feed_id', None)).first() - if (feed == None or feed.subscriber.id != g.user.id) and not g.user.is_admin(): - flash("This feed do not exist.", "danger") - return redirect(url_for('home')) - return func(*args, **kwargs) + if feed != None and (feed.subscriber.id == g.user.id or g.user.is_admin()): + return func(*args, **kwargs) + flash("This feed do not exist.", "danger") + return redirect(url_for('home')) + else: + return func(*args, **kwargs) return decorated |