2
2
from six import iteritems
3
3
4
4
from openapi_core .compat import lru_cache
5
+ from openapi_core .schema .content .factories import ContentFactory
5
6
from openapi_core .schema .extensions .generators import ExtensionsGenerator
6
7
from openapi_core .schema .links .generators import LinksGenerator
7
- from openapi_core .schema .media_types .generators import MediaTypeGenerator
8
8
from openapi_core .schema .parameters .generators import ParametersGenerator
9
9
from openapi_core .schema .responses .models import Response
10
10
@@ -20,30 +20,30 @@ def generate(self, responses):
20
20
response_deref = self .dereferencer .dereference (response )
21
21
description = response_deref ['description' ]
22
22
headers = response_deref .get ('headers' )
23
- content = response_deref .get ('content' )
23
+ content_spec = response_deref .get ('content' )
24
24
links_dict = response_deref .get ('links' , {})
25
25
links = self .links_generator .generate (links_dict )
26
26
27
27
extensions = self .extensions_generator .generate (response_deref )
28
28
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 )
32
32
33
33
parameters = None
34
34
if headers :
35
35
parameters = self .parameters_generator .generate (headers )
36
36
37
37
yield http_status , Response (
38
38
http_status , description ,
39
- content = media_types , headers = parameters , links = links ,
39
+ content = content , headers = parameters , links = links ,
40
40
extensions = extensions ,
41
41
)
42
42
43
43
@property
44
44
@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 )
47
47
48
48
@property
49
49
@lru_cache ()
0 commit comments