Media type finder

Author: p1c2u

openapi_core/contrib/falcon/handlers.py

@@ -5,7 +5,7 @@
 from falcon.status_codes import (
     HTTP_400, HTTP_404, HTTP_405, HTTP_415,
 )
-from openapi_core.schema.media_types.exceptions import InvalidContentType
+from openapi_core.templating.media_types.exceptions import MediaTypeNotFound
 from openapi_core.templating.paths.exceptions import (
     ServerNotFound, OperationNotFound, PathNotFound,
 )
@@ -17,7 +17,7 @@ class FalconOpenAPIErrorsHandler(object):
         ServerNotFound: 400,
         OperationNotFound: 405,
         PathNotFound: 404,
-        InvalidContentType: 415,
+        MediaTypeNotFound: 415,
     }
 
     FALCON_STATUS_CODES = {

openapi_core/contrib/flask/handlers.py

@@ -2,7 +2,7 @@
 from flask.globals import current_app
 from flask.json import dumps
 
-from openapi_core.schema.media_types.exceptions import InvalidContentType
+from openapi_core.templating.media_types.exceptions import MediaTypeNotFound
 from openapi_core.templating.paths.exceptions import (
     ServerNotFound, OperationNotFound, PathNotFound,
 )
@@ -14,7 +14,7 @@ class FlaskOpenAPIErrorsHandler(object):
         ServerNotFound: 400,
         OperationNotFound: 405,
         PathNotFound: 404,
-        InvalidContentType: 415,
+        MediaTypeNotFound: 415,
     }
 
     @classmethod

openapi_core/schema/content/exceptions.py

@@ -5,13 +5,3 @@
 
 class OpenAPIContentError(OpenAPIMappingError):
     pass
-
-
-@attr.s(hash=True)
-class MimeTypeNotFound(OpenAPIContentError):
-    mimetype = attr.ib()
-    availableMimetypes = attr.ib()
-
-    def __str__(self):
-        return "Mimetype not found: {0}. Valid mimetypes: {1}".format(
-            self.mimetype, self.availableMimetypes)

openapi_core/schema/content/models.py

@@ -1,21 +1,5 @@
 """OpenAPI core content models module"""
-import fnmatch
-
-from six import iteritems
-
-from openapi_core.schema.content.exceptions import MimeTypeNotFound
 
 
 class Content(dict):
-
-    def __getitem__(self, mimetype):
-        try:
-            return super(Content, self).__getitem__(mimetype)
-        except KeyError:
-            pass
-
-        for key, value in iteritems(self):
-            if fnmatch.fnmatch(mimetype, key):
-                return value
-
-        raise MimeTypeNotFound(mimetype, list(self.keys()))
+    pass

openapi_core/schema/request_bodies/models.py

@@ -1,6 +1,4 @@
 """OpenAPI core request bodies models module"""
-from openapi_core.schema.content.exceptions import MimeTypeNotFound
-from openapi_core.schema.media_types.exceptions import InvalidContentType
 
 
 class RequestBody(object):
@@ -11,9 +9,3 @@ def __init__(self, content, required=False, extensions=None):
         self.required = required
 
         self.extensions = extensions and dict(extensions) or {}
-
-    def __getitem__(self, mimetype):
-        try:
-            return self.content[mimetype]
-        except MimeTypeNotFound:
-            raise InvalidContentType(mimetype)

openapi_core/schema/responses/models.py

@@ -1,6 +1,4 @@
 """OpenAPI core responses models module"""
-from openapi_core.schema.content.exceptions import MimeTypeNotFound
-from openapi_core.schema.media_types.exceptions import InvalidContentType
 
 
 class Response(object):
@@ -15,12 +13,3 @@ def __init__(
         self.links = links and dict(links) or {}
 
         self.extensions = extensions and dict(extensions) or {}
-
-    def __getitem__(self, mimetype):
-        return self.get_content_type(mimetype)
-
-    def get_content_type(self, mimetype):
-        try:
-            return self.content[mimetype]
-        except MimeTypeNotFound:
-            raise InvalidContentType(mimetype)

openapi_core/templating/media_types/__init__.py

@@ -0,0 +1,19 @@
+import attr
+
+from openapi_core.exceptions import OpenAPIError
+
+
+class MediaTypeFinderError(OpenAPIError):
+    """Media type finder error"""
+
+
+@attr.s(hash=True)
+class MediaTypeNotFound(MediaTypeFinderError):
+    mimetype = attr.ib()
+    availableMimetypes = attr.ib()
+
+    def __str__(self):
+        return (
+            "Content for the following mimetype not found: {0}. "
+            "Valid mimetypes: {1}"
+        ).format(self.mimetype, self.availableMimetypes)

openapi_core/templating/media_types/exceptions.py

@@ -0,0 +1,24 @@
+"""OpenAPI core templating media types finders module"""
+import fnmatch
+
+from six import iteritems
+
+from openapi_core.templating.media_types.exceptions import MediaTypeNotFound
+
+
+class MediaTypeFinder(object):
+
+    def __init__(self, content):
+        self.content = content
+
+    def find(self, request):
+        try:
+            return self.content[request.mimetype]
+        except KeyError:
+            pass
+
+        for key, value in iteritems(self.content):
+            if fnmatch.fnmatch(request.mimetype, key):
+                return value
+
+        raise MediaTypeNotFound(request.mimetype, list(self.content.keys()))

openapi_core/templating/media_types/finders.py

@@ -4,12 +4,12 @@
 
 from openapi_core.casting.schemas.exceptions import CastError
 from openapi_core.deserializing.exceptions import DeserializeError
-from openapi_core.schema.media_types.exceptions import InvalidContentType
 from openapi_core.schema.parameters.exceptions import (
     MissingRequiredParameter, MissingParameter,
 )
 from openapi_core.schema.request_bodies.exceptions import MissingRequestBody
 from openapi_core.security.exceptions import SecurityError
+from openapi_core.templating.media_types.exceptions import MediaTypeFinderError
 from openapi_core.templating.paths.exceptions import PathError
 from openapi_core.unmarshalling.schemas.enums import UnmarshalContext
 from openapi_core.unmarshalling.schemas.exceptions import (
@@ -154,8 +154,9 @@ def _get_body(self, request, operation):
             return None, []
 
         try:
-            media_type = operation.request_body[request.mimetype]
-        except InvalidContentType as exc:
+            media_type = self._get_media_type(
+                operation.request_body.content, request)
+        except MediaTypeFinderError as exc:
             return None, [exc, ]
 
         try: