Skip to content

Commit dc011c2

Browse files
committed
docs: mock diracx when building docs
1 parent e350670 commit dc011c2

File tree

2 files changed

+26
-27
lines changed

2 files changed

+26
-27
lines changed

docs/diracdoctools/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
"_arc",
1010
"arc",
1111
"cmreslogging",
12+
"diracx",
1213
"fts3",
1314
"gfal2",
1415
"git",
Lines changed: 25 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -1,36 +1,34 @@
11
# pylint: disable=import-error
2-
import os
2+
from diracx.client import Dirac
3+
from diracx.client.models import JobSearchParams
34

4-
if os.getenv("DIRAC_ENABLE_DIRACX_JOB_MONITORING", "No").lower() in ("yes", "true"):
5-
from diracx.client import Dirac
6-
from diracx.client.models import JobSearchParams
5+
from diracx.cli.utils import get_auth_headers
6+
from diracx.core.preferences import DiracxPreferences
77

8-
from diracx.cli.utils import get_auth_headers
9-
from diracx.core.preferences import DiracxPreferences
8+
from DIRAC.Core.Utilities.ReturnValues import convertToReturnValue
109

11-
from DIRAC.Core.Utilities.ReturnValues import convertToReturnValue
1210

13-
class JobMonitoringClient:
14-
def __init__(self, *args, **kwargs):
15-
self.endpoint = DiracxPreferences().url
11+
class JobMonitoringClient:
12+
def __init__(self, *args, **kwargs):
13+
self.endpoint = DiracxPreferences().url
1614

17-
def fetch(self, parameters, jobIDs):
18-
with Dirac(endpoint=self.endpoint) as api:
19-
jobs = api.jobs.search(
20-
parameters=["JobID"] + parameters,
21-
search=[{"parameter": "JobID", "operator": "in", "values": jobIDs}],
22-
headers=get_auth_headers(),
23-
)
24-
return {j["JobID"]: {param: j[param] for param in parameters} for j in jobs}
15+
def fetch(self, parameters, jobIDs):
16+
with Dirac(endpoint=self.endpoint) as api:
17+
jobs = api.jobs.search(
18+
parameters=["JobID"] + parameters,
19+
search=[{"parameter": "JobID", "operator": "in", "values": jobIDs}],
20+
headers=get_auth_headers(),
21+
)
22+
return {j["JobID"]: {param: j[param] for param in parameters} for j in jobs}
2523

26-
@convertToReturnValue
27-
def getJobsMinorStatus(self, jobIDs):
28-
return self.fetch(["MinorStatus"], jobIDs)
24+
@convertToReturnValue
25+
def getJobsMinorStatus(self, jobIDs):
26+
return self.fetch(["MinorStatus"], jobIDs)
2927

30-
@convertToReturnValue
31-
def getJobsStates(self, jobIDs):
32-
return self.fetch(["Status", "MinorStatus", "ApplicationStatus"], jobIDs)
28+
@convertToReturnValue
29+
def getJobsStates(self, jobIDs):
30+
return self.fetch(["Status", "MinorStatus", "ApplicationStatus"], jobIDs)
3331

34-
@convertToReturnValue
35-
def getJobsSites(self, jobIDs):
36-
return self.fetch(["Site"], jobIDs)
32+
@convertToReturnValue
33+
def getJobsSites(self, jobIDs):
34+
return self.fetch(["Site"], jobIDs)

0 commit comments

Comments
 (0)