feat: Pilots and Users are splitted for jobs

Author: Robin-Van-de-Merghel

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

@@ -38,7 +38,6 @@
     build_auth_revoke_refresh_token_by_refresh_token_request,
     build_auth_userinfo_request,
     build_config_serve_config_request,
-    build_jobs_add_heartbeat_request,
     build_jobs_assign_sandbox_to_job_request,
     build_jobs_get_job_sandbox_request,
     build_jobs_get_job_sandboxes_request,
@@ -1580,123 +1579,6 @@ async def set_job_statuses(
 
         return deserialized  # type: ignore
 
-    @overload
-    async def add_heartbeat(
-        self, body: Dict[str, _models.HeartbeatData], *, content_type: str = "application/json", **kwargs: Any
-    ) -> List[_models.JobCommand]:
-        """Add Heartbeat.
-
-        Register a heartbeat from the job.
-
-        This endpoint is used by the JobAgent to send heartbeats to the WMS and to
-        receive job commands from the WMS. It also results in stalled jobs being
-        restored to the RUNNING status.
-
-        The ``data`` parameter and return value are mappings keyed by job ID.
-
-        :param body: Required.
-        :type body: dict[str, ~_generated.models.HeartbeatData]
-        :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 JobCommand
-        :rtype: list[~_generated.models.JobCommand]
-        :raises ~azure.core.exceptions.HttpResponseError:
-        """
-
-    @overload
-    async def add_heartbeat(
-        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
-    ) -> List[_models.JobCommand]:
-        """Add Heartbeat.
-
-        Register a heartbeat from the job.
-
-        This endpoint is used by the JobAgent to send heartbeats to the WMS and to
-        receive job commands from the WMS. It also results in stalled jobs being
-        restored to the RUNNING status.
-
-        The ``data`` parameter and return value are mappings keyed by job ID.
-
-        :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: list of JobCommand
-        :rtype: list[~_generated.models.JobCommand]
-        :raises ~azure.core.exceptions.HttpResponseError:
-        """
-
-    @distributed_trace_async
-    async def add_heartbeat(
-        self, body: Union[Dict[str, _models.HeartbeatData], IO[bytes]], **kwargs: Any
-    ) -> List[_models.JobCommand]:
-        """Add Heartbeat.
-
-        Register a heartbeat from the job.
-
-        This endpoint is used by the JobAgent to send heartbeats to the WMS and to
-        receive job commands from the WMS. It also results in stalled jobs being
-        restored to the RUNNING status.
-
-        The ``data`` parameter and return value are mappings keyed by job ID.
-
-        :param body: Is either a {str: HeartbeatData} type or a IO[bytes] type. Required.
-        :type body: dict[str, ~_generated.models.HeartbeatData] or IO[bytes]
-        :return: list of JobCommand
-        :rtype: list[~_generated.models.JobCommand]
-        :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[_models.JobCommand]] = 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, "{HeartbeatData}")
-
-        _request = build_jobs_add_heartbeat_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("[JobCommand]", pipeline_response.http_response)
-
-        if cls:
-            return cls(pipeline_response, deserialized, {})  # type: ignore
-
-        return deserialized  # type: ignore
-
     @distributed_trace_async
     async def reschedule_jobs(self, *, job_ids: List[int], reset_jobs: bool = False, **kwargs: Any) -> Dict[str, Any]:
         """Reschedule Jobs.

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

@@ -22,10 +22,8 @@
     BodyPilotsRefreshPilotTokens,
     GroupInfo,
     HTTPValidationError,
-    HeartbeatData,
     InitiateDeviceFlowResponse,
     InsertedJob,
-    JobCommand,
     JobStatusUpdate,
     JobSummaryParams,
     JobSummaryParamsSearchItem,
@@ -82,10 +80,8 @@
     "BodyPilotsRefreshPilotTokens",
     "GroupInfo",
     "HTTPValidationError",
-    "HeartbeatData",
     "InitiateDeviceFlowResponse",
     "InsertedJob",
-    "JobCommand",
     "JobStatusUpdate",
     "JobSummaryParams",
     "JobSummaryParamsSearchItem",

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

@@ -355,73 +355,6 @@ def __init__(self, *, properties: List[str], **kwargs: Any) -> None:
         self.properties = properties
 
 
-class HeartbeatData(_serialization.Model):
-    """HeartbeatData.
-
-    :ivar load_average: Loadaverage.
-    :vartype load_average: float
-    :ivar memory_used: Memoryused.
-    :vartype memory_used: float
-    :ivar vsize: Vsize.
-    :vartype vsize: float
-    :ivar available_disk_space: Availablediskspace.
-    :vartype available_disk_space: float
-    :ivar cpu_consumed: Cpuconsumed.
-    :vartype cpu_consumed: float
-    :ivar wall_clock_time: Wallclocktime.
-    :vartype wall_clock_time: float
-    :ivar standard_output: Standardoutput.
-    :vartype standard_output: str
-    """
-
-    _attribute_map = {
-        "load_average": {"key": "LoadAverage", "type": "float"},
-        "memory_used": {"key": "MemoryUsed", "type": "float"},
-        "vsize": {"key": "Vsize", "type": "float"},
-        "available_disk_space": {"key": "AvailableDiskSpace", "type": "float"},
-        "cpu_consumed": {"key": "CPUConsumed", "type": "float"},
-        "wall_clock_time": {"key": "WallClockTime", "type": "float"},
-        "standard_output": {"key": "StandardOutput", "type": "str"},
-    }
-
-    def __init__(
-        self,
-        *,
-        load_average: Optional[float] = None,
-        memory_used: Optional[float] = None,
-        vsize: Optional[float] = None,
-        available_disk_space: Optional[float] = None,
-        cpu_consumed: Optional[float] = None,
-        wall_clock_time: Optional[float] = None,
-        standard_output: Optional[str] = None,
-        **kwargs: Any
-    ) -> None:
-        """
-        :keyword load_average: Loadaverage.
-        :paramtype load_average: float
-        :keyword memory_used: Memoryused.
-        :paramtype memory_used: float
-        :keyword vsize: Vsize.
-        :paramtype vsize: float
-        :keyword available_disk_space: Availablediskspace.
-        :paramtype available_disk_space: float
-        :keyword cpu_consumed: Cpuconsumed.
-        :paramtype cpu_consumed: float
-        :keyword wall_clock_time: Wallclocktime.
-        :paramtype wall_clock_time: float
-        :keyword standard_output: Standardoutput.
-        :paramtype standard_output: str
-        """
-        super().__init__(**kwargs)
-        self.load_average = load_average
-        self.memory_used = memory_used
-        self.vsize = vsize
-        self.available_disk_space = available_disk_space
-        self.cpu_consumed = cpu_consumed
-        self.wall_clock_time = wall_clock_time
-        self.standard_output = standard_output
-
-
 class HTTPValidationError(_serialization.Model):
     """HTTPValidationError.
 
@@ -554,45 +487,6 @@ def __init__(
         self.time_stamp = time_stamp
 
 
-class JobCommand(_serialization.Model):
-    """JobCommand.
-
-    All required parameters must be populated in order to send to server.
-
-    :ivar job_id: Job Id. Required.
-    :vartype job_id: int
-    :ivar command: Command. Required.
-    :vartype command: str
-    :ivar arguments: Arguments.
-    :vartype arguments: str
-    """
-
-    _validation = {
-        "job_id": {"required": True},
-        "command": {"required": True},
-    }
-
-    _attribute_map = {
-        "job_id": {"key": "job_id", "type": "int"},
-        "command": {"key": "command", "type": "str"},
-        "arguments": {"key": "arguments", "type": "str"},
-    }
-
-    def __init__(self, *, job_id: int, command: str, arguments: Optional[str] = None, **kwargs: Any) -> None:
-        """
-        :keyword job_id: Job Id. Required.
-        :paramtype job_id: int
-        :keyword command: Command. Required.
-        :paramtype command: str
-        :keyword arguments: Arguments.
-        :paramtype arguments: str
-        """
-        super().__init__(**kwargs)
-        self.job_id = job_id
-        self.command = command
-        self.arguments = arguments
-
-
 class JobStatusUpdate(_serialization.Model):
     """JobStatusUpdate.