feat: make TaskQueueDB VO aware for diracx transition

DIRACGrid · Sep 6, 2023 · 9ce9a86 · 9ce9a86
1 parent 262eab1
commit 9ce9a86
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py b/src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py
@@ -97,6 +97,7 @@ def __initializeDB(self):
                 "Owner": "VARCHAR(255) NOT NULL",
                 "OwnerDN": "VARCHAR(255)",
                 "OwnerGroup": "VARCHAR(32) NOT NULL",
+                "VO": "VARCHAR(32) NOT NULL",
                 "CPUTime": "BIGINT(20) UNSIGNED NOT NULL",
                 "Priority": "FLOAT NOT NULL",
                 "Enabled": "TINYINT(1) NOT NULL DEFAULT 0",
@@ -251,6 +252,8 @@ def __createTaskQueue(self, tqDefDict, priority=1, connObj=False):
         for field in singleValueDefFields:
             sqlSingleFields.append(field)
             sqlValues.append(tqDefDict[field])
+        sqlSingleFields.append("VO")
+        sqlValues.append(Registry.getVOForGroup(tqDefDict["OwnerGroup"]))
         # Insert the TQ Disabled
         sqlSingleFields.append("Enabled")
         sqlValues.append("0")