Skip to content

Commit e35c42f

Browse files
committed
feat: more type annotations
1 parent d68ed87 commit e35c42f

1 file changed

Lines changed: 10 additions & 6 deletions

File tree

scim2_server/provider.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)