LibreGaming
/
contentdb
14
0
Fork 0
Code Issues 13 Pull Requests Projects Releases Wiki Activity

Add permission validation to Package.checkPerm()

This commit is contained in:
rubenwardy 2018-03-20 03:54:38 +00:00
parent bd58f9bee9
commit 73f24add61
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/models.py
View File

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