feat: Add more functions from JobMonitoring to DiracX

Author: Robin VAN DE MERGHEL

src/DIRAC/WorkloadManagementSystem/FutureClient/JobMonitoringClient.py

@@ -32,6 +32,16 @@ def getJobsSites(self, jobIDs):
     def getJobsStatus(self, jobIDs):
         return self.fetch(["Status"], jobIDs)
 
+    @convertToReturnValue
+    def getJobParameter(self, jobID, parName):
+        # with DiracXClient() as api:
+        #     parameters = api.jobs.get_job_parameters(
+        #         job_id=jobID
+        #     )
+
+        #     return parameters
+        pass
+
     @convertToReturnValue
     def getJobParameters(self, jobIDs):
         # if isinstance(jobIDs, int):
@@ -83,3 +93,73 @@ def cleanLFN(data):
                 return_dict[job_id].append(cleanLFN(data))
 
             return return_dict
+
+    @convertToReturnValue
+    def getJobs(self, attrDict, cutDate=None):
+        # TODO: Check, selectJobs is awful...
+        # See if we do it
+        pass
+
+    @convertToReturnValue
+    def getJobJDL(self, jobID, original):
+        with DiracXClient() as api:
+            jdl = api.jobs.get_job_jdl(job_id=jobID)
+
+            if not jdl:
+                return ""
+
+            if original:
+                return jdl["OriginalJDL"]
+
+            return jdl["JDL"]
+
+    @convertToReturnValue
+    def getJobLoggingInfo(self, jobID):
+        fields = ["DateTime", "Source", "Status", "MinorStatus", "ApplicationStatus"]  # CF Dirac.py
+
+        jobs = self.fetch(parameters=["LoggingInfo"] + fields, jobIDs=[jobID])
+
+        # Normally, only one job
+        if len(jobs) == 0:
+            return []
+        job = jobs[0]
+
+        # Rearrange into a list,  CF Dirac.py
+        res = []
+        for field in fields:
+            res.append(job.get(field, ""))
+
+        return res
+
+    @convertToReturnValue
+    def getJobSummary(self, jobID):
+        return self.fetch(parameters=[], jobIDs=[jobID])
+
+    @convertToReturnValue
+    def getJobsSummary(self, jobIDs):
+        return self.fetch(parameters=[], jobIDs=jobIDs)
+
+    @convertToReturnValue
+    def getJobAttributes(self, jobID, attrList=None):
+        if not attrList:
+            attrList = []
+
+        return self.fetch(parameters=attrList, jobIDs=[jobID])
+
+    @convertToReturnValue
+    def getJobAttribute(self, jobID, attribute):
+        return self.fetch(parameters=[attribute], jobIDs=[jobID])
+
+    @convertToReturnValue
+    def getJobHeartBeatData(self, jobID):
+        with DiracXClient() as api:
+            res = api.jobs.get_job_heartbeat_info(job_id=jobID)
+
+            result = []
+            for row in res:
+                name = row["Name"]
+                value = str(row["Value"])
+                heartbeattime = row["HeartBeatTime"]
+                # ('name', '"0.1"', 'time')
+                # :)
+                result.append((str(name), "%.01f" % (float(value.replace('"', ""))), str(heartbeattime)))