2020
2121
2222class FalconOpenAPIMiddleware (UnmarshallingProcessor ):
23- request_class = FalconOpenAPIRequest
24- response_class = FalconOpenAPIResponse
23+ request_cls = FalconOpenAPIRequest
24+ response_cls = FalconOpenAPIResponse
2525 errors_handler = FalconOpenAPIErrorsHandler ()
2626
2727 def __init__ (
2828 self ,
2929 spec : Spec ,
3030 request_unmarshaller_cls : Optional [RequestUnmarshallerType ] = None ,
3131 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 ,
3434 errors_handler : Optional [FalconOpenAPIErrorsHandler ] = None ,
3535 ** unmarshaller_kwargs : Any ,
3636 ):
@@ -40,8 +40,8 @@ def __init__(
4040 response_unmarshaller_cls = response_unmarshaller_cls ,
4141 ** unmarshaller_kwargs ,
4242 )
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
4545 self .errors_handler = errors_handler or self .errors_handler
4646
4747 @classmethod
@@ -50,17 +50,17 @@ def from_spec(
5050 spec : Spec ,
5151 request_unmarshaller_cls : Optional [RequestUnmarshallerType ] = None ,
5252 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 ,
5555 errors_handler : Optional [FalconOpenAPIErrorsHandler ] = None ,
5656 ** unmarshaller_kwargs : Any ,
5757 ) -> "FalconOpenAPIMiddleware" :
5858 return cls (
5959 spec ,
6060 request_unmarshaller_cls = request_unmarshaller_cls ,
6161 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 ,
6464 errors_handler = errors_handler ,
6565 ** unmarshaller_kwargs ,
6666 )
@@ -74,6 +74,8 @@ def process_request(self, req: Request, resp: Response) -> None: # type: ignore
7474 def process_response ( # type: ignore
7575 self , req : Request , resp : Response , resource : Any , req_succeeded : bool
7676 ) -> None :
77+ if self .response_cls is None :
78+ return resp
7779 openapi_req = self ._get_openapi_request (req )
7880 openapi_resp = self ._get_openapi_response (resp )
7981 resp .context .openapi = super ().process_response (
@@ -101,9 +103,10 @@ def _handle_response_errors(
101103 return self .errors_handler .handle (req , resp , response_result .errors )
102104
103105 def _get_openapi_request (self , request : Request ) -> FalconOpenAPIRequest :
104- return self .request_class (request )
106+ return self .request_cls (request )
105107
106108 def _get_openapi_response (
107109 self , response : Response
108110 ) -> FalconOpenAPIResponse :
109- return self .response_class (response )
111+ assert self .response_cls is not None
112+ return self .response_cls (response )
0 commit comments