darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-22 16:31:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

Transaction management added for one example. All tests passing.

Browse Source
This commit is contained in:
Nicholas Staples
2016-03-23 16:30:47 +00:00
parent f76af9acb7
commit 3a59c75257
4 changed files with 22 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/dao/permissions_dao.py
View File

@@ -61,17 +61,19 @@ class PermissionDAO(DAOClass):
query = self.get_query(filter_by_dict={'user': user.id, 'service': service.id})
query.delete()
def set_user_permission(self, user, permissions):
def set_user_service_permission(self, user, service, permissions, _commit=False):
try:
query = self.get_query(filter_by_dict={'user': user.id})
query = self.get_query(filter_by_dict={'user': user.id, 'service': service.id})
query.delete()
for p in permissions:
self.create_instance(p, _commit=False)
except Exception as e:
db.session.rollback()
if _commit:
db.session.rollback()
raise e
else:
db.session.commit()
if _commit:
db.session.commit()
permission_dao = PermissionDAO()