feat (RMS): getBulkRequestStatus

src/DIRAC/RequestManagementSystem/Client/ReqClient.py

@@ -5,6 +5,7 @@
  :synopsis: implementation of client for RequestDB using DISET framework
 
 """
+
 import os
 import time
 import random
@@ -470,6 +471,23 @@ def resetFailedRequest(self, requestID, allR=False):
  return self.putRequest(req)
  return S_OK("Not reset")
 
+ @ignoreEncodeWarning
+ def getBulkRequestStatus(self, requestIDs: list[int]):
+ """get the Status for the supplied request IDs.
+
+ :param self: self reference
+ :param list requestIDs: list of job IDs (integers)
+ :return: S_ERROR or S_OK( { reqID1:status, requID2:status2, ... })
+ """
+ res = self._getRPC().getBulkRequestStatus(requestIDs)
+ if not res["OK"]:
+ return res
+
+ # Cast the requestIDs back to int
+ statuses = strToIntDict(res["Value"])
+
+ return S_OK(statuses)
+
 
 # ============= Some useful functions to be shared ===========
 

src/DIRAC/RequestManagementSystem/DB/RequestDB.py

@@ -869,6 +869,16 @@ def getRequestStatus(self, requestID):
  session.close()
  return S_OK(status[0])
 
+ def getBulkRequestStatus(self, requestIDs):
+ """get requests statuses for given request IDs"""
+ session = self.DBSession()
+ try:
+ statuses = session.query(Request.RequestID, Request._Status).filter(Request.RequestID.in_(requestIDs)).all()
+ status_dict = {req_id: req_status for req_id, req_status in statuses}
+ finally:
+ session.close()
+ return S_OK(status_dict)
+
  def getRequestFileStatus(self, requestID, lfnList):
  """get status for files in request given its id
 

src/DIRAC/RequestManagementSystem/Service/ReqManagerHandler.py

@@ -340,6 +340,16 @@ def export_getRequestStatus(cls, requestID):
  gLogger.error(f"getRequestStatus: {status['Message']}")
  return status
 
+ types_getBulkRequestStatus = [list]
+
+ @classmethod
+ def export_getBulkRequestStatus(cls, requestIDs):
+ """get requests statuses given their ids"""
+ res = cls.__requestDB.getBulkRequestStatus(requestIDs)
+ if not res["OK"]:
+ gLogger.error(f"getRequestStatus: {res['Message']}")
+ return res
+
  types_getRequestFileStatus = [int, [str, list]]
 
  @classmethod