Skip to content

Commit

Permalink
regenerate client
Browse files Browse the repository at this point in the history
  • Loading branch information
@ryuwd
ryuwd committed Sep 21, 2024
1 parent 471dd9f commit 3bee6e1
Show file tree
Hide file tree
Showing 2 changed files with 38 additions and 7 deletions.
14 changes: 12 additions & 2 deletions diracx-client/src/diracx/client/aio/operations/_operations.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1997,13 +1997,17 @@ async def get_job_status_bulk(
return deserialized # type: ignore

@distributed_trace_async
async def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
async def reschedule_bulk_jobs(
self, *, job_ids: List[int], reset_jobs: bool = False, **kwargs: Any
) -> Any:
"""Reschedule Bulk Jobs.
Reschedule Bulk Jobs.
:keyword job_ids: Required.
:paramtype job_ids: list[int]
:keyword reset_jobs: Default value is False.
:paramtype reset_jobs: bool
:return: any
:rtype: any
:raises ~azure.core.exceptions.HttpResponseError:
Expand All @@ -2025,6 +2029,7 @@ async def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> An

_request = build_jobs_reschedule_bulk_jobs_request(
job_ids=job_ids,
reset_jobs=reset_jobs,
headers=_headers,
params=_params,
)
Expand Down Expand Up @@ -2053,13 +2058,17 @@ async def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> An
return deserialized # type: ignore

@distributed_trace_async
async def reschedule_single_job(self, job_id: int, **kwargs: Any) -> Any:
async def reschedule_single_job(
self, job_id: int, *, reset_job: bool = False, **kwargs: Any
) -> Any:
"""Reschedule Single Job.
Reschedule Single Job.
:param job_id: Required.
:type job_id: int
:keyword reset_job: Default value is False.
:paramtype reset_job: bool
:return: any
:rtype: any
:raises ~azure.core.exceptions.HttpResponseError:
Expand All @@ -2081,6 +2090,7 @@ async def reschedule_single_job(self, job_id: int, **kwargs: Any) -> Any:

_request = build_jobs_reschedule_single_job_request(
job_id=job_id,
reset_job=reset_job,
headers=_headers,
params=_params,
)
Expand Down
31 changes: 26 additions & 5 deletions diracx-client/src/diracx/client/operations/_operations.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -619,7 +619,7 @@ def build_jobs_get_job_status_bulk_request(


def build_jobs_reschedule_bulk_jobs_request(
*, job_ids: List[int], **kwargs: Any
*, job_ids: List[int], reset_jobs: bool = False, **kwargs: Any
) -> HttpRequest:
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})
Expand All @@ -631,6 +631,8 @@ def build_jobs_reschedule_bulk_jobs_request(

# Construct parameters
_params["job_ids"] = _SERIALIZER.query("job_ids", job_ids, "[int]")
if reset_jobs is not None:
_params["reset_jobs"] = _SERIALIZER.query("reset_jobs", reset_jobs, "bool")

# Construct headers
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
Expand All @@ -640,8 +642,11 @@ def build_jobs_reschedule_bulk_jobs_request(
)


def build_jobs_reschedule_single_job_request(job_id: int, **kwargs: Any) -> HttpRequest:
def build_jobs_reschedule_single_job_request(
job_id: int, *, reset_job: bool = False, **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")

Expand All @@ -653,10 +658,16 @@ def build_jobs_reschedule_single_job_request(job_id: int, **kwargs: Any) -> Http

_url: str = _url.format(**path_format_arguments) # type: ignore

# Construct parameters
if reset_job is not None:
_params["reset_job"] = _SERIALIZER.query("reset_job", reset_job, "bool")

# Construct headers
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")

return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
return HttpRequest(
method="POST", url=_url, params=_params, headers=_headers, **kwargs
)


def build_jobs_remove_single_job_request(job_id: int, **kwargs: Any) -> HttpRequest:
Expand Down Expand Up @@ -2741,13 +2752,17 @@ def get_job_status_bulk(
return deserialized # type: ignore

@distributed_trace
def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
def reschedule_bulk_jobs(
self, *, job_ids: List[int], reset_jobs: bool = False, **kwargs: Any
) -> Any:
"""Reschedule Bulk Jobs.
Reschedule Bulk Jobs.
:keyword job_ids: Required.
:paramtype job_ids: list[int]
:keyword reset_jobs: Default value is False.
:paramtype reset_jobs: bool
:return: any
:rtype: any
:raises ~azure.core.exceptions.HttpResponseError:
Expand All @@ -2769,6 +2784,7 @@ def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:

_request = build_jobs_reschedule_bulk_jobs_request(
job_ids=job_ids,
reset_jobs=reset_jobs,
headers=_headers,
params=_params,
)
Expand Down Expand Up @@ -2797,13 +2813,17 @@ def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
return deserialized # type: ignore

@distributed_trace
def reschedule_single_job(self, job_id: int, **kwargs: Any) -> Any:
def reschedule_single_job(
self, job_id: int, *, reset_job: bool = False, **kwargs: Any
) -> Any:
"""Reschedule Single Job.
Reschedule Single Job.
:param job_id: Required.
:type job_id: int
:keyword reset_job: Default value is False.
:paramtype reset_job: bool
:return: any
:rtype: any
:raises ~azure.core.exceptions.HttpResponseError:
Expand All @@ -2825,6 +2845,7 @@ def reschedule_single_job(self, job_id: int, **kwargs: Any) -> Any:

_request = build_jobs_reschedule_single_job_request(
job_id=job_id,
reset_job=reset_job,
headers=_headers,
params=_params,
)
Expand Down

0 comments on commit 3bee6e1

Please sign in to comment.