diff options
Diffstat (limited to 'backend/zmusic/login.py')
| -rw-r--r-- | backend/zmusic/login.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/backend/zmusic/login.py b/backend/zmusic/login.py index e625736..6e7b270 100644 --- a/backend/zmusic/login.py +++ b/backend/zmusic/login.py @@ -34,11 +34,14 @@ def login_required(fn): def admin_required(fn): @wraps(fn) def decorated_view(*args, **kwargs): - if query_is_admin_user(request.args) or (current_user.is_authenticated() and current_user.admin): + if is_admin(): return fn(*args, **kwargs) return app.login_manager.unauthorized() return decorated_view +def is_admin(): + return query_is_admin_user(request.args) or (current_user.is_authenticated() and current_user.admin) + def query_is_music_user(query): username = query.get("username", None) password = query.get("password", None) |