diracx-client 0.0.1a50__py3-none-any.whl → 0.0.2__py3-none-any.whl

diracx/client/_generated/aio/operations/_operations.py

@@ -35,7 +35,6 @@ from ...operations._operations import (
     build_auth_initiate_authorization_flow_request,
     build_auth_initiate_device_flow_request,
     build_auth_revoke_refresh_token_by_jti_request,
-    build_auth_revoke_refresh_token_by_refresh_token_request,
     build_auth_userinfo_request,
     build_config_serve_config_request,
     build_jobs_add_heartbeat_request,
@@ -293,6 +292,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         raise_if_not_implemented(
             self.__class__,
             [
+                "revoke_refresh_token_by_refresh_token",
                 "get_oidc_token",
             ],
         )
@@ -583,59 +583,6 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
 
         return deserialized  # type: ignore
 
-    @distributed_trace_async
-    async def revoke_refresh_token_by_refresh_token(self, *, refresh_token: str, client_id: str, **kwargs: Any) -> str:
-        """Revoke Refresh Token By Refresh Token.
-
-        Revoke a refresh token.
-
-        :keyword refresh_token: Required.
-        :paramtype refresh_token: str
-        :keyword client_id: Required.
-        :paramtype client_id: str
-        :return: str
-        :rtype: str
-        :raises ~azure.core.exceptions.HttpResponseError:
-        """
-        error_map: MutableMapping = {
-            401: ClientAuthenticationError,
-            404: ResourceNotFoundError,
-            409: ResourceExistsError,
-            304: ResourceNotModifiedError,
-        }
-        error_map.update(kwargs.pop("error_map", {}) or {})
-
-        _headers = kwargs.pop("headers", {}) or {}
-        _params = kwargs.pop("params", {}) or {}
-
-        cls: ClsType[str] = kwargs.pop("cls", None)
-
-        _request = build_auth_revoke_refresh_token_by_refresh_token_request(
-            refresh_token=refresh_token,
-            client_id=client_id,
-            headers=_headers,
-            params=_params,
-        )
-        _request.url = self._client.format_url(_request.url)
-
-        _stream = False
-        pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
-            _request, stream=_stream, **kwargs
-        )
-
-        response = pipeline_response.http_response
-
-        if response.status_code not in [200]:
-            map_error(status_code=response.status_code, response=response, error_map=error_map)
-            raise HttpResponseError(response=response)
-
-        deserialized = self._deserialize("str", pipeline_response.http_response)
-
-        if cls:
-            return cls(pipeline_response, deserialized, {})  # type: ignore
-
-        return deserialized  # type: ignore
-
     @distributed_trace_async
     async def revoke_refresh_token_by_jti(self, jti: str, **kwargs: Any) -> str:
         """Revoke Refresh Token By Jti.

diracx/client/_generated/models/__init__.py

@@ -14,6 +14,7 @@ if TYPE_CHECKING:
 from ._models import (  # type: ignore
     BodyAuthGetOidcToken,
     BodyAuthGetOidcTokenGrantType,
+    BodyAuthRevokeRefreshTokenByRefreshToken,
     BodyJobsRescheduleJobs,
     BodyJobsUnassignBulkJobsSandboxes,
     GroupInfo,
@@ -65,6 +66,7 @@ from ._patch import patch_sdk as _patch_sdk
 __all__ = [
     "BodyAuthGetOidcToken",
     "BodyAuthGetOidcTokenGrantType",
+    "BodyAuthRevokeRefreshTokenByRefreshToken",
     "BodyJobsRescheduleJobs",
     "BodyJobsUnassignBulkJobsSandboxes",
     "GroupInfo",

diracx/client/_generated/models/_models.py

@@ -94,6 +94,46 @@ class BodyAuthGetOidcTokenGrantType(_serialization.Model):
     """OAuth2 Grant type."""
 
 
+class BodyAuthRevokeRefreshTokenByRefreshToken(_serialization.Model):
+    """Body_auth_revoke_refresh_token_by_refresh_token.
+
+    All required parameters must be populated in order to send to server.
+
+    :ivar token: The refresh token to revoke. Required.
+    :vartype token: str
+    :ivar token_type_hint: Hint for the type of token being revoked.
+    :vartype token_type_hint: str
+    :ivar client_id: The client ID of the application requesting the revocation.
+    :vartype client_id: str
+    """
+
+    _validation = {
+        "token": {"required": True},
+    }
+
+    _attribute_map = {
+        "token": {"key": "token", "type": "str"},
+        "token_type_hint": {"key": "token_type_hint", "type": "str"},
+        "client_id": {"key": "client_id", "type": "str"},
+    }
+
+    def __init__(
+        self, *, token: str, token_type_hint: Optional[str] = None, client_id: str = "myDIRACClientID", **kwargs: Any
+    ) -> None:
+        """
+        :keyword token: The refresh token to revoke. Required.
+        :paramtype token: str
+        :keyword token_type_hint: Hint for the type of token being revoked.
+        :paramtype token_type_hint: str
+        :keyword client_id: The client ID of the application requesting the revocation.
+        :paramtype client_id: str
+        """
+        super().__init__(**kwargs)
+        self.token = token
+        self.token_type_hint = token_type_hint
+        self.client_id = client_id
+
+
 class BodyJobsRescheduleJobs(_serialization.Model):
     """Body_jobs_reschedule_jobs.
 

diracx/client/_generated/operations/_operations.py

@@ -83,7 +83,7 @@ def build_well_known_get_security_txt_request(**kwargs: Any) -> HttpRequest:  #
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
-    _url = "/.well-known/.well-known/security.txt"
+    _url = "/.well-known/security.txt"
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -175,27 +175,6 @@ def build_auth_get_refresh_tokens_request(**kwargs: Any) -> HttpRequest:
     return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
 
 
-def build_auth_revoke_refresh_token_by_refresh_token_request(  # pylint: disable=name-too-long
-    *, refresh_token: str, client_id: str, **kwargs: Any
-) -> HttpRequest:
-    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
-    _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
-
-    accept = _headers.pop("Accept", "application/json")
-
-    # Construct URL
-    _url = "/api/auth/revoke"
-
-    # Construct parameters
-    _params["refresh_token"] = _SERIALIZER.query("refresh_token", refresh_token, "str")
-    _params["client_id"] = _SERIALIZER.query("client_id", client_id, "str")
-
-    # Construct headers
-    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
-
-    return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs)
-
-
 def build_auth_revoke_refresh_token_by_jti_request(  # pylint: disable=name-too-long
     jti: str, **kwargs: Any
 ) -> HttpRequest:
@@ -816,6 +795,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         raise_if_not_implemented(
             self.__class__,
             [
+                "revoke_refresh_token_by_refresh_token",
                 "get_oidc_token",
             ],
         )
@@ -1104,59 +1084,6 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
 
         return deserialized  # type: ignore
 
-    @distributed_trace
-    def revoke_refresh_token_by_refresh_token(self, *, refresh_token: str, client_id: str, **kwargs: Any) -> str:
-        """Revoke Refresh Token By Refresh Token.
-
-        Revoke a refresh token.
-
-        :keyword refresh_token: Required.
-        :paramtype refresh_token: str
-        :keyword client_id: Required.
-        :paramtype client_id: str
-        :return: str
-        :rtype: str
-        :raises ~azure.core.exceptions.HttpResponseError:
-        """
-        error_map: MutableMapping = {
-            401: ClientAuthenticationError,
-            404: ResourceNotFoundError,
-            409: ResourceExistsError,
-            304: ResourceNotModifiedError,
-        }
-        error_map.update(kwargs.pop("error_map", {}) or {})
-
-        _headers = kwargs.pop("headers", {}) or {}
-        _params = kwargs.pop("params", {}) or {}
-
-        cls: ClsType[str] = kwargs.pop("cls", None)
-
-        _request = build_auth_revoke_refresh_token_by_refresh_token_request(
-            refresh_token=refresh_token,
-            client_id=client_id,
-            headers=_headers,
-            params=_params,
-        )
-        _request.url = self._client.format_url(_request.url)
-
-        _stream = False
-        pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            _request, stream=_stream, **kwargs
-        )
-
-        response = pipeline_response.http_response
-
-        if response.status_code not in [200]:
-            map_error(status_code=response.status_code, response=response, error_map=error_map)
-            raise HttpResponseError(response=response)
-
-        deserialized = self._deserialize("str", pipeline_response.http_response)
-
-        if cls:
-            return cls(pipeline_response, deserialized, {})  # type: ignore
-
-        return deserialized  # type: ignore
-
     @distributed_trace
     def revoke_refresh_token_by_jti(self, jti: str, **kwargs: Any) -> str:
         """Revoke Refresh Token By Jti.

diracx/client/patches/auth/aio.py

@@ -6,6 +6,7 @@ __all__ list in the corresponding file in the patches directory.
 """
 
 from __future__ import annotations
+from ast import Dict
 
 __all__ = [
     "AuthOperations",
@@ -19,7 +20,7 @@ from ..._generated.aio.operations._operations import (
     _models,
     AuthOperations as _AuthOperations,
 )
-from .common import prepare_request, handle_response
+from .common import handle_revoke_response, prepare_oidc_request, handle_oidc_response, prepare_revoke_request
 
 
 class AuthOperations(_AuthOperations):
@@ -27,7 +28,7 @@ class AuthOperations(_AuthOperations):
     async def get_oidc_token(
         self, device_code: str, client_id: str, **kwargs
     ) -> TokenResponse | _models.DeviceFlowErrorResponse:
-        request = prepare_request(
+        request = prepare_oidc_request(
             device_code=device_code,
             client_id=client_id,
             format_url=self._client.format_url,
@@ -39,7 +40,30 @@ class AuthOperations(_AuthOperations):
             )
         )
 
-        response = handle_response(pipeline_response, self._deserialize)
+        response = handle_oidc_response(pipeline_response, self._deserialize)
         if isinstance(response, _models.DeviceFlowErrorResponse):
             return response
         return TokenResponse.model_validate(response.as_dict())
+
+    @distributed_trace_async
+    async def revoke_refresh_token_by_refresh_token(
+        self,
+        *,
+        token: str,
+        client_id: str,
+        token_type_hint: str = "refresh_token",
+        **kwargs,
+    ) -> str:
+        request = prepare_revoke_request(
+            token=token,
+            client_id=client_id,
+            token_type_hint=token_type_hint,
+            format_url=self._client.format_url,
+        )
+
+        pipeline_response: PipelineResponse = (
+            await self._client._pipeline.run(  # pylint: disable=protected-access
+                request, stream=False, **kwargs
+            )
+        )
+        return handle_revoke_response(pipeline_response, self._deserialize)

diracx/client/patches/auth/common.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 
 __all__ = [
-    "prepare_request",
-    "handle_response",
+    "prepare_oidc_request",
+    "handle_oidc_response",
 ]
 
 from typing import Any
@@ -18,20 +18,24 @@ from ..._generated.models import TokenResponse, DeviceFlowErrorResponse
 from ..._generated.operations._operations import _SERIALIZER
 
 
-def build_token_request(**kwargs: Any) -> HttpRequest:
+def build_request(**kwargs: Any) -> HttpRequest:
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
 
     accept = _headers.pop("Accept", "application/json")
 
-    _url = "/api/auth/token"
+    _url = kwargs.pop("url")
 
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
 
-    return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
+    _method = kwargs.pop("method")
 
+    return HttpRequest(method=_method, url=_url, headers=_headers, **kwargs)
 
-def prepare_request(device_code, client_id, format_url) -> HttpRequest:
-    request = build_token_request(
+
+def prepare_oidc_request(device_code, client_id, format_url) -> HttpRequest:
+    request = build_request(
+        method="POST",
+        url="/api/auth/token",
         data={
             "grant_type": "urn:ietf:params:oauth:grant-type:device_code",
             "device_code": device_code,
@@ -42,7 +46,21 @@ def prepare_request(device_code, client_id, format_url) -> HttpRequest:
     return request
 
 
-def handle_response(
+def prepare_revoke_request(token, client_id, token_type_hint, format_url) -> HttpRequest:
+    request = build_request(
+        method="POST",
+        url="/api/auth/revoke",
+        data={
+            "token": token,
+            "client_id": client_id,
+            "token_type_hint": token_type_hint,
+        }
+    )
+    request.url = format_url(request.url)
+    return request
+
+
+def handle_oidc_response(
     pipeline_response: PipelineResponse, deserialize
 ) -> TokenResponse | DeviceFlowErrorResponse:
     response = pipeline_response.http_response
@@ -54,3 +72,14 @@ def handle_response(
     else:
         map_error(status_code=response.status_code, response=response, error_map={})
         raise HttpResponseError(response=response)
+
+
+def handle_revoke_response(
+    pipeline_response: PipelineResponse, deserialize
+) -> str:
+    response = pipeline_response.http_response
+
+    if response.status_code != 200:
+        map_error(status_code=response.status_code, response=response, error_map={})
+        raise HttpResponseError(response=response)
+    return deserialize("str", pipeline_response)

diracx/client/patches/auth/sync.py

@@ -18,15 +18,15 @@ from ..._generated.operations._operations import (
     _models,
     AuthOperations as _AuthOperations,
 )
-from .common import prepare_request, handle_response
+from .common import handle_revoke_response, prepare_oidc_request, handle_oidc_response, prepare_revoke_request
 
 
 class AuthOperations(_AuthOperations):
     @distributed_trace
-    async def get_oidc_token(
+    def get_oidc_token(
         self, device_code: str, client_id: str, **kwargs
     ) -> _models.TokenResponse | _models.DeviceFlowErrorResponse:
-        request = prepare_request(
+        request = prepare_oidc_request(
             device_code=device_code,
             client_id=client_id,
             format_url=self._client.format_url,
@@ -38,4 +38,29 @@ class AuthOperations(_AuthOperations):
             )
         )
 
-        return handle_response(pipeline_response, self._deserialize)
+        return handle_oidc_response(pipeline_response, self._deserialize)
+
+
+    @distributed_trace
+    def revoke_refresh_token_by_refresh_token(
+        self,
+        *,
+        token: str,
+        client_id: str,
+        token_type_hint: str = "refresh_token",
+        **kwargs,
+    ) -> str:
+        request = prepare_revoke_request(
+            token=token,
+            client_id=client_id,
+            token_type_hint=token_type_hint,
+            format_url=self._client.format_url,
+        )
+
+        pipeline_response: PipelineResponse = (
+            self._client._pipeline.run(  # pylint: disable=protected-access
+                request, stream=False, **kwargs
+            )
+        )
+
+        return handle_revoke_response(pipeline_response, self._deserialize)

{diracx_client-0.0.1a50.dist-info → diracx_client-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diracx-client
-Version: 0.0.1a50
+Version: 0.0.2
 Summary: TODO
 License: GPL-3.0-only
 Classifier: Intended Audience :: Science/Research

{diracx_client-0.0.1a50.dist-info → diracx_client-0.0.2.dist-info}/RECORD

@@ -20,19 +20,19 @@ diracx/client/_generated/aio/_configuration.py,sha256=JWKZSmE3mnX6JR3SIHvVed20_8
 diracx/client/_generated/aio/_patch.py,sha256=sKaNgM0-20jF_R-S87eusC0aZa38QJEsLLzW-HgOjOQ,686
 diracx/client/_generated/aio/_vendor.py,sha256=6EFsP3ValI_vbZ_ivnRYHg_oRjZ5jDiLlUqvAYyDMhM,1936
 diracx/client/_generated/aio/operations/__init__.py,sha256=gtvv0durn3Cr8FUdFImqDcCtCVp7RKULjprmO4jOoeI,1070
-diracx/client/_generated/aio/operations/_operations.py,sha256=jp9l6BLGSVjCxzQ1ifKyvppqP8JRXeOWNVNXOnD3FSY,85821
+diracx/client/_generated/aio/operations/_operations.py,sha256=nGCYU9HJHKK7M4b-pbRYQW__uFk2X-4hQuvnzgsxXV0,83943
 diracx/client/_generated/aio/operations/_patch.py,sha256=jyk2R4wUaClz07zK9TiH6V_SxKakrHNZAUm35lqUC6k,816
-diracx/client/_generated/models/__init__.py,sha256=fqroW7xud-0lcwH-faWFHBDvDxgKYPNG3qkvb7SDiOk,2928
+diracx/client/_generated/models/__init__.py,sha256=QRqgQrc6KxUjebhh4G9A43Su7U9JSu5_3lnr0xx6p8w,3022
 diracx/client/_generated/models/_enums.py,sha256=7ANOVuRfxsJLHSGGh34Nl4PwCBjK7FAbCO9djayU2_0,1808
-diracx/client/_generated/models/_models.py,sha256=KFOvKVO8jJcy4_Mh49SpjR6PrRnsVKPUQmTJs2c7Onw,57791
+diracx/client/_generated/models/_models.py,sha256=ToHQKKDfgPxmuqW4VTRNh17a8izu3UdxEWmcTKe06KA,59218
 diracx/client/_generated/models/_patch.py,sha256=D8bxd8gAeLH7RWZNUuzTT7P_dyot9Cey91Rb-UwI2BI,1501
 diracx/client/_generated/operations/__init__.py,sha256=gtvv0durn3Cr8FUdFImqDcCtCVp7RKULjprmO4jOoeI,1070
-diracx/client/_generated/operations/_operations.py,sha256=9LrZ2_Qz0CQJwKh3teML-w4qtia4SuMtrvb6rMZrfZw,103882
+diracx/client/_generated/operations/_operations.py,sha256=b-oL_31wFU355mAU_9l7-oUBXqqUlzp-Exbv7nUBYFA,101275
 diracx/client/_generated/operations/_patch.py,sha256=xE3M4iVZE5LKlg9XfMveXmOMxFQZCDrPE40nba9VUq8,816
 diracx/client/patches/utils.py,sha256=rsw76XnstQgTxjqs5p95jwFsPuR3zkt5JTBc8VhqoAs,10733
-diracx/client/patches/auth/aio.py,sha256=7QKUtoNRZeMes8I1zeyVpahh4rIT-MY7IqvB4amNBYA,1483
-diracx/client/patches/auth/common.py,sha256=nwnfdcEQ7BEkyRBZG0TRS88k6isXdXyd1--7qy7_css,1776
-diracx/client/patches/auth/sync.py,sha256=AUag3E3W9rajGkzXfBuaRp6-kBlS9Ooi7R8QxMQ03tQ,1256
+diracx/client/patches/auth/aio.py,sha256=uqnz06XKLEwb6Iu8kqf2I7JVokQNQSoRfIMKpsm8Dog,2286
+diracx/client/patches/auth/common.py,sha256=7YVuFsfD4OahbLILSgtaG6VsyN1qSA7WCieC524wATQ,2617
+diracx/client/patches/auth/sync.py,sha256=2WY-sEVbYIrGBG1BKGgkdfAoo_xYtJnuRl4Be7THrPg,2016
 diracx/client/patches/client/aio.py,sha256=GYUVCgSvlEeJnjVoqdorBAEdP4H6JoC_4L8tlvtAZIE,5330
 diracx/client/patches/client/common.py,sha256=sWQ4-2iMHl5kwQzfAhny89uOWo8paoYpgp15Ak0UP5c,6243
 diracx/client/patches/client/sync.py,sha256=AVfGtr0O8KsfX2TrtikftfSNFrZRIaujz_MOuaHKk3U,4720
@@ -40,6 +40,6 @@ diracx/client/patches/jobs/aio.py,sha256=8zUlwKPyx0yUkoxEjUkdYZcErs9jSv1ZEnZTHQe
 diracx/client/patches/jobs/common.py,sha256=fTlQKEg06RSGG4ccFn-Mk_zSlZofPbpAJCzNtAZrPJA,4263
 diracx/client/patches/jobs/sync.py,sha256=3tJDAm4Hq0X8DztJtjWPv69UzvIEcv195BG1fruoUN4,1977
 _diracx_client_importer.pth,sha256=wJwJieKpvzPdCVaM-1oKiKN8OL3wBaHgrl8qyexFoKY,31
-diracx_client-0.0.1a50.dist-info/METADATA,sha256=GQtbBQKuN5qzpSss-ZKiegByqy1RR9y6tLbXMrmfja8,606
-diracx_client-0.0.1a50.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-diracx_client-0.0.1a50.dist-info/RECORD,,
+diracx_client-0.0.2.dist-info/METADATA,sha256=MQmnvFT0ThQ6BcITWvPhddIHD8579tPfbVZ3T0ZLKZI,603
+diracx_client-0.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+diracx_client-0.0.2.dist-info/RECORD,,