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

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34100 additions and 23589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
app/dao/permissions_dao.py
View File

@@ -19,11 +19,11 @@ default_service_permissions = [
SEND_TEXTS,
SEND_EMAILS,
MANAGE_API_KEYS,
VIEW_ACTIVITY]
VIEW_ACTIVITY,
]
class PermissionDAO(DAOClass):
class Meta:
model = Permission
@@ -40,7 +40,9 @@ class PermissionDAO(DAOClass):
query = self.Meta.model.query.filter_by(user=user)
query.delete()
def set_user_service_permission(self, user, service, permissions, _commit=False, replace=False):
def set_user_service_permission(
self, user, service, permissions, _commit=False, replace=False
):
try:
if replace:
query = self.Meta.model.query.filter_by(user=user, service=service)
@@ -58,12 +60,20 @@ class PermissionDAO(DAOClass):
db.session.commit()
def get_permissions_by_user_id(self, user_id):
return self.Meta.model.query.filter_by(user_id=user_id)\
.join(Permission.service).filter_by(active=True).all()
return (
self.Meta.model.query.filter_by(user_id=user_id)
.join(Permission.service)
.filter_by(active=True)
.all()
)
def get_permissions_by_user_id_and_service_id(self, user_id, service_id):
return self.Meta.model.query.filter_by(user_id=user_id)\
.join(Permission.service).filter_by(active=True, id=service_id).all()
return (
self.Meta.model.query.filter_by(user_id=user_id)
.join(Permission.service)
.filter_by(active=True, id=service_id)
.all()
)
permission_dao = PermissionDAO()