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

Fix for git commit review.

Browse Source
This commit is contained in:
Nicholas Staples
2016-02-29 11:50:43 +00:00
parent 9b73b0d9f8
commit 2cf2b51cd9
6 changed files with 39 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
app/dao/permissions_dao.py
View File

@@ -1,5 +1,6 @@
from app.dao import DAOClass
from app.models import Permission
from app.models import (Permission, Service, User)
from werkzeug.datastructures import MultiDict
# Service Permissions
@@ -15,6 +16,26 @@ class PermissionDAO(DAOClass):
class Meta:
model = Permission
def get_query(self, filter_by_dict={}):
if isinstance(filter_by_dict, dict):
filter_by_dict = MultiDict(filter_by_dict)
query = self.Meta.model.query
if 'id' in filter_by_dict:
query = query.filter(Permission.id.in_(filter_by_dict.getlist('id')))
if 'service' in filter_by_dict:
service_ids = filter_by_dict.getlist('service')
if len(service_ids) == 1:
query.filter_by(service=Service.query.get(service_ids[0]))
# TODO the join method for multiple services
if 'user' in filter_by_dict:
user_ids = filter_by_dict.getlist('service')
if len(user_ids) == 1:
query = query.filter_by(user=User.query.get(user_ids[0]))
# TODO the join method for multiple users
if 'permission' in filter_by_dict:
query = query.filter(Permission.permission.in_(filter_by_dict.getlist('permission')))
return query
def add_default_service_permissions_for_user(self, user, service):
for name in default_service_permissions:
permission = Permission(permission=name, user=user, service=service)