darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-02 08:25:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

correct service_whitelist dao/rest functionality

Browse Source 
additionally added dao tests
This commit is contained in:
Leo Hemsted
2016-09-22 17:17:34 +01:00
parent 5c5ca26630
commit af0dbd14be
4 changed files with 58 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/dao/service_whitelist_dao.py
View File

@@ -1,11 +1,15 @@
from sqlalchemy import or_
from app import db
from app.models import 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_whitelisted_contact(obj):
db.session.add(obj)
def dao_add_and_commit_whitelisted_contacts(objs):
db.session.add_all(objs)
db.session.commit()
def dao_remove_service_whitelist(service_id):
return ServiceWhitelist.query.filter(ServiceWhitelist.service_id == service_id).delete()