diracx-client 0.0.1a20__py3-none-any.whl → 0.0.1a22__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- 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
|
@@ -202,9 +202,9 @@ def build_auth_userinfo_request(**kwargs: Any) -> HttpRequest:
|
|
202
202
|
|
203
203
|
def build_auth_authorization_flow_request(
|
204
204
|
*,
|
205
|
-
response_type:
|
205
|
+
response_type: str,
|
206
206
|
code_challenge: str,
|
207
|
-
code_challenge_method:
|
207
|
+
code_challenge_method: str,
|
208
208
|
client_id: str,
|
209
209
|
redirect_uri: str,
|
210
210
|
scope: str,
|
@@ -458,28 +458,7 @@ def build_jobs_assign_sandbox_to_job_request(
|
|
458
458
|
)
|
459
459
|
|
460
460
|
|
461
|
-
def
|
462
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
463
|
-
|
464
|
-
content_type: Optional[str] = kwargs.pop(
|
465
|
-
"content_type", _headers.pop("Content-Type", None)
|
466
|
-
)
|
467
|
-
accept = _headers.pop("Accept", "application/json")
|
468
|
-
|
469
|
-
# Construct URL
|
470
|
-
_url = "/api/jobs/"
|
471
|
-
|
472
|
-
# Construct headers
|
473
|
-
if content_type is not None:
|
474
|
-
_headers["Content-Type"] = _SERIALIZER.header(
|
475
|
-
"content_type", content_type, "str"
|
476
|
-
)
|
477
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
478
|
-
|
479
|
-
return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
|
480
|
-
|
481
|
-
|
482
|
-
def build_jobs_delete_bulk_jobs_request(
|
461
|
+
def build_jobs_remove_bulk_jobs_request(
|
483
462
|
*, job_ids: List[int], **kwargs: Any
|
484
463
|
) -> HttpRequest:
|
485
464
|
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
@@ -501,73 +480,7 @@ def build_jobs_delete_bulk_jobs_request(
|
|
501
480
|
)
|
502
481
|
|
503
482
|
|
504
|
-
def
|
505
|
-
*, job_ids: List[int], **kwargs: Any
|
506
|
-
) -> HttpRequest:
|
507
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
508
|
-
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})
|
509
|
-
|
510
|
-
accept = _headers.pop("Accept", "application/json")
|
511
|
-
|
512
|
-
# Construct URL
|
513
|
-
_url = "/api/jobs/kill"
|
514
|
-
|
515
|
-
# Construct parameters
|
516
|
-
_params["job_ids"] = _SERIALIZER.query("job_ids", job_ids, "[int]")
|
517
|
-
|
518
|
-
# Construct headers
|
519
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
520
|
-
|
521
|
-
return HttpRequest(
|
522
|
-
method="POST", url=_url, params=_params, headers=_headers, **kwargs
|
523
|
-
)
|
524
|
-
|
525
|
-
|
526
|
-
def build_jobs_remove_bulk_jobs_request(
|
527
|
-
*, job_ids: List[int], **kwargs: Any
|
528
|
-
) -> HttpRequest:
|
529
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
530
|
-
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})
|
531
|
-
|
532
|
-
accept = _headers.pop("Accept", "application/json")
|
533
|
-
|
534
|
-
# Construct URL
|
535
|
-
_url = "/api/jobs/remove"
|
536
|
-
|
537
|
-
# Construct parameters
|
538
|
-
_params["job_ids"] = _SERIALIZER.query("job_ids", job_ids, "[int]")
|
539
|
-
|
540
|
-
# Construct headers
|
541
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
542
|
-
|
543
|
-
return HttpRequest(
|
544
|
-
method="POST", url=_url, params=_params, headers=_headers, **kwargs
|
545
|
-
)
|
546
|
-
|
547
|
-
|
548
|
-
def build_jobs_get_job_status_bulk_request(
|
549
|
-
*, job_ids: List[int], **kwargs: Any
|
550
|
-
) -> HttpRequest:
|
551
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
552
|
-
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})
|
553
|
-
|
554
|
-
accept = _headers.pop("Accept", "application/json")
|
555
|
-
|
556
|
-
# Construct URL
|
557
|
-
_url = "/api/jobs/status"
|
558
|
-
|
559
|
-
# Construct parameters
|
560
|
-
_params["job_ids"] = _SERIALIZER.query("job_ids", job_ids, "[int]")
|
561
|
-
|
562
|
-
# Construct headers
|
563
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
564
|
-
|
565
|
-
return HttpRequest(
|
566
|
-
method="GET", url=_url, params=_params, headers=_headers, **kwargs
|
567
|
-
)
|
568
|
-
|
569
|
-
|
570
|
-
def build_jobs_set_job_status_bulk_request(
|
483
|
+
def build_jobs_set_job_statuses_request(
|
571
484
|
*, force: bool = False, **kwargs: Any
|
572
485
|
) -> HttpRequest:
|
573
486
|
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
@@ -597,30 +510,8 @@ def build_jobs_set_job_status_bulk_request(
|
|
597
510
|
)
|
598
511
|
|
599
512
|
|
600
|
-
def build_jobs_get_job_status_history_bulk_request( # pylint: disable=name-too-long
|
601
|
-
*, job_ids: List[int], **kwargs: Any
|
602
|
-
) -> HttpRequest:
|
603
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
604
|
-
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})
|
605
|
-
|
606
|
-
accept = _headers.pop("Accept", "application/json")
|
607
|
-
|
608
|
-
# Construct URL
|
609
|
-
_url = "/api/jobs/status/history"
|
610
|
-
|
611
|
-
# Construct parameters
|
612
|
-
_params["job_ids"] = _SERIALIZER.query("job_ids", job_ids, "[int]")
|
613
|
-
|
614
|
-
# Construct headers
|
615
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
616
|
-
|
617
|
-
return HttpRequest(
|
618
|
-
method="GET", url=_url, params=_params, headers=_headers, **kwargs
|
619
|
-
)
|
620
|
-
|
621
|
-
|
622
513
|
def build_jobs_reschedule_bulk_jobs_request(
|
623
|
-
*, job_ids: List[int], **kwargs: Any
|
514
|
+
*, job_ids: List[int], reset_jobs: bool = False, **kwargs: Any
|
624
515
|
) -> HttpRequest:
|
625
516
|
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
626
517
|
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})
|
@@ -632,6 +523,8 @@ def build_jobs_reschedule_bulk_jobs_request(
|
|
632
523
|
|
633
524
|
# Construct parameters
|
634
525
|
_params["job_ids"] = _SERIALIZER.query("job_ids", job_ids, "[int]")
|
526
|
+
if reset_jobs is not None:
|
527
|
+
_params["reset_jobs"] = _SERIALIZER.query("reset_jobs", reset_jobs, "bool")
|
635
528
|
|
636
529
|
# Construct headers
|
637
530
|
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
@@ -641,25 +534,6 @@ def build_jobs_reschedule_bulk_jobs_request(
|
|
641
534
|
)
|
642
535
|
|
643
536
|
|
644
|
-
def build_jobs_reschedule_single_job_request(job_id: int, **kwargs: Any) -> HttpRequest:
|
645
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
646
|
-
|
647
|
-
accept = _headers.pop("Accept", "application/json")
|
648
|
-
|
649
|
-
# Construct URL
|
650
|
-
_url = "/api/jobs/{job_id}/reschedule"
|
651
|
-
path_format_arguments = {
|
652
|
-
"job_id": _SERIALIZER.url("job_id", job_id, "int"),
|
653
|
-
}
|
654
|
-
|
655
|
-
_url: str = _url.format(**path_format_arguments) # type: ignore
|
656
|
-
|
657
|
-
# Construct headers
|
658
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
659
|
-
|
660
|
-
return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
|
661
|
-
|
662
|
-
|
663
537
|
def build_jobs_search_request(
|
664
538
|
*, page: int = 1, per_page: int = 100, **kwargs: Any
|
665
539
|
) -> HttpRequest:
|
@@ -713,49 +587,8 @@ def build_jobs_summary_request(**kwargs: Any) -> HttpRequest:
|
|
713
587
|
return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
|
714
588
|
|
715
589
|
|
716
|
-
def
|
717
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
718
|
-
|
719
|
-
accept = _headers.pop("Accept", "application/json")
|
720
|
-
|
721
|
-
# Construct URL
|
722
|
-
_url = "/api/jobs/{job_id}"
|
723
|
-
path_format_arguments = {
|
724
|
-
"job_id": _SERIALIZER.url("job_id", job_id, "int"),
|
725
|
-
}
|
726
|
-
|
727
|
-
_url: str = _url.format(**path_format_arguments) # type: ignore
|
728
|
-
|
729
|
-
# Construct headers
|
730
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
731
|
-
|
732
|
-
return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
|
733
|
-
|
734
|
-
|
735
|
-
def build_jobs_delete_single_job_request(job_id: int, **kwargs: Any) -> HttpRequest:
|
736
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
737
|
-
|
738
|
-
accept = _headers.pop("Accept", "application/json")
|
739
|
-
|
740
|
-
# Construct URL
|
741
|
-
_url = "/api/jobs/{job_id}"
|
742
|
-
path_format_arguments = {
|
743
|
-
"job_id": _SERIALIZER.url("job_id", job_id, "int"),
|
744
|
-
}
|
745
|
-
|
746
|
-
_url: str = _url.format(**path_format_arguments) # type: ignore
|
747
|
-
|
748
|
-
# Construct headers
|
749
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
750
|
-
|
751
|
-
return HttpRequest(method="DELETE", url=_url, headers=_headers, **kwargs)
|
752
|
-
|
753
|
-
|
754
|
-
def build_jobs_set_single_job_properties_request( # pylint: disable=name-too-long
|
755
|
-
job_id: int, *, json: JSON, update_timestamp: bool = False, **kwargs: Any
|
756
|
-
) -> HttpRequest:
|
590
|
+
def build_jobs_submit_bulk_jdl_jobs_request(**kwargs: Any) -> HttpRequest:
|
757
591
|
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
758
|
-
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})
|
759
592
|
|
760
593
|
content_type: Optional[str] = kwargs.pop(
|
761
594
|
"content_type", _headers.pop("Content-Type", None)
|
@@ -763,18 +596,7 @@ def build_jobs_set_single_job_properties_request( # pylint: disable=name-too-lo
|
|
763
596
|
accept = _headers.pop("Accept", "application/json")
|
764
597
|
|
765
598
|
# Construct URL
|
766
|
-
_url = "/api/jobs/
|
767
|
-
path_format_arguments = {
|
768
|
-
"job_id": _SERIALIZER.url("job_id", job_id, "int"),
|
769
|
-
}
|
770
|
-
|
771
|
-
_url: str = _url.format(**path_format_arguments) # type: ignore
|
772
|
-
|
773
|
-
# Construct parameters
|
774
|
-
if update_timestamp is not None:
|
775
|
-
_params["update_timestamp"] = _SERIALIZER.query(
|
776
|
-
"update_timestamp", update_timestamp, "bool"
|
777
|
-
)
|
599
|
+
_url = "/api/jobs/jdl"
|
778
600
|
|
779
601
|
# Construct headers
|
780
602
|
if content_type is not None:
|
@@ -783,201 +605,86 @@ def build_jobs_set_single_job_properties_request( # pylint: disable=name-too-lo
|
|
783
605
|
)
|
784
606
|
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
785
607
|
|
786
|
-
return HttpRequest(
|
787
|
-
method="PATCH", url=_url, params=_params, headers=_headers, json=json, **kwargs
|
788
|
-
)
|
608
|
+
return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
|
789
609
|
|
790
610
|
|
791
|
-
|
792
|
-
|
611
|
+
class WellKnownOperations:
|
612
|
+
"""
|
613
|
+
.. warning::
|
614
|
+
**DO NOT** instantiate this class directly.
|
793
615
|
|
794
|
-
|
616
|
+
Instead, you should access the following operations through
|
617
|
+
:class:`~generated.Dirac`'s
|
618
|
+
:attr:`well_known` attribute.
|
619
|
+
"""
|
795
620
|
|
796
|
-
|
797
|
-
_url = "/api/jobs/{job_id}/kill"
|
798
|
-
path_format_arguments = {
|
799
|
-
"job_id": _SERIALIZER.url("job_id", job_id, "int"),
|
800
|
-
}
|
621
|
+
models = _models
|
801
622
|
|
802
|
-
|
623
|
+
def __init__(self, *args, **kwargs):
|
624
|
+
input_args = list(args)
|
625
|
+
self._client = input_args.pop(0) if input_args else kwargs.pop("client")
|
626
|
+
self._config = input_args.pop(0) if input_args else kwargs.pop("config")
|
627
|
+
self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer")
|
628
|
+
self._deserialize = (
|
629
|
+
input_args.pop(0) if input_args else kwargs.pop("deserializer")
|
630
|
+
)
|
803
631
|
|
804
|
-
|
805
|
-
|
632
|
+
@distributed_trace
|
633
|
+
def openid_configuration(self, **kwargs: Any) -> Any:
|
634
|
+
"""Openid Configuration.
|
806
635
|
|
807
|
-
|
636
|
+
OpenID Connect discovery endpoint.
|
808
637
|
|
638
|
+
:return: any
|
639
|
+
:rtype: any
|
640
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
641
|
+
"""
|
642
|
+
error_map: MutableMapping = {
|
643
|
+
401: ClientAuthenticationError,
|
644
|
+
404: ResourceNotFoundError,
|
645
|
+
409: ResourceExistsError,
|
646
|
+
304: ResourceNotModifiedError,
|
647
|
+
}
|
648
|
+
error_map.update(kwargs.pop("error_map", {}) or {})
|
809
649
|
|
810
|
-
|
811
|
-
|
650
|
+
_headers = kwargs.pop("headers", {}) or {}
|
651
|
+
_params = kwargs.pop("params", {}) or {}
|
812
652
|
|
813
|
-
|
653
|
+
cls: ClsType[Any] = kwargs.pop("cls", None)
|
814
654
|
|
815
|
-
|
816
|
-
|
817
|
-
|
818
|
-
|
819
|
-
|
655
|
+
_request = build_well_known_openid_configuration_request(
|
656
|
+
headers=_headers,
|
657
|
+
params=_params,
|
658
|
+
)
|
659
|
+
_request.url = self._client.format_url(_request.url)
|
820
660
|
|
821
|
-
|
661
|
+
_stream = False
|
662
|
+
pipeline_response: PipelineResponse = (
|
663
|
+
self._client._pipeline.run( # pylint: disable=protected-access
|
664
|
+
_request, stream=_stream, **kwargs
|
665
|
+
)
|
666
|
+
)
|
822
667
|
|
823
|
-
|
824
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
668
|
+
response = pipeline_response.http_response
|
825
669
|
|
826
|
-
|
670
|
+
if response.status_code not in [200]:
|
671
|
+
map_error(
|
672
|
+
status_code=response.status_code, response=response, error_map=error_map
|
673
|
+
)
|
674
|
+
raise HttpResponseError(response=response)
|
827
675
|
|
676
|
+
deserialized = self._deserialize("object", pipeline_response.http_response)
|
828
677
|
|
829
|
-
|
830
|
-
|
678
|
+
if cls:
|
679
|
+
return cls(pipeline_response, deserialized, {}) # type: ignore
|
831
680
|
|
832
|
-
|
681
|
+
return deserialized # type: ignore
|
833
682
|
|
834
|
-
|
835
|
-
|
836
|
-
|
837
|
-
"job_id": _SERIALIZER.url("job_id", job_id, "int"),
|
838
|
-
}
|
683
|
+
@distributed_trace
|
684
|
+
def installation_metadata(self, **kwargs: Any) -> _models.Metadata:
|
685
|
+
"""Installation Metadata.
|
839
686
|
|
840
|
-
|
841
|
-
|
842
|
-
# Construct headers
|
843
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
844
|
-
|
845
|
-
return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
|
846
|
-
|
847
|
-
|
848
|
-
def build_jobs_set_single_job_status_request(
|
849
|
-
job_id: int, *, force: bool = False, **kwargs: Any
|
850
|
-
) -> HttpRequest:
|
851
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
852
|
-
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})
|
853
|
-
|
854
|
-
content_type: Optional[str] = kwargs.pop(
|
855
|
-
"content_type", _headers.pop("Content-Type", None)
|
856
|
-
)
|
857
|
-
accept = _headers.pop("Accept", "application/json")
|
858
|
-
|
859
|
-
# Construct URL
|
860
|
-
_url = "/api/jobs/{job_id}/status"
|
861
|
-
path_format_arguments = {
|
862
|
-
"job_id": _SERIALIZER.url("job_id", job_id, "int"),
|
863
|
-
}
|
864
|
-
|
865
|
-
_url: str = _url.format(**path_format_arguments) # type: ignore
|
866
|
-
|
867
|
-
# Construct parameters
|
868
|
-
if force is not None:
|
869
|
-
_params["force"] = _SERIALIZER.query("force", force, "bool")
|
870
|
-
|
871
|
-
# Construct headers
|
872
|
-
if content_type is not None:
|
873
|
-
_headers["Content-Type"] = _SERIALIZER.header(
|
874
|
-
"content_type", content_type, "str"
|
875
|
-
)
|
876
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
877
|
-
|
878
|
-
return HttpRequest(
|
879
|
-
method="PATCH", url=_url, params=_params, headers=_headers, **kwargs
|
880
|
-
)
|
881
|
-
|
882
|
-
|
883
|
-
def build_jobs_get_single_job_status_history_request( # pylint: disable=name-too-long
|
884
|
-
job_id: int, **kwargs: Any
|
885
|
-
) -> HttpRequest:
|
886
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
887
|
-
|
888
|
-
accept = _headers.pop("Accept", "application/json")
|
889
|
-
|
890
|
-
# Construct URL
|
891
|
-
_url = "/api/jobs/{job_id}/status/history"
|
892
|
-
path_format_arguments = {
|
893
|
-
"job_id": _SERIALIZER.url("job_id", job_id, "int"),
|
894
|
-
}
|
895
|
-
|
896
|
-
_url: str = _url.format(**path_format_arguments) # type: ignore
|
897
|
-
|
898
|
-
# Construct headers
|
899
|
-
_headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
|
900
|
-
|
901
|
-
return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
|
902
|
-
|
903
|
-
|
904
|
-
class WellKnownOperations:
|
905
|
-
"""
|
906
|
-
.. warning::
|
907
|
-
**DO NOT** instantiate this class directly.
|
908
|
-
|
909
|
-
Instead, you should access the following operations through
|
910
|
-
:class:`~generated.Dirac`'s
|
911
|
-
:attr:`well_known` attribute.
|
912
|
-
"""
|
913
|
-
|
914
|
-
models = _models
|
915
|
-
|
916
|
-
def __init__(self, *args, **kwargs):
|
917
|
-
input_args = list(args)
|
918
|
-
self._client = input_args.pop(0) if input_args else kwargs.pop("client")
|
919
|
-
self._config = input_args.pop(0) if input_args else kwargs.pop("config")
|
920
|
-
self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer")
|
921
|
-
self._deserialize = (
|
922
|
-
input_args.pop(0) if input_args else kwargs.pop("deserializer")
|
923
|
-
)
|
924
|
-
|
925
|
-
@distributed_trace
|
926
|
-
def openid_configuration(self, **kwargs: Any) -> Any:
|
927
|
-
"""Openid Configuration.
|
928
|
-
|
929
|
-
OpenID Connect discovery endpoint.
|
930
|
-
|
931
|
-
:return: any
|
932
|
-
:rtype: any
|
933
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
934
|
-
"""
|
935
|
-
error_map: MutableMapping = {
|
936
|
-
401: ClientAuthenticationError,
|
937
|
-
404: ResourceNotFoundError,
|
938
|
-
409: ResourceExistsError,
|
939
|
-
304: ResourceNotModifiedError,
|
940
|
-
}
|
941
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
942
|
-
|
943
|
-
_headers = kwargs.pop("headers", {}) or {}
|
944
|
-
_params = kwargs.pop("params", {}) or {}
|
945
|
-
|
946
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
947
|
-
|
948
|
-
_request = build_well_known_openid_configuration_request(
|
949
|
-
headers=_headers,
|
950
|
-
params=_params,
|
951
|
-
)
|
952
|
-
_request.url = self._client.format_url(_request.url)
|
953
|
-
|
954
|
-
_stream = False
|
955
|
-
pipeline_response: PipelineResponse = (
|
956
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
957
|
-
_request, stream=_stream, **kwargs
|
958
|
-
)
|
959
|
-
)
|
960
|
-
|
961
|
-
response = pipeline_response.http_response
|
962
|
-
|
963
|
-
if response.status_code not in [200]:
|
964
|
-
map_error(
|
965
|
-
status_code=response.status_code, response=response, error_map=error_map
|
966
|
-
)
|
967
|
-
raise HttpResponseError(response=response)
|
968
|
-
|
969
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
970
|
-
|
971
|
-
if cls:
|
972
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
973
|
-
|
974
|
-
return deserialized # type: ignore
|
975
|
-
|
976
|
-
@distributed_trace
|
977
|
-
def installation_metadata(self, **kwargs: Any) -> _models.Metadata:
|
978
|
-
"""Installation Metadata.
|
979
|
-
|
980
|
-
Get metadata about the dirac installation.
|
687
|
+
Get metadata about the dirac installation.
|
981
688
|
|
982
689
|
:return: Metadata
|
983
690
|
:rtype: ~generated.models.Metadata
|
@@ -1060,9 +767,25 @@ class AuthOperations: # pylint: disable=abstract-class-instantiated
|
|
1060
767
|
"""Initiate Device Flow.
|
1061
768
|
|
1062
769
|
Initiate the device flow against DIRAC authorization Server.
|
1063
|
-
|
1064
|
-
|
1065
|
-
|
770
|
+
|
771
|
+
Scope details:
|
772
|
+
|
773
|
+
|
774
|
+
*
|
775
|
+
If only VO is provided: Uses the default group and its properties for the VO.
|
776
|
+
|
777
|
+
*
|
778
|
+
If VO and group are provided: Uses the specified group and its properties for the VO.
|
779
|
+
|
780
|
+
*
|
781
|
+
If VO and properties are provided: Uses the default group and combines its properties with
|
782
|
+
the
|
783
|
+
provided properties.
|
784
|
+
|
785
|
+
*
|
786
|
+
If VO, group, and properties are provided: Uses the specified group and combines its
|
787
|
+
properties with the
|
788
|
+
provided properties.
|
1066
789
|
|
1067
790
|
Offers the user to go with the browser to
|
1068
791
|
``auth/<vo>/device?user_code=XYZ``.
|
@@ -1457,9 +1180,9 @@ class AuthOperations: # pylint: disable=abstract-class-instantiated
|
|
1457
1180
|
def authorization_flow(
|
1458
1181
|
self,
|
1459
1182
|
*,
|
1460
|
-
response_type:
|
1183
|
+
response_type: str,
|
1461
1184
|
code_challenge: str,
|
1462
|
-
code_challenge_method:
|
1185
|
+
code_challenge_method: str,
|
1463
1186
|
client_id: str,
|
1464
1187
|
redirect_uri: str,
|
1465
1188
|
scope: str,
|
@@ -1472,16 +1195,35 @@ class AuthOperations: # pylint: disable=abstract-class-instantiated
|
|
1472
1195
|
It will redirect to the actual OpenID server (IAM, CheckIn) to
|
1473
1196
|
perform a authorization code flow.
|
1474
1197
|
|
1198
|
+
Scope details:
|
1199
|
+
|
1200
|
+
|
1201
|
+
*
|
1202
|
+
If only VO is provided: Uses the default group and its properties for the VO.
|
1203
|
+
|
1204
|
+
*
|
1205
|
+
If VO and group are provided: Uses the specified group and its properties for the VO.
|
1206
|
+
|
1207
|
+
*
|
1208
|
+
If VO and properties are provided: Uses the default group and combines its properties with
|
1209
|
+
the
|
1210
|
+
provided properties.
|
1211
|
+
|
1212
|
+
*
|
1213
|
+
If VO, group, and properties are provided: Uses the specified group and combines its
|
1214
|
+
properties with the
|
1215
|
+
provided properties.
|
1216
|
+
|
1475
1217
|
We set the user details obtained from the user authorize flow in a cookie
|
1476
1218
|
to be able to map the authorization flow with the corresponding
|
1477
1219
|
user authorize flow.
|
1478
1220
|
|
1479
|
-
:keyword response_type:
|
1480
|
-
:paramtype response_type: str
|
1221
|
+
:keyword response_type: Required.
|
1222
|
+
:paramtype response_type: str
|
1481
1223
|
:keyword code_challenge: Required.
|
1482
1224
|
:paramtype code_challenge: str
|
1483
|
-
:keyword code_challenge_method:
|
1484
|
-
:paramtype code_challenge_method: str
|
1225
|
+
:keyword code_challenge_method: Required.
|
1226
|
+
:paramtype code_challenge_method: str
|
1485
1227
|
:keyword client_id: Required.
|
1486
1228
|
:paramtype client_id: str
|
1487
1229
|
:keyword redirect_uri: Required.
|
@@ -1709,7 +1451,7 @@ class ConfigOperations:
|
|
1709
1451
|
return deserialized # type: ignore
|
1710
1452
|
|
1711
1453
|
|
1712
|
-
class JobsOperations:
|
1454
|
+
class JobsOperations:
|
1713
1455
|
"""
|
1714
1456
|
.. warning::
|
1715
1457
|
**DO NOT** instantiate this class directly.
|
@@ -1945,1015 +1687,13 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
1945
1687
|
}
|
1946
1688
|
error_map.update(kwargs.pop("error_map", {}) or {})
|
1947
1689
|
|
1948
|
-
_headers = kwargs.pop("headers", {}) or {}
|
1949
|
-
_params = kwargs.pop("params", {}) or {}
|
1950
|
-
|
1951
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
1952
|
-
|
1953
|
-
_request = build_jobs_unassign_bulk_jobs_sandboxes_request(
|
1954
|
-
jobs_ids=jobs_ids,
|
1955
|
-
headers=_headers,
|
1956
|
-
params=_params,
|
1957
|
-
)
|
1958
|
-
_request.url = self._client.format_url(_request.url)
|
1959
|
-
|
1960
|
-
_stream = False
|
1961
|
-
pipeline_response: PipelineResponse = (
|
1962
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
1963
|
-
_request, stream=_stream, **kwargs
|
1964
|
-
)
|
1965
|
-
)
|
1966
|
-
|
1967
|
-
response = pipeline_response.http_response
|
1968
|
-
|
1969
|
-
if response.status_code not in [200]:
|
1970
|
-
map_error(
|
1971
|
-
status_code=response.status_code, response=response, error_map=error_map
|
1972
|
-
)
|
1973
|
-
raise HttpResponseError(response=response)
|
1974
|
-
|
1975
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
1976
|
-
|
1977
|
-
if cls:
|
1978
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
1979
|
-
|
1980
|
-
return deserialized # type: ignore
|
1981
|
-
|
1982
|
-
@distributed_trace
|
1983
|
-
def get_job_sandboxes(self, job_id: int, **kwargs: Any) -> Dict[str, List[Any]]:
|
1984
|
-
"""Get Job Sandboxes.
|
1985
|
-
|
1986
|
-
Get input and output sandboxes of given job.
|
1987
|
-
|
1988
|
-
:param job_id: Required.
|
1989
|
-
:type job_id: int
|
1990
|
-
:return: dict mapping str to list of any
|
1991
|
-
:rtype: dict[str, list[any]]
|
1992
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
1993
|
-
"""
|
1994
|
-
error_map: MutableMapping = {
|
1995
|
-
401: ClientAuthenticationError,
|
1996
|
-
404: ResourceNotFoundError,
|
1997
|
-
409: ResourceExistsError,
|
1998
|
-
304: ResourceNotModifiedError,
|
1999
|
-
}
|
2000
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2001
|
-
|
2002
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2003
|
-
_params = kwargs.pop("params", {}) or {}
|
2004
|
-
|
2005
|
-
cls: ClsType[Dict[str, List[Any]]] = kwargs.pop("cls", None)
|
2006
|
-
|
2007
|
-
_request = build_jobs_get_job_sandboxes_request(
|
2008
|
-
job_id=job_id,
|
2009
|
-
headers=_headers,
|
2010
|
-
params=_params,
|
2011
|
-
)
|
2012
|
-
_request.url = self._client.format_url(_request.url)
|
2013
|
-
|
2014
|
-
_stream = False
|
2015
|
-
pipeline_response: PipelineResponse = (
|
2016
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2017
|
-
_request, stream=_stream, **kwargs
|
2018
|
-
)
|
2019
|
-
)
|
2020
|
-
|
2021
|
-
response = pipeline_response.http_response
|
2022
|
-
|
2023
|
-
if response.status_code not in [200]:
|
2024
|
-
map_error(
|
2025
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2026
|
-
)
|
2027
|
-
raise HttpResponseError(response=response)
|
2028
|
-
|
2029
|
-
deserialized = self._deserialize("{[object]}", pipeline_response.http_response)
|
2030
|
-
|
2031
|
-
if cls:
|
2032
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2033
|
-
|
2034
|
-
return deserialized # type: ignore
|
2035
|
-
|
2036
|
-
@distributed_trace
|
2037
|
-
def unassign_job_sandboxes(self, job_id: int, **kwargs: Any) -> Any:
|
2038
|
-
"""Unassign Job Sandboxes.
|
2039
|
-
|
2040
|
-
Delete single job sandbox mapping.
|
2041
|
-
|
2042
|
-
:param job_id: Required.
|
2043
|
-
:type job_id: int
|
2044
|
-
:return: any
|
2045
|
-
:rtype: any
|
2046
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2047
|
-
"""
|
2048
|
-
error_map: MutableMapping = {
|
2049
|
-
401: ClientAuthenticationError,
|
2050
|
-
404: ResourceNotFoundError,
|
2051
|
-
409: ResourceExistsError,
|
2052
|
-
304: ResourceNotModifiedError,
|
2053
|
-
}
|
2054
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2055
|
-
|
2056
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2057
|
-
_params = kwargs.pop("params", {}) or {}
|
2058
|
-
|
2059
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2060
|
-
|
2061
|
-
_request = build_jobs_unassign_job_sandboxes_request(
|
2062
|
-
job_id=job_id,
|
2063
|
-
headers=_headers,
|
2064
|
-
params=_params,
|
2065
|
-
)
|
2066
|
-
_request.url = self._client.format_url(_request.url)
|
2067
|
-
|
2068
|
-
_stream = False
|
2069
|
-
pipeline_response: PipelineResponse = (
|
2070
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2071
|
-
_request, stream=_stream, **kwargs
|
2072
|
-
)
|
2073
|
-
)
|
2074
|
-
|
2075
|
-
response = pipeline_response.http_response
|
2076
|
-
|
2077
|
-
if response.status_code not in [200]:
|
2078
|
-
map_error(
|
2079
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2080
|
-
)
|
2081
|
-
raise HttpResponseError(response=response)
|
2082
|
-
|
2083
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2084
|
-
|
2085
|
-
if cls:
|
2086
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2087
|
-
|
2088
|
-
return deserialized # type: ignore
|
2089
|
-
|
2090
|
-
@distributed_trace
|
2091
|
-
def get_job_sandbox(
|
2092
|
-
self, job_id: int, sandbox_type: Union[str, _models.SandboxType], **kwargs: Any
|
2093
|
-
) -> List[Any]:
|
2094
|
-
"""Get Job Sandbox.
|
2095
|
-
|
2096
|
-
Get input or output sandbox of given job.
|
2097
|
-
|
2098
|
-
:param job_id: Required.
|
2099
|
-
:type job_id: int
|
2100
|
-
:param sandbox_type: Known values are: "input" and "output". Required.
|
2101
|
-
:type sandbox_type: str or ~generated.models.SandboxType
|
2102
|
-
:return: list of any
|
2103
|
-
:rtype: list[any]
|
2104
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2105
|
-
"""
|
2106
|
-
error_map: MutableMapping = {
|
2107
|
-
401: ClientAuthenticationError,
|
2108
|
-
404: ResourceNotFoundError,
|
2109
|
-
409: ResourceExistsError,
|
2110
|
-
304: ResourceNotModifiedError,
|
2111
|
-
}
|
2112
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2113
|
-
|
2114
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2115
|
-
_params = kwargs.pop("params", {}) or {}
|
2116
|
-
|
2117
|
-
cls: ClsType[List[Any]] = kwargs.pop("cls", None)
|
2118
|
-
|
2119
|
-
_request = build_jobs_get_job_sandbox_request(
|
2120
|
-
job_id=job_id,
|
2121
|
-
sandbox_type=sandbox_type,
|
2122
|
-
headers=_headers,
|
2123
|
-
params=_params,
|
2124
|
-
)
|
2125
|
-
_request.url = self._client.format_url(_request.url)
|
2126
|
-
|
2127
|
-
_stream = False
|
2128
|
-
pipeline_response: PipelineResponse = (
|
2129
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2130
|
-
_request, stream=_stream, **kwargs
|
2131
|
-
)
|
2132
|
-
)
|
2133
|
-
|
2134
|
-
response = pipeline_response.http_response
|
2135
|
-
|
2136
|
-
if response.status_code not in [200]:
|
2137
|
-
map_error(
|
2138
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2139
|
-
)
|
2140
|
-
raise HttpResponseError(response=response)
|
2141
|
-
|
2142
|
-
deserialized = self._deserialize("[object]", pipeline_response.http_response)
|
2143
|
-
|
2144
|
-
if cls:
|
2145
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2146
|
-
|
2147
|
-
return deserialized # type: ignore
|
2148
|
-
|
2149
|
-
@distributed_trace
|
2150
|
-
def assign_sandbox_to_job(self, job_id: int, body: str, **kwargs: Any) -> Any:
|
2151
|
-
"""Assign Sandbox To Job.
|
2152
|
-
|
2153
|
-
Map the pfn as output sandbox to job.
|
2154
|
-
|
2155
|
-
:param job_id: Required.
|
2156
|
-
:type job_id: int
|
2157
|
-
:param body: Required.
|
2158
|
-
:type body: str
|
2159
|
-
:return: any
|
2160
|
-
:rtype: any
|
2161
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2162
|
-
"""
|
2163
|
-
error_map: MutableMapping = {
|
2164
|
-
401: ClientAuthenticationError,
|
2165
|
-
404: ResourceNotFoundError,
|
2166
|
-
409: ResourceExistsError,
|
2167
|
-
304: ResourceNotModifiedError,
|
2168
|
-
}
|
2169
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2170
|
-
|
2171
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
2172
|
-
_params = kwargs.pop("params", {}) or {}
|
2173
|
-
|
2174
|
-
content_type: str = kwargs.pop(
|
2175
|
-
"content_type", _headers.pop("Content-Type", "application/json")
|
2176
|
-
)
|
2177
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2178
|
-
|
2179
|
-
_content = self._serialize.body(body, "str")
|
2180
|
-
|
2181
|
-
_request = build_jobs_assign_sandbox_to_job_request(
|
2182
|
-
job_id=job_id,
|
2183
|
-
content_type=content_type,
|
2184
|
-
content=_content,
|
2185
|
-
headers=_headers,
|
2186
|
-
params=_params,
|
2187
|
-
)
|
2188
|
-
_request.url = self._client.format_url(_request.url)
|
2189
|
-
|
2190
|
-
_stream = False
|
2191
|
-
pipeline_response: PipelineResponse = (
|
2192
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2193
|
-
_request, stream=_stream, **kwargs
|
2194
|
-
)
|
2195
|
-
)
|
2196
|
-
|
2197
|
-
response = pipeline_response.http_response
|
2198
|
-
|
2199
|
-
if response.status_code not in [200]:
|
2200
|
-
map_error(
|
2201
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2202
|
-
)
|
2203
|
-
raise HttpResponseError(response=response)
|
2204
|
-
|
2205
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2206
|
-
|
2207
|
-
if cls:
|
2208
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2209
|
-
|
2210
|
-
return deserialized # type: ignore
|
2211
|
-
|
2212
|
-
@overload
|
2213
|
-
def submit_bulk_jobs(
|
2214
|
-
self, body: List[str], *, content_type: str = "application/json", **kwargs: Any
|
2215
|
-
) -> List[_models.InsertedJob]:
|
2216
|
-
"""Submit Bulk Jobs.
|
2217
|
-
|
2218
|
-
Submit Bulk Jobs.
|
2219
|
-
|
2220
|
-
:param body: Required.
|
2221
|
-
:type body: list[str]
|
2222
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2223
|
-
Default value is "application/json".
|
2224
|
-
:paramtype content_type: str
|
2225
|
-
:return: list of InsertedJob
|
2226
|
-
:rtype: list[~generated.models.InsertedJob]
|
2227
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2228
|
-
"""
|
2229
|
-
|
2230
|
-
@overload
|
2231
|
-
def submit_bulk_jobs(
|
2232
|
-
self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
|
2233
|
-
) -> List[_models.InsertedJob]:
|
2234
|
-
"""Submit Bulk Jobs.
|
2235
|
-
|
2236
|
-
Submit Bulk Jobs.
|
2237
|
-
|
2238
|
-
:param body: Required.
|
2239
|
-
:type body: IO[bytes]
|
2240
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2241
|
-
Default value is "application/json".
|
2242
|
-
:paramtype content_type: str
|
2243
|
-
:return: list of InsertedJob
|
2244
|
-
:rtype: list[~generated.models.InsertedJob]
|
2245
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2246
|
-
"""
|
2247
|
-
|
2248
|
-
@distributed_trace
|
2249
|
-
def submit_bulk_jobs(
|
2250
|
-
self, body: Union[List[str], IO[bytes]], **kwargs: Any
|
2251
|
-
) -> List[_models.InsertedJob]:
|
2252
|
-
"""Submit Bulk Jobs.
|
2253
|
-
|
2254
|
-
Submit Bulk Jobs.
|
2255
|
-
|
2256
|
-
:param body: Is either a [str] type or a IO[bytes] type. Required.
|
2257
|
-
:type body: list[str] or IO[bytes]
|
2258
|
-
:return: list of InsertedJob
|
2259
|
-
:rtype: list[~generated.models.InsertedJob]
|
2260
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2261
|
-
"""
|
2262
|
-
error_map: MutableMapping = {
|
2263
|
-
401: ClientAuthenticationError,
|
2264
|
-
404: ResourceNotFoundError,
|
2265
|
-
409: ResourceExistsError,
|
2266
|
-
304: ResourceNotModifiedError,
|
2267
|
-
}
|
2268
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2269
|
-
|
2270
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
2271
|
-
_params = kwargs.pop("params", {}) or {}
|
2272
|
-
|
2273
|
-
content_type: Optional[str] = kwargs.pop(
|
2274
|
-
"content_type", _headers.pop("Content-Type", None)
|
2275
|
-
)
|
2276
|
-
cls: ClsType[List[_models.InsertedJob]] = kwargs.pop("cls", None)
|
2277
|
-
|
2278
|
-
content_type = content_type or "application/json"
|
2279
|
-
_json = None
|
2280
|
-
_content = None
|
2281
|
-
if isinstance(body, (IOBase, bytes)):
|
2282
|
-
_content = body
|
2283
|
-
else:
|
2284
|
-
_json = self._serialize.body(body, "[str]")
|
2285
|
-
|
2286
|
-
_request = build_jobs_submit_bulk_jobs_request(
|
2287
|
-
content_type=content_type,
|
2288
|
-
json=_json,
|
2289
|
-
content=_content,
|
2290
|
-
headers=_headers,
|
2291
|
-
params=_params,
|
2292
|
-
)
|
2293
|
-
_request.url = self._client.format_url(_request.url)
|
2294
|
-
|
2295
|
-
_stream = False
|
2296
|
-
pipeline_response: PipelineResponse = (
|
2297
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2298
|
-
_request, stream=_stream, **kwargs
|
2299
|
-
)
|
2300
|
-
)
|
2301
|
-
|
2302
|
-
response = pipeline_response.http_response
|
2303
|
-
|
2304
|
-
if response.status_code not in [200]:
|
2305
|
-
map_error(
|
2306
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2307
|
-
)
|
2308
|
-
raise HttpResponseError(response=response)
|
2309
|
-
|
2310
|
-
deserialized = self._deserialize(
|
2311
|
-
"[InsertedJob]", pipeline_response.http_response
|
2312
|
-
)
|
2313
|
-
|
2314
|
-
if cls:
|
2315
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2316
|
-
|
2317
|
-
return deserialized # type: ignore
|
2318
|
-
|
2319
|
-
@distributed_trace
|
2320
|
-
def delete_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
|
2321
|
-
"""Delete Bulk Jobs.
|
2322
|
-
|
2323
|
-
Delete Bulk Jobs.
|
2324
|
-
|
2325
|
-
:keyword job_ids: Required.
|
2326
|
-
:paramtype job_ids: list[int]
|
2327
|
-
:return: any
|
2328
|
-
:rtype: any
|
2329
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2330
|
-
"""
|
2331
|
-
error_map: MutableMapping = {
|
2332
|
-
401: ClientAuthenticationError,
|
2333
|
-
404: ResourceNotFoundError,
|
2334
|
-
409: ResourceExistsError,
|
2335
|
-
304: ResourceNotModifiedError,
|
2336
|
-
}
|
2337
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2338
|
-
|
2339
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2340
|
-
_params = kwargs.pop("params", {}) or {}
|
2341
|
-
|
2342
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2343
|
-
|
2344
|
-
_request = build_jobs_delete_bulk_jobs_request(
|
2345
|
-
job_ids=job_ids,
|
2346
|
-
headers=_headers,
|
2347
|
-
params=_params,
|
2348
|
-
)
|
2349
|
-
_request.url = self._client.format_url(_request.url)
|
2350
|
-
|
2351
|
-
_stream = False
|
2352
|
-
pipeline_response: PipelineResponse = (
|
2353
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2354
|
-
_request, stream=_stream, **kwargs
|
2355
|
-
)
|
2356
|
-
)
|
2357
|
-
|
2358
|
-
response = pipeline_response.http_response
|
2359
|
-
|
2360
|
-
if response.status_code not in [200]:
|
2361
|
-
map_error(
|
2362
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2363
|
-
)
|
2364
|
-
raise HttpResponseError(response=response)
|
2365
|
-
|
2366
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2367
|
-
|
2368
|
-
if cls:
|
2369
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2370
|
-
|
2371
|
-
return deserialized # type: ignore
|
2372
|
-
|
2373
|
-
@distributed_trace
|
2374
|
-
def kill_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
|
2375
|
-
"""Kill Bulk Jobs.
|
2376
|
-
|
2377
|
-
Kill Bulk Jobs.
|
2378
|
-
|
2379
|
-
:keyword job_ids: Required.
|
2380
|
-
:paramtype job_ids: list[int]
|
2381
|
-
:return: any
|
2382
|
-
:rtype: any
|
2383
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2384
|
-
"""
|
2385
|
-
error_map: MutableMapping = {
|
2386
|
-
401: ClientAuthenticationError,
|
2387
|
-
404: ResourceNotFoundError,
|
2388
|
-
409: ResourceExistsError,
|
2389
|
-
304: ResourceNotModifiedError,
|
2390
|
-
}
|
2391
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2392
|
-
|
2393
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2394
|
-
_params = kwargs.pop("params", {}) or {}
|
2395
|
-
|
2396
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2397
|
-
|
2398
|
-
_request = build_jobs_kill_bulk_jobs_request(
|
2399
|
-
job_ids=job_ids,
|
2400
|
-
headers=_headers,
|
2401
|
-
params=_params,
|
2402
|
-
)
|
2403
|
-
_request.url = self._client.format_url(_request.url)
|
2404
|
-
|
2405
|
-
_stream = False
|
2406
|
-
pipeline_response: PipelineResponse = (
|
2407
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2408
|
-
_request, stream=_stream, **kwargs
|
2409
|
-
)
|
2410
|
-
)
|
2411
|
-
|
2412
|
-
response = pipeline_response.http_response
|
2413
|
-
|
2414
|
-
if response.status_code not in [200]:
|
2415
|
-
map_error(
|
2416
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2417
|
-
)
|
2418
|
-
raise HttpResponseError(response=response)
|
2419
|
-
|
2420
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2421
|
-
|
2422
|
-
if cls:
|
2423
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2424
|
-
|
2425
|
-
return deserialized # type: ignore
|
2426
|
-
|
2427
|
-
@distributed_trace
|
2428
|
-
def remove_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
|
2429
|
-
"""Remove Bulk Jobs.
|
2430
|
-
|
2431
|
-
Fully remove a list of jobs from the WMS databases.
|
2432
|
-
|
2433
|
-
WARNING: This endpoint has been implemented for the compatibility with the legacy DIRAC WMS
|
2434
|
-
and the JobCleaningAgent. However, once this agent is ported to diracx, this endpoint should
|
2435
|
-
be removed, and the delete endpoint should be used instead for any other purpose.
|
2436
|
-
|
2437
|
-
:keyword job_ids: Required.
|
2438
|
-
:paramtype job_ids: list[int]
|
2439
|
-
:return: any
|
2440
|
-
:rtype: any
|
2441
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2442
|
-
"""
|
2443
|
-
error_map: MutableMapping = {
|
2444
|
-
401: ClientAuthenticationError,
|
2445
|
-
404: ResourceNotFoundError,
|
2446
|
-
409: ResourceExistsError,
|
2447
|
-
304: ResourceNotModifiedError,
|
2448
|
-
}
|
2449
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2450
|
-
|
2451
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2452
|
-
_params = kwargs.pop("params", {}) or {}
|
2453
|
-
|
2454
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2455
|
-
|
2456
|
-
_request = build_jobs_remove_bulk_jobs_request(
|
2457
|
-
job_ids=job_ids,
|
2458
|
-
headers=_headers,
|
2459
|
-
params=_params,
|
2460
|
-
)
|
2461
|
-
_request.url = self._client.format_url(_request.url)
|
2462
|
-
|
2463
|
-
_stream = False
|
2464
|
-
pipeline_response: PipelineResponse = (
|
2465
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2466
|
-
_request, stream=_stream, **kwargs
|
2467
|
-
)
|
2468
|
-
)
|
2469
|
-
|
2470
|
-
response = pipeline_response.http_response
|
2471
|
-
|
2472
|
-
if response.status_code not in [200]:
|
2473
|
-
map_error(
|
2474
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2475
|
-
)
|
2476
|
-
raise HttpResponseError(response=response)
|
2477
|
-
|
2478
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2479
|
-
|
2480
|
-
if cls:
|
2481
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2482
|
-
|
2483
|
-
return deserialized # type: ignore
|
2484
|
-
|
2485
|
-
@distributed_trace
|
2486
|
-
def get_job_status_bulk(
|
2487
|
-
self, *, job_ids: List[int], **kwargs: Any
|
2488
|
-
) -> Dict[str, _models.LimitedJobStatusReturn]:
|
2489
|
-
"""Get Job Status Bulk.
|
2490
|
-
|
2491
|
-
Get Job Status Bulk.
|
2492
|
-
|
2493
|
-
:keyword job_ids: Required.
|
2494
|
-
:paramtype job_ids: list[int]
|
2495
|
-
:return: dict mapping str to LimitedJobStatusReturn
|
2496
|
-
:rtype: dict[str, ~generated.models.LimitedJobStatusReturn]
|
2497
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2498
|
-
"""
|
2499
|
-
error_map: MutableMapping = {
|
2500
|
-
401: ClientAuthenticationError,
|
2501
|
-
404: ResourceNotFoundError,
|
2502
|
-
409: ResourceExistsError,
|
2503
|
-
304: ResourceNotModifiedError,
|
2504
|
-
}
|
2505
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2506
|
-
|
2507
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2508
|
-
_params = kwargs.pop("params", {}) or {}
|
2509
|
-
|
2510
|
-
cls: ClsType[Dict[str, _models.LimitedJobStatusReturn]] = kwargs.pop(
|
2511
|
-
"cls", None
|
2512
|
-
)
|
2513
|
-
|
2514
|
-
_request = build_jobs_get_job_status_bulk_request(
|
2515
|
-
job_ids=job_ids,
|
2516
|
-
headers=_headers,
|
2517
|
-
params=_params,
|
2518
|
-
)
|
2519
|
-
_request.url = self._client.format_url(_request.url)
|
2520
|
-
|
2521
|
-
_stream = False
|
2522
|
-
pipeline_response: PipelineResponse = (
|
2523
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2524
|
-
_request, stream=_stream, **kwargs
|
2525
|
-
)
|
2526
|
-
)
|
2527
|
-
|
2528
|
-
response = pipeline_response.http_response
|
2529
|
-
|
2530
|
-
if response.status_code not in [200]:
|
2531
|
-
map_error(
|
2532
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2533
|
-
)
|
2534
|
-
raise HttpResponseError(response=response)
|
2535
|
-
|
2536
|
-
deserialized = self._deserialize(
|
2537
|
-
"{LimitedJobStatusReturn}", pipeline_response.http_response
|
2538
|
-
)
|
2539
|
-
|
2540
|
-
if cls:
|
2541
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2542
|
-
|
2543
|
-
return deserialized # type: ignore
|
2544
|
-
|
2545
|
-
@overload
|
2546
|
-
def set_job_status_bulk(
|
2547
|
-
self,
|
2548
|
-
body: Dict[str, Dict[str, _models.JobStatusUpdate]],
|
2549
|
-
*,
|
2550
|
-
force: bool = False,
|
2551
|
-
content_type: str = "application/json",
|
2552
|
-
**kwargs: Any,
|
2553
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
2554
|
-
"""Set Job Status Bulk.
|
2555
|
-
|
2556
|
-
Set Job Status Bulk.
|
2557
|
-
|
2558
|
-
:param body: Required.
|
2559
|
-
:type body: dict[str, dict[str, ~generated.models.JobStatusUpdate]]
|
2560
|
-
:keyword force: Default value is False.
|
2561
|
-
:paramtype force: bool
|
2562
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2563
|
-
Default value is "application/json".
|
2564
|
-
:paramtype content_type: str
|
2565
|
-
:return: dict mapping str to SetJobStatusReturn
|
2566
|
-
:rtype: dict[str, ~generated.models.SetJobStatusReturn]
|
2567
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2568
|
-
"""
|
2569
|
-
|
2570
|
-
@overload
|
2571
|
-
def set_job_status_bulk(
|
2572
|
-
self,
|
2573
|
-
body: IO[bytes],
|
2574
|
-
*,
|
2575
|
-
force: bool = False,
|
2576
|
-
content_type: str = "application/json",
|
2577
|
-
**kwargs: Any,
|
2578
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
2579
|
-
"""Set Job Status Bulk.
|
2580
|
-
|
2581
|
-
Set Job Status Bulk.
|
2582
|
-
|
2583
|
-
:param body: Required.
|
2584
|
-
:type body: IO[bytes]
|
2585
|
-
:keyword force: Default value is False.
|
2586
|
-
:paramtype force: bool
|
2587
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2588
|
-
Default value is "application/json".
|
2589
|
-
:paramtype content_type: str
|
2590
|
-
:return: dict mapping str to SetJobStatusReturn
|
2591
|
-
:rtype: dict[str, ~generated.models.SetJobStatusReturn]
|
2592
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2593
|
-
"""
|
2594
|
-
|
2595
|
-
@distributed_trace
|
2596
|
-
def set_job_status_bulk(
|
2597
|
-
self,
|
2598
|
-
body: Union[Dict[str, Dict[str, _models.JobStatusUpdate]], IO[bytes]],
|
2599
|
-
*,
|
2600
|
-
force: bool = False,
|
2601
|
-
**kwargs: Any,
|
2602
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
2603
|
-
"""Set Job Status Bulk.
|
2604
|
-
|
2605
|
-
Set Job Status Bulk.
|
2606
|
-
|
2607
|
-
:param body: Is either a {str: {str: JobStatusUpdate}} type or a IO[bytes] type. Required.
|
2608
|
-
:type body: dict[str, dict[str, ~generated.models.JobStatusUpdate]] or IO[bytes]
|
2609
|
-
:keyword force: Default value is False.
|
2610
|
-
:paramtype force: bool
|
2611
|
-
:return: dict mapping str to SetJobStatusReturn
|
2612
|
-
:rtype: dict[str, ~generated.models.SetJobStatusReturn]
|
2613
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2614
|
-
"""
|
2615
|
-
error_map: MutableMapping = {
|
2616
|
-
401: ClientAuthenticationError,
|
2617
|
-
404: ResourceNotFoundError,
|
2618
|
-
409: ResourceExistsError,
|
2619
|
-
304: ResourceNotModifiedError,
|
2620
|
-
}
|
2621
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2622
|
-
|
2623
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
2624
|
-
_params = kwargs.pop("params", {}) or {}
|
2625
|
-
|
2626
|
-
content_type: Optional[str] = kwargs.pop(
|
2627
|
-
"content_type", _headers.pop("Content-Type", None)
|
2628
|
-
)
|
2629
|
-
cls: ClsType[Dict[str, _models.SetJobStatusReturn]] = kwargs.pop("cls", None)
|
2630
|
-
|
2631
|
-
content_type = content_type or "application/json"
|
2632
|
-
_json = None
|
2633
|
-
_content = None
|
2634
|
-
if isinstance(body, (IOBase, bytes)):
|
2635
|
-
_content = body
|
2636
|
-
else:
|
2637
|
-
_json = self._serialize.body(body, "{{JobStatusUpdate}}")
|
2638
|
-
|
2639
|
-
_request = build_jobs_set_job_status_bulk_request(
|
2640
|
-
force=force,
|
2641
|
-
content_type=content_type,
|
2642
|
-
json=_json,
|
2643
|
-
content=_content,
|
2644
|
-
headers=_headers,
|
2645
|
-
params=_params,
|
2646
|
-
)
|
2647
|
-
_request.url = self._client.format_url(_request.url)
|
2648
|
-
|
2649
|
-
_stream = False
|
2650
|
-
pipeline_response: PipelineResponse = (
|
2651
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2652
|
-
_request, stream=_stream, **kwargs
|
2653
|
-
)
|
2654
|
-
)
|
2655
|
-
|
2656
|
-
response = pipeline_response.http_response
|
2657
|
-
|
2658
|
-
if response.status_code not in [200]:
|
2659
|
-
map_error(
|
2660
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2661
|
-
)
|
2662
|
-
raise HttpResponseError(response=response)
|
2663
|
-
|
2664
|
-
deserialized = self._deserialize(
|
2665
|
-
"{SetJobStatusReturn}", pipeline_response.http_response
|
2666
|
-
)
|
2667
|
-
|
2668
|
-
if cls:
|
2669
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2670
|
-
|
2671
|
-
return deserialized # type: ignore
|
2672
|
-
|
2673
|
-
@distributed_trace
|
2674
|
-
def get_job_status_history_bulk(
|
2675
|
-
self, *, job_ids: List[int], **kwargs: Any
|
2676
|
-
) -> Dict[str, List[_models.JobStatusReturn]]:
|
2677
|
-
"""Get Job Status History Bulk.
|
2678
|
-
|
2679
|
-
Get Job Status History Bulk.
|
2680
|
-
|
2681
|
-
:keyword job_ids: Required.
|
2682
|
-
:paramtype job_ids: list[int]
|
2683
|
-
:return: dict mapping str to list of JobStatusReturn
|
2684
|
-
:rtype: dict[str, list[~generated.models.JobStatusReturn]]
|
2685
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2686
|
-
"""
|
2687
|
-
error_map: MutableMapping = {
|
2688
|
-
401: ClientAuthenticationError,
|
2689
|
-
404: ResourceNotFoundError,
|
2690
|
-
409: ResourceExistsError,
|
2691
|
-
304: ResourceNotModifiedError,
|
2692
|
-
}
|
2693
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2694
|
-
|
2695
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2696
|
-
_params = kwargs.pop("params", {}) or {}
|
2697
|
-
|
2698
|
-
cls: ClsType[Dict[str, List[_models.JobStatusReturn]]] = kwargs.pop("cls", None)
|
2699
|
-
|
2700
|
-
_request = build_jobs_get_job_status_history_bulk_request(
|
2701
|
-
job_ids=job_ids,
|
2702
|
-
headers=_headers,
|
2703
|
-
params=_params,
|
2704
|
-
)
|
2705
|
-
_request.url = self._client.format_url(_request.url)
|
2706
|
-
|
2707
|
-
_stream = False
|
2708
|
-
pipeline_response: PipelineResponse = (
|
2709
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2710
|
-
_request, stream=_stream, **kwargs
|
2711
|
-
)
|
2712
|
-
)
|
2713
|
-
|
2714
|
-
response = pipeline_response.http_response
|
2715
|
-
|
2716
|
-
if response.status_code not in [200]:
|
2717
|
-
map_error(
|
2718
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2719
|
-
)
|
2720
|
-
raise HttpResponseError(response=response)
|
2721
|
-
|
2722
|
-
deserialized = self._deserialize(
|
2723
|
-
"{[JobStatusReturn]}", pipeline_response.http_response
|
2724
|
-
)
|
2725
|
-
|
2726
|
-
if cls:
|
2727
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2728
|
-
|
2729
|
-
return deserialized # type: ignore
|
2730
|
-
|
2731
|
-
@distributed_trace
|
2732
|
-
def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
|
2733
|
-
"""Reschedule Bulk Jobs.
|
2734
|
-
|
2735
|
-
Reschedule Bulk Jobs.
|
2736
|
-
|
2737
|
-
:keyword job_ids: Required.
|
2738
|
-
:paramtype job_ids: list[int]
|
2739
|
-
:return: any
|
2740
|
-
:rtype: any
|
2741
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2742
|
-
"""
|
2743
|
-
error_map: MutableMapping = {
|
2744
|
-
401: ClientAuthenticationError,
|
2745
|
-
404: ResourceNotFoundError,
|
2746
|
-
409: ResourceExistsError,
|
2747
|
-
304: ResourceNotModifiedError,
|
2748
|
-
}
|
2749
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2750
|
-
|
2751
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2752
|
-
_params = kwargs.pop("params", {}) or {}
|
2753
|
-
|
2754
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2755
|
-
|
2756
|
-
_request = build_jobs_reschedule_bulk_jobs_request(
|
2757
|
-
job_ids=job_ids,
|
2758
|
-
headers=_headers,
|
2759
|
-
params=_params,
|
2760
|
-
)
|
2761
|
-
_request.url = self._client.format_url(_request.url)
|
2762
|
-
|
2763
|
-
_stream = False
|
2764
|
-
pipeline_response: PipelineResponse = (
|
2765
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2766
|
-
_request, stream=_stream, **kwargs
|
2767
|
-
)
|
2768
|
-
)
|
2769
|
-
|
2770
|
-
response = pipeline_response.http_response
|
2771
|
-
|
2772
|
-
if response.status_code not in [200]:
|
2773
|
-
map_error(
|
2774
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2775
|
-
)
|
2776
|
-
raise HttpResponseError(response=response)
|
2777
|
-
|
2778
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2779
|
-
|
2780
|
-
if cls:
|
2781
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2782
|
-
|
2783
|
-
return deserialized # type: ignore
|
2784
|
-
|
2785
|
-
@distributed_trace
|
2786
|
-
def reschedule_single_job(self, job_id: int, **kwargs: Any) -> Any:
|
2787
|
-
"""Reschedule Single Job.
|
2788
|
-
|
2789
|
-
Reschedule Single Job.
|
2790
|
-
|
2791
|
-
:param job_id: Required.
|
2792
|
-
:type job_id: int
|
2793
|
-
:return: any
|
2794
|
-
:rtype: any
|
2795
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2796
|
-
"""
|
2797
|
-
error_map: MutableMapping = {
|
2798
|
-
401: ClientAuthenticationError,
|
2799
|
-
404: ResourceNotFoundError,
|
2800
|
-
409: ResourceExistsError,
|
2801
|
-
304: ResourceNotModifiedError,
|
2802
|
-
}
|
2803
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2804
|
-
|
2805
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2806
|
-
_params = kwargs.pop("params", {}) or {}
|
2807
|
-
|
2808
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2809
|
-
|
2810
|
-
_request = build_jobs_reschedule_single_job_request(
|
2811
|
-
job_id=job_id,
|
2812
|
-
headers=_headers,
|
2813
|
-
params=_params,
|
2814
|
-
)
|
2815
|
-
_request.url = self._client.format_url(_request.url)
|
2816
|
-
|
2817
|
-
_stream = False
|
2818
|
-
pipeline_response: PipelineResponse = (
|
2819
|
-
self._client._pipeline.run( # pylint: disable=protected-access
|
2820
|
-
_request, stream=_stream, **kwargs
|
2821
|
-
)
|
2822
|
-
)
|
2823
|
-
|
2824
|
-
response = pipeline_response.http_response
|
2825
|
-
|
2826
|
-
if response.status_code not in [200]:
|
2827
|
-
map_error(
|
2828
|
-
status_code=response.status_code, response=response, error_map=error_map
|
2829
|
-
)
|
2830
|
-
raise HttpResponseError(response=response)
|
2831
|
-
|
2832
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2833
|
-
|
2834
|
-
if cls:
|
2835
|
-
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2836
|
-
|
2837
|
-
return deserialized # type: ignore
|
2838
|
-
|
2839
|
-
@overload
|
2840
|
-
def search(
|
2841
|
-
self,
|
2842
|
-
body: Optional[_models.JobSearchParams] = None,
|
2843
|
-
*,
|
2844
|
-
page: int = 1,
|
2845
|
-
per_page: int = 100,
|
2846
|
-
content_type: str = "application/json",
|
2847
|
-
**kwargs: Any,
|
2848
|
-
) -> List[JSON]:
|
2849
|
-
"""Search.
|
2850
|
-
|
2851
|
-
Retrieve information about jobs.
|
2852
|
-
|
2853
|
-
**TODO: Add more docs**.
|
2854
|
-
|
2855
|
-
:param body: Default value is None.
|
2856
|
-
:type body: ~generated.models.JobSearchParams
|
2857
|
-
:keyword page: Default value is 1.
|
2858
|
-
:paramtype page: int
|
2859
|
-
:keyword per_page: Default value is 100.
|
2860
|
-
:paramtype per_page: int
|
2861
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2862
|
-
Default value is "application/json".
|
2863
|
-
:paramtype content_type: str
|
2864
|
-
:return: list of JSON
|
2865
|
-
:rtype: list[JSON]
|
2866
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2867
|
-
"""
|
2868
|
-
|
2869
|
-
@overload
|
2870
|
-
def search(
|
2871
|
-
self,
|
2872
|
-
body: Optional[IO[bytes]] = None,
|
2873
|
-
*,
|
2874
|
-
page: int = 1,
|
2875
|
-
per_page: int = 100,
|
2876
|
-
content_type: str = "application/json",
|
2877
|
-
**kwargs: Any,
|
2878
|
-
) -> List[JSON]:
|
2879
|
-
"""Search.
|
2880
|
-
|
2881
|
-
Retrieve information about jobs.
|
2882
|
-
|
2883
|
-
**TODO: Add more docs**.
|
2884
|
-
|
2885
|
-
:param body: Default value is None.
|
2886
|
-
:type body: IO[bytes]
|
2887
|
-
:keyword page: Default value is 1.
|
2888
|
-
:paramtype page: int
|
2889
|
-
:keyword per_page: Default value is 100.
|
2890
|
-
:paramtype per_page: int
|
2891
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2892
|
-
Default value is "application/json".
|
2893
|
-
:paramtype content_type: str
|
2894
|
-
:return: list of JSON
|
2895
|
-
:rtype: list[JSON]
|
2896
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2897
|
-
"""
|
2898
|
-
|
2899
|
-
@distributed_trace
|
2900
|
-
def search(
|
2901
|
-
self,
|
2902
|
-
body: Optional[Union[_models.JobSearchParams, IO[bytes]]] = None,
|
2903
|
-
*,
|
2904
|
-
page: int = 1,
|
2905
|
-
per_page: int = 100,
|
2906
|
-
**kwargs: Any,
|
2907
|
-
) -> List[JSON]:
|
2908
|
-
"""Search.
|
2909
|
-
|
2910
|
-
Retrieve information about jobs.
|
2911
|
-
|
2912
|
-
**TODO: Add more docs**.
|
2913
|
-
|
2914
|
-
:param body: Is either a JobSearchParams type or a IO[bytes] type. Default value is None.
|
2915
|
-
:type body: ~generated.models.JobSearchParams or IO[bytes]
|
2916
|
-
:keyword page: Default value is 1.
|
2917
|
-
:paramtype page: int
|
2918
|
-
:keyword per_page: Default value is 100.
|
2919
|
-
:paramtype per_page: int
|
2920
|
-
:return: list of JSON
|
2921
|
-
:rtype: list[JSON]
|
2922
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
2923
|
-
"""
|
2924
|
-
error_map: MutableMapping = {
|
2925
|
-
401: ClientAuthenticationError,
|
2926
|
-
404: ResourceNotFoundError,
|
2927
|
-
409: ResourceExistsError,
|
2928
|
-
304: ResourceNotModifiedError,
|
2929
|
-
}
|
2930
|
-
error_map.update(kwargs.pop("error_map", {}) or {})
|
2931
|
-
|
2932
|
-
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
1690
|
+
_headers = kwargs.pop("headers", {}) or {}
|
2933
1691
|
_params = kwargs.pop("params", {}) or {}
|
2934
1692
|
|
2935
|
-
|
2936
|
-
"content_type", _headers.pop("Content-Type", None)
|
2937
|
-
)
|
2938
|
-
cls: ClsType[List[JSON]] = kwargs.pop("cls", None)
|
2939
|
-
|
2940
|
-
content_type = content_type or "application/json"
|
2941
|
-
_json = None
|
2942
|
-
_content = None
|
2943
|
-
if isinstance(body, (IOBase, bytes)):
|
2944
|
-
_content = body
|
2945
|
-
else:
|
2946
|
-
if body is not None:
|
2947
|
-
_json = self._serialize.body(body, "JobSearchParams")
|
2948
|
-
else:
|
2949
|
-
_json = None
|
1693
|
+
cls: ClsType[Any] = kwargs.pop("cls", None)
|
2950
1694
|
|
2951
|
-
_request =
|
2952
|
-
|
2953
|
-
per_page=per_page,
|
2954
|
-
content_type=content_type,
|
2955
|
-
json=_json,
|
2956
|
-
content=_content,
|
1695
|
+
_request = build_jobs_unassign_bulk_jobs_sandboxes_request(
|
1696
|
+
jobs_ids=jobs_ids,
|
2957
1697
|
headers=_headers,
|
2958
1698
|
params=_params,
|
2959
1699
|
)
|
@@ -2968,77 +1708,29 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
2968
1708
|
|
2969
1709
|
response = pipeline_response.http_response
|
2970
1710
|
|
2971
|
-
if response.status_code not in [200
|
1711
|
+
if response.status_code not in [200]:
|
2972
1712
|
map_error(
|
2973
1713
|
status_code=response.status_code, response=response, error_map=error_map
|
2974
1714
|
)
|
2975
1715
|
raise HttpResponseError(response=response)
|
2976
1716
|
|
2977
|
-
|
2978
|
-
if response.status_code == 206:
|
2979
|
-
response_headers["Content-Range"] = self._deserialize(
|
2980
|
-
"str", response.headers.get("Content-Range")
|
2981
|
-
)
|
2982
|
-
|
2983
|
-
deserialized = self._deserialize("[object]", pipeline_response.http_response)
|
1717
|
+
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2984
1718
|
|
2985
1719
|
if cls:
|
2986
|
-
return cls(pipeline_response, deserialized,
|
1720
|
+
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2987
1721
|
|
2988
1722
|
return deserialized # type: ignore
|
2989
1723
|
|
2990
|
-
@overload
|
2991
|
-
def summary(
|
2992
|
-
self,
|
2993
|
-
body: _models.JobSummaryParams,
|
2994
|
-
*,
|
2995
|
-
content_type: str = "application/json",
|
2996
|
-
**kwargs: Any,
|
2997
|
-
) -> Any:
|
2998
|
-
"""Summary.
|
2999
|
-
|
3000
|
-
Show information suitable for plotting.
|
3001
|
-
|
3002
|
-
:param body: Required.
|
3003
|
-
:type body: ~generated.models.JobSummaryParams
|
3004
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
3005
|
-
Default value is "application/json".
|
3006
|
-
:paramtype content_type: str
|
3007
|
-
:return: any
|
3008
|
-
:rtype: any
|
3009
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
3010
|
-
"""
|
3011
|
-
|
3012
|
-
@overload
|
3013
|
-
def summary(
|
3014
|
-
self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
|
3015
|
-
) -> Any:
|
3016
|
-
"""Summary.
|
3017
|
-
|
3018
|
-
Show information suitable for plotting.
|
3019
|
-
|
3020
|
-
:param body: Required.
|
3021
|
-
:type body: IO[bytes]
|
3022
|
-
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
3023
|
-
Default value is "application/json".
|
3024
|
-
:paramtype content_type: str
|
3025
|
-
:return: any
|
3026
|
-
:rtype: any
|
3027
|
-
:raises ~azure.core.exceptions.HttpResponseError:
|
3028
|
-
"""
|
3029
|
-
|
3030
1724
|
@distributed_trace
|
3031
|
-
def
|
3032
|
-
|
3033
|
-
) -> Any:
|
3034
|
-
"""Summary.
|
1725
|
+
def get_job_sandboxes(self, job_id: int, **kwargs: Any) -> Dict[str, List[Any]]:
|
1726
|
+
"""Get Job Sandboxes.
|
3035
1727
|
|
3036
|
-
|
1728
|
+
Get input and output sandboxes of given job.
|
3037
1729
|
|
3038
|
-
:param
|
3039
|
-
:type
|
3040
|
-
:return: any
|
3041
|
-
:rtype: any
|
1730
|
+
:param job_id: Required.
|
1731
|
+
:type job_id: int
|
1732
|
+
:return: dict mapping str to list of any
|
1733
|
+
:rtype: dict[str, list[any]]
|
3042
1734
|
:raises ~azure.core.exceptions.HttpResponseError:
|
3043
1735
|
"""
|
3044
1736
|
error_map: MutableMapping = {
|
@@ -3049,26 +1741,13 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3049
1741
|
}
|
3050
1742
|
error_map.update(kwargs.pop("error_map", {}) or {})
|
3051
1743
|
|
3052
|
-
_headers =
|
1744
|
+
_headers = kwargs.pop("headers", {}) or {}
|
3053
1745
|
_params = kwargs.pop("params", {}) or {}
|
3054
1746
|
|
3055
|
-
|
3056
|
-
"content_type", _headers.pop("Content-Type", None)
|
3057
|
-
)
|
3058
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
3059
|
-
|
3060
|
-
content_type = content_type or "application/json"
|
3061
|
-
_json = None
|
3062
|
-
_content = None
|
3063
|
-
if isinstance(body, (IOBase, bytes)):
|
3064
|
-
_content = body
|
3065
|
-
else:
|
3066
|
-
_json = self._serialize.body(body, "JobSummaryParams")
|
1747
|
+
cls: ClsType[Dict[str, List[Any]]] = kwargs.pop("cls", None)
|
3067
1748
|
|
3068
|
-
_request =
|
3069
|
-
|
3070
|
-
json=_json,
|
3071
|
-
content=_content,
|
1749
|
+
_request = build_jobs_get_job_sandboxes_request(
|
1750
|
+
job_id=job_id,
|
3072
1751
|
headers=_headers,
|
3073
1752
|
params=_params,
|
3074
1753
|
)
|
@@ -3089,7 +1768,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3089
1768
|
)
|
3090
1769
|
raise HttpResponseError(response=response)
|
3091
1770
|
|
3092
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
1771
|
+
deserialized = self._deserialize("{[object]}", pipeline_response.http_response)
|
3093
1772
|
|
3094
1773
|
if cls:
|
3095
1774
|
return cls(pipeline_response, deserialized, {}) # type: ignore
|
@@ -3097,10 +1776,10 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3097
1776
|
return deserialized # type: ignore
|
3098
1777
|
|
3099
1778
|
@distributed_trace
|
3100
|
-
def
|
3101
|
-
"""
|
1779
|
+
def unassign_job_sandboxes(self, job_id: int, **kwargs: Any) -> Any:
|
1780
|
+
"""Unassign Job Sandboxes.
|
3102
1781
|
|
3103
|
-
|
1782
|
+
Delete single job sandbox mapping.
|
3104
1783
|
|
3105
1784
|
:param job_id: Required.
|
3106
1785
|
:type job_id: int
|
@@ -3121,7 +1800,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3121
1800
|
|
3122
1801
|
cls: ClsType[Any] = kwargs.pop("cls", None)
|
3123
1802
|
|
3124
|
-
_request =
|
1803
|
+
_request = build_jobs_unassign_job_sandboxes_request(
|
3125
1804
|
job_id=job_id,
|
3126
1805
|
headers=_headers,
|
3127
1806
|
params=_params,
|
@@ -3151,15 +1830,19 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3151
1830
|
return deserialized # type: ignore
|
3152
1831
|
|
3153
1832
|
@distributed_trace
|
3154
|
-
def
|
3155
|
-
|
1833
|
+
def get_job_sandbox(
|
1834
|
+
self, job_id: int, sandbox_type: Union[str, _models.SandboxType], **kwargs: Any
|
1835
|
+
) -> List[Any]:
|
1836
|
+
"""Get Job Sandbox.
|
3156
1837
|
|
3157
|
-
|
1838
|
+
Get input or output sandbox of given job.
|
3158
1839
|
|
3159
1840
|
:param job_id: Required.
|
3160
1841
|
:type job_id: int
|
3161
|
-
:
|
3162
|
-
:
|
1842
|
+
:param sandbox_type: Known values are: "input" and "output". Required.
|
1843
|
+
:type sandbox_type: str or ~generated.models.SandboxType
|
1844
|
+
:return: list of any
|
1845
|
+
:rtype: list[any]
|
3163
1846
|
:raises ~azure.core.exceptions.HttpResponseError:
|
3164
1847
|
"""
|
3165
1848
|
error_map: MutableMapping = {
|
@@ -3173,10 +1856,11 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3173
1856
|
_headers = kwargs.pop("headers", {}) or {}
|
3174
1857
|
_params = kwargs.pop("params", {}) or {}
|
3175
1858
|
|
3176
|
-
cls: ClsType[Any] = kwargs.pop("cls", None)
|
1859
|
+
cls: ClsType[List[Any]] = kwargs.pop("cls", None)
|
3177
1860
|
|
3178
|
-
_request =
|
1861
|
+
_request = build_jobs_get_job_sandbox_request(
|
3179
1862
|
job_id=job_id,
|
1863
|
+
sandbox_type=sandbox_type,
|
3180
1864
|
headers=_headers,
|
3181
1865
|
params=_params,
|
3182
1866
|
)
|
@@ -3197,7 +1881,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3197
1881
|
)
|
3198
1882
|
raise HttpResponseError(response=response)
|
3199
1883
|
|
3200
|
-
deserialized = self._deserialize("object", pipeline_response.http_response)
|
1884
|
+
deserialized = self._deserialize("[object]", pipeline_response.http_response)
|
3201
1885
|
|
3202
1886
|
if cls:
|
3203
1887
|
return cls(pipeline_response, deserialized, {}) # type: ignore
|
@@ -3205,19 +1889,15 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3205
1889
|
return deserialized # type: ignore
|
3206
1890
|
|
3207
1891
|
@distributed_trace
|
3208
|
-
def
|
3209
|
-
|
3210
|
-
) -> Any:
|
3211
|
-
"""Set Single Job Properties.
|
1892
|
+
def assign_sandbox_to_job(self, job_id: int, body: str, **kwargs: Any) -> Any:
|
1893
|
+
"""Assign Sandbox To Job.
|
3212
1894
|
|
3213
|
-
|
1895
|
+
Map the pfn as output sandbox to job.
|
3214
1896
|
|
3215
1897
|
:param job_id: Required.
|
3216
1898
|
:type job_id: int
|
3217
1899
|
:param body: Required.
|
3218
|
-
:type body:
|
3219
|
-
:keyword update_timestamp: Default value is False.
|
3220
|
-
:paramtype update_timestamp: bool
|
1900
|
+
:type body: str
|
3221
1901
|
:return: any
|
3222
1902
|
:rtype: any
|
3223
1903
|
:raises ~azure.core.exceptions.HttpResponseError:
|
@@ -3238,13 +1918,12 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3238
1918
|
)
|
3239
1919
|
cls: ClsType[Any] = kwargs.pop("cls", None)
|
3240
1920
|
|
3241
|
-
|
1921
|
+
_content = self._serialize.body(body, "str")
|
3242
1922
|
|
3243
|
-
_request =
|
1923
|
+
_request = build_jobs_assign_sandbox_to_job_request(
|
3244
1924
|
job_id=job_id,
|
3245
|
-
update_timestamp=update_timestamp,
|
3246
1925
|
content_type=content_type,
|
3247
|
-
|
1926
|
+
content=_content,
|
3248
1927
|
headers=_headers,
|
3249
1928
|
params=_params,
|
3250
1929
|
)
|
@@ -3273,13 +1952,18 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3273
1952
|
return deserialized # type: ignore
|
3274
1953
|
|
3275
1954
|
@distributed_trace
|
3276
|
-
def
|
3277
|
-
"""
|
1955
|
+
def remove_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
|
1956
|
+
"""Remove Bulk Jobs.
|
3278
1957
|
|
3279
|
-
|
1958
|
+
Fully remove a list of jobs from the WMS databases.
|
3280
1959
|
|
3281
|
-
:
|
3282
|
-
|
1960
|
+
WARNING: This endpoint has been implemented for the compatibility with the legacy DIRAC WMS
|
1961
|
+
and the JobCleaningAgent. However, once this agent is ported to diracx, this endpoint should
|
1962
|
+
be removed, and a status change to Deleted (PATCH /jobs/status) should be used instead for any
|
1963
|
+
other purpose.
|
1964
|
+
|
1965
|
+
:keyword job_ids: Required.
|
1966
|
+
:paramtype job_ids: list[int]
|
3283
1967
|
:return: any
|
3284
1968
|
:rtype: any
|
3285
1969
|
:raises ~azure.core.exceptions.HttpResponseError:
|
@@ -3297,8 +1981,134 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3297
1981
|
|
3298
1982
|
cls: ClsType[Any] = kwargs.pop("cls", None)
|
3299
1983
|
|
3300
|
-
_request =
|
3301
|
-
|
1984
|
+
_request = build_jobs_remove_bulk_jobs_request(
|
1985
|
+
job_ids=job_ids,
|
1986
|
+
headers=_headers,
|
1987
|
+
params=_params,
|
1988
|
+
)
|
1989
|
+
_request.url = self._client.format_url(_request.url)
|
1990
|
+
|
1991
|
+
_stream = False
|
1992
|
+
pipeline_response: PipelineResponse = (
|
1993
|
+
self._client._pipeline.run( # pylint: disable=protected-access
|
1994
|
+
_request, stream=_stream, **kwargs
|
1995
|
+
)
|
1996
|
+
)
|
1997
|
+
|
1998
|
+
response = pipeline_response.http_response
|
1999
|
+
|
2000
|
+
if response.status_code not in [200]:
|
2001
|
+
map_error(
|
2002
|
+
status_code=response.status_code, response=response, error_map=error_map
|
2003
|
+
)
|
2004
|
+
raise HttpResponseError(response=response)
|
2005
|
+
|
2006
|
+
deserialized = self._deserialize("object", pipeline_response.http_response)
|
2007
|
+
|
2008
|
+
if cls:
|
2009
|
+
return cls(pipeline_response, deserialized, {}) # type: ignore
|
2010
|
+
|
2011
|
+
return deserialized # type: ignore
|
2012
|
+
|
2013
|
+
@overload
|
2014
|
+
def set_job_statuses(
|
2015
|
+
self,
|
2016
|
+
body: Dict[str, Dict[str, _models.JobStatusUpdate]],
|
2017
|
+
*,
|
2018
|
+
force: bool = False,
|
2019
|
+
content_type: str = "application/json",
|
2020
|
+
**kwargs: Any,
|
2021
|
+
) -> _models.SetJobStatusReturn:
|
2022
|
+
"""Set Job Statuses.
|
2023
|
+
|
2024
|
+
Set Job Statuses.
|
2025
|
+
|
2026
|
+
:param body: Required.
|
2027
|
+
:type body: dict[str, dict[str, ~generated.models.JobStatusUpdate]]
|
2028
|
+
:keyword force: Default value is False.
|
2029
|
+
:paramtype force: bool
|
2030
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2031
|
+
Default value is "application/json".
|
2032
|
+
:paramtype content_type: str
|
2033
|
+
:return: SetJobStatusReturn
|
2034
|
+
:rtype: ~generated.models.SetJobStatusReturn
|
2035
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
2036
|
+
"""
|
2037
|
+
|
2038
|
+
@overload
|
2039
|
+
def set_job_statuses(
|
2040
|
+
self,
|
2041
|
+
body: IO[bytes],
|
2042
|
+
*,
|
2043
|
+
force: bool = False,
|
2044
|
+
content_type: str = "application/json",
|
2045
|
+
**kwargs: Any,
|
2046
|
+
) -> _models.SetJobStatusReturn:
|
2047
|
+
"""Set Job Statuses.
|
2048
|
+
|
2049
|
+
Set Job Statuses.
|
2050
|
+
|
2051
|
+
:param body: Required.
|
2052
|
+
:type body: IO[bytes]
|
2053
|
+
:keyword force: Default value is False.
|
2054
|
+
:paramtype force: bool
|
2055
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2056
|
+
Default value is "application/json".
|
2057
|
+
:paramtype content_type: str
|
2058
|
+
:return: SetJobStatusReturn
|
2059
|
+
:rtype: ~generated.models.SetJobStatusReturn
|
2060
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
2061
|
+
"""
|
2062
|
+
|
2063
|
+
@distributed_trace
|
2064
|
+
def set_job_statuses(
|
2065
|
+
self,
|
2066
|
+
body: Union[Dict[str, Dict[str, _models.JobStatusUpdate]], IO[bytes]],
|
2067
|
+
*,
|
2068
|
+
force: bool = False,
|
2069
|
+
**kwargs: Any,
|
2070
|
+
) -> _models.SetJobStatusReturn:
|
2071
|
+
"""Set Job Statuses.
|
2072
|
+
|
2073
|
+
Set Job Statuses.
|
2074
|
+
|
2075
|
+
:param body: Is either a {str: {str: JobStatusUpdate}} type or a IO[bytes] type. Required.
|
2076
|
+
:type body: dict[str, dict[str, ~generated.models.JobStatusUpdate]] or IO[bytes]
|
2077
|
+
:keyword force: Default value is False.
|
2078
|
+
:paramtype force: bool
|
2079
|
+
:return: SetJobStatusReturn
|
2080
|
+
:rtype: ~generated.models.SetJobStatusReturn
|
2081
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
2082
|
+
"""
|
2083
|
+
error_map: MutableMapping = {
|
2084
|
+
401: ClientAuthenticationError,
|
2085
|
+
404: ResourceNotFoundError,
|
2086
|
+
409: ResourceExistsError,
|
2087
|
+
304: ResourceNotModifiedError,
|
2088
|
+
}
|
2089
|
+
error_map.update(kwargs.pop("error_map", {}) or {})
|
2090
|
+
|
2091
|
+
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
2092
|
+
_params = kwargs.pop("params", {}) or {}
|
2093
|
+
|
2094
|
+
content_type: Optional[str] = kwargs.pop(
|
2095
|
+
"content_type", _headers.pop("Content-Type", None)
|
2096
|
+
)
|
2097
|
+
cls: ClsType[_models.SetJobStatusReturn] = kwargs.pop("cls", None)
|
2098
|
+
|
2099
|
+
content_type = content_type or "application/json"
|
2100
|
+
_json = None
|
2101
|
+
_content = None
|
2102
|
+
if isinstance(body, (IOBase, bytes)):
|
2103
|
+
_content = body
|
2104
|
+
else:
|
2105
|
+
_json = self._serialize.body(body, "{{JobStatusUpdate}}")
|
2106
|
+
|
2107
|
+
_request = build_jobs_set_job_statuses_request(
|
2108
|
+
force=force,
|
2109
|
+
content_type=content_type,
|
2110
|
+
json=_json,
|
2111
|
+
content=_content,
|
3302
2112
|
headers=_headers,
|
3303
2113
|
params=_params,
|
3304
2114
|
)
|
@@ -3319,7 +2129,9 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3319
2129
|
)
|
3320
2130
|
raise HttpResponseError(response=response)
|
3321
2131
|
|
3322
|
-
deserialized = self._deserialize(
|
2132
|
+
deserialized = self._deserialize(
|
2133
|
+
"SetJobStatusReturn", pipeline_response.http_response
|
2134
|
+
)
|
3323
2135
|
|
3324
2136
|
if cls:
|
3325
2137
|
return cls(pipeline_response, deserialized, {}) # type: ignore
|
@@ -3327,17 +2139,17 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3327
2139
|
return deserialized # type: ignore
|
3328
2140
|
|
3329
2141
|
@distributed_trace
|
3330
|
-
def
|
3331
|
-
|
3332
|
-
|
3333
|
-
|
2142
|
+
def reschedule_bulk_jobs(
|
2143
|
+
self, *, job_ids: List[int], reset_jobs: bool = False, **kwargs: Any
|
2144
|
+
) -> Any:
|
2145
|
+
"""Reschedule Bulk Jobs.
|
3334
2146
|
|
3335
|
-
|
3336
|
-
and the JobCleaningAgent. However, once this agent is ported to diracx, this endpoint should
|
3337
|
-
be removed, and the delete endpoint should be used instead.
|
2147
|
+
Reschedule Bulk Jobs.
|
3338
2148
|
|
3339
|
-
:
|
3340
|
-
:
|
2149
|
+
:keyword job_ids: Required.
|
2150
|
+
:paramtype job_ids: list[int]
|
2151
|
+
:keyword reset_jobs: Default value is False.
|
2152
|
+
:paramtype reset_jobs: bool
|
3341
2153
|
:return: any
|
3342
2154
|
:rtype: any
|
3343
2155
|
:raises ~azure.core.exceptions.HttpResponseError:
|
@@ -3355,8 +2167,9 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3355
2167
|
|
3356
2168
|
cls: ClsType[Any] = kwargs.pop("cls", None)
|
3357
2169
|
|
3358
|
-
_request =
|
3359
|
-
|
2170
|
+
_request = build_jobs_reschedule_bulk_jobs_request(
|
2171
|
+
job_ids=job_ids,
|
2172
|
+
reset_jobs=reset_jobs,
|
3360
2173
|
headers=_headers,
|
3361
2174
|
params=_params,
|
3362
2175
|
)
|
@@ -3384,18 +2197,89 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3384
2197
|
|
3385
2198
|
return deserialized # type: ignore
|
3386
2199
|
|
2200
|
+
@overload
|
2201
|
+
def search(
|
2202
|
+
self,
|
2203
|
+
body: Optional[_models.JobSearchParams] = None,
|
2204
|
+
*,
|
2205
|
+
page: int = 1,
|
2206
|
+
per_page: int = 100,
|
2207
|
+
content_type: str = "application/json",
|
2208
|
+
**kwargs: Any,
|
2209
|
+
) -> List[JSON]:
|
2210
|
+
"""Search.
|
2211
|
+
|
2212
|
+
Retrieve information about jobs.
|
2213
|
+
|
2214
|
+
**TODO: Add more docs**.
|
2215
|
+
|
2216
|
+
:param body: Default value is None.
|
2217
|
+
:type body: ~generated.models.JobSearchParams
|
2218
|
+
:keyword page: Default value is 1.
|
2219
|
+
:paramtype page: int
|
2220
|
+
:keyword per_page: Default value is 100.
|
2221
|
+
:paramtype per_page: int
|
2222
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2223
|
+
Default value is "application/json".
|
2224
|
+
:paramtype content_type: str
|
2225
|
+
:return: list of JSON
|
2226
|
+
:rtype: list[JSON]
|
2227
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
2228
|
+
"""
|
2229
|
+
|
2230
|
+
@overload
|
2231
|
+
def search(
|
2232
|
+
self,
|
2233
|
+
body: Optional[IO[bytes]] = None,
|
2234
|
+
*,
|
2235
|
+
page: int = 1,
|
2236
|
+
per_page: int = 100,
|
2237
|
+
content_type: str = "application/json",
|
2238
|
+
**kwargs: Any,
|
2239
|
+
) -> List[JSON]:
|
2240
|
+
"""Search.
|
2241
|
+
|
2242
|
+
Retrieve information about jobs.
|
2243
|
+
|
2244
|
+
**TODO: Add more docs**.
|
2245
|
+
|
2246
|
+
:param body: Default value is None.
|
2247
|
+
:type body: IO[bytes]
|
2248
|
+
:keyword page: Default value is 1.
|
2249
|
+
:paramtype page: int
|
2250
|
+
:keyword per_page: Default value is 100.
|
2251
|
+
:paramtype per_page: int
|
2252
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2253
|
+
Default value is "application/json".
|
2254
|
+
:paramtype content_type: str
|
2255
|
+
:return: list of JSON
|
2256
|
+
:rtype: list[JSON]
|
2257
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
2258
|
+
"""
|
2259
|
+
|
3387
2260
|
@distributed_trace
|
3388
|
-
def
|
3389
|
-
self,
|
3390
|
-
|
3391
|
-
|
2261
|
+
def search(
|
2262
|
+
self,
|
2263
|
+
body: Optional[Union[_models.JobSearchParams, IO[bytes]]] = None,
|
2264
|
+
*,
|
2265
|
+
page: int = 1,
|
2266
|
+
per_page: int = 100,
|
2267
|
+
**kwargs: Any,
|
2268
|
+
) -> List[JSON]:
|
2269
|
+
"""Search.
|
2270
|
+
|
2271
|
+
Retrieve information about jobs.
|
3392
2272
|
|
3393
|
-
|
2273
|
+
**TODO: Add more docs**.
|
3394
2274
|
|
3395
|
-
:param
|
3396
|
-
:type
|
3397
|
-
:
|
3398
|
-
:
|
2275
|
+
:param body: Is either a JobSearchParams type or a IO[bytes] type. Default value is None.
|
2276
|
+
:type body: ~generated.models.JobSearchParams or IO[bytes]
|
2277
|
+
:keyword page: Default value is 1.
|
2278
|
+
:paramtype page: int
|
2279
|
+
:keyword per_page: Default value is 100.
|
2280
|
+
:paramtype per_page: int
|
2281
|
+
:return: list of JSON
|
2282
|
+
:rtype: list[JSON]
|
3399
2283
|
:raises ~azure.core.exceptions.HttpResponseError:
|
3400
2284
|
"""
|
3401
2285
|
error_map: MutableMapping = {
|
@@ -3406,15 +2290,31 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3406
2290
|
}
|
3407
2291
|
error_map.update(kwargs.pop("error_map", {}) or {})
|
3408
2292
|
|
3409
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2293
|
+
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
3410
2294
|
_params = kwargs.pop("params", {}) or {}
|
3411
2295
|
|
3412
|
-
|
3413
|
-
"
|
2296
|
+
content_type: Optional[str] = kwargs.pop(
|
2297
|
+
"content_type", _headers.pop("Content-Type", None)
|
3414
2298
|
)
|
2299
|
+
cls: ClsType[List[JSON]] = kwargs.pop("cls", None)
|
3415
2300
|
|
3416
|
-
|
3417
|
-
|
2301
|
+
content_type = content_type or "application/json"
|
2302
|
+
_json = None
|
2303
|
+
_content = None
|
2304
|
+
if isinstance(body, (IOBase, bytes)):
|
2305
|
+
_content = body
|
2306
|
+
else:
|
2307
|
+
if body is not None:
|
2308
|
+
_json = self._serialize.body(body, "JobSearchParams")
|
2309
|
+
else:
|
2310
|
+
_json = None
|
2311
|
+
|
2312
|
+
_request = build_jobs_search_request(
|
2313
|
+
page=page,
|
2314
|
+
per_page=per_page,
|
2315
|
+
content_type=content_type,
|
2316
|
+
json=_json,
|
2317
|
+
content=_content,
|
3418
2318
|
headers=_headers,
|
3419
2319
|
params=_params,
|
3420
2320
|
)
|
@@ -3429,98 +2329,77 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3429
2329
|
|
3430
2330
|
response = pipeline_response.http_response
|
3431
2331
|
|
3432
|
-
if response.status_code not in [200]:
|
2332
|
+
if response.status_code not in [200, 206]:
|
3433
2333
|
map_error(
|
3434
2334
|
status_code=response.status_code, response=response, error_map=error_map
|
3435
2335
|
)
|
3436
2336
|
raise HttpResponseError(response=response)
|
3437
2337
|
|
3438
|
-
|
3439
|
-
|
3440
|
-
|
2338
|
+
response_headers = {}
|
2339
|
+
if response.status_code == 206:
|
2340
|
+
response_headers["Content-Range"] = self._deserialize(
|
2341
|
+
"str", response.headers.get("Content-Range")
|
2342
|
+
)
|
2343
|
+
|
2344
|
+
deserialized = self._deserialize("[object]", pipeline_response.http_response)
|
3441
2345
|
|
3442
2346
|
if cls:
|
3443
|
-
return cls(pipeline_response, deserialized,
|
2347
|
+
return cls(pipeline_response, deserialized, response_headers) # type: ignore
|
3444
2348
|
|
3445
2349
|
return deserialized # type: ignore
|
3446
2350
|
|
3447
2351
|
@overload
|
3448
|
-
def
|
2352
|
+
def summary(
|
3449
2353
|
self,
|
3450
|
-
|
3451
|
-
body: Dict[str, _models.JobStatusUpdate],
|
2354
|
+
body: _models.JobSummaryParams,
|
3452
2355
|
*,
|
3453
|
-
force: bool = False,
|
3454
2356
|
content_type: str = "application/json",
|
3455
2357
|
**kwargs: Any,
|
3456
|
-
) ->
|
3457
|
-
"""
|
2358
|
+
) -> Any:
|
2359
|
+
"""Summary.
|
3458
2360
|
|
3459
|
-
|
2361
|
+
Show information suitable for plotting.
|
3460
2362
|
|
3461
|
-
:param job_id: Required.
|
3462
|
-
:type job_id: int
|
3463
2363
|
:param body: Required.
|
3464
|
-
:type body:
|
3465
|
-
:keyword force: Default value is False.
|
3466
|
-
:paramtype force: bool
|
2364
|
+
:type body: ~generated.models.JobSummaryParams
|
3467
2365
|
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
3468
2366
|
Default value is "application/json".
|
3469
2367
|
:paramtype content_type: str
|
3470
|
-
:return:
|
3471
|
-
:rtype:
|
2368
|
+
:return: any
|
2369
|
+
:rtype: any
|
3472
2370
|
:raises ~azure.core.exceptions.HttpResponseError:
|
3473
2371
|
"""
|
3474
2372
|
|
3475
2373
|
@overload
|
3476
|
-
def
|
3477
|
-
self,
|
3478
|
-
|
3479
|
-
|
3480
|
-
*,
|
3481
|
-
force: bool = False,
|
3482
|
-
content_type: str = "application/json",
|
3483
|
-
**kwargs: Any,
|
3484
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
3485
|
-
"""Set Single Job Status.
|
2374
|
+
def summary(
|
2375
|
+
self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
|
2376
|
+
) -> Any:
|
2377
|
+
"""Summary.
|
3486
2378
|
|
3487
|
-
|
2379
|
+
Show information suitable for plotting.
|
3488
2380
|
|
3489
|
-
:param job_id: Required.
|
3490
|
-
:type job_id: int
|
3491
2381
|
:param body: Required.
|
3492
2382
|
:type body: IO[bytes]
|
3493
|
-
:keyword force: Default value is False.
|
3494
|
-
:paramtype force: bool
|
3495
2383
|
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
3496
2384
|
Default value is "application/json".
|
3497
2385
|
:paramtype content_type: str
|
3498
|
-
:return:
|
3499
|
-
:rtype:
|
2386
|
+
:return: any
|
2387
|
+
:rtype: any
|
3500
2388
|
:raises ~azure.core.exceptions.HttpResponseError:
|
3501
2389
|
"""
|
3502
2390
|
|
3503
2391
|
@distributed_trace
|
3504
|
-
def
|
3505
|
-
self,
|
3506
|
-
|
3507
|
-
|
3508
|
-
*,
|
3509
|
-
force: bool = False,
|
3510
|
-
**kwargs: Any,
|
3511
|
-
) -> Dict[str, _models.SetJobStatusReturn]:
|
3512
|
-
"""Set Single Job Status.
|
2392
|
+
def summary(
|
2393
|
+
self, body: Union[_models.JobSummaryParams, IO[bytes]], **kwargs: Any
|
2394
|
+
) -> Any:
|
2395
|
+
"""Summary.
|
3513
2396
|
|
3514
|
-
|
2397
|
+
Show information suitable for plotting.
|
3515
2398
|
|
3516
|
-
:param
|
3517
|
-
:type
|
3518
|
-
:
|
3519
|
-
:
|
3520
|
-
:keyword force: Default value is False.
|
3521
|
-
:paramtype force: bool
|
3522
|
-
:return: dict mapping str to SetJobStatusReturn
|
3523
|
-
:rtype: dict[str, ~generated.models.SetJobStatusReturn]
|
2399
|
+
:param body: Is either a JobSummaryParams type or a IO[bytes] type. Required.
|
2400
|
+
:type body: ~generated.models.JobSummaryParams or IO[bytes]
|
2401
|
+
:return: any
|
2402
|
+
:rtype: any
|
3524
2403
|
:raises ~azure.core.exceptions.HttpResponseError:
|
3525
2404
|
"""
|
3526
2405
|
error_map: MutableMapping = {
|
@@ -3537,7 +2416,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3537
2416
|
content_type: Optional[str] = kwargs.pop(
|
3538
2417
|
"content_type", _headers.pop("Content-Type", None)
|
3539
2418
|
)
|
3540
|
-
cls: ClsType[
|
2419
|
+
cls: ClsType[Any] = kwargs.pop("cls", None)
|
3541
2420
|
|
3542
2421
|
content_type = content_type or "application/json"
|
3543
2422
|
_json = None
|
@@ -3545,11 +2424,9 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3545
2424
|
if isinstance(body, (IOBase, bytes)):
|
3546
2425
|
_content = body
|
3547
2426
|
else:
|
3548
|
-
_json = self._serialize.body(body, "
|
2427
|
+
_json = self._serialize.body(body, "JobSummaryParams")
|
3549
2428
|
|
3550
|
-
_request =
|
3551
|
-
job_id=job_id,
|
3552
|
-
force=force,
|
2429
|
+
_request = build_jobs_summary_request(
|
3553
2430
|
content_type=content_type,
|
3554
2431
|
json=_json,
|
3555
2432
|
content=_content,
|
@@ -3573,27 +2450,61 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3573
2450
|
)
|
3574
2451
|
raise HttpResponseError(response=response)
|
3575
2452
|
|
3576
|
-
deserialized = self._deserialize(
|
3577
|
-
"{SetJobStatusReturn}", pipeline_response.http_response
|
3578
|
-
)
|
2453
|
+
deserialized = self._deserialize("object", pipeline_response.http_response)
|
3579
2454
|
|
3580
2455
|
if cls:
|
3581
2456
|
return cls(pipeline_response, deserialized, {}) # type: ignore
|
3582
2457
|
|
3583
2458
|
return deserialized # type: ignore
|
3584
2459
|
|
2460
|
+
@overload
|
2461
|
+
def submit_bulk_jdl_jobs(
|
2462
|
+
self, body: List[str], *, content_type: str = "application/json", **kwargs: Any
|
2463
|
+
) -> List[_models.InsertedJob]:
|
2464
|
+
"""Submit Bulk Jdl Jobs.
|
2465
|
+
|
2466
|
+
Submit Bulk Jdl Jobs.
|
2467
|
+
|
2468
|
+
:param body: Required.
|
2469
|
+
:type body: list[str]
|
2470
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
|
2471
|
+
Default value is "application/json".
|
2472
|
+
:paramtype content_type: str
|
2473
|
+
:return: list of InsertedJob
|
2474
|
+
:rtype: list[~generated.models.InsertedJob]
|
2475
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
2476
|
+
"""
|
2477
|
+
|
2478
|
+
@overload
|
2479
|
+
def submit_bulk_jdl_jobs(
|
2480
|
+
self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
|
2481
|
+
) -> List[_models.InsertedJob]:
|
2482
|
+
"""Submit Bulk Jdl Jobs.
|
2483
|
+
|
2484
|
+
Submit Bulk Jdl Jobs.
|
2485
|
+
|
2486
|
+
:param body: Required.
|
2487
|
+
:type body: IO[bytes]
|
2488
|
+
:keyword content_type: Body Parameter content-type. Content type parameter for binary body.
|
2489
|
+
Default value is "application/json".
|
2490
|
+
:paramtype content_type: str
|
2491
|
+
:return: list of InsertedJob
|
2492
|
+
:rtype: list[~generated.models.InsertedJob]
|
2493
|
+
:raises ~azure.core.exceptions.HttpResponseError:
|
2494
|
+
"""
|
2495
|
+
|
3585
2496
|
@distributed_trace
|
3586
|
-
def
|
3587
|
-
self,
|
3588
|
-
) ->
|
3589
|
-
"""
|
2497
|
+
def submit_bulk_jdl_jobs(
|
2498
|
+
self, body: Union[List[str], IO[bytes]], **kwargs: Any
|
2499
|
+
) -> List[_models.InsertedJob]:
|
2500
|
+
"""Submit Bulk Jdl Jobs.
|
3590
2501
|
|
3591
|
-
|
2502
|
+
Submit Bulk Jdl Jobs.
|
3592
2503
|
|
3593
|
-
:param
|
3594
|
-
:type
|
3595
|
-
:return:
|
3596
|
-
:rtype:
|
2504
|
+
:param body: Is either a [str] type or a IO[bytes] type. Required.
|
2505
|
+
:type body: list[str] or IO[bytes]
|
2506
|
+
:return: list of InsertedJob
|
2507
|
+
:rtype: list[~generated.models.InsertedJob]
|
3597
2508
|
:raises ~azure.core.exceptions.HttpResponseError:
|
3598
2509
|
"""
|
3599
2510
|
error_map: MutableMapping = {
|
@@ -3604,13 +2515,26 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3604
2515
|
}
|
3605
2516
|
error_map.update(kwargs.pop("error_map", {}) or {})
|
3606
2517
|
|
3607
|
-
_headers = kwargs.pop("headers", {}) or {}
|
2518
|
+
_headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
|
3608
2519
|
_params = kwargs.pop("params", {}) or {}
|
3609
2520
|
|
3610
|
-
|
2521
|
+
content_type: Optional[str] = kwargs.pop(
|
2522
|
+
"content_type", _headers.pop("Content-Type", None)
|
2523
|
+
)
|
2524
|
+
cls: ClsType[List[_models.InsertedJob]] = kwargs.pop("cls", None)
|
2525
|
+
|
2526
|
+
content_type = content_type or "application/json"
|
2527
|
+
_json = None
|
2528
|
+
_content = None
|
2529
|
+
if isinstance(body, (IOBase, bytes)):
|
2530
|
+
_content = body
|
2531
|
+
else:
|
2532
|
+
_json = self._serialize.body(body, "[str]")
|
3611
2533
|
|
3612
|
-
_request =
|
3613
|
-
|
2534
|
+
_request = build_jobs_submit_bulk_jdl_jobs_request(
|
2535
|
+
content_type=content_type,
|
2536
|
+
json=_json,
|
2537
|
+
content=_content,
|
3614
2538
|
headers=_headers,
|
3615
2539
|
params=_params,
|
3616
2540
|
)
|
@@ -3632,7 +2556,7 @@ class JobsOperations: # pylint: disable=too-many-public-methods
|
|
3632
2556
|
raise HttpResponseError(response=response)
|
3633
2557
|
|
3634
2558
|
deserialized = self._deserialize(
|
3635
|
-
"
|
2559
|
+
"[InsertedJob]", pipeline_response.http_response
|
3636
2560
|
)
|
3637
2561
|
|
3638
2562
|
if cls:
|