fix: Lots of fixes to make it work

Author: Robin-Van-de-Merghel

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

@@ -59,9 +59,11 @@
     build_pilots_clear_pilots_request,
     build_pilots_create_pilot_secrets_request,
     build_pilots_delete_pilots_request,
+    build_pilots_get_logs_request,
     build_pilots_pilot_login_request,
     build_pilots_refresh_pilot_tokens_request,
     build_pilots_search_request,
+    build_pilots_send_message_request,
     build_pilots_update_pilot_fields_request,
     build_well_known_get_installation_metadata_request,
     build_well_known_get_jwks_request,
@@ -3470,3 +3472,166 @@ async def refresh_pilot_tokens(
             return cls(pipeline_response, deserialized, {})  # type: ignore
 
         return deserialized  # type: ignore
+
+    @overload
+    async def send_message(
+        self,
+        body: _models.LogMessage,
+        *,
+        content_type: str = "application/json",
+        **kwargs: Any,
+    ) -> int:
+        """Send Message.
+
+        Send Message.
+
+        :param body: Required.
+        :type body: ~_generated.models.LogMessage
+        :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
+         Default value is "application/json".
+        :paramtype content_type: str
+        :return: int
+        :rtype: int
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+
+    @overload
+    async def send_message(
+        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
+    ) -> int:
+        """Send Message.
+
+        Send Message.
+
+        :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: int
+        :rtype: int
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+
+    @distributed_trace_async
+    async def send_message(
+        self, body: Union[_models.LogMessage, IO[bytes]], **kwargs: Any
+    ) -> int:
+        """Send Message.
+
+        Send Message.
+
+        :param body: Is either a LogMessage type or a IO[bytes] type. Required.
+        :type body: ~_generated.models.LogMessage or IO[bytes]
+        :return: int
+        :rtype: int
+        :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[int] = 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, "LogMessage")
+
+        _request = build_pilots_send_message_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("int", pipeline_response.http_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
+
+    @distributed_trace_async
+    async def get_logs(self, *, pilot_id: int, **kwargs: Any) -> List[Dict[str, Any]]:
+        """Get Logs.
+
+        Get Logs.
+
+        :keyword pilot_id: Required.
+        :paramtype pilot_id: 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 = kwargs.pop("headers", {}) or {}
+        _params = kwargs.pop("params", {}) or {}
+
+        cls: ClsType[List[Dict[str, Any]]] = kwargs.pop("cls", None)
+
+        _request = build_pilots_get_logs_request(
+            pilot_id=pilot_id,
+            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

@@ -30,6 +30,8 @@
     JobStatusUpdate,
     JobSummaryParams,
     JobSummaryParamsSearchItem,
+    LogLine,
+    LogMessage,
     Metadata,
     OpenIDConfiguration,
     PilotCredentialsInfo,
@@ -90,6 +92,8 @@
     "JobStatusUpdate",
     "JobSummaryParams",
     "JobSummaryParamsSearchItem",
+    "LogLine",
+    "LogMessage",
     "Metadata",
     "OpenIDConfiguration",
     "PilotCredentialsInfo",

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

@@ -762,6 +762,86 @@ class JobSummaryParamsSearchItem(_serialization.Model):
     """JobSummaryParamsSearchItem."""
 
 
+class LogLine(_serialization.Model):
+    """LogLine.
+
+    All required parameters must be populated in order to send to server.
+
+    :ivar line_no: Line No. Required.
+    :vartype line_no: int
+    :ivar line: Line. Required.
+    :vartype line: str
+    """
+
+    _validation = {
+        "line_no": {"required": True},
+        "line": {"required": True},
+    }
+
+    _attribute_map = {
+        "line_no": {"key": "line_no", "type": "int"},
+        "line": {"key": "line", "type": "str"},
+    }
+
+    def __init__(self, *, line_no: int, line: str, **kwargs: Any) -> None:
+        """
+        :keyword line_no: Line No. Required.
+        :paramtype line_no: int
+        :keyword line: Line. Required.
+        :paramtype line: str
+        """
+        super().__init__(**kwargs)
+        self.line_no = line_no
+        self.line = line
+
+
+class LogMessage(_serialization.Model):
+    """LogMessage.
+
+    All required parameters must be populated in order to send to server.
+
+    :ivar pilot_stamp: Pilot Stamp. Required.
+    :vartype pilot_stamp: str
+    :ivar lines: Lines. Required.
+    :vartype lines: list[~_generated.models.LogLine]
+    :ivar vo: Vo. Required.
+    :vartype vo: str
+    """
+
+    _validation = {
+        "pilot_stamp": {"required": True},
+        "lines": {"required": True},
+        "vo": {"required": True},
+    }
+
+    _attribute_map = {
+        "pilot_stamp": {"key": "pilot_stamp", "type": "str"},
+        "lines": {"key": "lines", "type": "[LogLine]"},
+        "vo": {"key": "vo", "type": "str"},
+    }
+
+    def __init__(
+        self,
+        *,
+        pilot_stamp: str,
+        lines: List["_models.LogLine"],
+        vo: str,
+        **kwargs: Any,
+    ) -> None:
+        """
+        :keyword pilot_stamp: Pilot Stamp. Required.
+        :paramtype pilot_stamp: str
+        :keyword lines: Lines. Required.
+        :paramtype lines: list[~_generated.models.LogLine]
+        :keyword vo: Vo. Required.
+        :paramtype vo: str
+        """
+        super().__init__(**kwargs)
+        self.pilot_stamp = pilot_stamp
+        self.lines = lines
+        self.vo = vo
+
+
 class Metadata(_serialization.Model):
     """Metadata.