LukeParke/connexion
1
0
Fork 0
mirror of https://github.com/LukeHagar/connexion.git synced 2025-12-11 04:19:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

https://github.com/zalando/connexion/issues/32 Reduce code repetition

Browse Source
This commit is contained in:
João Santos
2015-07-13 15:46:34 +02:00
parent 6d5297fbe0
commit f5ef93bd2f
2 changed files with 47 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
connexion/decorators/validation.py
View File

@@ -29,7 +29,26 @@ TYPE_MAP = {'integer': int,
'boolean': bool} # map of swagger types to python types
def validate_schema(data, schema) -> flask.Response:
class RequestBodyValidator:
def __init__(self, schema):
self.schema = schema
def __call__(self, function: types.FunctionType) -> types.FunctionType:
@functools.wraps(function)
def wrapper(*args, **kwargs):
data = flask.request.json
logger.debug("%s validating schema...", flask.request.url)
error = self.validate_schema(data, self.schema)
if error:
return error
response = function(*args, **kwargs)
return response
return wrapper
def validate_schema(self, data, schema) -> flask.Response:
schema_type = schema.get('type')
log_extra = {'url': flask.request.url, 'schema_type': schema_type}
@@ -39,10 +58,9 @@ def validate_schema(data, schema) -> flask.Response:
logger.error("Wrong data type, expected 'list' got '%s'", actual_type_name, extra=log_extra)
return problem(400, 'Bad Request', "Wrong type, expected 'array' got '{}'".format(actual_type_name))
for item in data:
error = validate_schema(item, schema.get('items'))
error = self.validate_schema(item, schema.get('items'))
if error:
return error
elif schema_type == 'object':
if not isinstance(data, dict):
actual_type_name = type(data).__name__
@@ -62,13 +80,9 @@ def validate_schema(data, schema) -> flask.Response:
for key in data.keys():
key_properties = schema['properties'].get(key)
if key_properties:
expected_type = TYPE_MAP.get(key_properties['type']) # type: type
if expected_type and not isinstance(data[key], expected_type):
expected_type_name = expected_type.__name__
actual_type_name = type(data[key]).__name__
logger.error("'%s' is not a '%s'", key, expected_type_name)
return problem(400, 'Bad Request',
"Wrong type, expected '{}' got '{}'".format(expected_type_name, actual_type_name))
error = self.validate_schema(data[key], key_properties)
if error:
return error
else:
expected_type = TYPE_MAP.get(schema_type) # type: type
actual_type = type(data) # type: type
@@ -78,23 +92,3 @@ def validate_schema(data, schema) -> flask.Response:
logger.error("'%s' is not a '%s'", data, expected_type_name)
return problem(400, 'Bad Request',
"Wrong type, expected '{}' got '{}'".format(schema_type, actual_type_name))
class RequestBodyValidator:
def __init__(self, schema):
self.schema = schema
def __call__(self, function: types.FunctionType) -> types.FunctionType:
@functools.wraps(function)
def wrapper(*args, **kwargs):
data = flask.request.json
logger.debug("%s validating schema...", flask.request.url)
error = validate_schema(data, self.schema)
if error:
return error
response = function(*args, **kwargs)
return response
return wrapper

2
tests/test_app.py
View File

@@ -247,7 +247,7 @@ def test_schema(app):
assert bad_type.content_type == 'application/problem+json'
bad_type_response = json.loads(bad_type.data.decode()) # type: dict
assert bad_type_response['title'] == 'Bad Request'
assert bad_type_response['detail'] == "Wrong type, expected 'str' got 'int'"
assert bad_type_response['detail'] == "Wrong type, expected 'string' got 'int'"
good_request = app_client.post('/v1.0/test_schema', headers=headers,
data=json.dumps({'image_version': 'version'})) # type: flask.Response