20
20
21
21
22
22
class FalconOpenAPIMiddleware (UnmarshallingProcessor ):
23
- request_class = FalconOpenAPIRequest
24
- response_class = FalconOpenAPIResponse
23
+ request_cls = FalconOpenAPIRequest
24
+ response_cls = FalconOpenAPIResponse
25
25
errors_handler = FalconOpenAPIErrorsHandler ()
26
26
27
27
def __init__ (
28
28
self ,
29
29
spec : Spec ,
30
30
request_unmarshaller_cls : Optional [RequestUnmarshallerType ] = None ,
31
31
response_unmarshaller_cls : Optional [ResponseUnmarshallerType ] = None ,
32
- request_class : Type [FalconOpenAPIRequest ] = FalconOpenAPIRequest ,
33
- response_class : Type [FalconOpenAPIResponse ] = FalconOpenAPIResponse ,
32
+ request_cls : Type [FalconOpenAPIRequest ] = FalconOpenAPIRequest ,
33
+ response_cls : Type [FalconOpenAPIResponse ] = FalconOpenAPIResponse ,
34
34
errors_handler : Optional [FalconOpenAPIErrorsHandler ] = None ,
35
35
** unmarshaller_kwargs : Any ,
36
36
):
@@ -40,8 +40,8 @@ def __init__(
40
40
response_unmarshaller_cls = response_unmarshaller_cls ,
41
41
** unmarshaller_kwargs ,
42
42
)
43
- self .request_class = request_class or self .request_class
44
- self .response_class = response_class or self .response_class
43
+ self .request_cls = request_cls or self .request_cls
44
+ self .response_cls = response_cls or self .response_cls
45
45
self .errors_handler = errors_handler or self .errors_handler
46
46
47
47
@classmethod
@@ -50,17 +50,17 @@ def from_spec(
50
50
spec : Spec ,
51
51
request_unmarshaller_cls : Optional [RequestUnmarshallerType ] = None ,
52
52
response_unmarshaller_cls : Optional [ResponseUnmarshallerType ] = None ,
53
- request_class : Type [FalconOpenAPIRequest ] = FalconOpenAPIRequest ,
54
- response_class : Type [FalconOpenAPIResponse ] = FalconOpenAPIResponse ,
53
+ request_cls : Type [FalconOpenAPIRequest ] = FalconOpenAPIRequest ,
54
+ response_cls : Type [FalconOpenAPIResponse ] = FalconOpenAPIResponse ,
55
55
errors_handler : Optional [FalconOpenAPIErrorsHandler ] = None ,
56
56
** unmarshaller_kwargs : Any ,
57
57
) -> "FalconOpenAPIMiddleware" :
58
58
return cls (
59
59
spec ,
60
60
request_unmarshaller_cls = request_unmarshaller_cls ,
61
61
response_unmarshaller_cls = response_unmarshaller_cls ,
62
- request_class = request_class ,
63
- response_class = response_class ,
62
+ request_cls = request_cls ,
63
+ response_cls = response_cls ,
64
64
errors_handler = errors_handler ,
65
65
** unmarshaller_kwargs ,
66
66
)
@@ -74,6 +74,8 @@ def process_request(self, req: Request, resp: Response) -> None: # type: ignore
74
74
def process_response ( # type: ignore
75
75
self , req : Request , resp : Response , resource : Any , req_succeeded : bool
76
76
) -> None :
77
+ if self .response_cls is None :
78
+ return resp
77
79
openapi_req = self ._get_openapi_request (req )
78
80
openapi_resp = self ._get_openapi_response (resp )
79
81
resp .context .openapi = super ().process_response (
@@ -101,9 +103,10 @@ def _handle_response_errors(
101
103
return self .errors_handler .handle (req , resp , response_result .errors )
102
104
103
105
def _get_openapi_request (self , request : Request ) -> FalconOpenAPIRequest :
104
- return self .request_class (request )
106
+ return self .request_cls (request )
105
107
106
108
def _get_openapi_response (
107
109
self , response : Response
108
110
) -> FalconOpenAPIResponse :
109
- return self .response_class (response )
111
+ assert self .response_cls is not None
112
+ return self .response_cls (response )
0 commit comments