@@ -161,7 +161,7 @@ def continue_etag(request: Request, resource: Resource) -> bool:
161161
162162 def call_single_resource (
163163 self , request : Request , resource_endpoint : str , resource_id : str , ** kwargs
164- ):
164+ ) -> Response :
165165 find_endpoint = "/" + resource_endpoint
166166 resource_type = self .backend .get_resource_type_by_endpoint (find_endpoint )
167167 if not resource_type :
@@ -264,7 +264,7 @@ def get_attrs_from_request(request: Request) -> Dict:
264264 raise BadRequest
265265 return ret
266266
267- def build_search_request (self , request : Request ):
267+ def build_search_request (self , request : Request ) -> SearchRequest :
268268 """Constructs a SearchRequest object from a werkzeug request.
269269
270270 :param request: werkzeug request
@@ -327,7 +327,9 @@ def query_resource(self, request: Request, resource: Optional[ResourceType]):
327327 resources = resources ,
328328 )
329329
330- def call_resource (self , request : Request , resource_endpoint : str , ** kwargs ):
330+ def call_resource (
331+ self , request : Request , resource_endpoint : str , ** kwargs
332+ ) -> Response :
331333 resource_type = self .backend .get_resource_type_by_endpoint (
332334 "/" + resource_endpoint
333335 )
@@ -359,14 +361,16 @@ def call_resource(self, request: Request, resource_endpoint: str, **kwargs):
359361 headers = {"Location" : created_resource .meta .location },
360362 )
361363
362- def call_query_all (self , request : Request , ** kwargs ):
364+ def call_query_all (self , request : Request , ** kwargs ) -> Response :
363365 return self .make_response (
364366 self .query_resource (request , None ).model_dump (
365367 scim_ctx = Context .RESOURCE_QUERY_RESPONSE ,
366368 )
367369 )
368370
369- def call_resource_search (self , request : Request , resource_endpoint : str , ** kwargs ):
371+ def call_resource_search (
372+ self , request : Request , resource_endpoint : str , ** kwargs
373+ ) -> Response :
370374 resource_type = self .backend .get_resource_type_by_endpoint (
371375 "/" + resource_endpoint
372376 )
@@ -410,7 +414,7 @@ def check_auth(self, request: Request):
410414 raise Unauthorized
411415
412416 @staticmethod
413- def make_response (content , status = 200 , ** kwargs ):
417+ def make_response (content , status = 200 , ** kwargs ) -> Response :
414418 """Constructs a werkzeug response from any JSON-serializable
415419 content."""
416420 etag = None
0 commit comments