Fix crash on anonymous user in rank_required

This commit is contained in:
rubenwardy 2018-05-14 01:27:08 +01:00
parent d2b705b8bb
commit e2341aafd9
No known key found for this signature in database
GPG Key ID: A1E29D52FF81513C
1 changed files with 3 additions and 1 deletions

View File

@ -1,4 +1,4 @@
from flask import request, flash, abort
from flask import request, flash, abort, redirect
from flask_user import *
from flask_login import login_user, logout_user
from app.models import *
@ -84,6 +84,8 @@ def rank_required(rank):
def decorator(f):
@wraps(f)
def decorated_function(*args, **kwargs):
if not current_user.is_authenticated:
return redirect(url_for("user.login"))
if not current_user.rank.atLeast(rank):
abort(403)