Skip to content

Commit 949e9b5

Browse files
committed
chore: multipart typing
1 parent ea05d97 commit 949e9b5

1 file changed

Lines changed: 12 additions & 7 deletions

File tree

src/keycloak/connection.py

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -323,7 +323,7 @@ def raw_post(self, path: str, data: dict | str | MultipartEncoder, **kwargs: Any
323323
:param path: Path for request.
324324
:type path: str
325325
:param data: Payload for request.
326-
:type data: dict
326+
:type data: dict | str | MultipartEncoder
327327
:param kwargs: Additional arguments
328328
:type kwargs: dict
329329
:returns: Response the request.
@@ -347,14 +347,14 @@ def raw_post(self, path: str, data: dict | str | MultipartEncoder, **kwargs: Any
347347
msg = "Can't connect to server"
348348
raise KeycloakConnectionError(msg) from e
349349

350-
def raw_put(self, path: str, data: dict | str, **kwargs: Any) -> Response: # noqa: ANN401
350+
def raw_put(self, path: str, data: dict | str | MultipartEncoder, **kwargs: Any) -> Response: # noqa: ANN401
351351
"""
352352
Submit put request to the path.
353353
354354
:param path: Path for request.
355355
:type path: str
356356
:param data: Payload for request.
357-
:type data: dict
357+
:type data: dict | str | MultipartEncoder
358358
:param kwargs: Additional arguments
359359
:type kwargs: dict
360360
:returns: Response the request.
@@ -450,7 +450,7 @@ async def a_raw_post(
450450
:param path: Path for request.
451451
:type path: str
452452
:param data: Payload for request.
453-
:type data: dict
453+
:type data: dict | str | MultipartEncoder
454454
:param kwargs: Additional arguments
455455
:type kwargs: dict
456456
:returns: Response the request.
@@ -474,14 +474,19 @@ async def a_raw_post(
474474
msg = "Can't connect to server"
475475
raise KeycloakConnectionError(msg) from e
476476

477-
async def a_raw_put(self, path: str, data: dict | str, **kwargs: Any) -> AsyncResponse: # noqa: ANN401
477+
async def a_raw_put(
478+
self,
479+
path: str,
480+
data: dict | str | MultipartEncoder,
481+
**kwargs: Any, # noqa: ANN401
482+
) -> AsyncResponse:
478483
"""
479484
Submit put request to the path.
480485
481486
:param path: Path for request.
482487
:type path: str
483488
:param data: Payload for request.
484-
:type data: dict
489+
:type data: dict | str | MultipartEncoder
485490
:param kwargs: Additional arguments
486491
:type kwargs: dict
487492
:returns: Response the request.
@@ -548,7 +553,7 @@ def _prepare_httpx_request_content(data: dict | str | None | MultipartEncoder) -
548553
See https://www.python-httpx.org/compatibility/#request-content
549554
550555
:param data: the request content
551-
:type data: dict | str | None
556+
:type data: dict | str | None | MultipartEncoder
552557
:returns: A dict mapping the correct kwarg to the request content
553558
:rtype: dict
554559
"""

0 commit comments

Comments
 (0)