feat: Enhance _getTransformationID to support multiple transformation IDs and names

Author: chrisburr

src/DIRAC/TransformationSystem/DB/TransformationDB.py

@@ -406,21 +406,49 @@ def __updateTransformationParameter(self, transID, paramName, paramValue, connec
 
     def _getTransformationID(self, transName, connection=False):
         """Method returns ID of transformation with the name=<name>"""
-        try:
-            transName = int(transName)
-            cmd = f"SELECT TransformationID from Transformations WHERE TransformationID={transName};"
-        except ValueError:
-            if not isinstance(transName, str):
-                return S_ERROR("Transformation should be ID or name")
-            cmd = f"SELECT TransformationID from Transformations WHERE TransformationName='{transName}';"
-        res = self._query(cmd, conn=connection)
-        if not res["OK"]:
-            gLogger.error("Failed to obtain transformation ID for transformation", f"{transName}: {res['Message']}")
-            return res
-        elif not res["Value"]:
-            gLogger.verbose(f"Transformation {transName} does not exist")
-            return S_ERROR("Transformation does not exist")
-        return S_OK(res["Value"][0][0])
+        tids = []
+        tnames = []
+        for name in [transName] if isinstance(transName, (int, str)) else transName:
+            if isinstance(name, int):
+                tids.append(name)
+            elif name.isdigit():
+                tids.append(int(name))
+            else:
+                tnames.append(name)
+
+        result = []
+        if tids:
+            cmd = "SELECT TransformationID from Transformations"
+            cmd += f" WHERE TransformationID IN ({','.join(map(connection._escapeString, tids))})"
+            res = self._query(cmd, conn=connection)
+            if not res["OK"]:
+                gLogger.error(
+                    "Failed to obtain transformation IDs for transformations", f"{transName}: {res['Message']}"
+                )
+                return res
+            if len(res["Value"]) != len(tids):
+                missing = set(tids) - {row[0] for row in res["Value"]}
+                gLogger.verbose(f"Transformations {missing} do not exist")
+                return S_ERROR(f"Transformations {missing} do not exist")
+            result.extend(row[0] for row in res["Value"])
+        if tnames:
+            cmd = "SELECT TransformationID, TransformationName from Transformations"
+            cmd += f" WHERE TransformationName IN ({','.join(map(connection._escapeString, tnames))})"
+            res = self._query(cmd, conn=connection)
+            if not res["OK"]:
+                gLogger.error(
+                    "Failed to obtain transformation IDs for transformations", f"{transName}: {res['Message']}"
+                )
+                return res
+            if len(res["Value"]) != len(tnames):
+                missing = set(tnames) - {row[1] for row in res["Value"]}
+                gLogger.verbose(f"Transformations {missing} do not exist")
+                return S_ERROR(f"Transformations {missing} do not exist")
+            result.extend(row[0] for row in res["Value"])
+
+        if isinstance(transName, (int, str)):
+            return S_OK(result[0])
+        return S_OK(result)
 
     def __deleteTransformation(self, transID, connection=False):
         return self._update(f"DELETE FROM Transformations WHERE TransformationID={transID};", conn=connection)