Skip to content

Commit 82969a4

Browse files
committed
fix: simplification for sending error image
1 parent 57e334a commit 82969a4

File tree

4 files changed

+6
-18
lines changed

4 files changed

+6
-18
lines changed

src/DIRAC/AccountingSystem/Service/ReportGeneratorHandler.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -200,12 +200,8 @@ def __generatePlotFromFileId(self, fileId):
200200
return S_OK(result["Value"][fileToReturn])
201201

202202
def __sendErrorAsImg(self, msgText, fileHelper):
203-
retVal = generateErrorMessagePlot(msgText)
204-
retVal = fileHelper.sendData(retVal["Value"])
205-
if not retVal["OK"]:
206-
return retVal
203+
fileHelper.sendData(generateErrorMessagePlot(msgText))
207204
fileHelper.sendEOF()
208-
return S_OK()
209205

210206
def transfer_toClient(self, fileId, token, fileHelper):
211207
"""

src/DIRAC/Core/Utilities/Plotting/Plots.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def generateErrorMessagePlot(msgText):
5050
textGraph(msgText, fn, {})
5151
data = fn.getvalue()
5252
fn.close()
53-
return S_OK(data)
53+
return data
5454

5555

5656
def generateTimedStackedBarPlot(fileName, data, metadata):

src/DIRAC/MonitoringSystem/Service/MonitoringHandler.py

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -159,18 +159,11 @@ def __generatePlotFromFileId(self, fileId):
159159

160160
def __sendErrorAsImg(self, msgText, fileHelper):
161161
"""
162-
In case of an error message a whcite plot is created with the error message.
162+
In case of an error message a white plot is created with the error message.
163163
"""
164164

165-
retVal = generateErrorMessagePlot(msgText)
166-
if not retVal:
167-
retVal = fileHelper.sendData(retVal["Message"])
168-
else:
169-
retVal = fileHelper.sendData(retVal["Value"])
170-
if not retVal["OK"]:
171-
return retVal
172-
fileHelper.sendEOF()
173-
return S_OK()
165+
fileHelper.sendData(generateErrorMessagePlot(msgText))
166+
fileHelper.sendEOF()
174167

175168
def __checkPlotRequest(self, reportRequest):
176169
"""

tests/Integration/AccountingSystem/Test_Plots.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -290,9 +290,8 @@ def test_error():
290290
"""
291291

292292
res = generateErrorMessagePlot("testing error message")
293-
assert res["OK"] is True
294293
with open(filename, "wb") as out:
295-
out.write(res["Value"])
294+
out.write(res)
296295

297296
res = compare(filename, os.path.join(plots_directory, "error.png"))
298297
assert res == 0.0

0 commit comments

Comments
 (0)