Skip to content

Commit cce8c21

Browse files
committed
fix: align jobMonitoringClient with the new DiracClient
1 parent 208bb79 commit cce8c21

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

src/DIRAC/WorkloadManagementSystem/FutureClient/JobMonitoringClient.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,16 @@
11
# pylint: disable=import-error
2-
from diracx.client import Dirac
2+
from diracx.client import DiracClient
33
from diracx.client.models import JobSearchParams
44

5-
from diracx.cli.utils import get_auth_headers
6-
from diracx.core.preferences import DiracxPreferences
7-
85
from DIRAC.Core.Utilities.ReturnValues import convertToReturnValue
96

107

118
class JobMonitoringClient:
12-
def __init__(self, *args, **kwargs):
13-
self.endpoint = DiracxPreferences().url
14-
159
def fetch(self, parameters, jobIDs):
16-
with Dirac(endpoint=self.endpoint) as api:
10+
with DiracClient() as api:
1711
jobs = api.jobs.search(
1812
parameters=["JobID"] + parameters,
1913
search=[{"parameter": "JobID", "operator": "in", "values": jobIDs}],
20-
headers=get_auth_headers(),
2114
)
2215
return {j["JobID"]: {param: j[param] for param in parameters} for j in jobs}
2316

0 commit comments

Comments
 (0)