darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-10 01:56:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

More tests for rest and dao.

Browse Source
This commit is contained in:
Nicholas Staples
2016-01-13 12:14:21 +00:00
parent dad0fff4ba
commit 99820b3e85
5 changed files with 187 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/dao/templates_dao.py
View File

@@ -8,7 +8,10 @@ from app.models import (Template, Service)
def save_model_template(template, update_dict=None):
if update_dict:
update_dict.pop('id', None)
service_id = update_dict.pop('service')
Template.query.filter_by(id=template.id).update(update_dict)
template.service = Service.query.get(service_id)
else:
db.session.add(template)
db.session.commit()
@@ -23,7 +26,7 @@ def get_model_templates(template_id=None, service_id=None):
# TODO need better mapping from function params to sql query.
if template_id and service_id:
return Template.query.filter_by(
id=template_id, service=Service.get(service_id)).one()
id=template_id, service=Service.query.get(service_id)).one()
elif template_id:
return Template.query.filter_by(id=template_id).one()
elif service_id: