Skip to content

Commit 144a4f0

Browse files
committed
Schema content refactor 2
1 parent f6efa84 commit 144a4f0

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

openapi_core/schema/responses/generators.py

+8-8
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22
from six import iteritems
33

44
from openapi_core.compat import lru_cache
5+
from openapi_core.schema.content.factories import ContentFactory
56
from openapi_core.schema.extensions.generators import ExtensionsGenerator
67
from openapi_core.schema.links.generators import LinksGenerator
7-
from openapi_core.schema.media_types.generators import MediaTypeGenerator
88
from openapi_core.schema.parameters.generators import ParametersGenerator
99
from openapi_core.schema.responses.models import Response
1010

@@ -20,30 +20,30 @@ def generate(self, responses):
2020
response_deref = self.dereferencer.dereference(response)
2121
description = response_deref['description']
2222
headers = response_deref.get('headers')
23-
content = response_deref.get('content')
23+
content_spec = response_deref.get('content')
2424
links_dict = response_deref.get('links', {})
2525
links = self.links_generator.generate(links_dict)
2626

2727
extensions = self.extensions_generator.generate(response_deref)
2828

29-
media_types = None
30-
if content:
31-
media_types = self.media_types_generator.generate(content)
29+
content = None
30+
if content_spec:
31+
content = self.content_factory.create(content_spec)
3232

3333
parameters = None
3434
if headers:
3535
parameters = self.parameters_generator.generate(headers)
3636

3737
yield http_status, Response(
3838
http_status, description,
39-
content=media_types, headers=parameters, links=links,
39+
content=content, headers=parameters, links=links,
4040
extensions=extensions,
4141
)
4242

4343
@property
4444
@lru_cache()
45-
def media_types_generator(self):
46-
return MediaTypeGenerator(self.dereferencer, self.schemas_registry)
45+
def content_factory(self):
46+
return ContentFactory(self.dereferencer, self.schemas_registry)
4747

4848
@property
4949
@lru_cache()

openapi_core/schema/responses/models.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def __init__(
1212
self.http_status = http_status
1313
self.description = description
1414
self.headers = headers and dict(headers) or {}
15-
self.content = content and Content(content) or Content()
15+
self.content = content
1616
self.links = links and dict(links) or {}
1717

1818
self.extensions = extensions and dict(extensions) or {}

0 commit comments

Comments
 (0)