fix (TS): use TransformationStatus

arrabito · Aug 22, 2024 · 79c631b · 79c631b
1 parent 9648934
commit 79c631b
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/DIRAC/TransformationSystem/DB/TransformationDB.py b/src/DIRAC/TransformationSystem/DB/TransformationDB.py
@@ -15,6 +15,7 @@
 from DIRAC import gLogger, S_OK, S_ERROR
 from DIRAC.Core.Base.DB import DB
 from DIRAC.Core.Utilities.DErrno import cmpError
+from DIRAC.TransformationSystem.Client import TransformationStatus
 from DIRAC.Resources.Catalog.FileCatalog import FileCatalog
 from DIRAC.Core.Security.ProxyInfo import getProxyInfo
 from DIRAC.Core.Utilities.List import stringListToString, intListToString, breakListIntoChunks
@@ -1629,12 +1630,11 @@ def addFile(self, fileDicts, force=False, connection=False):
                     fileIDs = []
                     for fileDict in res["Value"]:
                         fileIDs.append(fileDict["FileID"])
-                        if fileDict["Status"] == "Deleted":
-                            res = self.__setTransformationFileStatus(list(fileIDs), "Unused", connection=connection)
+                        if fileDict["Status"] == TransformationStatus.DELETED:
+                            res = self.__setTransformationFileStatus(list(fileIDs), TransformationStatus.UNUSED, connection=connection)
                             if not res["OK"]:
                                 return res
-                            res = self.__setDataFileStatus(list(fileIDs), "New", connection=connection)
-                            if not res["OK"]:
+                            if not (res := self.__setDataFileStatus(list(fileIDs), TransformationStatus.NEW, connection=connection)["OK"]:
                                 return res
 
                     res = self.addFilesToTransformation(transID, lfns)
@@ -1776,11 +1776,11 @@ def setMetadata(self, path, usermetadatadict, connection=False):
                 fileIDs = []
                 for fileDict in res["Value"]:
                     fileIDs.append(fileDict["FileID"])
-                    if fileDict["Status"] == "Deleted":
-                        res = self.__setTransformationFileStatus(list(fileIDs), "Unused", connection=connection)
+                    if fileDict["Status"] == TransformationStatus.DELETED:
+                        res = self.__setTransformationFileStatus(list(fileIDs), TransformationStatus.UNUSED, connection=connection)
                         if not res["OK"]:
                             return res
-                        res = self.__setDataFileStatus(list(fileIDs), "New", connection=connection)
+                        res = self.__setDataFileStatus(list(fileIDs), TransformationStatus.NEW, connection=connection)
                         if not res["OK"]:
                             return res