feat: Add pilot summary, and refactor jobs one

Author: Robin VAN DE MERGHEL

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

@@ -58,6 +58,7 @@
     build_pilots_delete_pilots_request,
     build_pilots_get_pilot_jobs_request,
     build_pilots_search_request,
+    build_pilots_summary_request,
     build_pilots_update_pilot_fields_request,
     build_well_known_get_installation_metadata_request,
     build_well_known_get_jwks_request,
@@ -1982,14 +1983,14 @@ async def search(
 
     @overload
     async def summary(
-        self, body: _models.JobSummaryParams, *, content_type: str = "application/json", **kwargs: Any
+        self, body: _models.SummaryParams, *, content_type: str = "application/json", **kwargs: Any
     ) -> Any:
         """Summary.
 
         Show information suitable for plotting.
 
         :param body: Required.
-        :type body: ~_generated.models.JobSummaryParams
+        :type body: ~_generated.models.SummaryParams
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -2015,13 +2016,13 @@ async def summary(self, body: IO[bytes], *, content_type: str = "application/jso
         """
 
     @distributed_trace_async
-    async def summary(self, body: Union[_models.JobSummaryParams, IO[bytes]], **kwargs: Any) -> Any:
+    async def summary(self, body: Union[_models.SummaryParams, IO[bytes]], **kwargs: Any) -> Any:
         """Summary.
 
         Show information suitable for plotting.
 
-        :param body: Is either a JobSummaryParams type or a IO[bytes] type. Required.
-        :type body: ~_generated.models.JobSummaryParams or IO[bytes]
+        :param body: Is either a SummaryParams type or a IO[bytes] type. Required.
+        :type body: ~_generated.models.SummaryParams or IO[bytes]
         :return: any
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -2046,7 +2047,7 @@ async def summary(self, body: Union[_models.JobSummaryParams, IO[bytes]], **kwar
         if isinstance(body, (IOBase, bytes)):
             _content = body
         else:
-            _json = self._serialize.body(body, "JobSummaryParams")
+            _json = self._serialize.body(body, "SummaryParams")
 
         _request = build_jobs_summary_request(
             content_type=content_type,
@@ -2749,3 +2750,98 @@ async def search(
             return cls(pipeline_response, deserialized, response_headers)  # type: ignore
 
         return deserialized  # type: ignore
+
+    @overload
+    async def summary(
+        self, body: _models.SummaryParams, *, content_type: str = "application/json", **kwargs: Any
+    ) -> Any:
+        """Summary.
+
+        Show information suitable for plotting.
+
+        :param body: Required.
+        :type body: ~_generated.models.SummaryParams
+        :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
+         Default value is "application/json".
+        :paramtype content_type: str
+        :return: any
+        :rtype: any
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+
+    @overload
+    async def summary(self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any) -> Any:
+        """Summary.
+
+        Show information suitable for plotting.
+
+        :param body: Required.
+        :type body: IO[bytes]
+        :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
+         Default value is "application/json".
+        :paramtype content_type: str
+        :return: any
+        :rtype: any
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+
+    @distributed_trace_async
+    async def summary(self, body: Union[_models.SummaryParams, IO[bytes]], **kwargs: Any) -> Any:
+        """Summary.
+
+        Show information suitable for plotting.
+
+        :param body: Is either a SummaryParams type or a IO[bytes] type. Required.
+        :type body: ~_generated.models.SummaryParams or IO[bytes]
+        :return: any
+        :rtype: any
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map: MutableMapping = {
+            401: ClientAuthenticationError,
+            404: ResourceNotFoundError,
+            409: ResourceExistsError,
+            304: ResourceNotModifiedError,
+        }
+        error_map.update(kwargs.pop("error_map", {}) or {})
+
+        _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+        _params = kwargs.pop("params", {}) or {}
+
+        content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
+        cls: ClsType[Any] = kwargs.pop("cls", None)
+
+        content_type = content_type or "application/json"
+        _json = None
+        _content = None
+        if isinstance(body, (IOBase, bytes)):
+            _content = body
+        else:
+            _json = self._serialize.body(body, "SummaryParams")
+
+        _request = build_pilots_summary_request(
+            content_type=content_type,
+            json=_json,
+            content=_content,
+            headers=_headers,
+            params=_params,
+        )
+        _request.url = self._client.format_url(_request.url)
+
+        _stream = False
+        pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
+            _request, stream=_stream, **kwargs
+        )
+
+        response = pipeline_response.http_response
+
+        if response.status_code not in [200]:
+            map_error(status_code=response.status_code, response=response, error_map=error_map)
+            raise HttpResponseError(response=response)
+
+        deserialized = self._deserialize("object", pipeline_response.http_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore

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

@@ -24,8 +24,6 @@
     InsertedJob,
     JobCommand,
     JobStatusUpdate,
-    JobSummaryParams,
-    JobSummaryParamsSearchItem,
     Metadata,
     OpenIDConfiguration,
     PilotFieldsMapping,
@@ -39,6 +37,8 @@
     SetJobStatusReturn,
     SetJobStatusReturnSuccess,
     SortSpec,
+    SummaryParams,
+    SummaryParamsSearchItem,
     SupportInfo,
     TokenResponse,
     UserInfoResponse,
@@ -76,8 +76,6 @@
     "InsertedJob",
     "JobCommand",
     "JobStatusUpdate",
-    "JobSummaryParams",
-    "JobSummaryParamsSearchItem",
     "Metadata",
     "OpenIDConfiguration",
     "PilotFieldsMapping",
@@ -91,6 +89,8 @@
     "SetJobStatusReturn",
     "SetJobStatusReturnSuccess",
     "SortSpec",
+    "SummaryParams",
+    "SummaryParamsSearchItem",
     "SupportInfo",
     "TokenResponse",
     "UserInfoResponse",

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

@@ -536,44 +536,6 @@ def __init__(
         self.source = source
 
 
-class JobSummaryParams(_serialization.Model):
-    """JobSummaryParams.
-
-    All required parameters must be populated in order to send to server.
-
-    :ivar grouping: Grouping. Required.
-    :vartype grouping: list[str]
-    :ivar search: Search.
-    :vartype search: list[~_generated.models.JobSummaryParamsSearchItem]
-    """
-
-    _validation = {
-        "grouping": {"required": True},
-    }
-
-    _attribute_map = {
-        "grouping": {"key": "grouping", "type": "[str]"},
-        "search": {"key": "search", "type": "[JobSummaryParamsSearchItem]"},
-    }
-
-    def __init__(
-        self, *, grouping: List[str], search: List["_models.JobSummaryParamsSearchItem"] = [], **kwargs: Any
-    ) -> None:
-        """
-        :keyword grouping: Grouping. Required.
-        :paramtype grouping: list[str]
-        :keyword search: Search.
-        :paramtype search: list[~_generated.models.JobSummaryParamsSearchItem]
-        """
-        super().__init__(**kwargs)
-        self.grouping = grouping
-        self.search = search
-
-
-class JobSummaryParamsSearchItem(_serialization.Model):
-    """JobSummaryParamsSearchItem."""
-
-
 class Metadata(_serialization.Model):
     """Metadata.
 
@@ -1203,6 +1165,44 @@ def __init__(self, *, parameter: str, direction: Union[str, "_models.SortDirecti
         self.direction = direction
 
 
+class SummaryParams(_serialization.Model):
+    """SummaryParams.
+
+    All required parameters must be populated in order to send to server.
+
+    :ivar grouping: Grouping. Required.
+    :vartype grouping: list[str]
+    :ivar search: Search.
+    :vartype search: list[~_generated.models.SummaryParamsSearchItem]
+    """
+
+    _validation = {
+        "grouping": {"required": True},
+    }
+
+    _attribute_map = {
+        "grouping": {"key": "grouping", "type": "[str]"},
+        "search": {"key": "search", "type": "[SummaryParamsSearchItem]"},
+    }
+
+    def __init__(
+        self, *, grouping: List[str], search: List["_models.SummaryParamsSearchItem"] = [], **kwargs: Any
+    ) -> None:
+        """
+        :keyword grouping: Grouping. Required.
+        :paramtype grouping: list[str]
+        :keyword search: Search.
+        :paramtype search: list[~_generated.models.SummaryParamsSearchItem]
+        """
+        super().__init__(**kwargs)
+        self.grouping = grouping
+        self.search = search
+
+
+class SummaryParamsSearchItem(_serialization.Model):
+    """SummaryParamsSearchItem."""
+
+
 class SupportInfo(_serialization.Model):
     """SupportInfo.