Skip to content

Commit 31af99e

Browse files
fix: Regenerated client
1 parent 43039e8 commit 31af99e

File tree

8 files changed

+505
-42
lines changed

8 files changed

+505
-42
lines changed

diracx-client/src/diracx/client/_generated/_client.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
AuthOperations,
2020
ConfigOperations,
2121
JobsOperations,
22+
LollygagOperations,
2223
PilotsOperations,
2324
WellKnownOperations,
2425
)
@@ -35,6 +36,8 @@ class Dirac: # pylint: disable=client-accepts-api-version-keyword
3536
:vartype config: _generated.operations.ConfigOperations
3637
:ivar jobs: JobsOperations operations
3738
:vartype jobs: _generated.operations.JobsOperations
39+
:ivar lollygag: LollygagOperations operations
40+
:vartype lollygag: _generated.operations.LollygagOperations
3841
:ivar pilots: PilotsOperations operations
3942
:vartype pilots: _generated.operations.PilotsOperations
4043
:keyword endpoint: Service URL. Required. Default value is "".
@@ -88,6 +91,9 @@ def __init__( # pylint: disable=missing-client-constructor-parameter-credential
8891
self.jobs = JobsOperations(
8992
self._client, self._config, self._serialize, self._deserialize
9093
)
94+
self.lollygag = LollygagOperations(
95+
self._client, self._config, self._serialize, self._deserialize
96+
)
9197
self.pilots = PilotsOperations(
9298
self._client, self._config, self._serialize, self._deserialize
9399
)

diracx-client/src/diracx/client/_generated/aio/_client.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
AuthOperations,
2020
ConfigOperations,
2121
JobsOperations,
22+
LollygagOperations,
2223
PilotsOperations,
2324
WellKnownOperations,
2425
)
@@ -35,6 +36,8 @@ class Dirac: # pylint: disable=client-accepts-api-version-keyword
3536
:vartype config: _generated.aio.operations.ConfigOperations
3637
:ivar jobs: JobsOperations operations
3738
:vartype jobs: _generated.aio.operations.JobsOperations
39+
:ivar lollygag: LollygagOperations operations
40+
:vartype lollygag: _generated.aio.operations.LollygagOperations
3841
:ivar pilots: PilotsOperations operations
3942
:vartype pilots: _generated.aio.operations.PilotsOperations
4043
:keyword endpoint: Service URL. Required. Default value is "".
@@ -88,6 +91,9 @@ def __init__( # pylint: disable=missing-client-constructor-parameter-credential
8891
self.jobs = JobsOperations(
8992
self._client, self._config, self._serialize, self._deserialize
9093
)
94+
self.lollygag = LollygagOperations(
95+
self._client, self._config, self._serialize, self._deserialize
96+
)
9197
self.pilots = PilotsOperations(
9298
self._client, self._config, self._serialize, self._deserialize
9399
)

diracx-client/src/diracx/client/_generated/aio/operations/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from ._operations import AuthOperations # type: ignore
1515
from ._operations import ConfigOperations # type: ignore
1616
from ._operations import JobsOperations # type: ignore
17+
from ._operations import LollygagOperations # type: ignore
1718
from ._operations import PilotsOperations # type: ignore
1819

1920
from ._patch import __all__ as _patch_all
@@ -25,6 +26,7 @@
2526
"AuthOperations",
2627
"ConfigOperations",
2728
"JobsOperations",
29+
"LollygagOperations",
2830
"PilotsOperations",
2931
]
3032
__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore

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

Lines changed: 197 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,9 @@
5353
build_jobs_summary_request,
5454
build_jobs_unassign_bulk_jobs_sandboxes_request,
5555
build_jobs_unassign_job_sandboxes_request,
56+
build_lollygag_get_gubbins_secrets_request,
57+
build_lollygag_get_owner_object_request,
58+
build_lollygag_insert_owner_object_request,
5659
build_pilots_get_pilot_info_request,
5760
build_well_known_get_installation_metadata_request,
5861
build_well_known_get_openid_configuration_request,
@@ -153,13 +156,15 @@ async def get_openid_configuration(
153156
return deserialized # type: ignore
154157

155158
@distributed_trace_async
156-
async def get_installation_metadata(self, **kwargs: Any) -> _models.Metadata:
159+
async def get_installation_metadata(
160+
self, **kwargs: Any
161+
) -> _models.ExtendedMetadata:
157162
"""Get Installation Metadata.
158163
159-
Get metadata about the dirac installation.
164+
Get Installation Metadata.
160165
161-
:return: Metadata
162-
:rtype: ~_generated.models.Metadata
166+
:return: ExtendedMetadata
167+
:rtype: ~_generated.models.ExtendedMetadata
163168
:raises ~azure.core.exceptions.HttpResponseError:
164169
"""
165170
error_map: MutableMapping = {
@@ -173,7 +178,7 @@ async def get_installation_metadata(self, **kwargs: Any) -> _models.Metadata:
173178
_headers = kwargs.pop("headers", {}) or {}
174179
_params = kwargs.pop("params", {}) or {}
175180

176-
cls: ClsType[_models.Metadata] = kwargs.pop("cls", None)
181+
cls: ClsType[_models.ExtendedMetadata] = kwargs.pop("cls", None)
177182

178183
_request = build_well_known_get_installation_metadata_request(
179184
headers=_headers,
@@ -196,7 +201,9 @@ async def get_installation_metadata(self, **kwargs: Any) -> _models.Metadata:
196201
)
197202
raise HttpResponseError(response=response)
198203

199-
deserialized = self._deserialize("Metadata", pipeline_response.http_response)
204+
deserialized = self._deserialize(
205+
"ExtendedMetadata", pipeline_response.http_response
206+
)
200207

201208
if cls:
202209
return cls(pipeline_response, deserialized, {}) # type: ignore
@@ -2404,6 +2411,190 @@ async def submit_jdl_jobs(
24042411
return deserialized # type: ignore
24052412

24062413

2414+
class LollygagOperations:
2415+
"""
2416+
.. warning::
2417+
**DO NOT** instantiate this class directly.
2418+
2419+
Instead, you should access the following operations through
2420+
:class:`~_generated.aio.Dirac`'s
2421+
:attr:`lollygag` attribute.
2422+
"""
2423+
2424+
models = _models
2425+
2426+
def __init__(self, *args, **kwargs) -> None:
2427+
input_args = list(args)
2428+
self._client: AsyncPipelineClient = (
2429+
input_args.pop(0) if input_args else kwargs.pop("client")
2430+
)
2431+
self._config: DiracConfiguration = (
2432+
input_args.pop(0) if input_args else kwargs.pop("config")
2433+
)
2434+
self._serialize: Serializer = (
2435+
input_args.pop(0) if input_args else kwargs.pop("serializer")
2436+
)
2437+
self._deserialize: Deserializer = (
2438+
input_args.pop(0) if input_args else kwargs.pop("deserializer")
2439+
)
2440+
2441+
@distributed_trace_async
2442+
async def insert_owner_object(self, owner_name: str, **kwargs: Any) -> Any:
2443+
"""Insert Owner Object.
2444+
2445+
Insert Owner Object.
2446+
2447+
:param owner_name: Required.
2448+
:type owner_name: str
2449+
:return: any
2450+
:rtype: any
2451+
:raises ~azure.core.exceptions.HttpResponseError:
2452+
"""
2453+
error_map: MutableMapping = {
2454+
401: ClientAuthenticationError,
2455+
404: ResourceNotFoundError,
2456+
409: ResourceExistsError,
2457+
304: ResourceNotModifiedError,
2458+
}
2459+
error_map.update(kwargs.pop("error_map", {}) or {})
2460+
2461+
_headers = kwargs.pop("headers", {}) or {}
2462+
_params = kwargs.pop("params", {}) or {}
2463+
2464+
cls: ClsType[Any] = kwargs.pop("cls", None)
2465+
2466+
_request = build_lollygag_insert_owner_object_request(
2467+
owner_name=owner_name,
2468+
headers=_headers,
2469+
params=_params,
2470+
)
2471+
_request.url = self._client.format_url(_request.url)
2472+
2473+
_stream = False
2474+
pipeline_response: PipelineResponse = (
2475+
await self._client._pipeline.run( # pylint: disable=protected-access
2476+
_request, stream=_stream, **kwargs
2477+
)
2478+
)
2479+
2480+
response = pipeline_response.http_response
2481+
2482+
if response.status_code not in [200]:
2483+
map_error(
2484+
status_code=response.status_code, response=response, error_map=error_map
2485+
)
2486+
raise HttpResponseError(response=response)
2487+
2488+
deserialized = self._deserialize("object", pipeline_response.http_response)
2489+
2490+
if cls:
2491+
return cls(pipeline_response, deserialized, {}) # type: ignore
2492+
2493+
return deserialized # type: ignore
2494+
2495+
@distributed_trace_async
2496+
async def get_owner_object(self, **kwargs: Any) -> Any:
2497+
"""Get Owner Object.
2498+
2499+
Get Owner Object.
2500+
2501+
:return: any
2502+
:rtype: any
2503+
:raises ~azure.core.exceptions.HttpResponseError:
2504+
"""
2505+
error_map: MutableMapping = {
2506+
401: ClientAuthenticationError,
2507+
404: ResourceNotFoundError,
2508+
409: ResourceExistsError,
2509+
304: ResourceNotModifiedError,
2510+
}
2511+
error_map.update(kwargs.pop("error_map", {}) or {})
2512+
2513+
_headers = kwargs.pop("headers", {}) or {}
2514+
_params = kwargs.pop("params", {}) or {}
2515+
2516+
cls: ClsType[Any] = kwargs.pop("cls", None)
2517+
2518+
_request = build_lollygag_get_owner_object_request(
2519+
headers=_headers,
2520+
params=_params,
2521+
)
2522+
_request.url = self._client.format_url(_request.url)
2523+
2524+
_stream = False
2525+
pipeline_response: PipelineResponse = (
2526+
await self._client._pipeline.run( # pylint: disable=protected-access
2527+
_request, stream=_stream, **kwargs
2528+
)
2529+
)
2530+
2531+
response = pipeline_response.http_response
2532+
2533+
if response.status_code not in [200]:
2534+
map_error(
2535+
status_code=response.status_code, response=response, error_map=error_map
2536+
)
2537+
raise HttpResponseError(response=response)
2538+
2539+
deserialized = self._deserialize("object", pipeline_response.http_response)
2540+
2541+
if cls:
2542+
return cls(pipeline_response, deserialized, {}) # type: ignore
2543+
2544+
return deserialized # type: ignore
2545+
2546+
@distributed_trace_async
2547+
async def get_gubbins_secrets(self, **kwargs: Any) -> Any:
2548+
"""Get Gubbins Secrets.
2549+
2550+
Does nothing but expects a GUBBINS_SENSEI permission.
2551+
2552+
:return: any
2553+
:rtype: any
2554+
:raises ~azure.core.exceptions.HttpResponseError:
2555+
"""
2556+
error_map: MutableMapping = {
2557+
401: ClientAuthenticationError,
2558+
404: ResourceNotFoundError,
2559+
409: ResourceExistsError,
2560+
304: ResourceNotModifiedError,
2561+
}
2562+
error_map.update(kwargs.pop("error_map", {}) or {})
2563+
2564+
_headers = kwargs.pop("headers", {}) or {}
2565+
_params = kwargs.pop("params", {}) or {}
2566+
2567+
cls: ClsType[Any] = kwargs.pop("cls", None)
2568+
2569+
_request = build_lollygag_get_gubbins_secrets_request(
2570+
headers=_headers,
2571+
params=_params,
2572+
)
2573+
_request.url = self._client.format_url(_request.url)
2574+
2575+
_stream = False
2576+
pipeline_response: PipelineResponse = (
2577+
await self._client._pipeline.run( # pylint: disable=protected-access
2578+
_request, stream=_stream, **kwargs
2579+
)
2580+
)
2581+
2582+
response = pipeline_response.http_response
2583+
2584+
if response.status_code not in [200]:
2585+
map_error(
2586+
status_code=response.status_code, response=response, error_map=error_map
2587+
)
2588+
raise HttpResponseError(response=response)
2589+
2590+
deserialized = self._deserialize("object", pipeline_response.http_response)
2591+
2592+
if cls:
2593+
return cls(pipeline_response, deserialized, {}) # type: ignore
2594+
2595+
return deserialized # type: ignore
2596+
2597+
24072598
class PilotsOperations:
24082599
"""
24092600
.. warning::

diracx-client/src/diracx/client/_generated/models/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from ._models import ( # type: ignore
1515
BodyAuthGetOidcToken,
1616
BodyAuthGetOidcTokenGrantType,
17+
ExtendedMetadata,
1718
GroupInfo,
1819
HTTPValidationError,
1920
HeartbeatData,
@@ -25,7 +26,6 @@
2526
JobStatusUpdate,
2627
JobSummaryParams,
2728
JobSummaryParamsSearchItem,
28-
Metadata,
2929
OpenIDConfiguration,
3030
SandboxDownloadResponse,
3131
SandboxInfo,
@@ -61,6 +61,7 @@
6161
__all__ = [
6262
"BodyAuthGetOidcToken",
6363
"BodyAuthGetOidcTokenGrantType",
64+
"ExtendedMetadata",
6465
"GroupInfo",
6566
"HTTPValidationError",
6667
"HeartbeatData",
@@ -72,7 +73,6 @@
7273
"JobStatusUpdate",
7374
"JobSummaryParams",
7475
"JobSummaryParamsSearchItem",
75-
"Metadata",
7676
"OpenIDConfiguration",
7777
"SandboxDownloadResponse",
7878
"SandboxInfo",

0 commit comments

Comments
 (0)