darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-07-24 18:09:41 -04:00
Code Issues Packages Projects Releases Wiki Activity

Corrected code to handle template types

Browse Source
This commit is contained in:
Ken Tsang
2017-04-18 16:19:25 +01:00
parent ed953e992d
commit fc0cfa5dbf
3 changed files with 37 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
app/dao/templates_dao.py
View File

@@ -42,7 +42,16 @@ def dao_get_template_by_id(template_id, version=None):
return Template.query.filter_by(id=template_id).one()
def dao_get_all_templates_for_service(service_id):
def dao_get_all_templates_for_service(service_id, template_type=None):
if template_type is not None:
return Template.query.filter_by(
service_id=service_id,
template_type=template_type,
archived=False
).order_by(
desc(Template.created_at)
).all()
return Template.query.filter_by(
service_id=service_id,
archived=False

4
app/v2/templates/get_templates.py
View File

@@ -10,9 +10,9 @@ from app.v2.templates.templates_schemas import get_all_template_request
@v2_templates_blueprint.route("/", methods=['GET'])
def get_templates():
validate(request.args.to_dict(), get_all_template_request)
data = validate(request.args.to_dict(), get_all_template_request)
templates = templates_dao.dao_get_all_templates_for_service(api_user.service_id)
templates = templates_dao.dao_get_all_templates_for_service(api_user.service_id, data.get('type'))
return jsonify(
templates=[template.serialize() for template in templates]