From c47e805cf7d96e3261f249d629731127b2a13af3 Mon Sep 17 00:00:00 2001 From: Federico Stagni Date: Tue, 8 Sep 2026 12:32:28 +0200 Subject: [PATCH] fix: use PublisherClient to query selectSpaceTokenOccupancyCache --- .../WebApp/handler/SpaceOccupancyHandler.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/WebAppDIRAC/WebApp/handler/SpaceOccupancyHandler.py b/src/WebAppDIRAC/WebApp/handler/SpaceOccupancyHandler.py index 8f934b43..7a1a4996 100644 --- a/src/WebAppDIRAC/WebApp/handler/SpaceOccupancyHandler.py +++ b/src/WebAppDIRAC/WebApp/handler/SpaceOccupancyHandler.py @@ -1,10 +1,9 @@ -""" Handler for Space Occupancy web App -""" +"""Handler for Space Occupancy web App""" import json from DIRAC import gLogger -from DIRAC.ResourceStatusSystem.Client.ResourceManagementClient import ResourceManagementClient +from DIRAC.ResourceStatusSystem.Client.PublisherClient import PublisherClient from WebAppDIRAC.Lib.WebHandler import WebHandler, WErr @@ -12,7 +11,7 @@ class SpaceOccupancyHandler(WebHandler): DEFAULT_AUTHORIZATION = "authenticated" def initializeRequest(self): - self.rmc = ResourceManagementClient() + self.pc = PublisherClient() def web_getSelectionData(self, **kwargs): callback = { @@ -21,7 +20,7 @@ def web_getSelectionData(self, **kwargs): gLogger.info("Arguments to web_getSelectionData", kwargs) - if (result := self.rmc.selectSpaceTokenOccupancyCache())["OK"]: + if (result := self.pc.selectSpaceTokenOccupancyCache())["OK"]: for space in result["Value"]: callback["StorageElement"].add(space[1]) @@ -35,7 +34,7 @@ def web_getSelectionData(self, **kwargs): def web_getSpaceOccupancyData(self, StorageElement="null"): se = json.loads(StorageElement) - result = self.rmc.selectSpaceTokenOccupancyCache(None, list(se) if se else se) + result = self.pc.selectSpaceTokenOccupancyCache(list(se) if se else se) if not result["OK"]: raise WErr.fromSERROR(result)