diracx-client 0.0.1a20__py3-none-any.whl → 0.0.1a22__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- diracx/client/generated/__init__.py +9 -3
- diracx/client/generated/_client.py +1 -1
- diracx/client/generated/_configuration.py +1 -1
- diracx/client/generated/_serialization.py +11 -11
- diracx/client/generated/_vendor.py +1 -1
- diracx/client/generated/aio/__init__.py +9 -3
- diracx/client/generated/aio/_client.py +1 -1
- diracx/client/generated/aio/_configuration.py +1 -1
- diracx/client/generated/aio/_vendor.py +1 -1
- diracx/client/generated/aio/operations/__init__.py +12 -6
- diracx/client/generated/aio/operations/_operations.py +356 -1152
- diracx/client/generated/models/__init__.py +50 -51
- diracx/client/generated/models/_enums.py +1 -33
- diracx/client/generated/models/_models.py +40 -119
- diracx/client/generated/operations/__init__.py +12 -6
- diracx/client/generated/operations/_operations.py +513 -1589
- diracx/client/patches/aio/utils.py +19 -34
- diracx/client/patches/utils.py +140 -75
- {diracx_client-0.0.1a20.dist-info → diracx_client-0.0.1a22.dist-info}/METADATA +1 -1
- diracx_client-0.0.1a22.dist-info/RECORD +36 -0
- diracx_client-0.0.1a20.dist-info/RECORD +0 -36
- {diracx_client-0.0.1a20.dist-info → diracx_client-0.0.1a22.dist-info}/WHEEL +0 -0
- {diracx_client-0.0.1a20.dist-info → diracx_client-0.0.1a22.dist-info}/entry_points.txt +0 -0
- {diracx_client-0.0.1a20.dist-info → diracx_client-0.0.1a22.dist-info}/top_level.txt +0 -0
@@ -1,7 +1,7 @@
|
|
1
1
|
# pylint: disable=too-many-lines
|
2
2
|
# coding=utf-8
|
3
3
|
# --------------------------------------------------------------------------
|
4
|
-
# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.
|
4
|
+
# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.3, generator: @autorest/python@6.27.1)
|
5
5
|
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
|
6
6
|
# --------------------------------------------------------------------------
|
7
7
|
from io import IOBase
|
@@ -36,28 +36,15 @@ from ...operations._operations import (
|
|
36
36
|
build_auth_userinfo_request,
|
37
37
|
build_config_serve_config_request,
|
38
38
|
build_jobs_assign_sandbox_to_job_request,
|
39
|
-
build_jobs_delete_bulk_jobs_request,
|
40
|
-
build_jobs_delete_single_job_request,
|
41
39
|
build_jobs_get_job_sandbox_request,
|
42
40
|
build_jobs_get_job_sandboxes_request,
|
43
|
-
build_jobs_get_job_status_bulk_request,
|
44
|
-
build_jobs_get_job_status_history_bulk_request,
|
45
41
|
build_jobs_get_sandbox_file_request,
|
46
|
-
build_jobs_get_single_job_request,
|
47
|
-
build_jobs_get_single_job_status_history_request,
|
48
|
-
build_jobs_get_single_job_status_request,
|
49
42
|
build_jobs_initiate_sandbox_upload_request,
|
50
|
-
build_jobs_kill_bulk_jobs_request,
|
51
|
-
build_jobs_kill_single_job_request,
|
52
43
|
build_jobs_remove_bulk_jobs_request,
|
53
|
-
build_jobs_remove_single_job_request,
|
54
44
|
build_jobs_reschedule_bulk_jobs_request,
|
55
|
-
build_jobs_reschedule_single_job_request,
|
56
45
|
build_jobs_search_request,
|
57
|
-
|
58
|
-
|
59
|
-
build_jobs_set_single_job_status_request,
|
60
|
-
build_jobs_submit_bulk_jobs_request,
|
46
|
+
build_jobs_set_job_statuses_request,
|
47
|
+
build_jobs_submit_bulk_jdl_jobs_request,
|
61
48
|
build_jobs_summary_request,
|
62
49
|
build_jobs_unassign_bulk_jobs_sandboxes_request,
|
63
50
|
build_jobs_unassign_job_sandboxes_request,
|
@@ -236,9 +223,25 @@ class AuthOperations: # pylint: disable=abstract-class-instantiated
|
|
236
223
|
"""Initiate Device Flow.
|
237
224
|
|
238
225
|
Initiate the device flow against DIRAC authorization Server.
|
239
|
-
|
240
|
-
|
241
|
-
|
226
|
+
|
227
|
+
Scope details:
|
228
|
+
|
229
|
+
|
230
|
+
*
|
231
|
+
If only VO is provided: Uses the default group and its properties for the VO.
|
232
|
+
|
233
|
+
*
|
234
|
+
If VO and group are provided: Uses the specified group and its properties for the VO.
|
235
|
+
|
236
|
+
*
|
237
|
+
If VO and properties are provided: Uses the default group and combines its properties with
|
238
|
+
the
|
239
|
+
provided properties.
|
240
|
+
|
241
|
+
*
|
242
|
+
If VO, group, and properties are provided: Uses the specified group and combines its
|
243
|
+
properties with the
|
244
|
+
provided properties.
|
242
245
|
|
243
246
|
Offers the user to go with the browser to
|
244
247
|
``auth/<vo>/device?user_code=XYZ``.
|
@@ -633,9 +636,9 @@ class AuthOperations: # pylint: disable=abstract-class-instantiated
|
|
633
636
|
async def authorization_flow(
|
634
637
|
self,
|
635
638
|
*,
|
636
|
-
response_type:
|
639
|
+
response_type: str,
|
637
640
|
code_challenge: str,
|
638
|
-
code_challenge_method:
|
641
|
+
code_challenge_method: str,
|
639
642
|
client_id: str,
|
640
643
|
redirect_uri: str,
|
641
644
|
scope: str,
|
@@ -648,16 +651,35 @@ class AuthOperations: # pylint: disable=abstract-class-instantiated
|
|
648
651
|
It will redirect to the actual OpenID server (IAM, CheckIn) to
|
649
652
|
perform a authorization code flow.
|
650
653
|
|
654
|
+
Scope details:
|
655
|
+
|
656
|
+
|
657
|
+
*
|
658
|
+
If only VO is provided: Uses the default group and its properties for the VO.
|
659
|
+
|
660
|
+
*
|
661
|
+
If VO and group are provided: Uses the specified group and its properties for the VO.
|
662
|
+
|
663
|
+
*
|
664
|
+
If VO and properties are provided: Uses the default group and combines its properties with
|
665
|
+
the
|
666
|
+
provided properties.
|
667
|
+
|
668
|
+
*
|
669
|
+
If VO, group, and properties are provided: Uses the specified group and combines its
|
670
|
+
properties with the
|
671
|
+
provided properties.
|
672
|
+
|
651
673
|
We set the user details obtained from the user authorize flow in a cookie
|
652
674
|
to be able to map the authorization flow with the corresponding
|
653
675
|
user authorize flow.
|
654
676
|
|
655
|
-
:keyword response_type:
|
656
|
-
:paramtype response_type: str
|
677
|
+
:keyword response_type: Required.
|
678
|
+
:paramtype response_type: str
|
657
679
|
:keyword code_challenge: Required.
|
658
680
|
:paramtype code_challenge: str
|
659
|
-
:keyword code_challenge_method:
|
660
|
-
:paramtype code_challenge_method: str
|
681
|
+
:keyword code_challenge_method: Required.
|
682
|
+
:paramtype code_challenge_method: str
|
661
683
|
:keyword client_id: Required.
|
662
684
|
:paramtype client_id: str
|
663
685
|
:keyword redirect_uri: Required.
|
@@ -885,7 +907,7 @@ class ConfigOperations:
|
|
885
907
|
return deserialized # type: ignore
|
886
908
|
|
887
909
|
|
888
|
-
class JobsOperations:
|
910
|
+
class JobsOperations:
|
889
911
|
"""
|
890
912
|
.. warning::
|
891
913
|
**DO NOT** instantiate this class directly.
|
@@ -1387,54 +1409,133 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1387
1409
|
|
1388
1410
|
return deserialized # type: ignore
|
1389
1411
|
|
1412
|
+
@distributed_trace_async
|
1413
|
+
async def remove_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
|
1414
|
+
"""Remove Bulk Jobs.
|
1415
|
+
|
1416
|
+
Fully remove a list of jobs from the WMS databases.
|
1417
|
+
|
1418
|
+
WARNING: This endpoint has been implemented for the compatibility with the legacy DIRAC WMS
|
1419
|
+
and the JobCleaningAgent. However, once this agent is ported to diracx, this endpoint should
|
1420
|
+
be removed, and a status change to Deleted (PATCH /jobs/status) should be used instead for any
|
1421
|
+
other purpose.
|
1422
|
+
|
1423
|
+
:keyword job_ids: Required.
|
1424
|
+
:paramtype job_ids: list[int]
|
1425
|
+
:return: any
|
1426
|
+
:rtype: any
|
1427
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
1428
|
+
"""
|
1429
|
+
error_map: MutableMapping = {
|
1430
|
+
401: ClientAuthenticationError,
|
1431
|
+
404: ResourceNotFoundError,
|
1432
|
+
409: ResourceExistsError,
|
1433
|
+
304: ResourceNotModifiedError,
|
1434
|
+
}
|
1435
|
+
error_map.update(kwargs.pop("error_map", {}) or {})
|
1436
|
+
|
1437
|
+
_headers = kwargs.pop("headers", {}) or {}
|
1438
|
+
_params = kwargs.pop("params", {}) or {}
|
1439
|
+
|
1440
|
+
cls: ClsType[Any] = kwargs.pop("cls", None)
|
1441
|
+
|
1442
|
+
_request = build_jobs_remove_bulk_jobs_request(
|
1443
|
+
job_ids=job_ids,
|
1444
|
+
headers=_headers,
|
1445
|
+
params=_params,
|
1446
|
+
)
|
1447
|
+
_request.url = self._client.format_url(_request.url)
|
1448
|
+
|
1449
|
+
_stream = False
|
1450
|
+
pipeline_response: PipelineResponse = (
|
1451
|
+
await self._client._pipeline.run( # pylint: disable=protected-access
|
1452
|
+
_request, stream=_stream, **kwargs
|
1453
|
+
)
|
1454
|
+
)
|
1455
|
+
|
1456
|
+
response = pipeline_response.http_response
|
1457
|
+
|
1458
|
+
if response.status_code not in [200]:
|
1459
|
+
map_error(
|
1460
|
+
status_code=response.status_code, response=response, error_map=error_map
|
1461
|
+
)
|
1462
|
+
raise HttpResponseError(response=response)
|
1463
|
+
|
1464
|
+
deserialized = self._deserialize("object", pipeline_response.http_response)
|
1465
|
+
|
1466
|
+
if cls:
|
1467
|
+
return cls(pipeline_response, deserialized, {}) # type: ignore
|
1468
|
+
|
1469
|
+
return deserialized # type: ignore
|
1470
|
+
|
1390
1471
|
@overload
|
1391
|
-
async def
|
1392
|
-
self,
|
1393
|
-
|
1394
|
-
|
1472
|
+
async def set_job_statuses(
|
1473
|
+
self,
|
1474
|
+
body: Dict[str, Dict[str, _models.JobStatusUpdate]],
|
1475
|
+
*,
|
1476
|
+
force: bool = False,
|
1477
|
+
content_type: str = "application/json",
|
1478
|
+
**kwargs: Any,
|
1479
|
+
) -> _models.SetJobStatusReturn:
|
1480
|
+
"""Set Job Statuses.
|
1395
1481
|
|
1396
|
-
|
1482
|
+
Set Job Statuses.
|
1397
1483
|
|
1398
1484
|
:param body: Required.
|
1399
|
-
:type body:
|
1485
|
+
:type body: dict[str, dict[str, ~generated.models.JobStatusUpdate]]
|
1486
|
+
:keyword force: Default value is False.
|
1487
|
+
:paramtype force: bool
|
1400
1488
|
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
1401
1489
|
Default value is "application/json".
|
1402
1490
|
:paramtype content_type: str
|
1403
|
-
:return:
|
1404
|
-
:rtype:
|
1491
|
+
:return: SetJobStatusReturn
|
1492
|
+
:rtype: ~generated.models.SetJobStatusReturn
|
1405
1493
|
:raises ~azure.core.exceptions.HttpResponseError:
|
1406
1494
|
"""
|
1407
1495
|
|
1408
1496
|
@overload
|
1409
|
-
async def
|
1410
|
-
self,
|
1411
|
-
|
1412
|
-
|
1497
|
+
async def set_job_statuses(
|
1498
|
+
self,
|
1499
|
+
body: IO[bytes],
|
1500
|
+
*,
|
1501
|
+
force: bool = False,
|
1502
|
+
content_type: str = "application/json",
|
1503
|
+
**kwargs: Any,
|
1504
|
+
) -> _models.SetJobStatusReturn:
|
1505
|
+
"""Set Job Statuses.
|
1413
1506
|
|
1414
|
-
|
1507
|
+
Set Job Statuses.
|
1415
1508
|
|
1416
1509
|
:param body: Required.
|
1417
1510
|
:type body: IO[bytes]
|
1511
|
+
:keyword force: Default value is False.
|
1512
|
+
:paramtype force: bool
|
1418
1513
|
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
1419
1514
|
Default value is "application/json".
|
1420
1515
|
:paramtype content_type: str
|
1421
|
-
:return:
|
1422
|
-
:rtype:
|
1516
|
+
:return: SetJobStatusReturn
|
1517
|
+
:rtype: ~generated.models.SetJobStatusReturn
|
1423
1518
|
:raises ~azure.core.exceptions.HttpResponseError:
|
1424
1519
|
"""
|
1425
1520
|
|
1426
1521
|
@distributed_trace_async
|
1427
|
-
async def
|
1428
|
-
self,
|
1429
|
-
|
1430
|
-
|
1522
|
+
async def set_job_statuses(
|
1523
|
+
self,
|
1524
|
+
body: Union[Dict[str, Dict[str, _models.JobStatusUpdate]], IO[bytes]],
|
1525
|
+
*,
|
1526
|
+
force: bool = False,
|
1527
|
+
**kwargs: Any,
|
1528
|
+
) -> _models.SetJobStatusReturn:
|
1529
|
+
"""Set Job Statuses.
|
1431
1530
|
|
1432
|
-
|
1531
|
+
Set Job Statuses.
|
1433
1532
|
|
1434
|
-
:param body: Is either a
|
1435
|
-
:type body:
|
1436
|
-
:
|
1437
|
-
:
|
1533
|
+
:param body: Is either a {str: {str: JobStatusUpdate}} type or a IO[bytes] type. Required.
|
1534
|
+
:type body: dict[str, dict[str, ~generated.models.JobStatusUpdate]] or IO[bytes]
|
1535
|
+
:keyword force: Default value is False.
|
1536
|
+
:paramtype force: bool
|
1537
|
+
:return: SetJobStatusReturn
|
1538
|
+
:rtype: ~generated.models.SetJobStatusReturn
|
1438
1539
|
:raises ~azure.core.exceptions.HttpResponseError:
|
1439
1540
|
"""
|
1440
1541
|
error_map: MutableMapping = {
|
@@ -1451,7 +1552,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1451
1552
|
content_type: Optional[str] = kwargs.pop(
|
1452
1553
|
"content_type", _headers.pop("Content-Type", None)
|
1453
1554
|
)
|
1454
|
-
cls: ClsType[
|
1555
|
+
cls: ClsType[_models.SetJobStatusReturn] = kwargs.pop("cls", None)
|
1455
1556
|
|
1456
1557
|
content_type = content_type or "application/json"
|
1457
1558
|
_json = None
|
@@ -1459,9 +1560,10 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1459
1560
|
if isinstance(body, (IOBase, bytes)):
|
1460
1561
|
_content = body
|
1461
1562
|
else:
|
1462
|
-
_json = self._serialize.body(body, "
|
1563
|
+
_json = self._serialize.body(body, "{{JobStatusUpdate}}")
|
1463
1564
|
|
1464
|
-
_request =
|
1565
|
+
_request = build_jobs_set_job_statuses_request(
|
1566
|
+
force=force,
|
1465
1567
|
content_type=content_type,
|
1466
1568
|
json=_json,
|
1467
1569
|
content=_content,
|
@@ -1486,7 +1588,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1486
1588
|
raise HttpResponseError(response=response)
|
1487
1589
|
|
1488
1590
|
deserialized = self._deserialize(
|
1489
|
-
"
|
1591
|
+
"SetJobStatusReturn", pipeline_response.http_response
|
1490
1592
|
)
|
1491
1593
|
|
1492
1594
|
if cls:
|
@@ -1495,13 +1597,17 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1495
1597
|
return deserialized # type: ignore
|
1496
1598
|
|
1497
1599
|
@distributed_trace_async
|
1498
|
-
async def
|
1499
|
-
|
1600
|
+
async def reschedule_bulk_jobs(
|
1601
|
+
self, *, job_ids: List[int], reset_jobs: bool = False, **kwargs: Any
|
1602
|
+
) -> Any:
|
1603
|
+
"""Reschedule Bulk Jobs.
|
1500
1604
|
|
1501
|
-
|
1605
|
+
Reschedule Bulk Jobs.
|
1502
1606
|
|
1503
1607
|
:keyword job_ids: Required.
|
1504
1608
|
:paramtype job_ids: list[int]
|
1609
|
+
:keyword reset_jobs: Default value is False.
|
1610
|
+
:paramtype reset_jobs: bool
|
1505
1611
|
:return: any
|
1506
1612
|
:rtype: any
|
1507
1613
|
:raises ~azure.core.exceptions.HttpResponseError:
|
@@ -1519,8 +1625,9 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1519
1625
|
|
1520
1626
|
cls: ClsType[Any] = kwargs.pop("cls", None)
|
1521
1627
|
|
1522
|
-
_request =
|
1628
|
+
_request = build_jobs_reschedule_bulk_jobs_request(
|
1523
1629
|
job_ids=job_ids,
|
1630
|
+
reset_jobs=reset_jobs,
|
1524
1631
|
headers=_headers,
|
1525
1632
|
params=_params,
|
1526
1633
|
)
|
@@ -1548,16 +1655,89 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1548
1655
|
|
1549
1656
|
return deserialized # type: ignore
|
1550
1657
|
|
1658
|
+
@overload
|
1659
|
+
async def search(
|
1660
|
+
self,
|
1661
|
+
body: Optional[_models.JobSearchParams] = None,
|
1662
|
+
*,
|
1663
|
+
page: int = 1,
|
1664
|
+
per_page: int = 100,
|
1665
|
+
content_type: str = "application/json",
|
1666
|
+
**kwargs: Any,
|
1667
|
+
) -> List[JSON]:
|
1668
|
+
"""Search.
|
1669
|
+
|
1670
|
+
Retrieve information about jobs.
|
1671
|
+
|
1672
|
+
**TODO: Add more docs**.
|
1673
|
+
|
1674
|
+
:param body: Default value is None.
|
1675
|
+
:type body: ~generated.models.JobSearchParams
|
1676
|
+
:keyword page: Default value is 1.
|
1677
|
+
:paramtype page: int
|
1678
|
+
:keyword per_page: Default value is 100.
|
1679
|
+
:paramtype per_page: int
|
1680
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
1681
|
+
Default value is "application/json".
|
1682
|
+
:paramtype content_type: str
|
1683
|
+
:return: list of JSON
|
1684
|
+
:rtype: list[JSON]
|
1685
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
1686
|
+
"""
|
1687
|
+
|
1688
|
+
@overload
|
1689
|
+
async def search(
|
1690
|
+
self,
|
1691
|
+
body: Optional[IO[bytes]] = None,
|
1692
|
+
*,
|
1693
|
+
page: int = 1,
|
1694
|
+
per_page: int = 100,
|
1695
|
+
content_type: str = "application/json",
|
1696
|
+
**kwargs: Any,
|
1697
|
+
) -> List[JSON]:
|
1698
|
+
"""Search.
|
1699
|
+
|
1700
|
+
Retrieve information about jobs.
|
1701
|
+
|
1702
|
+
**TODO: Add more docs**.
|
1703
|
+
|
1704
|
+
:param body: Default value is None.
|
1705
|
+
:type body: IO[bytes]
|
1706
|
+
:keyword page: Default value is 1.
|
1707
|
+
:paramtype page: int
|
1708
|
+
:keyword per_page: Default value is 100.
|
1709
|
+
:paramtype per_page: int
|
1710
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
1711
|
+
Default value is "application/json".
|
1712
|
+
:paramtype content_type: str
|
1713
|
+
:return: list of JSON
|
1714
|
+
:rtype: list[JSON]
|
1715
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
1716
|
+
"""
|
1717
|
+
|
1551
1718
|
@distributed_trace_async
|
1552
|
-
async def
|
1553
|
-
|
1719
|
+
async def search(
|
1720
|
+
self,
|
1721
|
+
body: Optional[Union[_models.JobSearchParams, IO[bytes]]] = None,
|
1722
|
+
*,
|
1723
|
+
page: int = 1,
|
1724
|
+
per_page: int = 100,
|
1725
|
+
**kwargs: Any,
|
1726
|
+
) -> List[JSON]:
|
1727
|
+
"""Search.
|
1554
1728
|
|
1555
|
-
|
1729
|
+
Retrieve information about jobs.
|
1556
1730
|
|
1557
|
-
:
|
1558
|
-
|
1559
|
-
:
|
1560
|
-
:
|
1731
|
+
**TODO: Add more docs**.
|
1732
|
+
|
1733
|
+
:param body: Is either a JobSearchParams type or a IO[bytes] type. Default value is None.
|
1734
|
+
:type body: ~generated.models.JobSearchParams or IO[bytes]
|
1735
|
+
:keyword page: Default value is 1.
|
1736
|
+
:paramtype page: int
|
1737
|
+
:keyword per_page: Default value is 100.
|
1738
|
+
:paramtype per_page: int
|
1739
|
+
:return: list of JSON
|
1740
|
+
:rtype: list[JSON]
|
1561
1741
|
:raises ~azure.core.exceptions.HttpResponseError:
|
1562
1742
|
"""
|
1563
1743
|
error_map: MutableMapping = {
|
@@ -1568,13 +1748,31 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1568
1748
|
}
|
1569
1749
|
error_map.update(kwargs.pop("error_map", {}) or {})
|
1570
1750
|
|
1571
|
-
_headers = kwargs.pop("headers", {}) or {}
|
1751
|
+
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
1572
1752
|
_params = kwargs.pop("params", {}) or {}
|
1573
1753
|
|
1574
|
-
|
1754
|
+
content_type: Optional[str] = kwargs.pop(
|
1755
|
+
"content_type", _headers.pop("Content-Type", None)
|
1756
|
+
)
|
1757
|
+
cls: ClsType[List[JSON]] = kwargs.pop("cls", None)
|
1575
1758
|
|
1576
|
-
|
1577
|
-
|
1759
|
+
content_type = content_type or "application/json"
|
1760
|
+
_json = None
|
1761
|
+
_content = None
|
1762
|
+
if isinstance(body, (IOBase, bytes)):
|
1763
|
+
_content = body
|
1764
|
+
else:
|
1765
|
+
if body is not None:
|
1766
|
+
_json = self._serialize.body(body, "JobSearchParams")
|
1767
|
+
else:
|
1768
|
+
_json = None
|
1769
|
+
|
1770
|
+
_request = build_jobs_search_request(
|
1771
|
+
page=page,
|
1772
|
+
per_page=per_page,
|
1773
|
+
content_type=content_type,
|
1774
|
+
json=_json,
|
1775
|
+
content=_content,
|
1578
1776
|
headers=_headers,
|
1579
1777
|
params=_params,
|
1580
1778
|
)
|
@@ -1589,991 +1787,77 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1589
1787
|
|
1590
1788
|
response = pipeline_response.http_response
|
1591
1789
|
|
1592
|
-
if response.status_code not in [200]:
|
1790
|
+
if response.status_code not in [200, 206]:
|
1593
1791
|
map_error(
|
1594
1792
|
status_code=response.status_code, response=response, error_map=error_map
|
1595
1793
|
)
|
1596
1794
|
raise HttpResponseError(response=response)
|
1597
1795
|
|
1598
|
-
|
1796
|
+
response_headers = {}
|
1797
|
+
if response.status_code == 206:
|
1798
|
+
response_headers["Content-Range"] = self._deserialize(
|
1799
|
+
"str", response.headers.get("Content-Range")
|
1800
|
+
)
|
1801
|
+
|
1802
|
+
deserialized = self._deserialize("[object]", pipeline_response.http_response)
|
1599
1803
|
|
1600
1804
|
if cls:
|
1601
|
-
return cls(pipeline_response, deserialized,
|
1805
|
+
return cls(pipeline_response, deserialized, response_headers) # type: ignore
|
1602
1806
|
|
1603
1807
|
return deserialized # type: ignore
|
1604
1808
|
|
1605
|
-
@
|
1606
|
-
async def
|
1607
|
-
|
1809
|
+
@overload
|
1810
|
+
async def summary(
|
1811
|
+
self,
|
1812
|
+
body: _models.JobSummaryParams,
|
1813
|
+
*,
|
1814
|
+
content_type: str = "application/json",
|
1815
|
+
**kwargs: Any,
|
1816
|
+
) -> Any:
|
1817
|
+
"""Summary.
|
1608
1818
|
|
1609
|
-
|
1819
|
+
Show information suitable for plotting.
|
1610
1820
|
|
1611
|
-
|
1612
|
-
|
1613
|
-
|
1614
|
-
|
1615
|
-
:
|
1616
|
-
:paramtype job_ids: list[int]
|
1617
|
-
:return: any
|
1618
|
-
:rtype: any
|
1619
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1620
|
-
"""
|
1621
|
-
error_map: MutableMapping = {
|
1622
|
-
401: ClientAuthenticationError,
|
1623
|
-
404: ResourceNotFoundError,
|
1624
|
-
409: ResourceExistsError,
|
1625
|
-
304: ResourceNotModifiedError,
|
1626
|
-
}
|
1627
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
1628
|
-
|
1629
|
-
_headers = kwargs.pop("headers", {}) or {}
|
1630
|
-
_params = kwargs.pop("params", {}) or {}
|
1631
|
-
|
1632
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
1633
|
-
|
1634
|
-
_request = build_jobs_remove_bulk_jobs_request(
|
1635
|
-
job_ids=job_ids,
|
1636
|
-
headers=_headers,
|
1637
|
-
params=_params,
|
1638
|
-
)
|
1639
|
-
_request.url = self._client.format_url(_request.url)
|
1640
|
-
|
1641
|
-
_stream = False
|
1642
|
-
pipeline_response: PipelineResponse = (
|
1643
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
1644
|
-
_request, stream=_stream, **kwargs
|
1645
|
-
)
|
1646
|
-
)
|
1647
|
-
|
1648
|
-
response = pipeline_response.http_response
|
1649
|
-
|
1650
|
-
if response.status_code not in [200]:
|
1651
|
-
map_error(
|
1652
|
-
status_code=response.status_code, response=response, error_map=error_map
|
1653
|
-
)
|
1654
|
-
raise HttpResponseError(response=response)
|
1655
|
-
|
1656
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
1657
|
-
|
1658
|
-
if cls:
|
1659
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
1660
|
-
|
1661
|
-
return deserialized # type: ignore
|
1662
|
-
|
1663
|
-
@distributed_trace_async
|
1664
|
-
async def get_job_status_bulk(
|
1665
|
-
self, *, job_ids: List[int], **kwargs: Any
|
1666
|
-
) -> Dict[str, _models.LimitedJobStatusReturn]:
|
1667
|
-
"""Get Job Status Bulk.
|
1668
|
-
|
1669
|
-
Get Job Status Bulk.
|
1670
|
-
|
1671
|
-
:keyword job_ids: Required.
|
1672
|
-
:paramtype job_ids: list[int]
|
1673
|
-
:return: dict mapping str to LimitedJobStatusReturn
|
1674
|
-
:rtype: dict[str, ~generated.models.LimitedJobStatusReturn]
|
1675
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1676
|
-
"""
|
1677
|
-
error_map: MutableMapping = {
|
1678
|
-
401: ClientAuthenticationError,
|
1679
|
-
404: ResourceNotFoundError,
|
1680
|
-
409: ResourceExistsError,
|
1681
|
-
304: ResourceNotModifiedError,
|
1682
|
-
}
|
1683
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
1684
|
-
|
1685
|
-
_headers = kwargs.pop("headers", {}) or {}
|
1686
|
-
_params = kwargs.pop("params", {}) or {}
|
1687
|
-
|
1688
|
-
cls: ClsType[Dict[str, _models.LimitedJobStatusReturn]] = kwargs.pop(
|
1689
|
-
"cls", None
|
1690
|
-
)
|
1691
|
-
|
1692
|
-
_request = build_jobs_get_job_status_bulk_request(
|
1693
|
-
job_ids=job_ids,
|
1694
|
-
headers=_headers,
|
1695
|
-
params=_params,
|
1696
|
-
)
|
1697
|
-
_request.url = self._client.format_url(_request.url)
|
1698
|
-
|
1699
|
-
_stream = False
|
1700
|
-
pipeline_response: PipelineResponse = (
|
1701
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
1702
|
-
_request, stream=_stream, **kwargs
|
1703
|
-
)
|
1704
|
-
)
|
1705
|
-
|
1706
|
-
response = pipeline_response.http_response
|
1707
|
-
|
1708
|
-
if response.status_code not in [200]:
|
1709
|
-
map_error(
|
1710
|
-
status_code=response.status_code, response=response, error_map=error_map
|
1711
|
-
)
|
1712
|
-
raise HttpResponseError(response=response)
|
1713
|
-
|
1714
|
-
deserialized = self._deserialize(
|
1715
|
-
"{LimitedJobStatusReturn}", pipeline_response.http_response
|
1716
|
-
)
|
1717
|
-
|
1718
|
-
if cls:
|
1719
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
1720
|
-
|
1721
|
-
return deserialized # type: ignore
|
1722
|
-
|
1723
|
-
@overload
|
1724
|
-
async def set_job_status_bulk(
|
1725
|
-
self,
|
1726
|
-
body: Dict[str, Dict[str, _models.JobStatusUpdate]],
|
1727
|
-
*,
|
1728
|
-
force: bool = False,
|
1729
|
-
content_type: str = "application/json",
|
1730
|
-
**kwargs: Any,
|
1731
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
1732
|
-
"""Set Job Status Bulk.
|
1733
|
-
|
1734
|
-
Set Job Status Bulk.
|
1735
|
-
|
1736
|
-
:param body: Required.
|
1737
|
-
:type body: dict[str, dict[str, ~generated.models.JobStatusUpdate]]
|
1738
|
-
:keyword force: Default value is False.
|
1739
|
-
:paramtype force: bool
|
1740
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
1741
|
-
Default value is "application/json".
|
1742
|
-
:paramtype content_type: str
|
1743
|
-
:return: dict mapping str to SetJobStatusReturn
|
1744
|
-
:rtype: dict[str, ~generated.models.SetJobStatusReturn]
|
1745
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1746
|
-
"""
|
1747
|
-
|
1748
|
-
@overload
|
1749
|
-
async def set_job_status_bulk(
|
1750
|
-
self,
|
1751
|
-
body: IO[bytes],
|
1752
|
-
*,
|
1753
|
-
force: bool = False,
|
1754
|
-
content_type: str = "application/json",
|
1755
|
-
**kwargs: Any,
|
1756
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
1757
|
-
"""Set Job Status Bulk.
|
1758
|
-
|
1759
|
-
Set Job Status Bulk.
|
1760
|
-
|
1761
|
-
:param body: Required.
|
1762
|
-
:type body: IO[bytes]
|
1763
|
-
:keyword force: Default value is False.
|
1764
|
-
:paramtype force: bool
|
1765
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
1766
|
-
Default value is "application/json".
|
1767
|
-
:paramtype content_type: str
|
1768
|
-
:return: dict mapping str to SetJobStatusReturn
|
1769
|
-
:rtype: dict[str, ~generated.models.SetJobStatusReturn]
|
1770
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1771
|
-
"""
|
1772
|
-
|
1773
|
-
@distributed_trace_async
|
1774
|
-
async def set_job_status_bulk(
|
1775
|
-
self,
|
1776
|
-
body: Union[Dict[str, Dict[str, _models.JobStatusUpdate]], IO[bytes]],
|
1777
|
-
*,
|
1778
|
-
force: bool = False,
|
1779
|
-
**kwargs: Any,
|
1780
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
1781
|
-
"""Set Job Status Bulk.
|
1782
|
-
|
1783
|
-
Set Job Status Bulk.
|
1784
|
-
|
1785
|
-
:param body: Is either a {str: {str: JobStatusUpdate}} type or a IO[bytes] type. Required.
|
1786
|
-
:type body: dict[str, dict[str, ~generated.models.JobStatusUpdate]] or IO[bytes]
|
1787
|
-
:keyword force: Default value is False.
|
1788
|
-
:paramtype force: bool
|
1789
|
-
:return: dict mapping str to SetJobStatusReturn
|
1790
|
-
:rtype: dict[str, ~generated.models.SetJobStatusReturn]
|
1791
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1792
|
-
"""
|
1793
|
-
error_map: MutableMapping = {
|
1794
|
-
401: ClientAuthenticationError,
|
1795
|
-
404: ResourceNotFoundError,
|
1796
|
-
409: ResourceExistsError,
|
1797
|
-
304: ResourceNotModifiedError,
|
1798
|
-
}
|
1799
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
1800
|
-
|
1801
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
1802
|
-
_params = kwargs.pop("params", {}) or {}
|
1803
|
-
|
1804
|
-
content_type: Optional[str] = kwargs.pop(
|
1805
|
-
"content_type", _headers.pop("Content-Type", None)
|
1806
|
-
)
|
1807
|
-
cls: ClsType[Dict[str, _models.SetJobStatusReturn]] = kwargs.pop("cls", None)
|
1808
|
-
|
1809
|
-
content_type = content_type or "application/json"
|
1810
|
-
_json = None
|
1811
|
-
_content = None
|
1812
|
-
if isinstance(body, (IOBase, bytes)):
|
1813
|
-
_content = body
|
1814
|
-
else:
|
1815
|
-
_json = self._serialize.body(body, "{{JobStatusUpdate}}")
|
1816
|
-
|
1817
|
-
_request = build_jobs_set_job_status_bulk_request(
|
1818
|
-
force=force,
|
1819
|
-
content_type=content_type,
|
1820
|
-
json=_json,
|
1821
|
-
content=_content,
|
1822
|
-
headers=_headers,
|
1823
|
-
params=_params,
|
1824
|
-
)
|
1825
|
-
_request.url = self._client.format_url(_request.url)
|
1826
|
-
|
1827
|
-
_stream = False
|
1828
|
-
pipeline_response: PipelineResponse = (
|
1829
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
1830
|
-
_request, stream=_stream, **kwargs
|
1831
|
-
)
|
1832
|
-
)
|
1833
|
-
|
1834
|
-
response = pipeline_response.http_response
|
1835
|
-
|
1836
|
-
if response.status_code not in [200]:
|
1837
|
-
map_error(
|
1838
|
-
status_code=response.status_code, response=response, error_map=error_map
|
1839
|
-
)
|
1840
|
-
raise HttpResponseError(response=response)
|
1841
|
-
|
1842
|
-
deserialized = self._deserialize(
|
1843
|
-
"{SetJobStatusReturn}", pipeline_response.http_response
|
1844
|
-
)
|
1845
|
-
|
1846
|
-
if cls:
|
1847
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
1848
|
-
|
1849
|
-
return deserialized # type: ignore
|
1850
|
-
|
1851
|
-
@distributed_trace_async
|
1852
|
-
async def get_job_status_history_bulk(
|
1853
|
-
self, *, job_ids: List[int], **kwargs: Any
|
1854
|
-
) -> Dict[str, List[_models.JobStatusReturn]]:
|
1855
|
-
"""Get Job Status History Bulk.
|
1856
|
-
|
1857
|
-
Get Job Status History Bulk.
|
1858
|
-
|
1859
|
-
:keyword job_ids: Required.
|
1860
|
-
:paramtype job_ids: list[int]
|
1861
|
-
:return: dict mapping str to list of JobStatusReturn
|
1862
|
-
:rtype: dict[str, list[~generated.models.JobStatusReturn]]
|
1863
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1864
|
-
"""
|
1865
|
-
error_map: MutableMapping = {
|
1866
|
-
401: ClientAuthenticationError,
|
1867
|
-
404: ResourceNotFoundError,
|
1868
|
-
409: ResourceExistsError,
|
1869
|
-
304: ResourceNotModifiedError,
|
1870
|
-
}
|
1871
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
1872
|
-
|
1873
|
-
_headers = kwargs.pop("headers", {}) or {}
|
1874
|
-
_params = kwargs.pop("params", {}) or {}
|
1875
|
-
|
1876
|
-
cls: ClsType[Dict[str, List[_models.JobStatusReturn]]] = kwargs.pop("cls", None)
|
1877
|
-
|
1878
|
-
_request = build_jobs_get_job_status_history_bulk_request(
|
1879
|
-
job_ids=job_ids,
|
1880
|
-
headers=_headers,
|
1881
|
-
params=_params,
|
1882
|
-
)
|
1883
|
-
_request.url = self._client.format_url(_request.url)
|
1884
|
-
|
1885
|
-
_stream = False
|
1886
|
-
pipeline_response: PipelineResponse = (
|
1887
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
1888
|
-
_request, stream=_stream, **kwargs
|
1889
|
-
)
|
1890
|
-
)
|
1891
|
-
|
1892
|
-
response = pipeline_response.http_response
|
1893
|
-
|
1894
|
-
if response.status_code not in [200]:
|
1895
|
-
map_error(
|
1896
|
-
status_code=response.status_code, response=response, error_map=error_map
|
1897
|
-
)
|
1898
|
-
raise HttpResponseError(response=response)
|
1899
|
-
|
1900
|
-
deserialized = self._deserialize(
|
1901
|
-
"{[JobStatusReturn]}", pipeline_response.http_response
|
1902
|
-
)
|
1903
|
-
|
1904
|
-
if cls:
|
1905
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
1906
|
-
|
1907
|
-
return deserialized # type: ignore
|
1908
|
-
|
1909
|
-
@distributed_trace_async
|
1910
|
-
async def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
|
1911
|
-
"""Reschedule Bulk Jobs.
|
1912
|
-
|
1913
|
-
Reschedule Bulk Jobs.
|
1914
|
-
|
1915
|
-
:keyword job_ids: Required.
|
1916
|
-
:paramtype job_ids: list[int]
|
1917
|
-
:return: any
|
1918
|
-
:rtype: any
|
1919
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1920
|
-
"""
|
1921
|
-
error_map: MutableMapping = {
|
1922
|
-
401: ClientAuthenticationError,
|
1923
|
-
404: ResourceNotFoundError,
|
1924
|
-
409: ResourceExistsError,
|
1925
|
-
304: ResourceNotModifiedError,
|
1926
|
-
}
|
1927
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
1928
|
-
|
1929
|
-
_headers = kwargs.pop("headers", {}) or {}
|
1930
|
-
_params = kwargs.pop("params", {}) or {}
|
1931
|
-
|
1932
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
1933
|
-
|
1934
|
-
_request = build_jobs_reschedule_bulk_jobs_request(
|
1935
|
-
job_ids=job_ids,
|
1936
|
-
headers=_headers,
|
1937
|
-
params=_params,
|
1938
|
-
)
|
1939
|
-
_request.url = self._client.format_url(_request.url)
|
1940
|
-
|
1941
|
-
_stream = False
|
1942
|
-
pipeline_response: PipelineResponse = (
|
1943
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
1944
|
-
_request, stream=_stream, **kwargs
|
1945
|
-
)
|
1946
|
-
)
|
1947
|
-
|
1948
|
-
response = pipeline_response.http_response
|
1949
|
-
|
1950
|
-
if response.status_code not in [200]:
|
1951
|
-
map_error(
|
1952
|
-
status_code=response.status_code, response=response, error_map=error_map
|
1953
|
-
)
|
1954
|
-
raise HttpResponseError(response=response)
|
1955
|
-
|
1956
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
1957
|
-
|
1958
|
-
if cls:
|
1959
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
1960
|
-
|
1961
|
-
return deserialized # type: ignore
|
1962
|
-
|
1963
|
-
@distributed_trace_async
|
1964
|
-
async def reschedule_single_job(self, job_id: int, **kwargs: Any) -> Any:
|
1965
|
-
"""Reschedule Single Job.
|
1966
|
-
|
1967
|
-
Reschedule Single Job.
|
1968
|
-
|
1969
|
-
:param job_id: Required.
|
1970
|
-
:type job_id: int
|
1971
|
-
:return: any
|
1972
|
-
:rtype: any
|
1973
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1974
|
-
"""
|
1975
|
-
error_map: MutableMapping = {
|
1976
|
-
401: ClientAuthenticationError,
|
1977
|
-
404: ResourceNotFoundError,
|
1978
|
-
409: ResourceExistsError,
|
1979
|
-
304: ResourceNotModifiedError,
|
1980
|
-
}
|
1981
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
1982
|
-
|
1983
|
-
_headers = kwargs.pop("headers", {}) or {}
|
1984
|
-
_params = kwargs.pop("params", {}) or {}
|
1985
|
-
|
1986
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
1987
|
-
|
1988
|
-
_request = build_jobs_reschedule_single_job_request(
|
1989
|
-
job_id=job_id,
|
1990
|
-
headers=_headers,
|
1991
|
-
params=_params,
|
1992
|
-
)
|
1993
|
-
_request.url = self._client.format_url(_request.url)
|
1994
|
-
|
1995
|
-
_stream = False
|
1996
|
-
pipeline_response: PipelineResponse = (
|
1997
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
1998
|
-
_request, stream=_stream, **kwargs
|
1999
|
-
)
|
2000
|
-
)
|
2001
|
-
|
2002
|
-
response = pipeline_response.http_response
|
2003
|
-
|
2004
|
-
if response.status_code not in [200]:
|
2005
|
-
map_error(
|
2006
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2007
|
-
)
|
2008
|
-
raise HttpResponseError(response=response)
|
2009
|
-
|
2010
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2011
|
-
|
2012
|
-
if cls:
|
2013
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2014
|
-
|
2015
|
-
return deserialized # type: ignore
|
2016
|
-
|
2017
|
-
@overload
|
2018
|
-
async def search(
|
2019
|
-
self,
|
2020
|
-
body: Optional[_models.JobSearchParams] = None,
|
2021
|
-
*,
|
2022
|
-
page: int = 1,
|
2023
|
-
per_page: int = 100,
|
2024
|
-
content_type: str = "application/json",
|
2025
|
-
**kwargs: Any,
|
2026
|
-
) -> List[JSON]:
|
2027
|
-
"""Search.
|
2028
|
-
|
2029
|
-
Retrieve information about jobs.
|
2030
|
-
|
2031
|
-
**TODO: Add more docs**.
|
2032
|
-
|
2033
|
-
:param body: Default value is None.
|
2034
|
-
:type body: ~generated.models.JobSearchParams
|
2035
|
-
:keyword page: Default value is 1.
|
2036
|
-
:paramtype page: int
|
2037
|
-
:keyword per_page: Default value is 100.
|
2038
|
-
:paramtype per_page: int
|
2039
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2040
|
-
Default value is "application/json".
|
2041
|
-
:paramtype content_type: str
|
2042
|
-
:return: list of JSON
|
2043
|
-
:rtype: list[JSON]
|
2044
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2045
|
-
"""
|
2046
|
-
|
2047
|
-
@overload
|
2048
|
-
async def search(
|
2049
|
-
self,
|
2050
|
-
body: Optional[IO[bytes]] = None,
|
2051
|
-
*,
|
2052
|
-
page: int = 1,
|
2053
|
-
per_page: int = 100,
|
2054
|
-
content_type: str = "application/json",
|
2055
|
-
**kwargs: Any,
|
2056
|
-
) -> List[JSON]:
|
2057
|
-
"""Search.
|
2058
|
-
|
2059
|
-
Retrieve information about jobs.
|
2060
|
-
|
2061
|
-
**TODO: Add more docs**.
|
2062
|
-
|
2063
|
-
:param body: Default value is None.
|
2064
|
-
:type body: IO[bytes]
|
2065
|
-
:keyword page: Default value is 1.
|
2066
|
-
:paramtype page: int
|
2067
|
-
:keyword per_page: Default value is 100.
|
2068
|
-
:paramtype per_page: int
|
2069
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2070
|
-
Default value is "application/json".
|
2071
|
-
:paramtype content_type: str
|
2072
|
-
:return: list of JSON
|
2073
|
-
:rtype: list[JSON]
|
2074
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2075
|
-
"""
|
2076
|
-
|
2077
|
-
@distributed_trace_async
|
2078
|
-
async def search(
|
2079
|
-
self,
|
2080
|
-
body: Optional[Union[_models.JobSearchParams, IO[bytes]]] = None,
|
2081
|
-
*,
|
2082
|
-
page: int = 1,
|
2083
|
-
per_page: int = 100,
|
2084
|
-
**kwargs: Any,
|
2085
|
-
) -> List[JSON]:
|
2086
|
-
"""Search.
|
2087
|
-
|
2088
|
-
Retrieve information about jobs.
|
2089
|
-
|
2090
|
-
**TODO: Add more docs**.
|
2091
|
-
|
2092
|
-
:param body: Is either a JobSearchParams type or a IO[bytes] type. Default value is None.
|
2093
|
-
:type body: ~generated.models.JobSearchParams or IO[bytes]
|
2094
|
-
:keyword page: Default value is 1.
|
2095
|
-
:paramtype page: int
|
2096
|
-
:keyword per_page: Default value is 100.
|
2097
|
-
:paramtype per_page: int
|
2098
|
-
:return: list of JSON
|
2099
|
-
:rtype: list[JSON]
|
2100
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2101
|
-
"""
|
2102
|
-
error_map: MutableMapping = {
|
2103
|
-
401: ClientAuthenticationError,
|
2104
|
-
404: ResourceNotFoundError,
|
2105
|
-
409: ResourceExistsError,
|
2106
|
-
304: ResourceNotModifiedError,
|
2107
|
-
}
|
2108
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2109
|
-
|
2110
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
2111
|
-
_params = kwargs.pop("params", {}) or {}
|
2112
|
-
|
2113
|
-
content_type: Optional[str] = kwargs.pop(
|
2114
|
-
"content_type", _headers.pop("Content-Type", None)
|
2115
|
-
)
|
2116
|
-
cls: ClsType[List[JSON]] = kwargs.pop("cls", None)
|
2117
|
-
|
2118
|
-
content_type = content_type or "application/json"
|
2119
|
-
_json = None
|
2120
|
-
_content = None
|
2121
|
-
if isinstance(body, (IOBase, bytes)):
|
2122
|
-
_content = body
|
2123
|
-
else:
|
2124
|
-
if body is not None:
|
2125
|
-
_json = self._serialize.body(body, "JobSearchParams")
|
2126
|
-
else:
|
2127
|
-
_json = None
|
2128
|
-
|
2129
|
-
_request = build_jobs_search_request(
|
2130
|
-
page=page,
|
2131
|
-
per_page=per_page,
|
2132
|
-
content_type=content_type,
|
2133
|
-
json=_json,
|
2134
|
-
content=_content,
|
2135
|
-
headers=_headers,
|
2136
|
-
params=_params,
|
2137
|
-
)
|
2138
|
-
_request.url = self._client.format_url(_request.url)
|
2139
|
-
|
2140
|
-
_stream = False
|
2141
|
-
pipeline_response: PipelineResponse = (
|
2142
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
2143
|
-
_request, stream=_stream, **kwargs
|
2144
|
-
)
|
2145
|
-
)
|
2146
|
-
|
2147
|
-
response = pipeline_response.http_response
|
2148
|
-
|
2149
|
-
if response.status_code not in [200, 206]:
|
2150
|
-
map_error(
|
2151
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2152
|
-
)
|
2153
|
-
raise HttpResponseError(response=response)
|
2154
|
-
|
2155
|
-
response_headers = {}
|
2156
|
-
if response.status_code == 206:
|
2157
|
-
response_headers["Content-Range"] = self._deserialize(
|
2158
|
-
"str", response.headers.get("Content-Range")
|
2159
|
-
)
|
2160
|
-
|
2161
|
-
deserialized = self._deserialize("[object]", pipeline_response.http_response)
|
2162
|
-
|
2163
|
-
if cls:
|
2164
|
-
return cls(pipeline_response, deserialized, response_headers) # type: ignore
|
2165
|
-
|
2166
|
-
return deserialized # type: ignore
|
2167
|
-
|
2168
|
-
@overload
|
2169
|
-
async def summary(
|
2170
|
-
self,
|
2171
|
-
body: _models.JobSummaryParams,
|
2172
|
-
*,
|
2173
|
-
content_type: str = "application/json",
|
2174
|
-
**kwargs: Any,
|
2175
|
-
) -> Any:
|
2176
|
-
"""Summary.
|
2177
|
-
|
2178
|
-
Show information suitable for plotting.
|
2179
|
-
|
2180
|
-
:param body: Required.
|
2181
|
-
:type body: ~generated.models.JobSummaryParams
|
2182
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2183
|
-
Default value is "application/json".
|
2184
|
-
:paramtype content_type: str
|
2185
|
-
:return: any
|
2186
|
-
:rtype: any
|
2187
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2188
|
-
"""
|
2189
|
-
|
2190
|
-
@overload
|
2191
|
-
async def summary(
|
2192
|
-
self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
|
2193
|
-
) -> Any:
|
2194
|
-
"""Summary.
|
2195
|
-
|
2196
|
-
Show information suitable for plotting.
|
2197
|
-
|
2198
|
-
:param body: Required.
|
2199
|
-
:type body: IO[bytes]
|
2200
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2201
|
-
Default value is "application/json".
|
2202
|
-
:paramtype content_type: str
|
2203
|
-
:return: any
|
2204
|
-
:rtype: any
|
2205
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2206
|
-
"""
|
2207
|
-
|
2208
|
-
@distributed_trace_async
|
2209
|
-
async def summary(
|
2210
|
-
self, body: Union[_models.JobSummaryParams, IO[bytes]], **kwargs: Any
|
2211
|
-
) -> Any:
|
2212
|
-
"""Summary.
|
2213
|
-
|
2214
|
-
Show information suitable for plotting.
|
2215
|
-
|
2216
|
-
:param body: Is either a JobSummaryParams type or a IO[bytes] type. Required.
|
2217
|
-
:type body: ~generated.models.JobSummaryParams or IO[bytes]
|
2218
|
-
:return: any
|
2219
|
-
:rtype: any
|
2220
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2221
|
-
"""
|
2222
|
-
error_map: MutableMapping = {
|
2223
|
-
401: ClientAuthenticationError,
|
2224
|
-
404: ResourceNotFoundError,
|
2225
|
-
409: ResourceExistsError,
|
2226
|
-
304: ResourceNotModifiedError,
|
2227
|
-
}
|
2228
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2229
|
-
|
2230
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
2231
|
-
_params = kwargs.pop("params", {}) or {}
|
2232
|
-
|
2233
|
-
content_type: Optional[str] = kwargs.pop(
|
2234
|
-
"content_type", _headers.pop("Content-Type", None)
|
2235
|
-
)
|
2236
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2237
|
-
|
2238
|
-
content_type = content_type or "application/json"
|
2239
|
-
_json = None
|
2240
|
-
_content = None
|
2241
|
-
if isinstance(body, (IOBase, bytes)):
|
2242
|
-
_content = body
|
2243
|
-
else:
|
2244
|
-
_json = self._serialize.body(body, "JobSummaryParams")
|
2245
|
-
|
2246
|
-
_request = build_jobs_summary_request(
|
2247
|
-
content_type=content_type,
|
2248
|
-
json=_json,
|
2249
|
-
content=_content,
|
2250
|
-
headers=_headers,
|
2251
|
-
params=_params,
|
2252
|
-
)
|
2253
|
-
_request.url = self._client.format_url(_request.url)
|
2254
|
-
|
2255
|
-
_stream = False
|
2256
|
-
pipeline_response: PipelineResponse = (
|
2257
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
2258
|
-
_request, stream=_stream, **kwargs
|
2259
|
-
)
|
2260
|
-
)
|
2261
|
-
|
2262
|
-
response = pipeline_response.http_response
|
2263
|
-
|
2264
|
-
if response.status_code not in [200]:
|
2265
|
-
map_error(
|
2266
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2267
|
-
)
|
2268
|
-
raise HttpResponseError(response=response)
|
2269
|
-
|
2270
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2271
|
-
|
2272
|
-
if cls:
|
2273
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2274
|
-
|
2275
|
-
return deserialized # type: ignore
|
2276
|
-
|
2277
|
-
@distributed_trace_async
|
2278
|
-
async def get_single_job(self, job_id: int, **kwargs: Any) -> Any:
|
2279
|
-
"""Get Single Job.
|
2280
|
-
|
2281
|
-
Get Single Job.
|
2282
|
-
|
2283
|
-
:param job_id: Required.
|
2284
|
-
:type job_id: int
|
2285
|
-
:return: any
|
2286
|
-
:rtype: any
|
2287
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2288
|
-
"""
|
2289
|
-
error_map: MutableMapping = {
|
2290
|
-
401: ClientAuthenticationError,
|
2291
|
-
404: ResourceNotFoundError,
|
2292
|
-
409: ResourceExistsError,
|
2293
|
-
304: ResourceNotModifiedError,
|
2294
|
-
}
|
2295
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2296
|
-
|
2297
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2298
|
-
_params = kwargs.pop("params", {}) or {}
|
2299
|
-
|
2300
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2301
|
-
|
2302
|
-
_request = build_jobs_get_single_job_request(
|
2303
|
-
job_id=job_id,
|
2304
|
-
headers=_headers,
|
2305
|
-
params=_params,
|
2306
|
-
)
|
2307
|
-
_request.url = self._client.format_url(_request.url)
|
2308
|
-
|
2309
|
-
_stream = False
|
2310
|
-
pipeline_response: PipelineResponse = (
|
2311
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
2312
|
-
_request, stream=_stream, **kwargs
|
2313
|
-
)
|
2314
|
-
)
|
2315
|
-
|
2316
|
-
response = pipeline_response.http_response
|
2317
|
-
|
2318
|
-
if response.status_code not in [200]:
|
2319
|
-
map_error(
|
2320
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2321
|
-
)
|
2322
|
-
raise HttpResponseError(response=response)
|
2323
|
-
|
2324
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2325
|
-
|
2326
|
-
if cls:
|
2327
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2328
|
-
|
2329
|
-
return deserialized # type: ignore
|
2330
|
-
|
2331
|
-
@distributed_trace_async
|
2332
|
-
async def delete_single_job(self, job_id: int, **kwargs: Any) -> Any:
|
2333
|
-
"""Delete Single Job.
|
2334
|
-
|
2335
|
-
Delete a job by killing and setting the job status to DELETED.
|
2336
|
-
|
2337
|
-
:param job_id: Required.
|
2338
|
-
:type job_id: int
|
2339
|
-
:return: any
|
2340
|
-
:rtype: any
|
2341
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2342
|
-
"""
|
2343
|
-
error_map: MutableMapping = {
|
2344
|
-
401: ClientAuthenticationError,
|
2345
|
-
404: ResourceNotFoundError,
|
2346
|
-
409: ResourceExistsError,
|
2347
|
-
304: ResourceNotModifiedError,
|
2348
|
-
}
|
2349
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2350
|
-
|
2351
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2352
|
-
_params = kwargs.pop("params", {}) or {}
|
2353
|
-
|
2354
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2355
|
-
|
2356
|
-
_request = build_jobs_delete_single_job_request(
|
2357
|
-
job_id=job_id,
|
2358
|
-
headers=_headers,
|
2359
|
-
params=_params,
|
2360
|
-
)
|
2361
|
-
_request.url = self._client.format_url(_request.url)
|
2362
|
-
|
2363
|
-
_stream = False
|
2364
|
-
pipeline_response: PipelineResponse = (
|
2365
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
2366
|
-
_request, stream=_stream, **kwargs
|
2367
|
-
)
|
2368
|
-
)
|
2369
|
-
|
2370
|
-
response = pipeline_response.http_response
|
2371
|
-
|
2372
|
-
if response.status_code not in [200]:
|
2373
|
-
map_error(
|
2374
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2375
|
-
)
|
2376
|
-
raise HttpResponseError(response=response)
|
2377
|
-
|
2378
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2379
|
-
|
2380
|
-
if cls:
|
2381
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2382
|
-
|
2383
|
-
return deserialized # type: ignore
|
2384
|
-
|
2385
|
-
@distributed_trace_async
|
2386
|
-
async def set_single_job_properties(
|
2387
|
-
self, job_id: int, body: JSON, *, update_timestamp: bool = False, **kwargs: Any
|
2388
|
-
) -> Any:
|
2389
|
-
"""Set Single Job Properties.
|
2390
|
-
|
2391
|
-
Update the given job properties (MinorStatus, ApplicationStatus, etc).
|
2392
|
-
|
2393
|
-
:param job_id: Required.
|
2394
|
-
:type job_id: int
|
2395
|
-
:param body: Required.
|
2396
|
-
:type body: JSON
|
2397
|
-
:keyword update_timestamp: Default value is False.
|
2398
|
-
:paramtype update_timestamp: bool
|
2399
|
-
:return: any
|
2400
|
-
:rtype: any
|
2401
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2402
|
-
"""
|
2403
|
-
error_map: MutableMapping = {
|
2404
|
-
401: ClientAuthenticationError,
|
2405
|
-
404: ResourceNotFoundError,
|
2406
|
-
409: ResourceExistsError,
|
2407
|
-
304: ResourceNotModifiedError,
|
2408
|
-
}
|
2409
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2410
|
-
|
2411
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
2412
|
-
_params = kwargs.pop("params", {}) or {}
|
2413
|
-
|
2414
|
-
content_type: str = kwargs.pop(
|
2415
|
-
"content_type", _headers.pop("Content-Type", "application/json")
|
2416
|
-
)
|
2417
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2418
|
-
|
2419
|
-
_json = self._serialize.body(body, "object")
|
2420
|
-
|
2421
|
-
_request = build_jobs_set_single_job_properties_request(
|
2422
|
-
job_id=job_id,
|
2423
|
-
update_timestamp=update_timestamp,
|
2424
|
-
content_type=content_type,
|
2425
|
-
json=_json,
|
2426
|
-
headers=_headers,
|
2427
|
-
params=_params,
|
2428
|
-
)
|
2429
|
-
_request.url = self._client.format_url(_request.url)
|
2430
|
-
|
2431
|
-
_stream = False
|
2432
|
-
pipeline_response: PipelineResponse = (
|
2433
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
2434
|
-
_request, stream=_stream, **kwargs
|
2435
|
-
)
|
2436
|
-
)
|
2437
|
-
|
2438
|
-
response = pipeline_response.http_response
|
2439
|
-
|
2440
|
-
if response.status_code not in [200]:
|
2441
|
-
map_error(
|
2442
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2443
|
-
)
|
2444
|
-
raise HttpResponseError(response=response)
|
2445
|
-
|
2446
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2447
|
-
|
2448
|
-
if cls:
|
2449
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2450
|
-
|
2451
|
-
return deserialized # type: ignore
|
2452
|
-
|
2453
|
-
@distributed_trace_async
|
2454
|
-
async def kill_single_job(self, job_id: int, **kwargs: Any) -> Any:
|
2455
|
-
"""Kill Single Job.
|
2456
|
-
|
2457
|
-
Kill a job.
|
2458
|
-
|
2459
|
-
:param job_id: Required.
|
2460
|
-
:type job_id: int
|
1821
|
+
:param body: Required.
|
1822
|
+
:type body: ~generated.models.JobSummaryParams
|
1823
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
1824
|
+
Default value is "application/json".
|
1825
|
+
:paramtype content_type: str
|
2461
1826
|
:return: any
|
2462
1827
|
:rtype: any
|
2463
1828
|
:raises ~azure.core.exceptions.HttpResponseError:
|
2464
1829
|
"""
|
2465
|
-
error_map: MutableMapping = {
|
2466
|
-
401: ClientAuthenticationError,
|
2467
|
-
404: ResourceNotFoundError,
|
2468
|
-
409: ResourceExistsError,
|
2469
|
-
304: ResourceNotModifiedError,
|
2470
|
-
}
|
2471
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2472
|
-
|
2473
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2474
|
-
_params = kwargs.pop("params", {}) or {}
|
2475
|
-
|
2476
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2477
|
-
|
2478
|
-
_request = build_jobs_kill_single_job_request(
|
2479
|
-
job_id=job_id,
|
2480
|
-
headers=_headers,
|
2481
|
-
params=_params,
|
2482
|
-
)
|
2483
|
-
_request.url = self._client.format_url(_request.url)
|
2484
|
-
|
2485
|
-
_stream = False
|
2486
|
-
pipeline_response: PipelineResponse = (
|
2487
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
2488
|
-
_request, stream=_stream, **kwargs
|
2489
|
-
)
|
2490
|
-
)
|
2491
|
-
|
2492
|
-
response = pipeline_response.http_response
|
2493
|
-
|
2494
|
-
if response.status_code not in [200]:
|
2495
|
-
map_error(
|
2496
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2497
|
-
)
|
2498
|
-
raise HttpResponseError(response=response)
|
2499
|
-
|
2500
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2501
|
-
|
2502
|
-
if cls:
|
2503
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2504
|
-
|
2505
|
-
return deserialized # type: ignore
|
2506
|
-
|
2507
|
-
@distributed_trace_async
|
2508
|
-
async def remove_single_job(self, job_id: int, **kwargs: Any) -> Any:
|
2509
|
-
"""Remove Single Job.
|
2510
1830
|
|
2511
|
-
|
1831
|
+
@overload
|
1832
|
+
async def summary(
|
1833
|
+
self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
|
1834
|
+
) -> Any:
|
1835
|
+
"""Summary.
|
2512
1836
|
|
2513
|
-
|
2514
|
-
and the JobCleaningAgent. However, once this agent is ported to diracx, this endpoint should
|
2515
|
-
be removed, and the delete endpoint should be used instead.
|
1837
|
+
Show information suitable for plotting.
|
2516
1838
|
|
2517
|
-
:param
|
2518
|
-
:type
|
1839
|
+
:param body: Required.
|
1840
|
+
:type body: IO[bytes]
|
1841
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
1842
|
+
Default value is "application/json".
|
1843
|
+
:paramtype content_type: str
|
2519
1844
|
:return: any
|
2520
1845
|
:rtype: any
|
2521
1846
|
:raises ~azure.core.exceptions.HttpResponseError:
|
2522
1847
|
"""
|
2523
|
-
error_map: MutableMapping = {
|
2524
|
-
401: ClientAuthenticationError,
|
2525
|
-
404: ResourceNotFoundError,
|
2526
|
-
409: ResourceExistsError,
|
2527
|
-
304: ResourceNotModifiedError,
|
2528
|
-
}
|
2529
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2530
|
-
|
2531
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2532
|
-
_params = kwargs.pop("params", {}) or {}
|
2533
|
-
|
2534
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2535
|
-
|
2536
|
-
_request = build_jobs_remove_single_job_request(
|
2537
|
-
job_id=job_id,
|
2538
|
-
headers=_headers,
|
2539
|
-
params=_params,
|
2540
|
-
)
|
2541
|
-
_request.url = self._client.format_url(_request.url)
|
2542
|
-
|
2543
|
-
_stream = False
|
2544
|
-
pipeline_response: PipelineResponse = (
|
2545
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
2546
|
-
_request, stream=_stream, **kwargs
|
2547
|
-
)
|
2548
|
-
)
|
2549
|
-
|
2550
|
-
response = pipeline_response.http_response
|
2551
|
-
|
2552
|
-
if response.status_code not in [200]:
|
2553
|
-
map_error(
|
2554
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2555
|
-
)
|
2556
|
-
raise HttpResponseError(response=response)
|
2557
|
-
|
2558
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2559
|
-
|
2560
|
-
if cls:
|
2561
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2562
|
-
|
2563
|
-
return deserialized # type: ignore
|
2564
1848
|
|
2565
1849
|
@distributed_trace_async
|
2566
|
-
async def
|
2567
|
-
self,
|
2568
|
-
) ->
|
2569
|
-
"""
|
1850
|
+
async def summary(
|
1851
|
+
self, body: Union[_models.JobSummaryParams, IO[bytes]], **kwargs: Any
|
1852
|
+
) -> Any:
|
1853
|
+
"""Summary.
|
2570
1854
|
|
2571
|
-
|
1855
|
+
Show information suitable for plotting.
|
2572
1856
|
|
2573
|
-
:param
|
2574
|
-
:type
|
2575
|
-
:return:
|
2576
|
-
:rtype:
|
1857
|
+
:param body: Is either a JobSummaryParams type or a IO[bytes] type. Required.
|
1858
|
+
:type body: ~generated.models.JobSummaryParams or IO[bytes]
|
1859
|
+
:return: any
|
1860
|
+
:rtype: any
|
2577
1861
|
:raises ~azure.core.exceptions.HttpResponseError:
|
2578
1862
|
"""
|
2579
1863
|
error_map: MutableMapping = {
|
@@ -2584,15 +1868,26 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
2584
1868
|
}
|
2585
1869
|
error_map.update(kwargs.pop("error_map", {}) or {})
|
2586
1870
|
|
2587
|
-
_headers = kwargs.pop("headers", {}) or {}
|
1871
|
+
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
2588
1872
|
_params = kwargs.pop("params", {}) or {}
|
2589
1873
|
|
2590
|
-
|
2591
|
-
"
|
1874
|
+
content_type: Optional[str] = kwargs.pop(
|
1875
|
+
"content_type", _headers.pop("Content-Type", None)
|
2592
1876
|
)
|
1877
|
+
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2593
1878
|
|
2594
|
-
|
2595
|
-
|
1879
|
+
content_type = content_type or "application/json"
|
1880
|
+
_json = None
|
1881
|
+
_content = None
|
1882
|
+
if isinstance(body, (IOBase, bytes)):
|
1883
|
+
_content = body
|
1884
|
+
else:
|
1885
|
+
_json = self._serialize.body(body, "JobSummaryParams")
|
1886
|
+
|
1887
|
+
_request = build_jobs_summary_request(
|
1888
|
+
content_type=content_type,
|
1889
|
+
json=_json,
|
1890
|
+
content=_content,
|
2596
1891
|
headers=_headers,
|
2597
1892
|
params=_params,
|
2598
1893
|
)
|
@@ -2613,9 +1908,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
2613
1908
|
)
|
2614
1909
|
raise HttpResponseError(response=response)
|
2615
1910
|
|
2616
|
-
deserialized = self._deserialize(
|
2617
|
-
"{LimitedJobStatusReturn}", pipeline_response.http_response
|
2618
|
-
)
|
1911
|
+
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2619
1912
|
|
2620
1913
|
if cls:
|
2621
1914
|
return cls(pipeline_response, deserialized, {}) # type: ignore
|
@@ -2623,82 +1916,53 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
2623
1916
|
return deserialized # type: ignore
|
2624
1917
|
|
2625
1918
|
@overload
|
2626
|
-
async def
|
2627
|
-
self,
|
2628
|
-
|
2629
|
-
|
2630
|
-
*,
|
2631
|
-
force: bool = False,
|
2632
|
-
content_type: str = "application/json",
|
2633
|
-
**kwargs: Any,
|
2634
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
2635
|
-
"""Set Single Job Status.
|
1919
|
+
async def submit_bulk_jdl_jobs(
|
1920
|
+
self, body: List[str], *, content_type: str = "application/json", **kwargs: Any
|
1921
|
+
) -> List[_models.InsertedJob]:
|
1922
|
+
"""Submit Bulk Jdl Jobs.
|
2636
1923
|
|
2637
|
-
|
1924
|
+
Submit Bulk Jdl Jobs.
|
2638
1925
|
|
2639
|
-
:param job_id: Required.
|
2640
|
-
:type job_id: int
|
2641
1926
|
:param body: Required.
|
2642
|
-
:type body:
|
2643
|
-
:keyword force: Default value is False.
|
2644
|
-
:paramtype force: bool
|
1927
|
+
:type body: list[str]
|
2645
1928
|
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2646
1929
|
Default value is "application/json".
|
2647
1930
|
:paramtype content_type: str
|
2648
|
-
:return:
|
2649
|
-
:rtype:
|
1931
|
+
:return: list of InsertedJob
|
1932
|
+
:rtype: list[~generated.models.InsertedJob]
|
2650
1933
|
:raises ~azure.core.exceptions.HttpResponseError:
|
2651
1934
|
"""
|
2652
1935
|
|
2653
1936
|
@overload
|
2654
|
-
async def
|
2655
|
-
self,
|
2656
|
-
|
2657
|
-
|
2658
|
-
*,
|
2659
|
-
force: bool = False,
|
2660
|
-
content_type: str = "application/json",
|
2661
|
-
**kwargs: Any,
|
2662
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
2663
|
-
"""Set Single Job Status.
|
1937
|
+
async def submit_bulk_jdl_jobs(
|
1938
|
+
self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
|
1939
|
+
) -> List[_models.InsertedJob]:
|
1940
|
+
"""Submit Bulk Jdl Jobs.
|
2664
1941
|
|
2665
|
-
|
1942
|
+
Submit Bulk Jdl Jobs.
|
2666
1943
|
|
2667
|
-
:param job_id: Required.
|
2668
|
-
:type job_id: int
|
2669
1944
|
:param body: Required.
|
2670
1945
|
:type body: IO[bytes]
|
2671
|
-
:keyword force: Default value is False.
|
2672
|
-
:paramtype force: bool
|
2673
1946
|
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2674
1947
|
Default value is "application/json".
|
2675
1948
|
:paramtype content_type: str
|
2676
|
-
:return:
|
2677
|
-
:rtype:
|
1949
|
+
:return: list of InsertedJob
|
1950
|
+
:rtype: list[~generated.models.InsertedJob]
|
2678
1951
|
:raises ~azure.core.exceptions.HttpResponseError:
|
2679
1952
|
"""
|
2680
1953
|
|
2681
1954
|
@distributed_trace_async
|
2682
|
-
async def
|
2683
|
-
self,
|
2684
|
-
|
2685
|
-
|
2686
|
-
*,
|
2687
|
-
force: bool = False,
|
2688
|
-
**kwargs: Any,
|
2689
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
2690
|
-
"""Set Single Job Status.
|
1955
|
+
async def submit_bulk_jdl_jobs(
|
1956
|
+
self, body: Union[List[str], IO[bytes]], **kwargs: Any
|
1957
|
+
) -> List[_models.InsertedJob]:
|
1958
|
+
"""Submit Bulk Jdl Jobs.
|
2691
1959
|
|
2692
|
-
|
1960
|
+
Submit Bulk Jdl Jobs.
|
2693
1961
|
|
2694
|
-
:param
|
2695
|
-
:type
|
2696
|
-
:
|
2697
|
-
:
|
2698
|
-
:keyword force: Default value is False.
|
2699
|
-
:paramtype force: bool
|
2700
|
-
:return: dict mapping str to SetJobStatusReturn
|
2701
|
-
:rtype: dict[str, ~generated.models.SetJobStatusReturn]
|
1962
|
+
:param body: Is either a [str] type or a IO[bytes] type. Required.
|
1963
|
+
:type body: list[str] or IO[bytes]
|
1964
|
+
:return: list of InsertedJob
|
1965
|
+
:rtype: list[~generated.models.InsertedJob]
|
2702
1966
|
:raises ~azure.core.exceptions.HttpResponseError:
|
2703
1967
|
"""
|
2704
1968
|
error_map: MutableMapping = {
|
@@ -2715,7 +1979,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
2715
1979
|
content_type: Optional[str] = kwargs.pop(
|
2716
1980
|
"content_type", _headers.pop("Content-Type", None)
|
2717
1981
|
)
|
2718
|
-
cls: ClsType[
|
1982
|
+
cls: ClsType[List[_models.InsertedJob]] = kwargs.pop("cls", None)
|
2719
1983
|
|
2720
1984
|
content_type = content_type or "application/json"
|
2721
1985
|
_json = None
|
@@ -2723,11 +1987,9 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
2723
1987
|
if isinstance(body, (IOBase, bytes)):
|
2724
1988
|
_content = body
|
2725
1989
|
else:
|
2726
|
-
_json = self._serialize.body(body, "
|
1990
|
+
_json = self._serialize.body(body, "[str]")
|
2727
1991
|
|
2728
|
-
_request =
|
2729
|
-
job_id=job_id,
|
2730
|
-
force=force,
|
1992
|
+
_request = build_jobs_submit_bulk_jdl_jobs_request(
|
2731
1993
|
content_type=content_type,
|
2732
1994
|
json=_json,
|
2733
1995
|
content=_content,
|
@@ -2752,65 +2014,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
2752
2014
|
raise HttpResponseError(response=response)
|
2753
2015
|
|
2754
2016
|
deserialized = self._deserialize(
|
2755
|
-
"
|
2756
|
-
)
|
2757
|
-
|
2758
|
-
if cls:
|
2759
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2760
|
-
|
2761
|
-
return deserialized # type: ignore
|
2762
|
-
|
2763
|
-
@distributed_trace_async
|
2764
|
-
async def get_single_job_status_history(
|
2765
|
-
self, job_id: int, **kwargs: Any
|
2766
|
-
) -> Dict[str, List[_models.JobStatusReturn]]:
|
2767
|
-
"""Get Single Job Status History.
|
2768
|
-
|
2769
|
-
Get Single Job Status History.
|
2770
|
-
|
2771
|
-
:param job_id: Required.
|
2772
|
-
:type job_id: int
|
2773
|
-
:return: dict mapping str to list of JobStatusReturn
|
2774
|
-
:rtype: dict[str, list[~generated.models.JobStatusReturn]]
|
2775
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2776
|
-
"""
|
2777
|
-
error_map: MutableMapping = {
|
2778
|
-
401: ClientAuthenticationError,
|
2779
|
-
404: ResourceNotFoundError,
|
2780
|
-
409: ResourceExistsError,
|
2781
|
-
304: ResourceNotModifiedError,
|
2782
|
-
}
|
2783
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2784
|
-
|
2785
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2786
|
-
_params = kwargs.pop("params", {}) or {}
|
2787
|
-
|
2788
|
-
cls: ClsType[Dict[str, List[_models.JobStatusReturn]]] = kwargs.pop("cls", None)
|
2789
|
-
|
2790
|
-
_request = build_jobs_get_single_job_status_history_request(
|
2791
|
-
job_id=job_id,
|
2792
|
-
headers=_headers,
|
2793
|
-
params=_params,
|
2794
|
-
)
|
2795
|
-
_request.url = self._client.format_url(_request.url)
|
2796
|
-
|
2797
|
-
_stream = False
|
2798
|
-
pipeline_response: PipelineResponse = (
|
2799
|
-
await self._client._pipeline.run( # pylint: disable=protected-access
|
2800
|
-
_request, stream=_stream, **kwargs
|
2801
|
-
)
|
2802
|
-
)
|
2803
|
-
|
2804
|
-
response = pipeline_response.http_response
|
2805
|
-
|
2806
|
-
if response.status_code not in [200]:
|
2807
|
-
map_error(
|
2808
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2809
|
-
)
|
2810
|
-
raise HttpResponseError(response=response)
|
2811
|
-
|
2812
|
-
deserialized = self._deserialize(
|
2813
|
-
"{[JobStatusReturn]}", pipeline_response.http_response
|
2017
|
+
"[InsertedJob]", pipeline_response.http_response
|
2814
2018
|
)
|
2815
2019
|
|
2816
2020
|
if cls:
|