Triplehx3/contentdb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add permission validation to Package.checkPerm()

Browse Source
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
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