-
-
Notifications
You must be signed in to change notification settings - Fork 773
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
274 additions
and
150 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,123 @@ | ||
| import logging | ||
| import re | ||
|
|
||
| from jsonschema import ValidationError | ||
|
|
||
| from ..exceptions import ExtraParameterProblem | ||
| from ..problem import problem | ||
| from ..types import coerce_type | ||
| from ..utils import is_null | ||
|
|
||
| logger = logging.getLogger('connexion.serialization.deserializers') | ||
|
|
||
|
|
||
| class ContentHandler(object): | ||
|
|
||
| def __init__(self, validator, schema, strict, is_null_value_valid): | ||
| self.schema = schema | ||
| self.strict_validation = strict | ||
| self.is_null_value_valid = is_null_value_valid | ||
| self.validator = validator | ||
| self.has_default = schema.get('default', False) | ||
|
|
||
| def validate_schema(self, data, url): | ||
| # type: (dict, AnyStr) -> Union[ConnexionResponse, None] | ||
| if self.is_null_value_valid and is_null(data): | ||
| return None | ||
|
|
||
| try: | ||
| self.validator.validate(data) | ||
| except ValidationError as exception: | ||
| logger.error("{url} validation error: {error}".format(url=url, | ||
| error=exception.message), | ||
| extra={'validator': 'body'}) | ||
| return problem(400, 'Bad Request', str(exception.message)) | ||
|
|
||
| return None | ||
|
|
||
| def _deser(self, request): | ||
| return request.body | ||
|
|
||
| def validate(self, request): | ||
| data = self._deser(request) | ||
| errors = self.validate_schema(data, request.url) | ||
| if errors and not self.has_default: | ||
| return errors | ||
|
|
||
|
|
||
| class StreamingContentHandler(ContentHandler): | ||
| name = "application/octet-stream" | ||
| regex = re.compile(r'^application\/octet-stream.*') | ||
|
|
||
| def validate(self, request): | ||
| # Don't validate, leave stream for user to read | ||
| pass | ||
|
|
||
|
|
||
| class JSONContentHandler(ContentHandler): | ||
| name = "application/json" | ||
| regex = re.compile(r'^application\/json.*|^.*\+json$') | ||
|
|
||
| def _deser(self, request): | ||
| data = request.json | ||
| empty_body = not(request.body or request.form or request.files) | ||
| if data is None and not empty_body and not self.is_null_value_valid: | ||
| # Content-Type is json but actual body was not parsed | ||
| return problem(400, | ||
| "Bad Request", | ||
| "Request body is not valid JSON" | ||
| ) | ||
| return data | ||
|
|
||
|
|
||
| def validate_parameter_list(request_params, spec_params): | ||
| request_params = set(request_params) | ||
| spec_params = set(spec_params) | ||
|
|
||
| return request_params.difference(spec_params) | ||
|
|
||
|
|
||
| class FormDataContentHandler(ContentHandler): | ||
| name = "application/x-www-form-urlencoded" | ||
| regex = re.compile( | ||
| r'^application\/x-www-form-urlencoded.*' | ||
| ) | ||
|
|
||
| def validate_formdata_parameter_list(self, request): | ||
| request_params = request.form.keys() | ||
| spec_params = self.schema.get('properties', {}).keys() | ||
| return validate_parameter_list(request_params, spec_params) | ||
|
|
||
| def _deser(self, request): | ||
| data = dict(request.form.items()) or \ | ||
| (request.body if len(request.body) > 0 else {}) | ||
| data.update(dict.fromkeys(request.files, '')) # validator expects string.. | ||
| logger.debug('%s validating schema...', request.url) | ||
|
|
||
| if self.strict_validation: | ||
| formdata_errors = self.validate_formdata_parameter_list(request) | ||
| if formdata_errors: | ||
| raise ExtraParameterProblem(formdata_errors, []) | ||
|
|
||
| if data: | ||
| props = self.schema.get("properties", {}) | ||
| for k, param_defn in props.items(): | ||
| if k in data: | ||
| data[k] = coerce_type(param_defn, data[k], 'requestBody', k) | ||
|
|
||
| return data | ||
|
|
||
|
|
||
| class MultiPartFormDataContentHandler(FormDataContentHandler): | ||
| name = "multipart/form-data" | ||
| regex = re.compile( | ||
| r'^multipart\/form-data.*' | ||
| ) | ||
|
|
||
|
|
||
| DEFAULT_DESERIALIZERS = [ | ||
| StreamingContentHandler, | ||
| JSONContentHandler, | ||
| FormDataContentHandler, | ||
| MultiPartFormDataContentHandler | ||
| ] |
Oops, something went wrong.