@@ -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