feat: added PilotManager legacy adaptor

Author: fstagni

src/DIRAC/WorkloadManagementSystem/Client/PilotManagerClient.py

@@ -1,13 +1,17 @@
-""" Module that contains client access to the Pilots handler.
-"""
+"""Module that contains client access to the Pilots handler."""
 
 from DIRAC.Core.Base.Client import Client, createClient
+from DIRAC.WorkloadManagementSystem.FutureClient.PilotManagerClient import (
+    PilotManagerClient as futurePilotManagerClient,
+)
 
 
 @createClient("WorkloadManagement/PilotManager")
 class PilotManagerClient(Client):
     """PilotManagerClient sets url for the PilotManagerHandler."""
 
+    diracxClient = futurePilotManagerClient
+
     def __init__(self, url=None, **kwargs):
         """
         Sets URL for PilotManager handler

src/DIRAC/WorkloadManagementSystem/FutureClient/PilotManagerClient.py

@@ -0,0 +1,78 @@
+from DIRAC.Core.Security.DiracX import DiracXClient, FutureClient
+from DIRAC.Core.Utilities.ReturnValues import convertToReturnValue
+
+
+class PilotManagerClient(FutureClient):
+    @convertToReturnValue
+    def addPilotReferences(
+        self,
+        pilot_reference: str,
+        VO: str,
+        grid_type: str = "DIRAC",
+        pilot_stamp_dict: dict = {},
+        grid_site: str = "Unknown",
+        destination_site: str = "NotAssigned",
+    ):
+        """Add a new pilot to the database.
+        Uses pilots/management/register_pilot
+        """
+
+        with DiracXClient() as api:
+            api.pilots.register_pilot(
+                pilot_stamp=pilot_stamp_dict[pilot_reference],
+                vo=VO,
+                grid_type=grid_type,
+                grid_site=grid_site,
+                destination_site=destination_site,
+            )
+
+    @convertToReturnValue
+    def getPilotInfo(self, pilot_reference: str):
+        """Get the info about a given pilot job reference"""
+
+    #     with DiracXClient() as api:
+
+    @convertToReturnValue
+    def selectPilots(self, conditions_dict: dict):
+        """Select pilots given the selection conditions"""
+
+    #     with DiracXClient() as api:
+
+    @convertToReturnValue
+    def getPilotSummary(self, start_date: str, end_date: str):
+        """Get summary of the status of the Pilot Jobs"""
+
+    #   with DiracXClient() as api:
+
+    @convertToReturnValue
+    def getPilots(self, job_id: str | int):
+        """Get pilots executing/having executed a Job"""
+
+    # with DiracXClient() as api:
+
+    @convertToReturnValue
+    def setPilotStatus(
+        self,
+        pilot_reference: str,
+        status: str,
+        destination: str | None = None,
+        reason: str | None = None,
+        grid_site: str | None = None,
+        queue: str | None = None,
+        pilot_stamp: str | None = None,
+    ):
+        """Set the pilot status"""
+
+        with DiracXClient() as api:
+            values_dict = (
+                {
+                    "PilotStamp": pilot_stamp,
+                    "Status": status,
+                    "DestinationSite": destination,
+                    "StatusReason": reason,
+                    "GridSite": grid_site,
+                    "Queue": queue,
+                },
+            )
+
+            return api.pilots.update_pilot_metadata(pilot_stamps_to_fields_mapping=[values_dict])