Skip to content

Commit 6b061b1

Browse files
authored
Merge pull request #784 from fstagni/60_fixes_6
fix: do not fix the clients
2 parents dfd676f + f66babf commit 6b061b1

File tree

1 file changed

+7
-23
lines changed

1 file changed

+7
-23
lines changed

src/WebAppDIRAC/WebApp/handler/AccountingHandler.py

Lines changed: 7 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -21,22 +21,6 @@ class AccountingHandler(WebHandler):
2121
# Key: (user, group, setup, accounting type)
2222
__keysCache = DictCache.DictCache()
2323

24-
# Reports client
25-
repClient = None
26-
# Transfer client
27-
transClient = None
28-
29-
@classmethod
30-
def initializeHandler(cls, serviceInfo):
31-
"""This may be overwritten when you write a DIRAC service handler
32-
And it must be a class method. This method is called only one time,
33-
at the first request
34-
35-
:param dict serviceInfo: infos about service
36-
"""
37-
cls.repClient = ReportsClient()
38-
cls.transClient = TransferClient("Accounting/ReportGenerator")
39-
4024
def __getUniqueKeyValues(self, typeName):
4125
"""Get unique key values for accounting type
4226
@@ -47,7 +31,7 @@ def __getUniqueKeyValues(self, typeName):
4731
cacheKey = (self.getUserName(), self.getUserGroup(), self.getUserSetup(), typeName)
4832
data = AccountingHandler.__keysCache.get(cacheKey)
4933
if not data:
50-
retVal = self.repClient.listUniqueKeyValues(typeName)
34+
retVal = ReportsClient().listUniqueKeyValues(typeName)
5135
if "rpcStub" in retVal:
5236
del retVal["rpcStub"]
5337
if not retVal["OK"]:
@@ -104,7 +88,7 @@ def web_getSelectionData(self, type, **kwargs):
10488
# Cache for plotsList?
10589
data = AccountingHandler.__keysCache.get(f"reportsList:{typeName}")
10690
if not data:
107-
retVal = self.repClient.listReports(typeName)
91+
retVal = ReportsClient().listReports(typeName)
10892
if not retVal["OK"]:
10993
return {"success": "false", "result": "", "error": retVal["Message"]}
11094
data = retVal["Value"]
@@ -206,7 +190,7 @@ def web_generatePlot(self, typeName, plotName, timeSelector: int, grouping, **kw
206190
retVal = self.__parseFormParams(timeSelector, grouping=[grouping], typeName=typeName, **kwargs)
207191
if retVal["OK"]:
208192
start, end, pD, kwargs = retVal["Value"]
209-
retVal = self.repClient.generateDelayedPlot(typeName, plotName, start, end, pD, grouping, kwargs)
193+
retVal = ReportsClient().generateDelayedPlot(typeName, plotName, start, end, pD, grouping, kwargs)
210194
if retVal["OK"]:
211195
return {"success": True, "data": retVal["Value"]["plot"]}
212196
return {"success": False, "errors": retVal["Message"]}
@@ -225,7 +209,7 @@ def web_getPlotImg(self, fileName=None, **kwargs):
225209
plotImageFile = os.path.normpath("/" + plotImageFile).lstrip("/")
226210

227211
tempFile = tempfile.TemporaryFile()
228-
retVal = self.transClient.receiveFile(tempFile, plotImageFile)
212+
retVal = TransferClient("Accounting/ReportGenerator").receiveFile(tempFile, plotImageFile)
229213
if not retVal["OK"]:
230214
return {"success": "false", "error": retVal["Message"]}
231215

@@ -260,7 +244,7 @@ def web_getPlotImgFromCache(self, fileName=None, **kwargs):
260244

261245
plotImageFile = retVal["Value"]["plot"]
262246
tempFile = tempfile.TemporaryFile()
263-
retVal = self.transClient.receiveFile(tempFile, plotImageFile)
247+
retVal = TransferClient("Accounting/ReportGenerator").receiveFile(tempFile, plotImageFile)
264248
if not retVal["OK"]:
265249
return {"success": "false", "error": retVal["Message"]}
266250

@@ -283,7 +267,7 @@ def web_getCsvPlotData(self, typeName, plotName, timeSelector: int, grouping, **
283267
return {"success": "false", "error": retVal["Message"]}
284268
start, end, pD, kwargs = retVal["Value"]
285269
params = (typeName, plotName, start, end, pD, grouping, kwargs)
286-
retVal = self.repClient.getReport(*params)
270+
retVal = ReportsClient().getReport(*params)
287271
if not retVal["OK"]:
288272
return {"success": "false", "error": retVal["Message"]}
289273

@@ -319,6 +303,6 @@ def web_getPlotData(self, typeName, plotName, timeSelector: int, grouping, **kwa
319303
retVal = self.__parseFormParams(timeSelector, grouping=[grouping], typeName=typeName, **kwargs)
320304
if retVal["OK"]:
321305
start, end, pD, kwargs = retVal["Value"]
322-
retVal = self.repClient.getReport(typeName, plotName, start, end, pD, grouping, kwargs)
306+
retVal = ReportsClient().getReport(typeName, plotName, start, end, pD, grouping, kwargs)
323307

324308
return retVal["Value"] if retVal["OK"] else {"success": "false", "error": retVal["Message"]}

0 commit comments

Comments
 (0)