refactoring: Lots of refactoring of the code

Author: Robin-Van-de-Merghel

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

@@ -61,7 +61,7 @@
     build_pilots_clear_pilots_request,
     build_pilots_create_pilot_secrets_request,
     build_pilots_delete_pilots_request,
-    build_pilots_get_pilot_info_request,
+    build_pilots_search_request,
     build_pilots_update_pilot_fields_request,
     build_well_known_get_installation_metadata_request,
     build_well_known_get_jwks_request,
@@ -2883,151 +2883,6 @@ async def clear_pilots(
         if cls:
             return cls(pipeline_response, None, {})  # type: ignore
 
-    @overload
-    async def get_pilot_info(
-        self,
-        body: Optional[_models.SearchParams] = None,
-        *,
-        page: int = 1,
-        per_page: int = 100,
-        content_type: str = "application/json",
-        **kwargs: Any,
-    ) -> List[Dict[str, Any]]:
-        """Get Pilot Info.
-
-        Retrieve information about pilots.
-
-        :param body: Default value is None.
-        :type body: ~_generated.models.SearchParams
-        :keyword page: Default value is 1.
-        :paramtype page: int
-        :keyword per_page: Default value is 100.
-        :paramtype per_page: int
-        :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
-         Default value is "application/json".
-        :paramtype content_type: str
-        :return: list of dict mapping str to any
-        :rtype: list[dict[str, any]]
-        :raises ~azure.core.exceptions.HttpResponseError:
-        """
-
-    @overload
-    async def get_pilot_info(
-        self,
-        body: Optional[IO[bytes]] = None,
-        *,
-        page: int = 1,
-        per_page: int = 100,
-        content_type: str = "application/json",
-        **kwargs: Any,
-    ) -> List[Dict[str, Any]]:
-        """Get Pilot Info.
-
-        Retrieve information about pilots.
-
-        :param body: Default value is None.
-        :type body: IO[bytes]
-        :keyword page: Default value is 1.
-        :paramtype page: int
-        :keyword per_page: Default value is 100.
-        :paramtype per_page: int
-        :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
-         Default value is "application/json".
-        :paramtype content_type: str
-        :return: list of dict mapping str to any
-        :rtype: list[dict[str, any]]
-        :raises ~azure.core.exceptions.HttpResponseError:
-        """
-
-    @distributed_trace_async
-    async def get_pilot_info(
-        self,
-        body: Optional[Union[_models.SearchParams, IO[bytes]]] = None,
-        *,
-        page: int = 1,
-        per_page: int = 100,
-        **kwargs: Any,
-    ) -> List[Dict[str, Any]]:
-        """Get Pilot Info.
-
-        Retrieve information about pilots.
-
-        :param body: Is either a SearchParams type or a IO[bytes] type. Default value is None.
-        :type body: ~_generated.models.SearchParams or IO[bytes]
-        :keyword page: Default value is 1.
-        :paramtype page: int
-        :keyword per_page: Default value is 100.
-        :paramtype per_page: int
-        :return: list of dict mapping str to any
-        :rtype: list[dict[str, 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[List[Dict[str, Any]]] = kwargs.pop("cls", None)
-
-        content_type = content_type or "application/json"
-        _json = None
-        _content = None
-        if isinstance(body, (IOBase, bytes)):
-            _content = body
-        else:
-            if body is not None:
-                _json = self._serialize.body(body, "SearchParams")
-            else:
-                _json = None
-
-        _request = build_pilots_get_pilot_info_request(
-            page=page,
-            per_page=per_page,
-            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, 206]:
-            map_error(
-                status_code=response.status_code, response=response, error_map=error_map
-            )
-            raise HttpResponseError(response=response)
-
-        response_headers = {}
-        if response.status_code == 206:
-            response_headers["Content-Range"] = self._deserialize(
-                "str", response.headers.get("Content-Range")
-            )
-
-        deserialized = self._deserialize("[{object}]", pipeline_response.http_response)
-
-        if cls:
-            return cls(pipeline_response, deserialized, response_headers)  # type: ignore
-
-        return deserialized  # type: ignore
-
     @overload
     async def create_pilot_secrets(
         self,
@@ -3460,3 +3315,148 @@ async def associate_pilot_with_jobs(
 
         if cls:
             return cls(pipeline_response, None, {})  # type: ignore
+
+    @overload
+    async def search(
+        self,
+        body: Optional[_models.SearchParams] = None,
+        *,
+        page: int = 1,
+        per_page: int = 100,
+        content_type: str = "application/json",
+        **kwargs: Any,
+    ) -> List[Dict[str, Any]]:
+        """Search.
+
+        Retrieve information about pilots.
+
+        :param body: Default value is None.
+        :type body: ~_generated.models.SearchParams
+        :keyword page: Default value is 1.
+        :paramtype page: int
+        :keyword per_page: Default value is 100.
+        :paramtype per_page: int
+        :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
+         Default value is "application/json".
+        :paramtype content_type: str
+        :return: list of dict mapping str to any
+        :rtype: list[dict[str, any]]
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+
+    @overload
+    async def search(
+        self,
+        body: Optional[IO[bytes]] = None,
+        *,
+        page: int = 1,
+        per_page: int = 100,
+        content_type: str = "application/json",
+        **kwargs: Any,
+    ) -> List[Dict[str, Any]]:
+        """Search.
+
+        Retrieve information about pilots.
+
+        :param body: Default value is None.
+        :type body: IO[bytes]
+        :keyword page: Default value is 1.
+        :paramtype page: int
+        :keyword per_page: Default value is 100.
+        :paramtype per_page: int
+        :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
+         Default value is "application/json".
+        :paramtype content_type: str
+        :return: list of dict mapping str to any
+        :rtype: list[dict[str, any]]
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+
+    @distributed_trace_async
+    async def search(
+        self,
+        body: Optional[Union[_models.SearchParams, IO[bytes]]] = None,
+        *,
+        page: int = 1,
+        per_page: int = 100,
+        **kwargs: Any,
+    ) -> List[Dict[str, Any]]:
+        """Search.
+
+        Retrieve information about pilots.
+
+        :param body: Is either a SearchParams type or a IO[bytes] type. Default value is None.
+        :type body: ~_generated.models.SearchParams or IO[bytes]
+        :keyword page: Default value is 1.
+        :paramtype page: int
+        :keyword per_page: Default value is 100.
+        :paramtype per_page: int
+        :return: list of dict mapping str to any
+        :rtype: list[dict[str, 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[List[Dict[str, Any]]] = kwargs.pop("cls", None)
+
+        content_type = content_type or "application/json"
+        _json = None
+        _content = None
+        if isinstance(body, (IOBase, bytes)):
+            _content = body
+        else:
+            if body is not None:
+                _json = self._serialize.body(body, "SearchParams")
+            else:
+                _json = None
+
+        _request = build_pilots_search_request(
+            page=page,
+            per_page=per_page,
+            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, 206]:
+            map_error(
+                status_code=response.status_code, response=response, error_map=error_map
+            )
+            raise HttpResponseError(response=response)
+
+        response_headers = {}
+        if response.status_code == 206:
+            response_headers["Content-Range"] = self._deserialize(
+                "str", response.headers.get("Content-Range")
+            )
+
+        deserialized = self._deserialize("[{object}]", pipeline_response.http_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, response_headers)  # type: ignore
+
+        return deserialized  # type: ignore