darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-01-31 23:26:23 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update ServiceWhitelist dao to conform to new data model

Browse Source
This commit is contained in:
Imdad Ahad
2016-09-27 13:45:36 +01:00
parent 8184eff15a
commit fbf266be87
2 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/dao/service_whitelist_dao.py
View File

@@ -1,9 +1,10 @@
from app import db
from app.models import ServiceWhitelist
from app.models import Service, ServiceWhitelist
def dao_fetch_service_whitelist(service_id):
return ServiceWhitelist.query.filter(ServiceWhitelist.service_id == service_id).all()
return ServiceWhitelist.query.filter(
ServiceWhitelist.service_id == service_id).all()
def dao_add_and_commit_whitelisted_contacts(objs):
@@ -12,4 +13,5 @@ def dao_add_and_commit_whitelisted_contacts(objs):
def dao_remove_service_whitelist(service_id):
return ServiceWhitelist.query.filter(ServiceWhitelist.service_id == service_id).delete()
return ServiceWhitelist.query.filter(
ServiceWhitelist.service_id == service_id).delete()