Merge pull request #7238 from chaen/v8.1_FIX_sandboxAssign

fix (DiracX Sandboxstore): assign output SB
DIRACGrid · Oct 11, 2023 · 3d7415f · 3d7415f
2 parents 37690cd + ff8b071
commit 3d7415f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/DIRAC/WorkloadManagementSystem/Service/SandboxStoreHandler.py b/src/DIRAC/WorkloadManagementSystem/Service/SandboxStoreHandler.py
@@ -151,6 +151,11 @@ def _getFromClient(self, fileId, token, fileSize, fileHelper=None, data=""):
                         return S_ERROR("Error uploading sandbox", repr(e))
                 else:
                     gLogger.debug("Sandbox already exists in storage backend", res.pfn)
+
+                assignTo = {key: [(res.pfn, assignTo[key])] for key in assignTo}
+                result = self.export_assignSandboxesToEntities(assignTo)
+                if not result["OK"]:
+                    return result
                 return S_OK(res.pfn)
 
         sbPath = self.__getSandboxPath(f"{aHash}.{extension}")