Add permission validation to Package.checkPerm()

1 files changed, 3 insertions, 1 deletions

diff --git a/app/models.py b/app/models.py
index ca77979..0021e1d 100644
--- a/app/models.py
+++ b/app/models.py
@@ -141,6 +141,8 @@ class Package(db.Model):
 
 		if type(perm) == str:
 			perm = Permission[perm]
+		elif type(perm) != Permission:
+			raise Exception("Unknown permission given to Package.checkPerm()")
 
 		isOwner = user == self.author
 
@@ -158,7 +160,7 @@ class Package(db.Model):
 			return user.rank.atLeast(UserRank.MODERATOR)
 
 		else:
-			return False
+			raise Exception("Permission {} is not related to packages".format(perm.name))
 
 # Setup Flask-User
 db_adapter = SQLAlchemyAdapter(db, User)        # Register the User model