Refactored tests and get template endpoint

This commit is contained in:
Ken Tsang
2017-03-14 17:51:30 +00:00
parent 9e4b1b2bfc
commit a596a1bb87
2 changed files with 21 additions and 22 deletions

View File

@@ -1,6 +1,7 @@
import uuid
from flask import jsonify, request
from jsonschema.exceptions import ValidationError
from werkzeug.exceptions import abort
from app import api_user
@@ -11,21 +12,19 @@ from app.v2.template.template_schemas import get_template_by_id_request
@template_blueprint.route("/<template_id>", methods=['GET'])
@template_blueprint.route("/<template_id>/version/<version>", methods=['GET'])
@template_blueprint.route("/<template_id>/version/<int:version>", methods=['GET'])
def get_template_by_id(template_id, version=None):
try:
casted_id = uuid.UUID(template_id)
_data = {}
_data['id'] = template_id
if version:
_data['version'] = int(version)
_data['version'] = version
data = validate(_data, get_template_by_id_request)
except ValueError or AttributeError:
abort(404)
template = templates_dao.dao_get_template_by_id_and_service_id(
casted_id, api_user.service_id, data.get('version'))
template_id, api_user.service_id, data.get('version'))
return jsonify(template.serialize()), 200