From bc4f63da45a60b1d075935d4824d1c8f32df595c Mon Sep 17 00:00:00 2001 From: Steffen Schwebel Date: Thu, 10 Jun 2021 14:51:42 +0200 Subject: [PATCH] remove unused import --- powerdnsadmin/decorators.py | 4 ++-- powerdnsadmin/models/user_permissions.py | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/powerdnsadmin/decorators.py b/powerdnsadmin/decorators.py index 4030eb6..c68795c 100644 --- a/powerdnsadmin/decorators.py +++ b/powerdnsadmin/decorators.py @@ -107,7 +107,7 @@ def can_remove_domain(f): if Setting().get('per_user_permissions'): userperms = UserPermissions.query.filter(UserPermissions.user_id == current_user.id).first() - if userperms.remove_domain: + if userperms and userperms.remove_domain: return f(*args, **kwargs) abort(403) @@ -132,7 +132,7 @@ def can_create_domain(f): if Setting().get('per_user_permissions'): userperms = UserPermissions.query.filter(UserPermissions.user_id == current_user.id).first() - if userperms.create_domain: + if userperms and userperms.create_domain: return f(*args, **kwargs) abort(403) diff --git a/powerdnsadmin/models/user_permissions.py b/powerdnsadmin/models/user_permissions.py index 061b396..6ff3337 100644 --- a/powerdnsadmin/models/user_permissions.py +++ b/powerdnsadmin/models/user_permissions.py @@ -1,5 +1,4 @@ from .base import db -from flask import current_app class UserPermissions(db.Model): __tablename__ = 'user_permissions'